Added withElement, setElement, and made Store a Functor

1 files changed, 21 insertions, 0 deletions

diff --git a/Spear/Sys/Store.hs b/Spear/Sys/Store.hs
index 65381ca..3c1e720 100644
--- a/Spear/Sys/Store.hs
+++ b/Spear/Sys/Store.hs
@@ -8,6 +8,8 @@ module Spear.Sys.Store
 ,   storeFree
 ,   storeFreel
 ,   element
+,   setElement
+,   withElement
 )
 where
 
@@ -29,6 +31,10 @@ data Store a = Store
     deriving Show
 
 
+instance Functor Store where
+    fmap f (Store objects last) = Store (fmap (fmap f) objects) last
+
+
 -- | Create an empty store.
 emptyStore :: Store a
 emptyStore = Store V.empty (-1)
@@ -135,6 +141,21 @@ element :: Index -> Store a -> Maybe a
 element index (Store objects _) = objects V.! index
 
 
+-- | Set the element in the given slot.
+setElement :: Index -> a -> Store a -> Store a
+setElement index elem s = s { objects = objects s // [(index,Just elem)] }
+
+
+-- | Apply a function to the element in the given slot.
+withElement :: Index -> Store a -> (a -> a) -> Store a
+withElement index store f = store { objects = objects' }
+    where
+        objects' = objects store // [(index, obj')]
+        obj' = case element index store of
+            Nothing -> Nothing
+            Just x  -> Just $ f x
+
+
 -- test
 test :: IO ()
 test = evalStateT test' emptyStore