diff options
| author | Edward Z. Yang <ezyang@mit.edu> | 2011-04-11 16:04:14 +0100 |
|---|---|---|
| committer | Edward Z. Yang <ezyang@mit.edu> | 2011-04-11 17:49:23 +0100 |
| commit | 856448e1809237ef76e66691f71fbb6a7bd62b54 (patch) | |
| tree | c0d9c4100ff211cfc9e15343c158ecadce4e0c87 | |
| parent | 8a0ab97b1daefb57b53d6cf08a01bd597d09e32d (diff) | |
| download | haskell-856448e1809237ef76e66691f71fbb6a7bd62b54.tar.gz | |
Implement mapUFM_Directly.
Signed-off-by: Edward Z. Yang <ezyang@mit.edu>
| -rw-r--r-- | compiler/utils/UniqFM.lhs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/compiler/utils/UniqFM.lhs b/compiler/utils/UniqFM.lhs index 293e48ed14..31d1e878c6 100644 --- a/compiler/utils/UniqFM.lhs +++ b/compiler/utils/UniqFM.lhs @@ -45,7 +45,7 @@ module UniqFM ( intersectUFM, intersectUFM_C, foldUFM, foldUFM_Directly, - mapUFM, + mapUFM, mapUFM_Directly, elemUFM, elemUFM_Directly, filterUFM, filterUFM_Directly, sizeUFM, @@ -122,6 +122,7 @@ intersectUFM_C :: (elt1 -> elt2 -> elt3) foldUFM :: (elt -> a -> a) -> a -> UniqFM elt -> a foldUFM_Directly:: (Unique -> elt -> a -> a) -> a -> UniqFM elt -> a mapUFM :: (elt1 -> elt2) -> UniqFM elt1 -> UniqFM elt2 +mapUFM_Directly :: (Unique -> elt1 -> elt2) -> UniqFM elt1 -> UniqFM elt2 filterUFM :: (elt -> Bool) -> UniqFM elt -> UniqFM elt filterUFM_Directly :: (Unique -> elt -> Bool) -> UniqFM elt -> UniqFM elt @@ -188,6 +189,7 @@ intersectUFM_C f (UFM x) (UFM y) = UFM (M.intersectionWith f x y) foldUFM k z (UFM m) = M.fold k z m foldUFM_Directly k z (UFM m) = M.foldWithKey (k . getUnique) z m mapUFM f (UFM m) = UFM (M.map f m) +mapUFM_Directly f (UFM m) = UFM (M.mapWithKey (f . getUnique) m) filterUFM p (UFM m) = UFM (M.filter p m) filterUFM_Directly p (UFM m) = UFM (M.filterWithKey (p . getUnique) m) |