CristalCaveGem » LustreC

let report = Log.report ~plugin:"tiny"

module Ast = Tiny.Ast

let gen_loc () = Tiny.Location.dummy ()

let lloc_to_tloc loc = Tiny.Location.location_of_positions loc.Location.loc_start loc.Location.loc_end

let tloc_to_lloc loc = assert false (*Location.dummy_loc (*TODO*) *)

let ltyp_to_ttyp t =

  if Types.is_real_type t then Tiny.Ast.RealT

  else if Types.is_int_type t then Tiny.Ast.IntT

  else if Types.is_bool_type t then Tiny.Ast.BoolT

  else assert false (* not covered yet *)

let cst_bool loc b =

  { Ast.expr_desc =

      if b then

        Ast.Cst(Q.of_int 1, "true")

      else

        Ast.Cst(Q.of_int 0, "false");

    expr_loc = loc;

    expr_type = Ast.BoolT }

let cst_num loc t q =

  let s = Q.to_string q in

  let s =  if t = Tiny.Ast.RealT && Z.equal (Q.den q) Z.one then s ^ "."

             else s in

{ Ast.expr_desc =

    Ast.Cst(q, s);

  expr_loc = loc;

  expr_type = t }

let rec real_to_q man exp =

  if exp = 0 then

        Q.of_string (Num.string_of_num man)

  else

    if exp > 0 then Q.div (real_to_q man (exp-1)) (Q.of_int 10)

    else (* if exp<0 then *)

      Q.mul

        (real_to_q man (exp+1))

        (Q.of_int 10)

let instr_loc i =

  match i.Machine_code_types.lustre_eq with

  | None -> gen_loc ()

  | Some eq -> lloc_to_tloc eq.eq_loc

let build_instr d v =

  Ast.{ expr_desc = d;

        expr_loc = gen_loc ();

        expr_type = v }

let cst_false =  Ast.Cst (Q.zero, "false") 

let cst_true =  Ast.Cst (Q.one, "true")

let rec lval_to_texpr m loc _val =

  let build = build_instr in

  let new_desc =

    match _val.Machine_code_types.value_desc with

    | Machine_code_types.Cst cst -> (

      match cst with

        Lustre_types.Const_int n -> Ast.Cst (Q.of_int n, string_of_int n)

      | Const_real r -> Ast.Cst (Real.to_q r, Real.to_string r) 

      | Const_tag "true" -> cst_true

      | Const_tag "false" -> cst_false

      | _ -> Format.eprintf "No tiny translation for constant %a@.@?"

               (Machine_code_common.pp_val m) _val; assert false

    )

    | Var v -> Ast.Var (v.var_id)

    | Fun (op, vl) ->

       let t_arg = match vl with (* t_arg denotes the type of first argument. *)

         | hd::_ -> ltyp_to_ttyp hd.value_type

         | _ -> assert false

       in

       (

         match op, List.map (lval_to_texpr m loc) vl with

         | "+", [v1;v2] -> Ast.Binop (Ast.Plus, v1, v2)

         | "-", [v1;v2] -> Ast.Binop (Ast.Minus, v1, v2)

         | "*", [v1;v2] -> Ast.Binop (Ast.Times, v1, v2)

         | "/", [v1;v2] -> Ast.Binop (Ast.Div, v1, v2)

         | "<", [v1;v2] ->

            Ast.Cond (build (Ast.Binop (Ast.Minus, v2, v1)) t_arg, Ast.Strict)

         | "<=", [v1;v2] ->

            Ast.Cond (build(Ast.Binop (Ast.Minus, v2, v1)) t_arg, Ast.Loose)

         | ">", [v1;v2] ->

            Ast.Cond (build(Ast.Binop (Ast.Minus, v1, v2)) t_arg, Ast.Strict)

         | ">=", [v1;v2] ->

            Ast.Cond (build (Ast.Binop (Ast.Minus, v1, v2)) t_arg, Ast.Loose)

         | "uminus", [v1] -> Ast.Binop (Ast.Minus, cst_num loc t_arg Q.zero, v1)

         (* | "=", [v1;v2] when v1.xxxtype != BoolT (\* if arguments are numerical then basic comparison *\)  

          *    Ast.Cond (build (Ast.Binop (Ast.Minus, v1, v2)) t_arg, Ast.Loose)

          *    Ast.Binop (Ast.Eq, v1, v2) *)

         | "=", [v1;v2] 

         | "equi", [v1;v2] -> Ast.Binop (Ast.Eq, v1, v2)

         | "!=", [v1;v2] -> Ast.Unop (Ast.Not, build (Ast.Binop (Ast.Eq, v1, v2)) Ast.BoolT)

         | "not", [v1] -> Ast.Unop (Ast.Not, v1)

         | "&&", [v1;v2] -> Ast.Binop (Ast.And, v1, v2)

         | "||", [v1;v2] -> Ast.Binop (Ast.Or, v1, v2)

         | "impl", [v1;v2] ->

            let neg_v1 = Ast.neg_guard v1 in

            Ast.Binop (Ast.Or, neg_v1, v2)

         | _ -> Format.eprintf "No tiny translation for operator %s@.@?" op; assert false    

       )

    | _ -> assert false (* no array. access or power *)

  in

  build new_desc (ltyp_to_ttyp _val.value_type)

                       (*

let machine_init_to_ast m =

  let build_reset i =

    match i.Machine_code_types.instr_desc with

    | MReset id ->

    | _ -> Format.eprintf "Init instr: %a@." (Machine_code_common.pp_instr m) i; assert false

  in

build_seq build_reset m.Machine_code_types.minit

                        *)

let rec find_init m il =

  List.fold_left (fun res i ->

      match res with

      | Some _ -> res

      | _ -> (

        match i.Machine_code_types.instr_desc with

        | MLocalAssign _ | MStateAssign _ | MReset _| MNoReset _| MComment _|MSpec _

          -> None

        | MStep (ol, id, args) ->

           if List.mem_assoc id m.Machine_code_types.minstances then

             let fun_name, _ = List.assoc id m.minstances in

             match Corelang.node_name fun_name, ol with

             | "_arrow", [o] -> Some o.var_id

             | _ -> None

           else

             None

        | MBranch (_, guarded_expr) ->

           List.fold_left (

               fun res (_, il') ->

               match res with Some _ -> res | None -> find_init m il'

             ) res guarded_expr

      )

    )

    None il

let machine_body_to_ast init m =

  let init_var = find_init m m.Machine_code_types.mstep.step_instrs in

  let rec instrl_to_stm il  = 

    match il with

      [] -> assert false

    | [i] -> instr_to_stm i

    | i::il -> Ast.Seq (gen_loc (), instr_to_stm i, instrl_to_stm il)

  and guarded_expr_to_stm loc te guarded_instrs =

    match guarded_instrs with

    | [] -> assert false

    | [_,il] -> instrl_to_stm il

    | (label, il)::tl ->

       let stmt = instrl_to_stm il in

       let guard= match label with

           "true" -> te

         | "false" -> Ast.neg_guard te

         | _ -> assert false (* TODO: don't deal with non boolean

                                guards. Could just treturn true and

                                over-approximate behavior *)

       in

       if (match init_var, te.Ast.expr_desc with

           | Some v, Var v2 ->

              (* Format.eprintf "Building init (possibly if %b)@." (v=v2);  *)v = v2

           | _ ->

              (* Format.eprintf "Building if init def? %b @." (match init_var with None -> false | _ -> true);

               *)

              false

          )

       then 

         instrl_to_stm (

             if init then

               (List.assoc "true" guarded_instrs)

             else

               (List.assoc "false" guarded_instrs)

           )

       else

         Ast.Ite (loc, guard, stmt, guarded_expr_to_stm loc te tl)

  and instr_to_stm i =

    let loc = instr_loc i in

    match i.instr_desc with

    | MLocalAssign (v, e) | MStateAssign (v, e) ->

       Ast.Asn (loc, v.var_id, (lval_to_texpr m loc) e)

    | MBranch (e, guarded_instrs) -> (

      let te = lval_to_texpr m loc e in

      guarded_expr_to_stm loc te guarded_instrs

    )

    | MStep (ol, id, args) ->

       if List.mem_assoc id m.Machine_code_types.minstances then

         let fun_name, _ = List.assoc id m.minstances in

         match Corelang.node_name fun_name, ol with

         | "_arrow", [o] -> (

           (* init_var := Some o.var_id; *)

           Ast.Asn (loc, o.var_id, build_instr (if init then cst_true else cst_false) Ast.BoolT);

         (* We set the arrow to

            false: we are not anymore  

            in init state *)

         (*           Format.eprintf "%a = arrow()@." Printers.pp_var o;*)

         (*  Ast.Nop (loc); *)

         (* Ast.Asn (loc, o.var_id, 

          *        { expr_desc =              if init then Ast.Cst(Q.of_int 1, "true") else Ast.Cst(Q.of_int 0, "false");

          *          expr_loc = loc;

          *          expr_type = Ast.BoolT }

          * ) *)

         )

         | name, _ -> 

            (

              Format.eprintf "No tiny translation for stateful node call  %s@.@?" name;

              assert false

            )

       else (

         Format.eprintf "No tiny translation for function call  %s@.@?" id;

         assert false

       )

    | MReset id

      | MNoReset id -> assert false (* no more calls or functions, ie. no reset *)

    | MComment s 

      | MSpec s -> assert false

  in

  instrl_to_stm m.Machine_code_types.mstep.step_instrs 

let read_var bounds_opt v =

  let min, max =

    match bounds_opt with

      Some (min,max) -> min, max

    | None ->

       let one, minus_one =

         match Types.is_real_type v.Lustre_types.var_type with

         | true -> "1.", "-1."

         | false -> "1", "-1"

       in

       (Q.of_int (-1), minus_one), (Q.of_int 1, one)

  in

  let range = {

      Ast.expr_desc = Ast.Rand (min,max);

      expr_loc = gen_loc ();

      expr_type = ltyp_to_ttyp (v.Lustre_types.var_type)

    }

  in

  Ast.Asn (gen_loc (), v.var_id, range)

let rec read_vars bounds_inputs vl =

  match vl with

    [] -> Ast.Nop (gen_loc ())

  | [v] -> read_var

             (if List.mem_assoc v.Lustre_types.var_id bounds_inputs then

                Some (List.assoc v.Lustre_types.var_id bounds_inputs)

              else

                None)

             v

  | v::tl ->

     Ast.Seq (gen_loc (),

              read_var

                (if List.mem_assoc v.Lustre_types.var_id bounds_inputs then

                   Some (List.assoc v.Lustre_types.var_id bounds_inputs)

                 else

                   None)

                v,

              read_vars bounds_inputs tl

       )

let machine_to_ast bounds_input m =

  let read_vars = read_vars bounds_input m.Machine_code_types.mstep.step_inputs in

  (* let ast_init = machine_init_to_ast m in *)

  let ast_loop_first = machine_body_to_ast true m in

  let ast_loop_run = machine_body_to_ast false m in

  let ast_loop_body = Ast.Seq (gen_loc (), read_vars, ast_loop_run) in

  let loop = Ast.While(gen_loc (), cst_bool (gen_loc ()) true, ast_loop_body) in

  (* Ast.Seq (gen_loc(),

   *          ast_init, *)

           Ast.Seq (gen_loc (), read_vars,

                    Ast.Seq (gen_loc (),

                             ast_loop_first,

                             loop))

         (* ) *)

let machine_to_env m =

  List.fold_left (fun accu v ->

      report ~level:3 (fun fmt -> Format.fprintf fmt "Adding variable %a to env@." Printers.pp_var v);

      let typ =

        ltyp_to_ttyp (v.Lustre_types.var_type)

      in

      Ast.Var.Set.add (v.var_id, typ) accu)

    Ast.Var.Set.empty

    (Machine_code_common.machine_vars m)