compiler/basicTypes/NameEnv.lhs

   1 %
   2 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
   3 %
   4 \section[NameEnv]{@NameEnv@: name environments}
   5
   6 \begin{code}
   7 module NameEnv (
   8         NameEnv, mkNameEnv,
   9         emptyNameEnv, unitNameEnv, nameEnvElts,
  10         extendNameEnv_C, extendNameEnv_Acc, extendNameEnv,
  11         extendNameEnvList, extendNameEnvList_C,
  12         foldNameEnv, filterNameEnv,
  13         plusNameEnv, plusNameEnv_C,
  14         lookupNameEnv, lookupNameEnv_NF, delFromNameEnv, delListFromNameEnv,
  15         elemNameEnv, mapNameEnv
  16     ) where
  17
  18 #include "HsVersions.h"
  19
  20 import Name     ( Name )
  21 import UniqFM
  22 import Maybes   ( expectJust )
  23 \end{code}
  24
  25 %************************************************************************
  26 %*                                                                      *
  27 \subsection{Name environment}
  28 %*                                                                      *
  29 %************************************************************************
  30
  31 \begin{code}
  32 type NameEnv a = UniqFM a       -- Domain is Name
  33
  34 emptyNameEnv       :: NameEnv a
  35 mkNameEnv          :: [(Name,a)] -> NameEnv a
  36 nameEnvElts        :: NameEnv a -> [a]
  37 extendNameEnv_C    :: (a->a->a) -> NameEnv a -> Name -> a -> NameEnv a
  38 extendNameEnv_Acc  :: (a->b->b) -> (a->b) -> NameEnv b -> Name -> a -> NameEnv b
  39 extendNameEnv      :: NameEnv a -> Name -> a -> NameEnv a
  40 plusNameEnv        :: NameEnv a -> NameEnv a -> NameEnv a
  41 plusNameEnv_C      :: (a->a->a) -> NameEnv a -> NameEnv a -> NameEnv a
  42 extendNameEnvList  :: NameEnv a -> [(Name,a)] -> NameEnv a
  43 extendNameEnvList_C :: (a->a->a) -> NameEnv a -> [(Name,a)] -> NameEnv a
  44 delFromNameEnv     :: NameEnv a -> Name -> NameEnv a
  45 delListFromNameEnv :: NameEnv a -> [Name] -> NameEnv a
  46 elemNameEnv        :: Name -> NameEnv a -> Bool
  47 unitNameEnv        :: Name -> a -> NameEnv a
  48 lookupNameEnv      :: NameEnv a -> Name -> Maybe a
  49 lookupNameEnv_NF   :: NameEnv a -> Name -> a
  50 foldNameEnv        :: (a -> b -> b) -> b -> NameEnv a -> b
  51 filterNameEnv      :: (elt -> Bool) -> NameEnv elt -> NameEnv elt
  52 mapNameEnv         :: (elt1 -> elt2) -> NameEnv elt1 -> NameEnv elt2
  53
  54 emptyNameEnv        = emptyUFM
  55 foldNameEnv         = foldUFM
  56 mkNameEnv           = listToUFM
  57 nameEnvElts         = eltsUFM
  58 extendNameEnv_C     = addToUFM_C
  59 extendNameEnv_Acc   = addToUFM_Acc
  60 extendNameEnv       = addToUFM
  61 plusNameEnv         = plusUFM
  62 plusNameEnv_C       = plusUFM_C
  63 extendNameEnvList   = addListToUFM
  64 extendNameEnvList_C = addListToUFM_C
  65 delFromNameEnv      = delFromUFM
  66 delListFromNameEnv  = delListFromUFM
  67 elemNameEnv         = elemUFM
  68 unitNameEnv         = unitUFM
  69 filterNameEnv       = filterUFM
  70 mapNameEnv          = mapUFM
  71
  72 lookupNameEnv          = lookupUFM
  73 lookupNameEnv_NF env n = expectJust "lookupNameEnv_NF" (lookupUFM env n)
  74 \end{code}
  75