Implement expression_to_string

1 files changed, 48 insertions, 0 deletions

diff --git a/src/Print.ml b/src/Print.ml
index 24e038f9..441e76e6 100644
--- a/src/Print.ml
+++ b/src/Print.ml
@@ -668,4 +668,52 @@ module CfimAst = struct
     | A.Break i -> "break " ^ string_of_int i
     | A.Continue i -> "continue " ^ string_of_int i
     | A.Nop -> "nop"
+
+  let rec expression_to_string (fmt : ast_formatter) (indent : string)
+      (indent_incr : string) (e : A.expression) : string =
+    match e with
+    | A.Statement st -> indent ^ statement_to_string fmt st ^ ";"
+    | A.Sequence (e1, e2) ->
+        expression_to_string fmt indent indent_incr e1
+        ^ "\n"
+        ^ expression_to_string fmt indent indent_incr e2
+    | A.Switch (op, tgts) -> (
+        let op = operand_to_string fmt op in
+        match tgts with
+        | A.If (true_e, false_e) ->
+            let inner_indent = indent ^ indent_incr in
+            let inner_to_string =
+              expression_to_string fmt inner_indent indent_incr
+            in
+            let true_e = inner_to_string true_e in
+            let false_e = inner_to_string false_e in
+            indent ^ "if (" ^ op ^ ") {\n" ^ true_e ^ "\n" ^ indent ^ "}\n"
+            ^ indent ^ "else {\n" ^ false_e ^ "\n" ^ indent ^ "}"
+        | A.SwitchInt (_ty, branches, otherwise) ->
+            let indent1 = indent ^ indent_incr in
+            let indent2 = indent1 ^ indent_incr in
+            let inner_to_string1 =
+              expression_to_string fmt indent1 indent_incr
+            in
+            let inner_to_string2 =
+              expression_to_string fmt indent2 indent_incr
+            in
+            let branches =
+              List.map
+                (fun (sv, be) ->
+                  indent1
+                  ^ PV.scalar_value_to_string sv
+                  ^ " => {\n" ^ inner_to_string2 be ^ "\n" ^ indent1 ^ "}")
+                branches
+            in
+            let branches = String.concat "\n" branches in
+            let branches =
+              branches ^ "\n" ^ indent1 ^ "_ => {\n"
+              ^ inner_to_string2 otherwise ^ "\n" ^ indent1 ^ "}"
+            in
+            indent ^ "switch (" ^ op ^ ") {\n" ^ branches ^ "\n" ^ indent ^ "}")
+    | A.Loop loop_e ->
+        indent ^ "loop {\n"
+        ^ expression_to_string fmt (indent ^ indent_incr) indent_incr loop_e
+        ^ "\n" ^ indent ^ "}"
 end