[general] renaming (pp -> lustre_pp ; c_* -> intermediate_*)
This commit is contained in:
dsac
184
src/lustre_pp.ml
Normal file
184
src/lustre_pp.ml
Normal file
@@ -0,0 +1,184 @@
|
||||
open Ast
|
||||
|
||||
let rec debug_type_pp fmt = function
|
||||
| [] -> ()
|
||||
| TInt :: t -> Format.fprintf fmt "int %a" debug_type_pp t
|
||||
| TBool :: t -> Format.fprintf fmt "bool %a" debug_type_pp t
|
||||
| TReal :: t -> Format.fprintf fmt "real %a" debug_type_pp t
|
||||
|
||||
let pp_loc fmt ((start, stop), file) =
|
||||
let spos, epos =
|
||||
Lexing.(start.pos_cnum, stop.pos_cnum) in
|
||||
let f = open_in file in
|
||||
try
|
||||
begin
|
||||
let rec aux linenum curpos =
|
||||
let line = input_line f in
|
||||
let nextpos = curpos + (String.length line) + 1 in
|
||||
if nextpos >= epos then
|
||||
Format.fprintf fmt "<line %d: %s >" linenum line
|
||||
else
|
||||
aux (linenum + 1) nextpos
|
||||
in
|
||||
aux 1 0;
|
||||
close_in f
|
||||
end
|
||||
with e ->
|
||||
(close_in_noerr f; Format.fprintf fmt "???")
|
||||
|
||||
let rec pp_varlist fmt : t_varlist -> unit = function
|
||||
| ([], []) -> ()
|
||||
| ([TInt] , IVar h :: []) -> Format.fprintf fmt "%s: int" h
|
||||
| ([TReal], RVar h :: []) -> Format.fprintf fmt "%s: real" h
|
||||
| ([TBool], BVar h :: []) -> Format.fprintf fmt "%s: bool" h
|
||||
| (TInt :: tl, IVar h :: h' :: l) ->
|
||||
Format.fprintf fmt "%s: int, %a" h pp_varlist (tl, h' :: l)
|
||||
| (TBool :: tl, BVar h :: h' :: l) ->
|
||||
Format.fprintf fmt "%s: bool, %a" h pp_varlist (tl, h' :: l)
|
||||
| (TReal :: tl, RVar h :: h' :: l) ->
|
||||
Format.fprintf fmt "%s: real, %a" h pp_varlist (tl, h' :: l)
|
||||
| _ -> raise (MyTypeError "This exception should not have beed be raised.")
|
||||
|
||||
let pp_expression =
|
||||
let upd_prefix s = s ^ " | " in
|
||||
let rec pp_expression_aux prefix fmt expression =
|
||||
let rec pp_expression_list prefix fmt exprs =
|
||||
match exprs with
|
||||
| ETuple([], []) -> ()
|
||||
| ETuple (_ :: tt, expr :: exprs) ->
|
||||
Format.fprintf fmt "%a%a"
|
||||
(pp_expression_aux (prefix^" |> ")) expr
|
||||
(pp_expression_list prefix) (ETuple (tt, exprs))
|
||||
| _ -> raise (MyTypeError "This exception should not have been raised.")
|
||||
in
|
||||
match expression with
|
||||
| EWhen (_, e1, e2) ->
|
||||
begin
|
||||
Format.fprintf fmt "\t\t\t%sWHEN\n%a\t\t\tWHEN\n%a"
|
||||
prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) e1
|
||||
(pp_expression_aux (upd_prefix prefix)) e2
|
||||
end
|
||||
| EReset (_, e1, e2) ->
|
||||
begin
|
||||
Format.fprintf fmt "\t\t\t%sRESET\n%a\t\t\tRESET\n%a"
|
||||
prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) e1
|
||||
(pp_expression_aux (upd_prefix prefix)) e2
|
||||
end
|
||||
| EConst (_, c) ->
|
||||
begin match c with
|
||||
| CBool b -> Format.fprintf fmt "\t\t\t%s<%s : bool>\n" prefix (Bool.to_string b)
|
||||
| CInt i -> Format.fprintf fmt "\t\t\t%s<%5d: int>\n" prefix i
|
||||
| CReal r -> Format.fprintf fmt "\t\t\t%s<%5f: float>\n" prefix r
|
||||
end
|
||||
| EVar (_, IVar v) -> Format.fprintf fmt "\t\t\t%s<int var %s>\n" prefix v
|
||||
| EVar (_, BVar v) -> Format.fprintf fmt "\t\t\t%s<bool var %s>\n" prefix v
|
||||
| EVar (_, RVar v) -> Format.fprintf fmt "\t\t\t%s<real var %s>\n" prefix v
|
||||
| EMonOp (_, mop, arg) ->
|
||||
begin match mop with
|
||||
| MOp_not ->
|
||||
Format.fprintf fmt "\t\t\t%s ¬ \n%a" prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
| MOp_minus ->
|
||||
Format.fprintf fmt "\t\t\t%s — \n%a" prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
| MOp_pre ->
|
||||
Format.fprintf fmt "\t\t\t%spre\n%a" prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
end
|
||||
| EBinOp (_, bop, arg, arg') ->
|
||||
begin
|
||||
let s = match bop with
|
||||
| BOp_add -> " + " | BOp_sub -> " - "
|
||||
| BOp_mul -> " ∗ " | BOp_div -> " / " | BOp_mod -> "% "
|
||||
| BOp_and -> "&& " | BOp_or -> "|| " | BOp_arrow -> "-> " in
|
||||
Format.fprintf fmt "\t\t\t%s%s\n%a%a" prefix s
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
(pp_expression_aux (upd_prefix prefix)) arg'
|
||||
end
|
||||
| EComp (_, cop, arg, arg') ->
|
||||
begin
|
||||
let s = match cop with
|
||||
| COp_eq -> "== "
|
||||
| COp_neq -> " ≠ "
|
||||
| COp_le -> " ≤ " | COp_lt -> " < "
|
||||
| COp_ge -> " ≥ " | COp_gt -> " > " in
|
||||
Format.fprintf fmt "\t\t\t%s%s\n%a%a" prefix s
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
(pp_expression_aux (upd_prefix prefix)) arg'
|
||||
end
|
||||
| ETriOp (_, top, arg, arg', arg'') ->
|
||||
begin match top with
|
||||
| TOp_if ->
|
||||
Format.fprintf fmt "\t\t\t%sIF\n%a\t\t\tTHEN\n%a\t\t\tELSE\n%a"
|
||||
prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
(pp_expression_aux (upd_prefix prefix)) arg'
|
||||
(pp_expression_aux (upd_prefix prefix)) arg''
|
||||
| TOp_merge ->
|
||||
Format.fprintf fmt "\t\t\t%sMERGE ON CLK\n%a\t\t\tE1\n%a\t\t\tE2\n%a"
|
||||
prefix
|
||||
(pp_expression_aux (upd_prefix prefix)) arg
|
||||
(pp_expression_aux (upd_prefix prefix)) arg'
|
||||
(pp_expression_aux (upd_prefix prefix)) arg''
|
||||
end
|
||||
| EApp (_, f, args) ->
|
||||
Format.fprintf fmt "\t\t\t%sApp %s\n%a"
|
||||
prefix f.n_name
|
||||
(pp_expression_list prefix) args
|
||||
| ETuple _ ->
|
||||
Format.fprintf fmt "\t\t\t%sTuple\n%a" prefix
|
||||
(pp_expression_list prefix) expression
|
||||
in
|
||||
pp_expression_aux ""
|
||||
|
||||
let rec pp_equations fmt: t_eqlist -> unit = function
|
||||
| [] -> ()
|
||||
| (patt, expr) :: eqs ->
|
||||
Format.fprintf fmt "\t\t∗ Equation of type : %a\n\t\t left side: %a\n\
|
||||
\t\t right side:\n%a\n\n%a"
|
||||
debug_type_pp (Utils.type_exp expr)
|
||||
pp_varlist patt
|
||||
pp_expression expr
|
||||
pp_equations eqs
|
||||
|
||||
let rec pp_automata fmt: t_automaton list -> unit = function
|
||||
| [] -> ()
|
||||
| (_, trans)::tail ->
|
||||
Format.fprintf fmt "\t\t* Automaton : \n%a%a"
|
||||
pp_translist trans
|
||||
pp_automata tail
|
||||
|
||||
and pp_translist fmt: t_state list -> unit = function
|
||||
| [] -> ()
|
||||
| State(name, eqs, cond, next)::q ->
|
||||
Format.fprintf fmt "\t\t\t|%s -> do\n%a\n\t\t\tdone until %a \t\t\tthen %s\n%a"
|
||||
name
|
||||
pp_equations eqs
|
||||
pp_expression cond
|
||||
next
|
||||
pp_translist q
|
||||
|
||||
let pp_node fmt node =
|
||||
Format.fprintf fmt "\t∗ Node name : %s\n\t Inputs:\n%a\n\t Outputs:\n%a\n\t\
|
||||
\ \ Local variables:\n%a\n\t Equations:\n%a\n\t Automata:\n%a\n"
|
||||
node.n_name
|
||||
pp_varlist node.n_inputs
|
||||
pp_varlist node.n_outputs
|
||||
pp_varlist node.n_local_vars
|
||||
pp_equations node.n_equations
|
||||
pp_automata node.n_automata
|
||||
|
||||
let rec pp_nodes fmt nodes =
|
||||
match nodes with
|
||||
| [] -> ()
|
||||
| node :: nodes ->
|
||||
Format.fprintf fmt "%a\n%a" pp_node node pp_nodes nodes
|
||||
|
||||
let pp_ast fmt prog =
|
||||
Format.fprintf fmt
|
||||
"The program is made of %d node(s), listed below :\n%a"
|
||||
(List.length prog)
|
||||
pp_nodes prog
|
||||
|
||||
Reference in New Issue
Block a user