CristalCaveGem » LustreC

module VarDecl = struct

  type t = var_decl

  let compare v1 v2 = compare v1.var_id v2.var_id

end

module VDSet = Set.Make(VarDecl)

module Live = Map.Make(Int)

let pp_exists pp_l pp_r fmt (l, r) =

  fprintf fmt "@[<v 2>\\exists %a;@,%a@]"

    pp_l l

    pp_r r

  fprintf fmt "@[<v>%a ==>@ %a@]"

  fprintf fmt "@[<v>%a @ && %a@]"

let pp_paren pp fmt v =

  fprintf fmt "(%a)" pp v

let pp_locals m fmt vs =

  pp_print_list

    ~pp_open_box:pp_open_hbox

    ~pp_sep:pp_print_comma

    (pp_c_decl_local_var m) fmt vs

let rec assigned s instr =

  let open VDSet in

  match instr.instr_desc with

  | MLocalAssign (x, _) ->

    add x s

  | MStep (xs, _, _) ->

    union s (of_list xs)

  | MBranch (_, brs) ->

    List.(fold_left (fun s (_, instrs) -> fold_left assigned s instrs) s brs)

  | _ -> s

let rec occur_val s v =

  let open VDSet in

  match v.value_desc with

  | Var x ->

    add x s

  | Fun (_, vs)

  | Array vs ->

    List.fold_left occur_val s vs

  | Access (v1, v2)

  | Power (v1, v2) ->

    occur_val (occur_val s v1) v2

  | _ -> s

let rec occur s instr =

  let open VDSet in

  match instr.instr_desc with

  | MLocalAssign (_, v)

  | MStateAssign (_, v) ->

    occur_val s v

  | MStep (_, _, vs) ->

    List.fold_left occur_val s vs

  | MBranch (v, brs) ->

    List.(fold_left (fun s (_, instrs) -> fold_left occur s instrs)

            (occur_val s v) brs)

  | _ -> s

let live = Hashtbl.create 32

let set_live_of m =

  let open VDSet in

  let locals = of_list m.mstep.step_locals in

  let outputs = of_list m.mstep.step_outputs in

  let vars = union locals outputs in

  let no_occur_after i =

    let occ, _ = List.fold_left (fun (s, j) instr ->

        if j <= i then (s, j+1) else (occur s instr, j+1))

        (empty, 0) m.mstep.step_instrs in

    diff locals occ

  in

  let l, _, _ = List.fold_left (fun (l, asg, i) instr ->

      let asg = inter (assigned asg instr) vars in

      Live.add (i + 1) (diff asg (no_occur_after i)) l, asg, i + 1)

      (Live.empty, empty, 0) m.mstep.step_instrs in

  Hashtbl.add live m.mname.node_id (Live.add 0 empty l)

let live_i m i =

  Live.find i (Hashtbl.find live m.mname.node_id)

    (name, inputs, locals, outputs, mem_in, mem_out) =

  fprintf fmt "%s_trans%a(@[<hov>%a,@ %a%a%a%a@])"

    (pp_print_option

       (fun fmt _ ->

          pp_print_list

            ~pp_epilogue:pp_print_comma

            ~pp_sep:pp_print_comma

            pp_input fmt locals))

    i

let pp_mem_trans ?i pp_mem_in pp_mem_out pp_input pp_output fmt

    (m, mem_in, mem_out) =

  let locals, outputs = match i with

    | Some 0 ->

      [], []

    | Some i when i < List.length m.mstep.step_instrs ->

      let li = live_i m i in

      VDSet.(inter (of_list m.mstep.step_locals) li |> elements,

             inter (of_list m.mstep.step_outputs) li |> elements)

    | Some _ ->

      [], m.mstep.step_outputs

    | _ ->

      m.mstep.step_locals, m.mstep.step_outputs

  in

  pp_mem_trans_aux ?i pp_mem_in pp_mem_out pp_input pp_output fmt

    (m.mname.node_id,

     m.mstep.step_inputs,

     locals,

     outputs,

     mem_in,

     mem_out)

let pp_mem_trans' ?i fmt =

  pp_mem_trans ?i pp_print_string pp_print_string pp_var_decl pp_ptr_decl fmt

let print_ghost_simulation dependencies m fmt (i, instr) =

  let rec aux fmt instr =

    match instr.instr_desc with

    | MStateAssign (m, _) ->

      pp_equal

        (pp_access (pp_access pp_var_decl))

        (pp_indirect (pp_access pp_var_decl))

        fmt

        ((mem, ("_reg", m)), (self, ("_reg", m)))

    | MStep ([i0], i, vl)

      when Basic_library.is_value_internal_fun

          (mk_val (Fun (i, vl)) i0.var_type)  ->

      pp_true fmt ()

    | MStep (_, i, _) when !Options.mpfr && Mpfr.is_homomorphic_fun i ->

      pp_true fmt ()

    | MStep ([_], i, _) when has_c_prototype i dependencies ->

      pp_true fmt ()

    | MStep (_, i, _)

    | MReset i ->

      begin try

          let n, _ = List.assoc i m.minstances in

          pp_mem_ghost pp_access' pp_indirect' fmt

            (node_name n, (mem, i), (self, i))

        with Not_found -> pp_true fmt ()

      end

    | MBranch (_, brs) ->

      (* TODO: handle branches *)

      pp_and_l aux fmt List.(flatten (map snd brs))

    | _ -> pp_true fmt ()

  in

  pred aux instr

let print_machine_ghost_simulation dependencies m fmt i instr =

    (print_ghost_simulation dependencies m)

    ~i:(i+1)

    fmt (i, instr)

let pp_basic_assign_spec pp_var fmt typ var_name value =

  if Types.is_real_type typ && !Options.mpfr

  then

    assert false

    (* Mpfr.pp_inject_assign pp_var fmt (var_name, value) *)

  else

    pp_equal pp_var pp_var fmt (var_name, value)

let pp_assign_spec m self pp_var fmt (var_type, var_name, value) =

  let depth = expansion_depth value in

  let loop_vars = mk_loop_variables m var_type depth in

  let reordered_loop_vars = reorder_loop_variables loop_vars in

  let rec aux typ fmt vars =

    match vars with

    | [] ->

      pp_basic_assign_spec

        (pp_value_suffix ~indirect:false m self var_type loop_vars pp_var)

        fmt typ var_name value

    | (d, LVar i) :: q ->

      assert false

      (* let typ' = Types.array_element_type typ in

       * fprintf fmt "@[<v 2>{@,int %s;@,for(%s=0;%s<%a;%s++)@,%a @]@,}"

       *   i i i pp_c_dimension d i

       *   (aux typ') q *)

    | (d, LInt r) :: q ->

      assert false

      (* let typ' = Types.array_element_type typ in

       * let szl = Utils.enumerate (Dimension.size_const_dimension d) in

       * fprintf fmt "@[<v 2>{@,%a@]@,}"

       *   (pp_print_list (fun fmt i -> r := i; aux typ' fmt q)) szl *)

    | _ -> assert false

  in

  begin

    reset_loop_counter ();

    aux var_type fmt reordered_loop_vars;

  end

let print_machine_trans_simulation_aux ?i m pp fmt v =

  let name = m.mname.node_id in

  let mem_in = mk_mem_in m in

  let mem_out = mk_mem_out m in

  pp_spec

    (pp_predicate

       (pp_mem_trans pp_machine_decl pp_machine_decl

          (pp_c_decl_local_var m) pp_c_decl_output_var ?i)

       pp)

    fmt

    ((m, (name, "mem_ghost", mem_in), (name, "mem_ghost", mem_out)),

     v)

let print_trans_simulation machines dependencies m fmt (i, instr) =

  let mem_in = mk_mem_in m in

  let mem_out = mk_mem_out m in

  let d = VDSet.(diff (live_i m i) (live_i m (i+1))) in

  printf "%d : %a\n%d : %a\n\n"

    i

    (pp_print_parenthesized pp_var_decl) (VDSet.elements (live_i m i))

    (i+1)

    (pp_print_parenthesized pp_var_decl) (VDSet.elements (live_i m (i+1)));

  let prev_trans fmt () = pp_mem_trans' ~i fmt (m, mem_in, mem_out) in

  let pred pp v =

    let locals = VDSet.(inter (of_list m.mstep.step_locals) d |> elements) in

    if locals = []

    then pp_and prev_trans pp fmt ((), v)

    else pp_exists (pp_locals m) (pp_and prev_trans pp) fmt (locals, ((), v))

  in

  let rec aux fmt instr = match instr.instr_desc with

    | MLocalAssign (x, v)

    | MStateAssign (x, v) ->

      pp_assign_spec m mem_in (pp_c_var_read m) fmt

        (x.var_type, mk_val (Var x) x.var_type, v)

    | MStep ([i0], i, vl)

      when Basic_library.is_value_internal_fun

          (mk_val (Fun (i, vl)) i0.var_type)  ->

      pp_true fmt ()

    | MStep (_, i, _) when !Options.mpfr && Mpfr.is_homomorphic_fun i ->

      pp_true fmt ()

    | MStep ([_], i, _) when has_c_prototype i dependencies ->

      pp_true fmt ()

    | MStep (xs, f, ys) ->

      begin try

          let n, _ = List.assoc f m.minstances in

            pp_mem_trans_aux

              pp_access' pp_access'

              (pp_c_val m mem_in (pp_c_var_read m))

              pp_var_decl

              fmt

              (node_name n, ys, [], xs, (mem_in, f), (mem_out, f))

        with Not_found -> pp_true fmt ()

      end

    | MReset f ->

      begin try

          let n, _ = List.assoc f m.minstances in

          pp_mem_init' fmt (node_name n, mem_out)

        with Not_found -> pp_true fmt ()

      end

    | MBranch (v, brs) ->

      (* TODO: handle branches *)

      pp_and_l (fun fmt (l, instrs) ->

          pp_paren (pp_implies

                      (pp_equal

                         (pp_c_val m mem_in (pp_c_var_read m))

                         pp_print_string)

                      (pp_and_l aux))

            fmt

            ((v, l), instrs))

        fmt brs

    | _ -> pp_true fmt ()

  in

  pred aux instr

let print_machine_trans_simulation machines dependencies m fmt i instr =

  print_machine_trans_simulation_aux m

    (print_trans_simulation machines dependencies m)

    ~i:(i+1)

    fmt (i, instr)

let print_machine_core_annotations machines dependencies fmt m =

  if not (fst (Machine_code_common.get_stateless_status m)) then begin

    set_live_of m;

    let i = List.length m.mstep.step_instrs in

    let mem_in = mk_mem_in m in

    let mem_out = mk_mem_out m in

    let last_trans fmt () =

      let locals = VDSet.(inter

                            (of_list m.mstep.step_locals)

                            (live_i m i)

                          |> elements) in

      if locals = []

      then pp_mem_trans' ~i fmt (m, mem_in, mem_out)

      else pp_exists (pp_locals m) (pp_mem_trans' ~i) fmt

          (locals, (m, mem_in, mem_out))

    in

    fprintf fmt "%a@,%a%a"

      (print_machine_trans_simulation_aux m pp_true ~i:0) ()

        (print_machine_trans_simulation machines dependencies m))

      (print_machine_trans_simulation_aux m last_trans) ()

  end

          (pp_ite

             (pp_access pp_access')

             (pp_paren

                (pp_and

                   (pp_equal

                      (pp_access pp_access')

                      pp_print_string)

                   (pp_equal pp_print_string pp_print_string)))

                (pp_paren

                   (pp_and

                      (pp_equal

                         (pp_access pp_access')

                         (pp_access pp_access'))

                      (pp_equal pp_print_string pp_print_string))))))

    ((name, ["int x"; "int y"], [], ["_Bool out"],

      (name, "mem_ghost", mem_in), (name, "mem_ghost", mem_out)),

     (* (("out", mem_in_first), *)

     (mem_in_first, ((mem_out_first, "false"), ("out", "x")),

      ((mem_out_first, mem_in_first), ("out", "y"))))

         (print_machine_core_annotations machines dependencies)

                     (pp_mem_ghost' ~i)

                     (pp_mem_trans' ~i))))))