2 % (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
4 \section{@VarSet@: Variable sets}
8 VarSet, IdSet, TyVarSet, IdOrTyVarSet,
9 emptyVarSet, unitVarSet, mkVarSet,
11 elemVarSet, varSetElems, subVarSet,
12 unionVarSet, unionVarSets,
13 intersectVarSet, intersectsVarSet,
14 isEmptyVarSet, delVarSet, delVarSetByKey,
15 minusVarSet, foldVarSet, filterVarSet,
16 lookupVarSet, mapVarSet,
21 #include "HsVersions.h"
23 import CmdLineOpts ( opt_PprStyle_Debug )
24 import Var ( Var, Id, TyVar, IdOrTyVar, setVarUnique )
25 import Unique ( Unique, Uniquable(..), incrUnique )
27 import UniqFM ( delFromUFM_Directly )
31 %************************************************************************
33 \subsection{@VarSet@s}
35 %************************************************************************
38 type VarSet = UniqSet Var
39 type IdSet = UniqSet Id
40 type TyVarSet = UniqSet TyVar
41 type IdOrTyVarSet = UniqSet IdOrTyVar
44 intersectVarSet :: VarSet -> VarSet -> VarSet
45 intersectsVarSet:: VarSet -> VarSet -> Bool -- True if non-empty intersection
46 unionVarSet :: VarSet -> VarSet -> VarSet
47 unionVarSets :: [VarSet] -> VarSet
48 varSetElems :: VarSet -> [Var]
49 unitVarSet :: Var -> VarSet
50 extendVarSet :: VarSet -> Var -> VarSet
51 elemVarSet :: Var -> VarSet -> Bool
52 delVarSet :: VarSet -> Var -> VarSet
53 minusVarSet :: VarSet -> VarSet -> VarSet
54 isEmptyVarSet :: VarSet -> Bool
55 mkVarSet :: [Var] -> VarSet
56 foldVarSet :: (Var -> a -> a) -> a -> VarSet -> a
57 lookupVarSet :: VarSet -> Var -> Maybe Var
58 -- Returns the set element, which may be
59 -- (==) to the argument, but not the same as
60 mapVarSet :: (Var -> Var) -> VarSet -> VarSet
61 filterVarSet :: (Var -> Bool) -> VarSet -> VarSet
62 subVarSet :: VarSet -> VarSet -> Bool
64 delVarSetByKey :: VarSet -> Unique -> VarSet
66 emptyVarSet = emptyUniqSet
67 unitVarSet = unitUniqSet
68 extendVarSet = addOneToUniqSet
69 intersectVarSet = intersectUniqSets
70 intersectsVarSet s1 s2 = not (isEmptyVarSet (s1 `intersectVarSet` s2))
71 unionVarSet = unionUniqSets
72 unionVarSets = unionManyUniqSets
73 varSetElems = uniqSetToList
74 elemVarSet = elementOfUniqSet
75 minusVarSet = minusUniqSet
76 delVarSet = delOneFromUniqSet
77 isEmptyVarSet = isEmptyUniqSet
79 foldVarSet = foldUniqSet
80 lookupVarSet = lookupUniqSet
81 mapVarSet = mapUniqSet
82 filterVarSet = filterUniqSet
83 a `subVarSet` b = isEmptyVarSet (a `minusVarSet` b)
84 delVarSetByKey = delFromUFM_Directly -- Can't be bothered to add this to UniqSet
88 uniqAway :: VarSet -> Var -> Var
89 -- Give the Var a new unique, different to any in the VarSet
91 | not (var `elemVarSet` set) = var -- Nothing to do
94 = try 1 (incrUnique (getUnique var))
96 try n uniq | uniq `elemUniqSet_Directly` set = try ((n+1)::Int) (incrUnique uniq)
98 | opt_PprStyle_Debug && n > 3
99 = pprTrace "uniqAway:" (ppr n <+> text "tries" <+> ppr var)
100 setVarUnique var uniq
102 | otherwise = setVarUnique var uniq