diff options
Diffstat (limited to 'discocaml/draw_tree.ml')
| -rw-r--r-- | discocaml/draw_tree.ml | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/discocaml/draw_tree.ml b/discocaml/draw_tree.ml index 6ea4ff2..e04a59c 100644 --- a/discocaml/draw_tree.ml +++ b/discocaml/draw_tree.ml @@ -9,9 +9,9 @@ let add_node (fmt : Format.formatter) (i : expr index) (expr : expr) : unit = i.index | Int n -> Format.fprintf fmt " expr%d [label=\"%d\"];\n" i.index n | Lam (_, _) -> Format.fprintf fmt " expr%d [label=\"λ\"];\n" i.index - | Prim (`Add, _) -> Format.fprintf fmt " expr%d [label=\"+\"];\n" i.index - | Prim (`Sub, _) -> Format.fprintf fmt " expr%d [label=\"-\"];\n" i.index - | Prim (`Mul, _) -> Format.fprintf fmt " expr%d [label=\"*\"];\n" i.index + | Prim (Add, _) -> Format.fprintf fmt " expr%d [label=\"+\"];\n" i.index + | Prim (Sub, _) -> Format.fprintf fmt " expr%d [label=\"-\"];\n" i.index + | Prim (Mul, _) -> Format.fprintf fmt " expr%d [label=\"*\"];\n" i.index | Var n -> Format.fprintf fmt " expr%d [label=%S];\n" i.index n let add_expr_edges (ast : 'a ast) (fmt : Format.formatter) @@ -31,7 +31,15 @@ let add_expr_edges (ast : 'a ast) (fmt : Format.formatter) Format.fprintf fmt " expr%d -> expr%d_var;\n" i.index i.index; Format.fprintf fmt " expr%d_var [label=%S];\n" i.index x; edge_to b - | Prim (_, xs) -> Array.iter edge_to xs + | Prim (Add, (l, r)) -> + edge_to l; + edge_to r + | Prim (Sub, (l, r)) -> + edge_to l; + edge_to r + | Prim (Mul, (l, r)) -> + edge_to l; + edge_to r | Var _ -> () in loop |