Updated Obc printer
This commit is contained in:
parent
6f2d5175e5
commit
db6344921a
|
@ -140,44 +140,7 @@ let lhs_of_exp = function
|
||||||
module Printer =
|
module Printer =
|
||||||
struct
|
struct
|
||||||
open Format
|
open Format
|
||||||
|
open Pp_tools
|
||||||
let rec print_list ff print sep = function
|
|
||||||
| [] -> ()
|
|
||||||
| [x] -> print ff x
|
|
||||||
| x :: l ->
|
|
||||||
print ff x;
|
|
||||||
fprintf ff "%s@ " sep;
|
|
||||||
print_list ff print sep l
|
|
||||||
|
|
||||||
(* Infix chars are surrounded by parenthesis *)
|
|
||||||
let is_infix =
|
|
||||||
let module StrSet = Set.Make(String) in
|
|
||||||
let set_infix =
|
|
||||||
List.fold_right
|
|
||||||
StrSet.add
|
|
||||||
["or"; "quo"; "mod"; "land"; "lor"; "lxor"; "lsl"; "lsr"; "asr"]
|
|
||||||
StrSet.empty in
|
|
||||||
fun s -> StrSet.mem s set_infix
|
|
||||||
|
|
||||||
let print_name ff s =
|
|
||||||
let c = String.get s 0 in
|
|
||||||
let s = if is_infix s then "(" ^ s ^ ")"
|
|
||||||
else match c with
|
|
||||||
| 'a' .. 'z' | 'A' .. 'Z' | '_' | '`' -> s
|
|
||||||
| '*' -> "( " ^ s ^ " )"
|
|
||||||
| _ -> if s = "()" then s else "(" ^ s ^ ")" in
|
|
||||||
fprintf ff "%s" s
|
|
||||||
|
|
||||||
let print_longname ff ln =
|
|
||||||
let ln = (* currentname ln*) ln in
|
|
||||||
match ln with
|
|
||||||
| Name(m) -> print_name ff m
|
|
||||||
| Modname({ qual = "Pervasives"; id = m }) -> print_name ff m
|
|
||||||
| Modname({ qual = m1; id = m2 }) ->
|
|
||||||
fprintf ff "%s." m1; print_name ff m2
|
|
||||||
|
|
||||||
let print_ident ff id =
|
|
||||||
fprintf ff "%s" (name id)
|
|
||||||
|
|
||||||
let rec print_type ff = function
|
let rec print_type ff = function
|
||||||
| Tint -> fprintf ff "int"
|
| Tint -> fprintf ff "int"
|
||||||
|
@ -185,8 +148,8 @@ struct
|
||||||
| Tbool -> fprintf ff "bool"
|
| Tbool -> fprintf ff "bool"
|
||||||
| Tid(id) -> print_longname ff id
|
| Tid(id) -> print_longname ff id
|
||||||
| Tarray(ty, n) ->
|
| Tarray(ty, n) ->
|
||||||
print_type ff ty;
|
print_type ff ty;
|
||||||
fprintf ff "^%d" n
|
fprintf ff "^%d" n
|
||||||
|
|
||||||
let print_vd ff vd =
|
let print_vd ff vd =
|
||||||
fprintf ff "@[<v>";
|
fprintf ff "@[<v>";
|
||||||
|
@ -221,28 +184,27 @@ struct
|
||||||
print_exp ff idx;
|
print_exp ff idx;
|
||||||
fprintf ff "]"
|
fprintf ff "]"
|
||||||
|
|
||||||
and print_exps ff e_list = print_list ff print_exp "," e_list
|
and print_exps ff e_list = print_list_r print_exp "" "," "" ff e_list
|
||||||
|
|
||||||
and print_exp ff = function
|
and print_exp ff = function
|
||||||
| Lhs lhs -> print_lhs ff lhs
|
| Lhs lhs -> print_lhs ff lhs
|
||||||
| Const c -> print_c ff c
|
| Const c -> print_c ff c
|
||||||
| Op(op, e_list) -> print_op ff op e_list
|
| Op(op, e_list) -> print_op ff op e_list
|
||||||
| Struct_lit(_,f_e_list) ->
|
| Struct_lit(_,f_e_list) ->
|
||||||
fprintf ff "@[<v 1>{";
|
fprintf ff "@[<v 1>";
|
||||||
print_list ff
|
print_list_r
|
||||||
(fun ff (field, e) -> print_longname ff field;fprintf ff " = ";
|
(fun ff (field, e) -> print_longname ff field;fprintf ff " = ";
|
||||||
print_exp ff e)
|
print_exp ff e)
|
||||||
";" f_e_list;
|
"{" ";" "}" ff f_e_list;
|
||||||
fprintf ff "}@]"
|
fprintf ff "@]"
|
||||||
| Array_lit e_list ->
|
| Array_lit e_list ->
|
||||||
fprintf ff "@[[";
|
fprintf ff "@[";
|
||||||
print_list ff print_exp ";" e_list;
|
print_list_r print_exp "[" ";" "]" ff e_list;
|
||||||
fprintf ff "]@]"
|
fprintf ff "@]"
|
||||||
|
|
||||||
and print_op ff op e_list =
|
and print_op ff op e_list =
|
||||||
print_longname ff op;
|
print_longname ff op;
|
||||||
fprintf ff "(@["; print_list ff print_exp ", " e_list;
|
print_list_r print_exp "(" "," ")" ff e_list
|
||||||
fprintf ff ")@]"
|
|
||||||
|
|
||||||
let print_asgn ff pref x e =
|
let print_asgn ff pref x e =
|
||||||
fprintf ff "@[%s" pref; print_lhs ff x; fprintf ff " = ";
|
fprintf ff "@[%s" pref; print_lhs ff x; fprintf ff " = ";
|
||||||
|
@ -275,9 +237,7 @@ struct
|
||||||
(name x) i1 i2
|
(name x) i1 i2
|
||||||
print_act act
|
print_act act
|
||||||
| Step_ap (var_list, o, es) ->
|
| Step_ap (var_list, o, es) ->
|
||||||
fprintf ff "@[(";
|
print_list print_lhs "(" "," ")" ff var_list;
|
||||||
print_list ff print_lhs "," var_list;
|
|
||||||
fprintf ff "@])";
|
|
||||||
fprintf ff " = "; print_obj_call ff o; fprintf ff ".step(";
|
fprintf ff " = "; print_obj_call ff o; fprintf ff ".step(";
|
||||||
fprintf ff "@["; print_exps ff es; fprintf ff "@]";
|
fprintf ff "@["; print_exps ff es; fprintf ff "@]";
|
||||||
fprintf ff ")"
|
fprintf ff ")"
|
||||||
|
@ -286,24 +246,24 @@ struct
|
||||||
| Nothing -> fprintf ff "()"
|
| Nothing -> fprintf ff "()"
|
||||||
|
|
||||||
and print_tag_act_list ff tag_act_list =
|
and print_tag_act_list ff tag_act_list =
|
||||||
print_list ff
|
print_list
|
||||||
(fun ff (tag, a) ->
|
(fun ff (tag, a) ->
|
||||||
fprintf ff "@[<hov 2>case@ ";
|
fprintf ff "@[<hov 2>case@ ";
|
||||||
print_longname ff tag;
|
print_longname ff tag;
|
||||||
fprintf ff ":@ ";
|
fprintf ff ":@ ";
|
||||||
print_act ff a;
|
print_act ff a;
|
||||||
fprintf ff "@]") "" tag_act_list
|
fprintf ff "@]") "" "" "" ff tag_act_list
|
||||||
|
|
||||||
let print_step ff { inp = inp; out = out; local = nl; bd = bd } =
|
let print_step ff { inp = inp; out = out; local = nl; bd = bd } =
|
||||||
fprintf ff "@[<v 2>";
|
fprintf ff "@[<v 2>";
|
||||||
fprintf ff "step(@[";
|
fprintf ff "step(@[";
|
||||||
print_list ff print_vd ";" inp;
|
print_list_r print_vd "(" ";" ")" ff inp;
|
||||||
fprintf ff "@]) returns (@[";
|
fprintf ff "@]) returns ";
|
||||||
print_list ff print_vd ";" out;
|
print_list_r print_vd "(" ";" ")" ff out;
|
||||||
fprintf ff "@]){@,";
|
fprintf ff "@]){@,";
|
||||||
if nl <> [] then begin
|
if nl <> [] then begin
|
||||||
fprintf ff "@[<hov 4>var ";
|
fprintf ff "@[<hov 4>var ";
|
||||||
print_list ff print_vd ";" nl;
|
print_list_r print_vd "" ";" "" ff nl;
|
||||||
fprintf ff ";@]@,"
|
fprintf ff ";@]@,"
|
||||||
end;
|
end;
|
||||||
print_act ff bd;
|
print_act ff bd;
|
||||||
|
@ -320,12 +280,12 @@ struct
|
||||||
fprintf ff "@[<v 2>machine "; print_name ff id; fprintf ff " =@,";
|
fprintf ff "@[<v 2>machine "; print_name ff id; fprintf ff " =@,";
|
||||||
if mem <> [] then begin
|
if mem <> [] then begin
|
||||||
fprintf ff "@[<hov 4>var ";
|
fprintf ff "@[<hov 4>var ";
|
||||||
print_list ff print_vd ";" mem;
|
print_list_r print_vd "" ";" "" ff mem;
|
||||||
fprintf ff ";@]@,"
|
fprintf ff ";@]@,"
|
||||||
end;
|
end;
|
||||||
if objs <> [] then begin
|
if objs <> [] then begin
|
||||||
fprintf ff "@[<hov 4>obj ";
|
fprintf ff "@[<hov 4>obj ";
|
||||||
print_list ff print_obj ";" objs;
|
print_list print_obj "" ";" "" ff objs;
|
||||||
fprintf ff ";@]@,"
|
fprintf ff ";@]@,"
|
||||||
end;
|
end;
|
||||||
print_reset ff reset;
|
print_reset ff reset;
|
||||||
|
@ -338,17 +298,17 @@ struct
|
||||||
| Type_abs -> fprintf ff "@[type %s@\n@]" name
|
| Type_abs -> fprintf ff "@[type %s@\n@]" name
|
||||||
| Type_enum(tag_name_list) ->
|
| Type_enum(tag_name_list) ->
|
||||||
fprintf ff "@[type %s = " name;
|
fprintf ff "@[type %s = " name;
|
||||||
print_list ff print_name "| " tag_name_list;
|
print_list_r print_name "" "|" "" ff tag_name_list;
|
||||||
fprintf ff "@\n@]"
|
fprintf ff "@\n@]"
|
||||||
| Type_struct(f_ty_list) ->
|
| Type_struct(f_ty_list) ->
|
||||||
fprintf ff "@[type %s = " name;
|
fprintf ff "@[type %s = " name;
|
||||||
fprintf ff "@[<v 1>{";
|
fprintf ff "@[<v 1>";
|
||||||
print_list ff
|
print_list
|
||||||
(fun ff (field, ty) ->
|
(fun ff (field, ty) ->
|
||||||
print_name ff field;
|
print_name ff field;
|
||||||
fprintf ff ": ";
|
fprintf ff ": ";
|
||||||
print_type ff ty) ";" f_ty_list;
|
print_type ff ty) "{" ";" "}" ff f_ty_list;
|
||||||
fprintf ff "}@]@.@]"
|
fprintf ff "@]@.@]"
|
||||||
|
|
||||||
let print_open_module ff name =
|
let print_open_module ff name =
|
||||||
fprintf ff "@[open ";
|
fprintf ff "@[open ";
|
||||||
|
|
Loading…
Reference in a new issue