module GHC.Types.Var.Env (
VarEnv, IdEnv, TyVarEnv, CoVarEnv, TyCoVarEnv,
emptyVarEnv, unitVarEnv, mkVarEnv, mkVarEnv_Directly,
elemVarEnv, disjointVarEnv,
extendVarEnv, extendVarEnv_C, extendVarEnv_Acc,
extendVarEnvList,
plusVarEnv, plusVarEnv_C, plusVarEnv_CD, plusMaybeVarEnv_C,
plusVarEnvList, alterVarEnv,
delVarEnvList, delVarEnv,
minusVarEnv,
lookupVarEnv, lookupVarEnv_NF, lookupWithDefaultVarEnv,
mapVarEnv, zipVarEnv,
modifyVarEnv, modifyVarEnv_Directly,
isEmptyVarEnv,
elemVarEnvByKey,
filterVarEnv, restrictVarEnv,
partitionVarEnv,
DVarEnv, DIdEnv, DTyVarEnv,
emptyDVarEnv, mkDVarEnv,
dVarEnvElts,
extendDVarEnv, extendDVarEnv_C,
extendDVarEnvList,
lookupDVarEnv, elemDVarEnv,
isEmptyDVarEnv, foldDVarEnv, nonDetStrictFoldDVarEnv,
mapDVarEnv, filterDVarEnv,
modifyDVarEnv,
alterDVarEnv,
plusDVarEnv, plusDVarEnv_C,
unitDVarEnv,
delDVarEnv,
delDVarEnvList,
minusDVarEnv,
partitionDVarEnv,
anyDVarEnv,
InScopeSet,
emptyInScopeSet, mkInScopeSet, delInScopeSet,
extendInScopeSet, extendInScopeSetList, extendInScopeSetSet,
getInScopeVars, lookupInScope, lookupInScope_Directly,
unionInScope, elemInScopeSet, uniqAway,
varSetInScope,
unsafeGetFreshLocalUnique,
RnEnv2,
mkRnEnv2, rnBndr2, rnBndrs2, rnBndr2_var,
rnOccL, rnOccR, inRnEnvL, inRnEnvR, rnOccL_maybe, rnOccR_maybe,
rnBndrL, rnBndrR, nukeRnEnvL, nukeRnEnvR, rnSwap,
delBndrL, delBndrR, delBndrsL, delBndrsR,
extendRnInScopeSetList,
rnEtaL, rnEtaR,
rnInScope, rnInScopeSet, lookupRnInScope,
rnEnvL, rnEnvR,
TidyEnv,
emptyTidyEnv, mkEmptyTidyEnv, delTidyEnvList
) where
import GHC.Prelude
import qualified Data.IntMap.Strict as IntMap
import GHC.Types.Name.Occurrence
import GHC.Types.Name
import GHC.Types.Var as Var
import GHC.Types.Var.Set
import GHC.Types.Unique.Set
import GHC.Types.Unique.FM
import GHC.Types.Unique.DFM
import GHC.Types.Unique
import GHC.Utils.Misc
import GHC.Utils.Panic
import GHC.Data.Maybe
import GHC.Utils.Outputable
newtype InScopeSet = InScope VarSet
instance Outputable InScopeSet where
ppr :: InScopeSet -> SDoc
ppr (InScope VarSet
s) =
String -> SDoc
text String
"InScope" SDoc -> SDoc -> SDoc
<+>
SDoc -> SDoc
braces ([SDoc] -> SDoc
fsep (forall a b. (a -> b) -> [a] -> [b]
map (forall a. Outputable a => a -> SDoc
ppr forall b c a. (b -> c) -> (a -> b) -> a -> c
. Var -> Name
Var.varName) (forall elt. UniqSet elt -> [elt]
nonDetEltsUniqSet VarSet
s)))
emptyInScopeSet :: InScopeSet
emptyInScopeSet :: InScopeSet
emptyInScopeSet = VarSet -> InScopeSet
InScope VarSet
emptyVarSet
getInScopeVars :: InScopeSet -> VarSet
getInScopeVars :: InScopeSet -> VarSet
getInScopeVars (InScope VarSet
vs) = VarSet
vs
mkInScopeSet :: VarSet -> InScopeSet
mkInScopeSet :: VarSet -> InScopeSet
mkInScopeSet VarSet
in_scope = VarSet -> InScopeSet
InScope VarSet
in_scope
extendInScopeSet :: InScopeSet -> Var -> InScopeSet
extendInScopeSet :: InScopeSet -> Var -> InScopeSet
extendInScopeSet (InScope VarSet
in_scope) Var
v
= VarSet -> InScopeSet
InScope (VarSet -> Var -> VarSet
extendVarSet VarSet
in_scope Var
v)
extendInScopeSetList :: InScopeSet -> [Var] -> InScopeSet
extendInScopeSetList :: InScopeSet -> [Var] -> InScopeSet
extendInScopeSetList (InScope VarSet
in_scope) [Var]
vs
= VarSet -> InScopeSet
InScope forall a b. (a -> b) -> a -> b
$ forall (t :: * -> *) b a.
Foldable t =>
(b -> a -> b) -> b -> t a -> b
foldl' VarSet -> Var -> VarSet
extendVarSet VarSet
in_scope [Var]
vs
extendInScopeSetSet :: InScopeSet -> VarSet -> InScopeSet
extendInScopeSetSet :: InScopeSet -> VarSet -> InScopeSet
extendInScopeSetSet (InScope VarSet
in_scope) VarSet
vs
= VarSet -> InScopeSet
InScope (VarSet
in_scope VarSet -> VarSet -> VarSet
`unionVarSet` VarSet
vs)
delInScopeSet :: InScopeSet -> Var -> InScopeSet
delInScopeSet :: InScopeSet -> Var -> InScopeSet
delInScopeSet (InScope VarSet
in_scope) Var
v = VarSet -> InScopeSet
InScope (VarSet
in_scope VarSet -> Var -> VarSet
`delVarSet` Var
v)
elemInScopeSet :: Var -> InScopeSet -> Bool
elemInScopeSet :: Var -> InScopeSet -> Bool
elemInScopeSet Var
v (InScope VarSet
in_scope) = Var
v Var -> VarSet -> Bool
`elemVarSet` VarSet
in_scope
lookupInScope :: InScopeSet -> Var -> Maybe Var
lookupInScope :: InScopeSet -> Var -> Maybe Var
lookupInScope (InScope VarSet
in_scope) Var
v = VarSet -> Var -> Maybe Var
lookupVarSet VarSet
in_scope Var
v
lookupInScope_Directly :: InScopeSet -> Unique -> Maybe Var
lookupInScope_Directly :: InScopeSet -> Unique -> Maybe Var
lookupInScope_Directly (InScope VarSet
in_scope) Unique
uniq
= VarSet -> Unique -> Maybe Var
lookupVarSet_Directly VarSet
in_scope Unique
uniq
unionInScope :: InScopeSet -> InScopeSet -> InScopeSet
unionInScope :: InScopeSet -> InScopeSet -> InScopeSet
unionInScope (InScope VarSet
s1) (InScope VarSet
s2)
= VarSet -> InScopeSet
InScope (VarSet
s1 VarSet -> VarSet -> VarSet
`unionVarSet` VarSet
s2)
varSetInScope :: VarSet -> InScopeSet -> Bool
varSetInScope :: VarSet -> InScopeSet -> Bool
varSetInScope VarSet
vars (InScope VarSet
s1) = VarSet
vars VarSet -> VarSet -> Bool
`subVarSet` VarSet
s1
uniqAway :: InScopeSet -> Var -> Var
uniqAway :: InScopeSet -> Var -> Var
uniqAway InScopeSet
in_scope Var
var
| Var
var Var -> InScopeSet -> Bool
`elemInScopeSet` InScopeSet
in_scope = InScopeSet -> Var -> Var
uniqAway' InScopeSet
in_scope Var
var
| Bool
otherwise = Var
var
uniqAway' :: InScopeSet -> Var -> Var
uniqAway' :: InScopeSet -> Var -> Var
uniqAway' InScopeSet
in_scope Var
var
= Var -> Unique -> Var
setVarUnique Var
var (InScopeSet -> Unique
unsafeGetFreshLocalUnique InScopeSet
in_scope)
unsafeGetFreshLocalUnique :: InScopeSet -> Unique
unsafeGetFreshLocalUnique :: InScopeSet -> Unique
unsafeGetFreshLocalUnique (InScope VarSet
set)
| Just (Key
uniq,Var
_) <- forall a. Key -> IntMap a -> Maybe (Key, a)
IntMap.lookupLT (Unique -> Key
getKey Unique
maxLocalUnique) (forall key elt. UniqFM key elt -> IntMap elt
ufmToIntMap forall a b. (a -> b) -> a -> b
$ forall a. UniqSet a -> UniqFM a a
getUniqSet VarSet
set)
, let uniq' :: Unique
uniq' = Key -> Unique
mkLocalUnique Key
uniq
, Bool -> Bool
not forall a b. (a -> b) -> a -> b
$ Unique
uniq' Unique -> Unique -> Bool
`ltUnique` Unique
minLocalUnique
= Unique -> Unique
incrUnique Unique
uniq'
| Bool
otherwise
= Unique
minLocalUnique
data RnEnv2
= RV2 { RnEnv2 -> VarEnv Var
envL :: VarEnv Var
, RnEnv2 -> VarEnv Var
envR :: VarEnv Var
, RnEnv2 -> InScopeSet
in_scope :: InScopeSet }
mkRnEnv2 :: InScopeSet -> RnEnv2
mkRnEnv2 :: InScopeSet -> RnEnv2
mkRnEnv2 InScopeSet
vars = RV2 { envL :: VarEnv Var
envL = forall a. VarEnv a
emptyVarEnv
, envR :: VarEnv Var
envR = forall a. VarEnv a
emptyVarEnv
, in_scope :: InScopeSet
in_scope = InScopeSet
vars }
extendRnInScopeSetList :: RnEnv2 -> [Var] -> RnEnv2
extendRnInScopeSetList :: RnEnv2 -> [Var] -> RnEnv2
extendRnInScopeSetList RnEnv2
env [Var]
vs
| forall (t :: * -> *) a. Foldable t => t a -> Bool
null [Var]
vs = RnEnv2
env
| Bool
otherwise = RnEnv2
env { in_scope :: InScopeSet
in_scope = InScopeSet -> [Var] -> InScopeSet
extendInScopeSetList (RnEnv2 -> InScopeSet
in_scope RnEnv2
env) [Var]
vs }
rnInScope :: Var -> RnEnv2 -> Bool
rnInScope :: Var -> RnEnv2 -> Bool
rnInScope Var
x RnEnv2
env = Var
x Var -> InScopeSet -> Bool
`elemInScopeSet` RnEnv2 -> InScopeSet
in_scope RnEnv2
env
rnInScopeSet :: RnEnv2 -> InScopeSet
rnInScopeSet :: RnEnv2 -> InScopeSet
rnInScopeSet = RnEnv2 -> InScopeSet
in_scope
rnEnvL :: RnEnv2 -> VarEnv Var
rnEnvL :: RnEnv2 -> VarEnv Var
rnEnvL = RnEnv2 -> VarEnv Var
envL
rnEnvR :: RnEnv2 -> VarEnv Var
rnEnvR :: RnEnv2 -> VarEnv Var
rnEnvR = RnEnv2 -> VarEnv Var
envR
rnBndrs2 :: RnEnv2 -> [Var] -> [Var] -> RnEnv2
rnBndrs2 :: RnEnv2 -> [Var] -> [Var] -> RnEnv2
rnBndrs2 RnEnv2
env [Var]
bsL [Var]
bsR = forall acc a b. (acc -> a -> b -> acc) -> acc -> [a] -> [b] -> acc
foldl2 RnEnv2 -> Var -> Var -> RnEnv2
rnBndr2 RnEnv2
env [Var]
bsL [Var]
bsR
rnBndr2 :: RnEnv2 -> Var -> Var -> RnEnv2
rnBndr2 :: RnEnv2 -> Var -> Var -> RnEnv2
rnBndr2 RnEnv2
env Var
bL Var
bR = forall a b. (a, b) -> a
fst forall a b. (a -> b) -> a -> b
$ RnEnv2 -> Var -> Var -> (RnEnv2, Var)
rnBndr2_var RnEnv2
env Var
bL Var
bR
rnBndr2_var :: RnEnv2 -> Var -> Var -> (RnEnv2, Var)
rnBndr2_var :: RnEnv2 -> Var -> Var -> (RnEnv2, Var)
rnBndr2_var (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
bL Var
bR
= (RV2 { envL :: VarEnv Var
envL = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envL Var
bL Var
new_b
, envR :: VarEnv Var
envR = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envR Var
bR Var
new_b
, in_scope :: InScopeSet
in_scope = InScopeSet -> Var -> InScopeSet
extendInScopeSet InScopeSet
in_scope Var
new_b }, Var
new_b)
where
new_b :: Var
new_b | Bool -> Bool
not (Var
bL Var -> InScopeSet -> Bool
`elemInScopeSet` InScopeSet
in_scope) = Var
bL
| Bool -> Bool
not (Var
bR Var -> InScopeSet -> Bool
`elemInScopeSet` InScopeSet
in_scope) = Var
bR
| Bool
otherwise = InScopeSet -> Var -> Var
uniqAway' InScopeSet
in_scope Var
bL
rnBndrL :: RnEnv2 -> Var -> (RnEnv2, Var)
rnBndrL :: RnEnv2 -> Var -> (RnEnv2, Var)
rnBndrL (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
bL
= (RV2 { envL :: VarEnv Var
envL = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envL Var
bL Var
new_b
, envR :: VarEnv Var
envR = VarEnv Var
envR
, in_scope :: InScopeSet
in_scope = InScopeSet -> Var -> InScopeSet
extendInScopeSet InScopeSet
in_scope Var
new_b }, Var
new_b)
where
new_b :: Var
new_b = InScopeSet -> Var -> Var
uniqAway InScopeSet
in_scope Var
bL
rnBndrR :: RnEnv2 -> Var -> (RnEnv2, Var)
rnBndrR :: RnEnv2 -> Var -> (RnEnv2, Var)
rnBndrR (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
bR
= (RV2 { envR :: VarEnv Var
envR = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envR Var
bR Var
new_b
, envL :: VarEnv Var
envL = VarEnv Var
envL
, in_scope :: InScopeSet
in_scope = InScopeSet -> Var -> InScopeSet
extendInScopeSet InScopeSet
in_scope Var
new_b }, Var
new_b)
where
new_b :: Var
new_b = InScopeSet -> Var -> Var
uniqAway InScopeSet
in_scope Var
bR
rnEtaL :: RnEnv2 -> Var -> (RnEnv2, Var)
rnEtaL :: RnEnv2 -> Var -> (RnEnv2, Var)
rnEtaL (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
bL
= (RV2 { envL :: VarEnv Var
envL = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envL Var
bL Var
new_b
, envR :: VarEnv Var
envR = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envR Var
new_b Var
new_b
, in_scope :: InScopeSet
in_scope = InScopeSet -> Var -> InScopeSet
extendInScopeSet InScopeSet
in_scope Var
new_b }, Var
new_b)
where
new_b :: Var
new_b = InScopeSet -> Var -> Var
uniqAway InScopeSet
in_scope Var
bL
rnEtaR :: RnEnv2 -> Var -> (RnEnv2, Var)
rnEtaR :: RnEnv2 -> Var -> (RnEnv2, Var)
rnEtaR (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
bR
= (RV2 { envL :: VarEnv Var
envL = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envL Var
new_b Var
new_b
, envR :: VarEnv Var
envR = forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv Var
envR Var
bR Var
new_b
, in_scope :: InScopeSet
in_scope = InScopeSet -> Var -> InScopeSet
extendInScopeSet InScopeSet
in_scope Var
new_b }, Var
new_b)
where
new_b :: Var
new_b = InScopeSet -> Var -> Var
uniqAway InScopeSet
in_scope Var
bR
delBndrL, delBndrR :: RnEnv2 -> Var -> RnEnv2
delBndrL :: RnEnv2 -> Var -> RnEnv2
delBndrL rn :: RnEnv2
rn@(RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
env, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
v
= RnEnv2
rn { envL :: VarEnv Var
envL = VarEnv Var
env forall a. VarEnv a -> Var -> VarEnv a
`delVarEnv` Var
v, in_scope :: InScopeSet
in_scope = InScopeSet
in_scope InScopeSet -> Var -> InScopeSet
`extendInScopeSet` Var
v }
delBndrR :: RnEnv2 -> Var -> RnEnv2
delBndrR rn :: RnEnv2
rn@(RV2 { envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
env, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) Var
v
= RnEnv2
rn { envR :: VarEnv Var
envR = VarEnv Var
env forall a. VarEnv a -> Var -> VarEnv a
`delVarEnv` Var
v, in_scope :: InScopeSet
in_scope = InScopeSet
in_scope InScopeSet -> Var -> InScopeSet
`extendInScopeSet` Var
v }
delBndrsL, delBndrsR :: RnEnv2 -> [Var] -> RnEnv2
delBndrsL :: RnEnv2 -> [Var] -> RnEnv2
delBndrsL rn :: RnEnv2
rn@(RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
env, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) [Var]
v
= RnEnv2
rn { envL :: VarEnv Var
envL = VarEnv Var
env forall a. VarEnv a -> [Var] -> VarEnv a
`delVarEnvList` [Var]
v, in_scope :: InScopeSet
in_scope = InScopeSet
in_scope InScopeSet -> [Var] -> InScopeSet
`extendInScopeSetList` [Var]
v }
delBndrsR :: RnEnv2 -> [Var] -> RnEnv2
delBndrsR rn :: RnEnv2
rn@(RV2 { envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
env, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope }) [Var]
v
= RnEnv2
rn { envR :: VarEnv Var
envR = VarEnv Var
env forall a. VarEnv a -> [Var] -> VarEnv a
`delVarEnvList` [Var]
v, in_scope :: InScopeSet
in_scope = InScopeSet
in_scope InScopeSet -> [Var] -> InScopeSet
`extendInScopeSetList` [Var]
v }
rnOccL, rnOccR :: RnEnv2 -> Var -> Var
rnOccL :: RnEnv2 -> Var -> Var
rnOccL (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
env }) Var
v = forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv Var
env Var
v forall a. Maybe a -> a -> a
`orElse` Var
v
rnOccR :: RnEnv2 -> Var -> Var
rnOccR (RV2 { envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
env }) Var
v = forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv Var
env Var
v forall a. Maybe a -> a -> a
`orElse` Var
v
rnOccL_maybe, rnOccR_maybe :: RnEnv2 -> Var -> Maybe Var
rnOccL_maybe :: RnEnv2 -> Var -> Maybe Var
rnOccL_maybe (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
env }) Var
v = forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv Var
env Var
v
rnOccR_maybe :: RnEnv2 -> Var -> Maybe Var
rnOccR_maybe (RV2 { envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
env }) Var
v = forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv Var
env Var
v
inRnEnvL, inRnEnvR :: RnEnv2 -> Var -> Bool
inRnEnvL :: RnEnv2 -> Var -> Bool
inRnEnvL (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
env }) Var
v = Var
v forall a. Var -> VarEnv a -> Bool
`elemVarEnv` VarEnv Var
env
inRnEnvR :: RnEnv2 -> Var -> Bool
inRnEnvR (RV2 { envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
env }) Var
v = Var
v forall a. Var -> VarEnv a -> Bool
`elemVarEnv` VarEnv Var
env
lookupRnInScope :: RnEnv2 -> Var -> Var
lookupRnInScope :: RnEnv2 -> Var -> Var
lookupRnInScope RnEnv2
env Var
v = InScopeSet -> Var -> Maybe Var
lookupInScope (RnEnv2 -> InScopeSet
in_scope RnEnv2
env) Var
v forall a. Maybe a -> a -> a
`orElse` Var
v
nukeRnEnvL, nukeRnEnvR :: RnEnv2 -> RnEnv2
nukeRnEnvL :: RnEnv2 -> RnEnv2
nukeRnEnvL RnEnv2
env = RnEnv2
env { envL :: VarEnv Var
envL = forall a. VarEnv a
emptyVarEnv }
nukeRnEnvR :: RnEnv2 -> RnEnv2
nukeRnEnvR RnEnv2
env = RnEnv2
env { envR :: VarEnv Var
envR = forall a. VarEnv a
emptyVarEnv }
rnSwap :: RnEnv2 -> RnEnv2
rnSwap :: RnEnv2 -> RnEnv2
rnSwap (RV2 { envL :: RnEnv2 -> VarEnv Var
envL = VarEnv Var
envL, envR :: RnEnv2 -> VarEnv Var
envR = VarEnv Var
envR, in_scope :: RnEnv2 -> InScopeSet
in_scope = InScopeSet
in_scope })
= RV2 { envL :: VarEnv Var
envL = VarEnv Var
envR, envR :: VarEnv Var
envR = VarEnv Var
envL, in_scope :: InScopeSet
in_scope = InScopeSet
in_scope }
type TidyEnv = (TidyOccEnv, VarEnv Var)
emptyTidyEnv :: TidyEnv
emptyTidyEnv :: TidyEnv
emptyTidyEnv = (TidyOccEnv
emptyTidyOccEnv, forall a. VarEnv a
emptyVarEnv)
mkEmptyTidyEnv :: TidyOccEnv -> TidyEnv
mkEmptyTidyEnv :: TidyOccEnv -> TidyEnv
mkEmptyTidyEnv TidyOccEnv
occ_env = (TidyOccEnv
occ_env, forall a. VarEnv a
emptyVarEnv)
delTidyEnvList :: TidyEnv -> [Var] -> TidyEnv
delTidyEnvList :: TidyEnv -> [Var] -> TidyEnv
delTidyEnvList (TidyOccEnv
occ_env, VarEnv Var
var_env) [Var]
vs = (TidyOccEnv
occ_env', VarEnv Var
var_env')
where
occ_env' :: TidyOccEnv
occ_env' = TidyOccEnv
occ_env TidyOccEnv -> [FastString] -> TidyOccEnv
`delTidyOccEnvList` forall a b. (a -> b) -> [a] -> [b]
map (OccName -> FastString
occNameFS forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. NamedThing a => a -> OccName
getOccName) [Var]
vs
var_env' :: VarEnv Var
var_env' = VarEnv Var
var_env forall a. VarEnv a -> [Var] -> VarEnv a
`delVarEnvList` [Var]
vs
type VarEnv elt = UniqFM Var elt
type IdEnv elt = UniqFM Id elt
type TyVarEnv elt = UniqFM Var elt
type TyCoVarEnv elt = UniqFM TyCoVar elt
type CoVarEnv elt = UniqFM CoVar elt
emptyVarEnv :: VarEnv a
mkVarEnv :: [(Var, a)] -> VarEnv a
mkVarEnv_Directly :: [(Unique, a)] -> VarEnv a
zipVarEnv :: [Var] -> [a] -> VarEnv a
unitVarEnv :: Var -> a -> VarEnv a
alterVarEnv :: (Maybe a -> Maybe a) -> VarEnv a -> Var -> VarEnv a
extendVarEnv :: VarEnv a -> Var -> a -> VarEnv a
extendVarEnv_C :: (a->a->a) -> VarEnv a -> Var -> a -> VarEnv a
extendVarEnv_Acc :: (a->b->b) -> (a->b) -> VarEnv b -> Var -> a -> VarEnv b
plusVarEnv :: VarEnv a -> VarEnv a -> VarEnv a
plusVarEnvList :: [VarEnv a] -> VarEnv a
extendVarEnvList :: VarEnv a -> [(Var, a)] -> VarEnv a
partitionVarEnv :: (a -> Bool) -> VarEnv a -> (VarEnv a, VarEnv a)
restrictVarEnv :: VarEnv a -> VarSet -> VarEnv a
delVarEnvList :: VarEnv a -> [Var] -> VarEnv a
delVarEnv :: VarEnv a -> Var -> VarEnv a
minusVarEnv :: VarEnv a -> VarEnv b -> VarEnv a
plusVarEnv_C :: (a -> a -> a) -> VarEnv a -> VarEnv a -> VarEnv a
plusVarEnv_CD :: (a -> a -> a) -> VarEnv a -> a -> VarEnv a -> a -> VarEnv a
plusMaybeVarEnv_C :: (a -> a -> Maybe a) -> VarEnv a -> VarEnv a -> VarEnv a
mapVarEnv :: (a -> b) -> VarEnv a -> VarEnv b
modifyVarEnv :: (a -> a) -> VarEnv a -> Var -> VarEnv a
isEmptyVarEnv :: VarEnv a -> Bool
lookupVarEnv :: VarEnv a -> Var -> Maybe a
filterVarEnv :: (a -> Bool) -> VarEnv a -> VarEnv a
lookupVarEnv_NF :: VarEnv a -> Var -> a
lookupWithDefaultVarEnv :: VarEnv a -> a -> Var -> a
elemVarEnv :: Var -> VarEnv a -> Bool
elemVarEnvByKey :: Unique -> VarEnv a -> Bool
disjointVarEnv :: VarEnv a -> VarEnv a -> Bool
elemVarEnv :: forall a. Var -> VarEnv a -> Bool
elemVarEnv = forall key elt. Uniquable key => key -> UniqFM key elt -> Bool
elemUFM
elemVarEnvByKey :: forall a. Unique -> VarEnv a -> Bool
elemVarEnvByKey = forall key elt. Unique -> UniqFM key elt -> Bool
elemUFM_Directly
disjointVarEnv :: forall a. VarEnv a -> VarEnv a -> Bool
disjointVarEnv = forall key elt1 elt2. UniqFM key elt1 -> UniqFM key elt2 -> Bool
disjointUFM
alterVarEnv :: forall a. (Maybe a -> Maybe a) -> VarEnv a -> Var -> VarEnv a
alterVarEnv = forall key elt.
Uniquable key =>
(Maybe elt -> Maybe elt) -> UniqFM key elt -> key -> UniqFM key elt
alterUFM
extendVarEnv :: forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv = forall key elt.
Uniquable key =>
UniqFM key elt -> key -> elt -> UniqFM key elt
addToUFM
extendVarEnv_C :: forall a. (a -> a -> a) -> VarEnv a -> Var -> a -> VarEnv a
extendVarEnv_C = forall key elt.
Uniquable key =>
(elt -> elt -> elt)
-> UniqFM key elt -> key -> elt -> UniqFM key elt
addToUFM_C
extendVarEnv_Acc :: forall a b.
(a -> b -> b) -> (a -> b) -> VarEnv b -> Var -> a -> VarEnv b
extendVarEnv_Acc = forall key elt elts.
Uniquable key =>
(elt -> elts -> elts)
-> (elt -> elts)
-> UniqFM key elts
-> key
-> elt
-> UniqFM key elts
addToUFM_Acc
extendVarEnvList :: forall a. VarEnv a -> [(Var, a)] -> VarEnv a
extendVarEnvList = forall key elt.
Uniquable key =>
UniqFM key elt -> [(key, elt)] -> UniqFM key elt
addListToUFM
plusVarEnv_C :: forall a. (a -> a -> a) -> VarEnv a -> VarEnv a -> VarEnv a
plusVarEnv_C = forall elt key.
(elt -> elt -> elt)
-> UniqFM key elt -> UniqFM key elt -> UniqFM key elt
plusUFM_C
plusVarEnv_CD :: forall a.
(a -> a -> a) -> VarEnv a -> a -> VarEnv a -> a -> VarEnv a
plusVarEnv_CD = forall elta eltb eltc key.
(elta -> eltb -> eltc)
-> UniqFM key elta
-> elta
-> UniqFM key eltb
-> eltb
-> UniqFM key eltc
plusUFM_CD
plusMaybeVarEnv_C :: forall a. (a -> a -> Maybe a) -> VarEnv a -> VarEnv a -> VarEnv a
plusMaybeVarEnv_C = forall elt key.
(elt -> elt -> Maybe elt)
-> UniqFM key elt -> UniqFM key elt -> UniqFM key elt
plusMaybeUFM_C
delVarEnvList :: forall a. VarEnv a -> [Var] -> VarEnv a
delVarEnvList = forall key elt.
Uniquable key =>
UniqFM key elt -> [key] -> UniqFM key elt
delListFromUFM
delVarEnv :: forall a. VarEnv a -> Var -> VarEnv a
delVarEnv = forall key elt.
Uniquable key =>
UniqFM key elt -> key -> UniqFM key elt
delFromUFM
minusVarEnv :: forall a b. VarEnv a -> VarEnv b -> VarEnv a
minusVarEnv = forall key elt1 elt2.
UniqFM key elt1 -> UniqFM key elt2 -> UniqFM key elt1
minusUFM
plusVarEnv :: forall a. VarEnv a -> VarEnv a -> VarEnv a
plusVarEnv = forall key elt. UniqFM key elt -> UniqFM key elt -> UniqFM key elt
plusUFM
plusVarEnvList :: forall a. [VarEnv a] -> VarEnv a
plusVarEnvList = forall key elt. [UniqFM key elt] -> UniqFM key elt
plusUFMList
lookupVarEnv :: forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv = forall key elt. Uniquable key => UniqFM key elt -> key -> Maybe elt
lookupUFM
filterVarEnv :: forall a. (a -> Bool) -> VarEnv a -> VarEnv a
filterVarEnv = forall elt key. (elt -> Bool) -> UniqFM key elt -> UniqFM key elt
filterUFM
lookupWithDefaultVarEnv :: forall a. VarEnv a -> a -> Var -> a
lookupWithDefaultVarEnv = forall key elt.
Uniquable key =>
UniqFM key elt -> elt -> key -> elt
lookupWithDefaultUFM
mapVarEnv :: forall a b. (a -> b) -> VarEnv a -> VarEnv b
mapVarEnv = forall elt1 elt2 key.
(elt1 -> elt2) -> UniqFM key elt1 -> UniqFM key elt2
mapUFM
mkVarEnv :: forall a. [(Var, a)] -> VarEnv a
mkVarEnv = forall key elt. Uniquable key => [(key, elt)] -> UniqFM key elt
listToUFM
mkVarEnv_Directly :: forall a. [(Unique, a)] -> VarEnv a
mkVarEnv_Directly= forall elt key. [(Unique, elt)] -> UniqFM key elt
listToUFM_Directly
emptyVarEnv :: forall a. VarEnv a
emptyVarEnv = forall key elt. UniqFM key elt
emptyUFM
unitVarEnv :: forall a. Var -> a -> VarEnv a
unitVarEnv = forall key elt. Uniquable key => key -> elt -> UniqFM key elt
unitUFM
isEmptyVarEnv :: forall a. VarEnv a -> Bool
isEmptyVarEnv = forall key elt. UniqFM key elt -> Bool
isNullUFM
partitionVarEnv :: forall a. (a -> Bool) -> VarEnv a -> (VarEnv a, VarEnv a)
partitionVarEnv = forall elt key.
(elt -> Bool) -> UniqFM key elt -> (UniqFM key elt, UniqFM key elt)
partitionUFM
restrictVarEnv :: forall a. VarEnv a -> VarSet -> VarEnv a
restrictVarEnv VarEnv a
env VarSet
vs = forall elt key.
(Unique -> elt -> Bool) -> UniqFM key elt -> UniqFM key elt
filterUFM_Directly Unique -> a -> Bool
keep VarEnv a
env
where
keep :: Unique -> a -> Bool
keep Unique
u a
_ = Unique
u Unique -> VarSet -> Bool
`elemVarSetByKey` VarSet
vs
zipVarEnv :: forall a. [Var] -> [a] -> VarEnv a
zipVarEnv [Var]
tyvars [a]
tys = forall a. [(Var, a)] -> VarEnv a
mkVarEnv (forall a b. String -> [a] -> [b] -> [(a, b)]
zipEqual String
"zipVarEnv" [Var]
tyvars [a]
tys)
lookupVarEnv_NF :: forall a. VarEnv a -> Var -> a
lookupVarEnv_NF VarEnv a
env Var
id = case forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv a
env Var
id of
Just a
xx -> a
xx
Maybe a
Nothing -> forall a. String -> a
panic String
"lookupVarEnv_NF: Nothing"
modifyVarEnv :: forall a. (a -> a) -> VarEnv a -> Var -> VarEnv a
modifyVarEnv a -> a
mangle_fn VarEnv a
env Var
key
= case (forall a. VarEnv a -> Var -> Maybe a
lookupVarEnv VarEnv a
env Var
key) of
Maybe a
Nothing -> VarEnv a
env
Just a
xx -> forall a. VarEnv a -> Var -> a -> VarEnv a
extendVarEnv VarEnv a
env Var
key (a -> a
mangle_fn a
xx)
modifyVarEnv_Directly :: (a -> a) -> UniqFM key a -> Unique -> UniqFM key a
modifyVarEnv_Directly :: forall a key. (a -> a) -> UniqFM key a -> Unique -> UniqFM key a
modifyVarEnv_Directly a -> a
mangle_fn UniqFM key a
env Unique
key
= case (forall key elt. UniqFM key elt -> Unique -> Maybe elt
lookupUFM_Directly UniqFM key a
env Unique
key) of
Maybe a
Nothing -> UniqFM key a
env
Just a
xx -> forall key elt. UniqFM key elt -> Unique -> elt -> UniqFM key elt
addToUFM_Directly UniqFM key a
env Unique
key (a -> a
mangle_fn a
xx)
type DVarEnv elt = UniqDFM Var elt
type DIdEnv elt = UniqDFM Var elt
type DTyVarEnv elt = UniqDFM TyVar elt
emptyDVarEnv :: DVarEnv a
emptyDVarEnv :: forall a. DVarEnv a
emptyDVarEnv = forall key elt. UniqDFM key elt
emptyUDFM
dVarEnvElts :: DVarEnv a -> [a]
dVarEnvElts :: forall a. DVarEnv a -> [a]
dVarEnvElts = forall key elt. UniqDFM key elt -> [elt]
eltsUDFM
mkDVarEnv :: [(Var, a)] -> DVarEnv a
mkDVarEnv :: forall a. [(Var, a)] -> DVarEnv a
mkDVarEnv = forall key elt. Uniquable key => [(key, elt)] -> UniqDFM key elt
listToUDFM
extendDVarEnv :: DVarEnv a -> Var -> a -> DVarEnv a
extendDVarEnv :: forall a. DVarEnv a -> Var -> a -> DVarEnv a
extendDVarEnv = forall key elt.
Uniquable key =>
UniqDFM key elt -> key -> elt -> UniqDFM key elt
addToUDFM
minusDVarEnv :: DVarEnv a -> DVarEnv a' -> DVarEnv a
minusDVarEnv :: forall a a'. DVarEnv a -> DVarEnv a' -> DVarEnv a
minusDVarEnv = forall key elt1 elt2.
UniqDFM key elt1 -> UniqDFM key elt2 -> UniqDFM key elt1
minusUDFM
lookupDVarEnv :: DVarEnv a -> Var -> Maybe a
lookupDVarEnv :: forall a. DVarEnv a -> Var -> Maybe a
lookupDVarEnv = forall key elt.
Uniquable key =>
UniqDFM key elt -> key -> Maybe elt
lookupUDFM
foldDVarEnv :: (a -> b -> b) -> b -> DVarEnv a -> b
foldDVarEnv :: forall a b. (a -> b -> b) -> b -> DVarEnv a -> b
foldDVarEnv = forall elt a key. (elt -> a -> a) -> a -> UniqDFM key elt -> a
foldUDFM
nonDetStrictFoldDVarEnv :: (a -> b -> b) -> b -> DVarEnv a -> b
nonDetStrictFoldDVarEnv :: forall a b. (a -> b -> b) -> b -> DVarEnv a -> b
nonDetStrictFoldDVarEnv = forall elt a key. (elt -> a -> a) -> a -> UniqDFM key elt -> a
nonDetStrictFoldUDFM
mapDVarEnv :: (a -> b) -> DVarEnv a -> DVarEnv b
mapDVarEnv :: forall a b. (a -> b) -> DVarEnv a -> DVarEnv b
mapDVarEnv = forall elt1 elt2 key.
(elt1 -> elt2) -> UniqDFM key elt1 -> UniqDFM key elt2
mapUDFM
filterDVarEnv :: (a -> Bool) -> DVarEnv a -> DVarEnv a
filterDVarEnv :: forall a. (a -> Bool) -> DVarEnv a -> DVarEnv a
filterDVarEnv = forall elt key. (elt -> Bool) -> UniqDFM key elt -> UniqDFM key elt
filterUDFM
alterDVarEnv :: (Maybe a -> Maybe a) -> DVarEnv a -> Var -> DVarEnv a
alterDVarEnv :: forall a. (Maybe a -> Maybe a) -> DVarEnv a -> Var -> DVarEnv a
alterDVarEnv = forall key elt.
Uniquable key =>
(Maybe elt -> Maybe elt)
-> UniqDFM key elt -> key -> UniqDFM key elt
alterUDFM
plusDVarEnv :: DVarEnv a -> DVarEnv a -> DVarEnv a
plusDVarEnv :: forall a. DVarEnv a -> DVarEnv a -> DVarEnv a
plusDVarEnv = forall key elt.
UniqDFM key elt -> UniqDFM key elt -> UniqDFM key elt
plusUDFM
plusDVarEnv_C :: (a -> a -> a) -> DVarEnv a -> DVarEnv a -> DVarEnv a
plusDVarEnv_C :: forall a. (a -> a -> a) -> DVarEnv a -> DVarEnv a -> DVarEnv a
plusDVarEnv_C = forall elt key.
(elt -> elt -> elt)
-> UniqDFM key elt -> UniqDFM key elt -> UniqDFM key elt
plusUDFM_C
unitDVarEnv :: Var -> a -> DVarEnv a
unitDVarEnv :: forall a. Var -> a -> DVarEnv a
unitDVarEnv = forall key elt. Uniquable key => key -> elt -> UniqDFM key elt
unitUDFM
delDVarEnv :: DVarEnv a -> Var -> DVarEnv a
delDVarEnv :: forall a. DVarEnv a -> Var -> DVarEnv a
delDVarEnv = forall key elt.
Uniquable key =>
UniqDFM key elt -> key -> UniqDFM key elt
delFromUDFM
delDVarEnvList :: DVarEnv a -> [Var] -> DVarEnv a
delDVarEnvList :: forall a. DVarEnv a -> [Var] -> DVarEnv a
delDVarEnvList = forall key elt.
Uniquable key =>
UniqDFM key elt -> [key] -> UniqDFM key elt
delListFromUDFM
isEmptyDVarEnv :: DVarEnv a -> Bool
isEmptyDVarEnv :: forall a. DVarEnv a -> Bool
isEmptyDVarEnv = forall key elt. UniqDFM key elt -> Bool
isNullUDFM
elemDVarEnv :: Var -> DVarEnv a -> Bool
elemDVarEnv :: forall a. Var -> DVarEnv a -> Bool
elemDVarEnv = forall key elt. Uniquable key => key -> UniqDFM key elt -> Bool
elemUDFM
extendDVarEnv_C :: (a -> a -> a) -> DVarEnv a -> Var -> a -> DVarEnv a
extendDVarEnv_C :: forall a. (a -> a -> a) -> DVarEnv a -> Var -> a -> DVarEnv a
extendDVarEnv_C = forall key elt.
Uniquable key =>
(elt -> elt -> elt)
-> UniqDFM key elt -> key -> elt -> UniqDFM key elt
addToUDFM_C
modifyDVarEnv :: (a -> a) -> DVarEnv a -> Var -> DVarEnv a
modifyDVarEnv :: forall a. (a -> a) -> DVarEnv a -> Var -> DVarEnv a
modifyDVarEnv a -> a
mangle_fn DVarEnv a
env Var
key
= case (forall a. DVarEnv a -> Var -> Maybe a
lookupDVarEnv DVarEnv a
env Var
key) of
Maybe a
Nothing -> DVarEnv a
env
Just a
xx -> forall a. DVarEnv a -> Var -> a -> DVarEnv a
extendDVarEnv DVarEnv a
env Var
key (a -> a
mangle_fn a
xx)
partitionDVarEnv :: (a -> Bool) -> DVarEnv a -> (DVarEnv a, DVarEnv a)
partitionDVarEnv :: forall a. (a -> Bool) -> DVarEnv a -> (DVarEnv a, DVarEnv a)
partitionDVarEnv = forall elt key.
(elt -> Bool)
-> UniqDFM key elt -> (UniqDFM key elt, UniqDFM key elt)
partitionUDFM
extendDVarEnvList :: DVarEnv a -> [(Var, a)] -> DVarEnv a
extendDVarEnvList :: forall a. DVarEnv a -> [(Var, a)] -> DVarEnv a
extendDVarEnvList = forall key elt.
Uniquable key =>
UniqDFM key elt -> [(key, elt)] -> UniqDFM key elt
addListToUDFM
anyDVarEnv :: (a -> Bool) -> DVarEnv a -> Bool
anyDVarEnv :: forall a. (a -> Bool) -> DVarEnv a -> Bool
anyDVarEnv = forall elt key. (elt -> Bool) -> UniqDFM key elt -> Bool
anyUDFM