stable-maps 0.0.4 → 0.0.5

raw patch · 2 files changed

+27/−20 lines, 2 files

Files

• System/Mem/StableName/Map.hs +26/−19
• stable-maps.cabal +1/−1

@@ -1,7 +1,10 @@ {-# LANGUAGE CPP #-}-#if defined(__GLASGOW_HASKELL__) && __GLASGOW_HASKELL__ >= 702+#if __GLASGOW_HASKELL__ >= 702 {-# LANGUAGE Unsafe #-} #endif+#if __GLASGOW_HASKELL__ >= 707+{-# LANGUAGE RoleAnnotations #-}+#endif module System.Mem.StableName.Map     ( Map     , empty@@ -27,13 +30,17 @@ import Data.IntMap (IntMap) import Unsafe.Coerce (unsafeCoerce) -newtype Map f = Map { getMap :: IntMap [(DynamicStableName, f Any)] } +-- | Note: this can be generally unsound when `f` is a GADT!+newtype Map f = Map { getMap :: IntMap [(DynamicStableName, f Any)] }+#if __GLASGOW_HASKELL__ >= 707+type role Map nominal+#endif  -- unsafe combinators any :: f a -> f Any any = unsafeCoerce -liftAny1 :: (f a -> f a) -> f Any -> f Any +liftAny1 :: (f a -> f a) -> f Any -> f Any liftAny1 f a = unsafeCoerce $ f (unsafeCoerce a)  empty :: Map f@@ -53,14 +60,14 @@     Just _ -> True  notMember :: StableName a -> Map f -> Bool-notMember k m = not $ member k m +notMember k m = not $ member k m  insert :: StableName a -> f a -> Map f -> Map f-insert k v = -    Map . -    IntMap.insertWith (++) (hashDynamicStableName dk) [(dk,any v)] . +insert k v =+    Map .+    IntMap.insertWith (++) (hashDynamicStableName dk) [(dk,any v)] .     getMap-    where +    where         dk = wrapStableName k  -- | /O(log n)/. Insert with a function for combining the new value and old value.@@ -69,20 +76,20 @@ -- in the map. If the key does exist, the function will insert the pair -- @(key, f new_value old_value)@ insertWith :: (f a -> f a -> f a) -> StableName a -> f a -> Map f -> Map f-insertWith f k v = Map . IntMap.insertWith go (hashDynamicStableName dk) [(dk,any v)] . getMap -    where +insertWith f k v = Map . IntMap.insertWith go (hashDynamicStableName dk) [(dk,any v)] . getMap+    where         dk = wrapStableName k-        go _ ((k',v'):kvs) +        go _ ((k',v'):kvs)             | dk == k' = (k', liftAny1 (f v) v') : kvs             | otherwise = (k',v') : go undefined kvs         go _ [] = [(dk, any v)]  -- | Same as 'insertWith', but with the combining function applied strictly. insertWith' :: (f a -> f a -> f a) -> StableName a -> f a -> Map f -> Map f-insertWith' f k v = Map . IntMap.insertWith go (hashDynamicStableName dk) [(dk, any v)] . getMap -    where +insertWith' f k v = Map . IntMap.insertWith go (hashDynamicStableName dk) [(dk, any v)] . getMap+    where         dk = wrapStableName k-        go _ ((k',v'):kvs) +        go _ ((k',v'):kvs)             | dk == k' = let v'' = liftAny1 (f v) v' in v'' `seq` (k', v'') : kvs             | otherwise = (k', v') : go undefined kvs         go _ [] = [(dk, any v)]@@ -97,7 +104,7 @@         go [] = []  -- | /O(log n)/. Lookup the value at a key in the map.--- +-- -- The function will return the corresponding value as a @('Just' value)@ -- or 'Nothing' if the key isn't in the map. lookup :: StableName a -> Map f -> Maybe (f a)@@ -105,15 +112,15 @@     pairs <- IntMap.lookup (hashDynamicStableName dk) m     unsafeCoerce $ Prelude.lookup dk pairs     where-        dk = wrapStableName k +        dk = wrapStableName k -find :: StableName a -> Map f -> f a +find :: StableName a -> Map f -> f a find k m = case lookup k m of     Nothing -> error "Map.find: element not in the map"-    Just x -> x +    Just x -> x  -- | /O(log n)/. The expression @('findWithDefault' def k map)@ returns -- the value at key @k@ or returns the default value @def@ -- when the key is not in the map. findWithDefault :: f a -> StableName a -> Map f -> f a-findWithDefault dflt k m = maybe dflt id $ lookup k m +findWithDefault dflt k m = maybe dflt id $ lookup k m

@@ -1,5 +1,5 @@ Name:              stable-maps-Version:           0.0.4+Version:           0.0.5 Synopsis:          Heterogeneous maps keyed by StableNames Description:       Provides an unsafe API for inserting heterogeneous data in a collection keyed by StableNames and for later retrieving it. Homepage:          http://github.com/ekmett/stable-maps