@@ -1,8 +1,9 @@
11module Replace (replace, depth) where
22
33import Ast
4−import Data.Set (singleton,empty,unions,member)
4+import Data.Set (singleton,empty,unions,member, Set)
55
6+replace :: String -> Expr -> Expr -> Expr
67replace y v expr =
78 let f = replace y v in
89 case expr of
@@ -14,22 +15,25 @@ replace y v expr =
1415 Lift e es -> Lift (f e) (map f es)
1516 Fold e1 e2 e3 -> Fold (f e1) (f e2) (f e3)
1617 Async e -> Async (f e)
17− Let defs e -> if elem y vs then Let defs e else Let (zip vs (map f es)) (f e)
18+ Let defs e -> if y `elem` vs then Let defs e else Let (zip vs (map f es)) (f e)
1819 where (vs,es) = unzip defs
1920 Var x -> if x == y then v else Var x
2021 Data name es -> Data name (map f es)
21− Case e cases -> Case (f e) $ map (case_replace y v) cases
22+ Case e cases -> Case (f e) $ map (caseReplace y v) cases
2223 _ -> expr
2324
24−case_replace y v (p,e) =
25− if member y (pattern_vars p) then (p,e) else (p, replace y v e)
25+caseReplace :: String -> Expr -> (Pattern, Expr) -> (Pattern, Expr)
26+caseReplace y v (p,e) =
27+ if member y (patternVars p) then (p,e) else (p, replace y v e)
2628
27−pattern_vars pattern =
29+patternVars :: Pattern -> Set String
30+patternVars pattern =
2831 case pattern of
29− PData _ ps -> unions (map pattern_vars ps)
32+ PData _ ps -> unions (map patternVars ps)
3033 PVar x -> singleton x
3134 PAnything -> empty
3235
36+depth :: Expr -> Integer
3337depth = depth' 0
3438depth' d expr =
3539 let f = depth' (d+1) in