summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorSon Ho2022-01-28 15:10:40 +0100
committerSon Ho2022-01-28 15:10:40 +0100
commit32fac8c22454186aa6327474e63cde99958e3d4b (patch)
treed8c912be5b876f7cce12588b0c15ef3700dec896 /src
parentf75b76f8b1e7995fd7afa049e0cba4a1a6b4d52c (diff)
Implement the unit_vars_to_unit pass
Diffstat (limited to '')
-rw-r--r--src/PureMicroPasses.ml26
1 files changed, 23 insertions, 3 deletions
diff --git a/src/PureMicroPasses.ml b/src/PureMicroPasses.ml
index b3e65055..c75d9f9b 100644
--- a/src/PureMicroPasses.ml
+++ b/src/PureMicroPasses.ml
@@ -336,10 +336,30 @@ let to_monadic (def : fun_def) : fun_def =
{ def with signature }
(** Convert the unit variables to `()` if they are used as right-values or
- `_` if they are used as left values. *)
+ `_` if they are used as left values in patterns. *)
let unit_vars_to_unit (def : fun_def) : fun_def =
- (* TODO *)
- def
+ (* The map visitor *)
+ let obj =
+ object
+ inherit [_] map_expression as super
+
+ method! visit_var_or_dummy _ v =
+ match v with
+ | Dummy -> Dummy
+ | Var (v, mp) -> if v.ty = unit_ty then Dummy else Var (v, mp)
+ (** Replace in lvalues *)
+
+ method! visit_typed_rvalue env rv =
+ if rv.ty = unit_ty then unit_rvalue else super#visit_typed_rvalue env rv
+ (** Replace in rvalues *)
+ end
+ in
+ (* Update the body *)
+ let body = obj#visit_expression () def.body in
+ (* Update the input parameters *)
+ let inputs_lvs = List.map (obj#visit_typed_lvalue ()) def.inputs_lvs in
+ (* Return *)
+ { def with body; inputs_lvs }
(** Apply all the micro-passes to a function.