utils/ext-core/Env.hs

   1 {- Environments.
   2   The original version used lists. I changed it to use Data.Map.
   3   Sadly it doesn't seem to matter much. --tjc
   4 -}
   5
   6 module Env (Env,
   7             eempty,
   8             elookup,
   9             eextend,
  10             edomain,
  11             efromlist,
  12             etolist,
  13             efilter,
  14             eremove)
  15 where
  16
  17 import qualified Data.Map as M
  18 import Data.List
  19
  20 data Env a b = Env (M.Map a b)
  21  deriving Show
  22
  23 eempty :: Env a b
  24 eempty = Env M.empty
  25
  26 {- In case of duplicates, returns most recently added entry. -}
  27 elookup :: (Eq a, Ord a) => Env a b -> a -> Maybe b
  28 elookup (Env l) k = M.lookup k l
  29
  30 {- May hide existing entries. -}
  31 eextend :: Ord a => Env a b -> (a,b) -> Env a b
  32 eextend (Env l) (k,d) = Env (M.insert k d l)
  33
  34 edomain :: (Eq a) => Env a b -> [a]
  35 edomain (Env l) = M.keys l
  36
  37 {- In case of duplicates, first entry hides others. -}
  38 efromlist :: Ord a => [(a,b)] -> Env a b
  39 efromlist = Env . M.fromList
  40
  41 etolist :: Env a b -> [(a,b)]
  42 etolist (Env l) = M.toList l
  43
  44 eremove :: (Eq a, Ord a)  => Env a b -> a -> Env a b
  45 eremove (Env l) k = Env (M.delete k l)
  46
  47 efilter :: Ord a => Env a b -> (a -> Bool) -> Env a b
  48 efilter (Env l) p = Env (M.filterWithKey (\ k a -> p k) l)
  49