CristalCaveGem » LustreC

let rec pp_emf_dim fmt dim_expr =

  fprintf fmt "{";

  (let open Dimension in

   match dim_expr.dim_desc with

   | Dbool b -> fprintf fmt "\"kind\": \"bool\",@ \"value\": \"%b\"" b

   | Dint i -> fprintf fmt "\"kind\": \"int\",@ \"value\": \"%i\"" i

   | Dident s -> fprintf fmt "\"kind\": \"ident\",@ \"value\": \"%s\"" s

   | Dappl(f, args) -> fprintf fmt "\"kind\": \"fun\",@ \"id\": \"%s\",@ \"args\": [@[%a@]]"

                         f (Utils.fprintf_list ~sep:",@ " pp_emf_dim) args 

   | Dite(i,t,e) -> fprintf fmt "\"kind\": \"ite\",@ \"guard\": \"%a\",@ \"then\": %a,@ \"else\": %a"

                      pp_emf_dim i pp_emf_dim t pp_emf_dim e 

   | Dlink e -> pp_emf_dim fmt e

   | Dvar

   | Dunivar -> assert false (* unresolved *)

  );

  fprintf fmt "}"

(* First try to print the declared one *)

  | Tydec_any -> (* Dynamical built variable. No declared type. Shall

                    use the infered one. *)

     pp_infered_type fmt infered_t

  | Tydec_int -> fprintf fmt "{ \"kind\": \"int\" }" (* !Options.int_type *)

  | Tydec_real -> fprintf fmt "{ \"kind\": \"real\" }" (* !Options.real_type *)

  | Tydec_bool -> fprintf fmt "{ \"kind\": \"bool\" }"

    (* This is an alias type *)

    (* id for a enumerated type, eg. introduced by automata *)

    (* Print the type name associated to this enumerated type. This is

       basically an integer *)

    pp_tag_type id typ infered_t fmt

  )

  | Tydec_struct _ | Tydec_enum _ ->

     assert false (* should not happen. These type are only built when

                     declaring a type in the prefix of the lustre

                     file. They shall not be associated to variables

                   *)

  | Tydec_array (dim, e) -> (

    let inf_base = match infered_t.Typing.tdesc with

      | Typing.Tarray(_,t) -> t

      | _ ->   (* returing something useless, hoping that the concrete

                  datatype will return something usefull *)

         Typing.new_var ()

    in

    fprintf fmt "{ \"kind\": \"array\", \"base_type\": %t, \"dim\": %a }"

      (pp_concrete_type e inf_base)

      pp_emf_dim dim

(* | _ -> eprintf

 *          "unhandled construct in type printing for EMF backend: %a@."

 *          Printers.pp_var_type_dec_desc dec_t; raise (Failure "var") *)

and pp_tag_type id typ inf fmt =

  (* We ought to represent these types as values: enum will become int, we keep the name for structs *)

  let rec aux tydec_desc =

    match tydec_desc with  

    | Tydec_int 

      | Tydec_real 

      | Tydec_bool

      | Tydec_array _ -> pp_concrete_type tydec_desc inf fmt

    | Tydec_const id ->

       (* Alias of an alias: unrolling definitions *)

       let typ = (Corelang.typedef_of_top

                    (Hashtbl.find Corelang.type_table tydec_desc))

       in

       pp_tag_type id typ inf fmt

    | Tydec_clock ty -> aux ty

    | Tydec_enum const_list -> ( (* enum can be mapped to int *)

      let size = List.length const_list in

      fprintf fmt "{ \"name\": \"%s\", \"kind\": \"enum\", \"size\": \"%i\" }" id size

    )

    | Tydec_struct _ ->

       fprintf fmt "{ \"name\": \"%s\", \"kind\": \"struct\" }" id

    | Tydec_any -> (* shall not happen: a declared type cannot be

                      bound to type any *)

       assert false

  in

  aux typ.tydef_desc

and pp_infered_type fmt t =

  (* Shall only be used for variable types that were not properly declared. Ie generated at compile time. *)

  let open Types in

  if is_bool_type t  then fprintf fmt "bool" else

    if is_int_type t then fprintf fmt "int" else (* !Options.int_type *)

      if is_real_type t then fprintf fmt "real" else (* !Options.real_type *)

        match t.tdesc with

        | Tclock t ->

           pp_infered_type fmt t

        | Tstatic (_, t) ->

           fprintf fmt "%a" pp_infered_type t

        | Tconst id ->

           (* This is a type id for a enumerated type, eg. introduced by automata *)

           let typ =

             (Corelang.typedef_of_top

                (Hashtbl.find Corelang.type_table (Tydec_const id)))

           in

           pp_tag_type id typ t fmt

        | Tlink ty -> 

           pp_infered_type fmt ty 

        | _ -> eprintf "unhandled type: %a@." Types.print_node_ty t; assert false

(* Provide both the declared type and the infered one. *)

  with Failure msg -> eprintf "failed var: %a@.%s@." Printers.pp_var v msg; assert false

let pp_emf_list ?(eol:('a, formatter, unit) Pervasives.format="") pp fmt l =

  match l with

    [] -> ()

  | _ -> fprintf fmt "@[";

         Utils.fprintf_list ~sep:",@ " pp fmt l;

         fprintf fmt "@]%(%)" eol

(* Print the variable declaration *)

let pp_emf_vars_decl = pp_emf_list pp_emf_var_decl

let pp_cst_type c inf fmt (*infered_typ*) =

  match c with

  | Const_tag t ->

     let typ = (Corelang.typedef_of_top (Hashtbl.find Corelang.tag_table t)) in

     if typ.tydef_id = "bool" then

       fprintf fmt "bool"

     else

       pp_tag_type t typ inf fmt

  | Const_int _ -> fprintf fmt "int" (*!Options.int_type*)

  | Const_real _ -> fprintf fmt "real" (*!Options.real_type*)

  | Const_string _ -> fprintf fmt "string" 

  | _ -> eprintf "cst: %a@." Printers.pp_const c; assert false

let pp_emf_cst c inf fmt =

  let pp_typ fmt =

    fprintf fmt "\"datatype\": \"";

    pp_cst_type c inf fmt;

    fprintf fmt "\"@ "

  in

       pp_typ fmt;

       pp_typ fmt;

     fprintf fmt "{@[\"type\": \"constant\",@ \"value\": \"%s\",@ " s;

     pp_typ fmt;

     fprintf fmt "@]}"

    pp_typ fmt;

(* Print a value: either a constant or a variable value *)

  | Cst c -> pp_emf_cst c v.value_type fmt 

  | Expr_const c -> pp_emf_cst c e.expr_type fmt 

 (* Missing these 

  | Expr_ite   of expr * expr * expr

  | Expr_arrow of expr * expr

  | Expr_fby of expr * expr

  | Expr_array of expr list

  | Expr_access of expr * Dimension.dim_expr

  | Expr_power of expr * Dimension.dim_expr

  | Expr_pre of expr

  | Expr_when of expr * ident * label

  | Expr_merge of ident * (label * expr) list

  | Expr_appl of call_t

  *)

let pp_emf_exprs = pp_emf_list pp_emf_expr

let pp_emf_const fmt v =

  fprintf fmt "@[{\"name\": \"%a\", \"datatype\":\"%a\", \"original_name\": \"%a\", \"value\": \"%a\"}@]"

    pp_var_name v

    pp_var_type v

    Printers.pp_var_name v

    pp_emf_expr (match v.var_dec_value with None -> assert false | Some e -> e)

let pp_emf_consts = pp_emf_list pp_emf_const

let pp_emf_eexprs = pp_emf_list pp_emf_eexpr

(*

                      TODO Thanksgiving

                      trouver un moyen de transformer en machine code les instructions de chaque spec

                      peut etre associer a chaque imported node une minimachine

                      et rajouter un champ a spec dans machine code pour stoquer memoire et instr

 *)                

let pp_emf_stmt fmt stmt =

  match stmt with

  | Aut _ -> assert false

  | Eq eq -> (

    fprintf fmt "@[ @[<v 2>\"%a\": {@ " (Utils.fprintf_list ~sep:"_" pp_print_string) eq.eq_lhs;

    fprintf fmt "\"lhs\": [%a],@ " (Utils.fprintf_list ~sep:", " (fun fmt vid -> fprintf fmt "\"%s\"" vid)) eq.eq_lhs;

    fprintf fmt "\"rhs\": \"%a\",@ " pp_emf_expr eq.eq_rhs;

    fprintf fmt "@]@]@ }"

  )

let pp_emf_stmts = pp_emf_list pp_emf_stmt 

(* Printing the type declaration, not its use *)

let rec pp_emf_typ_dec fmt tydef_dec =

  fprintf fmt "{";

  (match tydef_dec with

   | Tydec_any -> fprintf fmt "\"kind\": \"any\""

   | Tydec_int -> fprintf fmt "\"kind\": \"int\""

   | Tydec_real -> fprintf fmt "\"kind\": \"real\""

   | Tydec_bool-> fprintf fmt "\"kind\": \"bool\""

   | Tydec_clock ck -> pp_emf_typ_dec fmt ck

   | Tydec_const c -> fprintf fmt "\"kind\": \"alias\",@ \"value\": \"%s\"" c

   | Tydec_enum el -> fprintf fmt "\"kind\": \"enum\",@ \"elements\": [%a]"

                        (Utils.fprintf_list ~sep:", " (fun fmt e -> fprintf fmt "\"%s\"" e)) el

   | Tydec_struct s -> fprintf fmt "\"kind\": \"struct\",@ \"fields\": [%a]"

                         (Utils.fprintf_list ~sep:", " (fun fmt (id,typ) ->

                              fprintf fmt "\"%s\": %a" id pp_emf_typ_dec typ)) s

   | Tydec_array (dim, typ) -> fprintf fmt "\"kind\": \"array\",@ \"dim\": @[%a@],@ \"base\": %a"

                               pp_emf_dim dim

                               pp_emf_typ_dec typ

  );

  fprintf fmt "}"

let pp_emf_typedef fmt typdef_top =

  let typedef = Corelang.typedef_of_top typdef_top in

  fprintf fmt "\"%s\": @[%a@]" typedef.tydef_id pp_emf_typ_dec typedef.tydef_desc 

let pp_emf_top_const fmt const_top = 

  let const = Corelang.const_of_top const_top in

  fprintf fmt "\"%s\": " const.const_id;

  pp_emf_cst const.const_value const.const_type fmt