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