X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=ghc%2Fcompiler%2FbasicTypes%2FVarSet.lhs;h=55e82a8515ae17871946e676343a1ed94dbeb249;hb=f451d5c104dcd21712ccd8fc5af02246d4894f83;hp=277c5d3c4e819cf72cc50d5e57bf8928b5850c69;hpb=451a8613203721d344e26eb043e8af827c58cd7b;p=ghc-hetmet.git diff --git a/ghc/compiler/basicTypes/VarSet.lhs b/ghc/compiler/basicTypes/VarSet.lhs index 277c5d3..55e82a8 100644 --- a/ghc/compiler/basicTypes/VarSet.lhs +++ b/ghc/compiler/basicTypes/VarSet.lhs @@ -5,27 +5,24 @@ \begin{code} module VarSet ( - VarSet, IdSet, TyVarSet, IdOrTyVarSet, + VarSet, IdSet, TyVarSet, emptyVarSet, unitVarSet, mkVarSet, - extendVarSet, + extendVarSet, extendVarSetList, extendVarSet_C, elemVarSet, varSetElems, subVarSet, unionVarSet, unionVarSets, intersectVarSet, intersectsVarSet, - isEmptyVarSet, delVarSet, delVarSetByKey, + isEmptyVarSet, delVarSet, delVarSetList, delVarSetByKey, minusVarSet, foldVarSet, filterVarSet, - lookupVarSet, mapVarSet, - - uniqAway + lookupVarSet, mapVarSet, sizeVarSet, seqVarSet, + elemVarSetByKey ) where #include "HsVersions.h" -import CmdLineOpts ( opt_PprStyle_Debug ) -import Var ( Var, Id, TyVar, IdOrTyVar, setVarUnique ) -import Unique ( Unique, Uniquable(..), incrUnique, deriveUnique ) +import Var ( Var, Id, TyVar ) +import Unique ( Unique ) import UniqSet -import UniqFM ( delFromUFM_Directly ) -import Outputable +import UniqFM ( delFromUFM_Directly, addToUFM_C ) \end{code} %************************************************************************ @@ -38,18 +35,18 @@ import Outputable type VarSet = UniqSet Var type IdSet = UniqSet Id type TyVarSet = UniqSet TyVar -type IdOrTyVarSet = UniqSet IdOrTyVar emptyVarSet :: VarSet intersectVarSet :: VarSet -> VarSet -> VarSet -intersectsVarSet:: VarSet -> VarSet -> Bool -- True if non-empty intersection unionVarSet :: VarSet -> VarSet -> VarSet unionVarSets :: [VarSet] -> VarSet varSetElems :: VarSet -> [Var] unitVarSet :: Var -> VarSet extendVarSet :: VarSet -> Var -> VarSet +extendVarSetList:: VarSet -> [Var] -> VarSet elemVarSet :: Var -> VarSet -> Bool delVarSet :: VarSet -> Var -> VarSet +delVarSetList :: VarSet -> [Var] -> VarSet minusVarSet :: VarSet -> VarSet -> VarSet isEmptyVarSet :: VarSet -> Bool mkVarSet :: [Var] -> VarSet @@ -58,46 +55,51 @@ lookupVarSet :: VarSet -> Var -> Maybe Var -- Returns the set element, which may be -- (==) to the argument, but not the same as mapVarSet :: (Var -> Var) -> VarSet -> VarSet +sizeVarSet :: VarSet -> Int filterVarSet :: (Var -> Bool) -> VarSet -> VarSet -subVarSet :: VarSet -> VarSet -> Bool +extendVarSet_C :: (Var->Var->Var) -> VarSet -> Var -> VarSet delVarSetByKey :: VarSet -> Unique -> VarSet +elemVarSetByKey :: Unique -> VarSet -> Bool emptyVarSet = emptyUniqSet unitVarSet = unitUniqSet extendVarSet = addOneToUniqSet +extendVarSetList= addListToUniqSet intersectVarSet = intersectUniqSets -intersectsVarSet s1 s2 = not (isEmptyVarSet (s1 `intersectVarSet` s2)) + +intersectsVarSet:: VarSet -> VarSet -> Bool -- True if non-empty intersection + -- (s1 `intersectsVarSet` s2) doesn't compute s2 if s1 is empty +subVarSet :: VarSet -> VarSet -> Bool -- True if first arg is subset of second + -- (s1 `subVarSet` s2) doesn't compute s2 if s1 is empty + unionVarSet = unionUniqSets unionVarSets = unionManyUniqSets varSetElems = uniqSetToList elemVarSet = elementOfUniqSet minusVarSet = minusUniqSet delVarSet = delOneFromUniqSet +delVarSetList = delListFromUniqSet isEmptyVarSet = isEmptyUniqSet mkVarSet = mkUniqSet foldVarSet = foldUniqSet lookupVarSet = lookupUniqSet mapVarSet = mapUniqSet +sizeVarSet = sizeUniqSet filterVarSet = filterUniqSet -a `subVarSet` b = isEmptyVarSet (a `minusVarSet` b) +extendVarSet_C combine s x = addToUFM_C combine s x x delVarSetByKey = delFromUFM_Directly -- Can't be bothered to add this to UniqSet +elemVarSetByKey = elemUniqSet_Directly \end{code} \begin{code} -uniqAway :: VarSet -> Var -> Var --- Give the Var a new unique, different to any in the VarSet -uniqAway set var - | not (var `elemVarSet` set) = var -- Nothing to do +-- See comments with type signatures +intersectsVarSet s1 s2 = not (isEmptyVarSet (s1 `intersectVarSet` s2)) +a `subVarSet` b = isEmptyVarSet (a `minusVarSet` b) +\end{code} - | otherwise - = try 1 (deriveUnique (getUnique var) (hashUniqSet set)) - where - try n uniq | uniq `elemUniqSet_Directly` set = try ((n+1)::Int) (incrUnique uniq) -#ifdef DEBUG - | opt_PprStyle_Debug && n > 3 - = pprTrace "uniqAway:" (ppr n <+> text "tries" <+> ppr var) - setVarUnique var uniq -#endif - | otherwise = setVarUnique var uniq +\begin{code} +seqVarSet :: VarSet -> () +seqVarSet s = sizeVarSet s `seq` () \end{code} +