Project

General

Profile

Download (8.6 KB) Statistics
| Branch: | Tag: | Revision:
1
(********************************************************************)
2
(*                                                                  *)
3
(*  The LustreC compiler toolset   /  The LustreC Development Team  *)
4
(*  Copyright 2012 -    --   ONERA - CNRS - INPT - ISAE-SUPAERO     *)
5
(*                                                                  *)
6
(*  LustreC is free software, distributed WITHOUT ANY WARRANTY      *)
7
(*  under the terms of the GNU Lesser General Public License        *)
8
(*  version 2.1.                                                    *)
9
(*                                                                  *)
10
(********************************************************************)
11

    
12
open Format
13

    
14
open Machine_code_types
15
open Lustre_types
16
open Corelang
17
open Machine_code_common
18

    
19
open Misc_printer
20
open Misc_lustre_function
21
open Ada_printer
22
open Ada_backend_common
23

    
24
(** Main module for generating packages bodies
25
 **)
26
module Main =
27
struct
28

    
29
  (** Printing function for basic assignement [var := value].
30

    
31
      @param fmt the formater to print on
32
      @param var_name the name of the variable
33
      @param value the value to be assigned
34
   **)
35
  let pp_assign env fmt var value =
36
    fprintf fmt "%a := %a"
37
      (pp_var env) var
38
      (pp_value env) value
39

    
40
  (** Printing function for instruction. See
41
      {!type:Machine_code_types.instr_t} for more details on
42
      machine types.
43

    
44
      @param typed_submachines list of all typed machine instances of this machine
45
      @param machine the current machine
46
      @param fmt the formater to print on
47
      @param instr the instruction to print
48
   **)
49
  let rec pp_machine_instr typed_submachines env instr fmt =
50
    let pp_instr = pp_machine_instr typed_submachines env in
51
    (* Print args for a step call *)
52
    let pp_state i fmt = fprintf fmt "%t.%s" pp_state_name i in
53
    (* Print a when branch of a case *)
54
    let pp_when (cond, instrs) fmt =
55
      fprintf fmt "when %s =>@,%a" cond pp_block (List.map pp_instr instrs)
56
    in
57
    (* Print a case *)
58
    let pp_case fmt (g, hl) =
59
      fprintf fmt "case %a is@,%aend case"
60
        (pp_value env) g
61
        pp_block (List.map pp_when hl)
62
    in
63
    (* Print a if *)
64
    (* If neg is true the we must test for the negation of the condition. It
65
       first check that we don't have a negation and a else case, if so it
66
       inverses the two branch and remove the negation doing a recursive
67
       call. *)
68
    let pp_if fmt (neg, g, instrs1, instrs2) =
69
      let pp_cond =
70
        if neg then
71
          fun fmt x -> fprintf fmt "! (%a)" (pp_value env) x
72
        else
73
          pp_value env
74
      in
75
      let pp_else = match instrs2 with
76
        | None -> fun fmt -> fprintf fmt ""
77
        | Some i2 -> fun fmt ->
78
            fprintf fmt "else@,%a" pp_block (List.map pp_instr i2)
79
      in
80
      fprintf fmt "if %a then@,%a%tend if"
81
        pp_cond g
82
        pp_block (List.map pp_instr instrs1)
83
        pp_else
84
    in
85
    match get_instr_desc instr with
86
      (* no reset *)
87
      | MNoReset _ -> ()
88
      (* TODO: handle clear_reset *)
89
      | MClearReset -> ()
90
      (* reset  *)
91
      | MSetReset i when List.mem_assoc i typed_submachines ->
92
          let (substitution, submachine) = get_instance i typed_submachines in
93
          let pp_package = pp_package_name_with_polymorphic substitution submachine in
94
          let args = if is_machine_statefull submachine then [[pp_state i]] else [] in
95
          pp_call fmt (pp_package_access (pp_package, pp_reset_procedure_name), args)
96
      | MLocalAssign (ident, value) ->
97
          pp_assign env fmt ident value
98
      | MStateAssign (ident, value) ->
99
          pp_assign env fmt ident value
100
      | MStep ([i0], i, vl) when is_builtin_fun i ->
101
          let value = mk_val (Fun (i, vl)) i0.var_type in
102
            pp_assign env fmt i0 value
103
      | MStep (il, i, vl) when List.mem_assoc i typed_submachines ->
104
          let (substitution, submachine) = get_instance i typed_submachines in
105
          let pp_package = pp_package_name_with_polymorphic substitution submachine in
106
          let input = List.map (fun x fmt -> pp_value env fmt x) vl in
107
          let output = List.map pp_var_name il in
108
          let args =
109
            (if is_machine_statefull submachine then [[pp_state i]] else [])
110
              @(if input!=[] then [input] else [])
111
              @(if output!=[] then [output] else [])
112
          in
113
          pp_call fmt (pp_package_access (pp_package, pp_step_procedure_name), args)
114
      | MBranch (_, []) -> assert false
115
      | MBranch (g, (c1, i1)::tl) when c1=tag_false || c1=tag_true ->
116
          pp_if fmt (build_if g c1 i1 tl)
117
      | MBranch (g, hl) -> pp_case fmt (g, hl)
118
      | MComment s  ->
119
          let lines = String.split_on_char '\n' s in
120
          fprintf fmt "%a" (Utils.fprintf_list ~sep:"" pp_oneline_comment) lines
121
      | _ -> assert false
122

    
123
  (** Print the definition of the step procedure from a machine.
124

    
125
     @param typed_submachines list of all typed machine instances of this machine
126
     @param fmt the formater to print on
127
     @param machine the machine
128
  **)
129
  let pp_step_definition env typed_submachines fmt (m, m_spec_opt, guarantees) =
130
    let transform_local_to_state_assign instr = match instr.instr_desc with
131
      | MLocalAssign (ident, value) -> 
132
        { instr with instr_desc = MStateAssign (ident, value) }
133
      | _ -> instr
134
    in
135
    let pp_local_ghost_list, spec_instrs = match m_spec_opt with
136
      | None -> [], []
137
      | Some m_spec ->
138
          List.map (build_pp_var_decl_local (Some (true, false, [], []))) (List.filter (fun x -> not (List.mem x.var_id guarantees)) m_spec.mstep.step_locals),
139
          List.map transform_local_to_state_assign m_spec.mstep.step_instrs
140
    in
141
    let pp_local_list = List.map (build_pp_var_decl_local None) (m.mstep.step_locals) in
142
    let pp_instr_list = List.map (pp_machine_instr typed_submachines env) (m.mstep.step_instrs@spec_instrs) in
143
    let content = AdaProcedureContent ((if pp_local_ghost_list = [] then [] else [pp_local_ghost_list])@(if pp_local_list = [] then [] else [pp_local_list]), pp_instr_list) in
144
    pp_procedure pp_step_procedure_name (build_pp_arg_step m) None fmt content
145

    
146
  (** Print the definition of the reset procedure from a machine.
147

    
148
     @param typed_submachines list of all typed machine instances of this machine
149
     @param fmt the formater to print on
150
     @param machine the machine
151
  **)
152
  let pp_reset_definition env typed_submachines fmt (m, m_spec_opt) =
153
    let build_assign = function var ->
154
      mkinstr (MStateAssign (var, mk_default_value var.var_type))
155
    in
156
    let env, memory = match m_spec_opt with
157
      | None -> env, m.mmemory
158
      | Some _ ->
159
          env,
160
          (m.mmemory)
161
    in
162
    let assigns = List.map build_assign memory in
163
    let pp_instr_list = List.map (pp_machine_instr typed_submachines env) (assigns@m.minit) in
164
    pp_procedure pp_reset_procedure_name (build_pp_arg_reset m) None fmt (AdaProcedureContent ([], pp_instr_list))
165

    
166
  (** Print the package definition(ads) of a machine.
167
    It requires the list of all typed instance.
168
    A typed submachine instance is (ident, type_machine) with ident
169
    the instance name and typed_machine is (substitution, machine) with machine
170
    the machine associated to the instance and substitution the instanciation of
171
    all its polymorphic types.
172
     @param fmt the formater to print on
173
     @param typed_submachines list of all typed machine instances of this machine
174
     @param m the machine
175
  **)
176
  let pp_file fmt (typed_submachines, ((opt_spec_machine, guarantees), machine)) =
177
    let env = List.map (fun x -> x.var_id, pp_state_name) machine.mmemory in
178
    let pp_reset fmt =
179
      if is_machine_statefull machine then
180
        fprintf fmt "%a;@,@," (pp_reset_definition env typed_submachines) (machine, opt_spec_machine)
181
      else
182
        fprintf fmt ""
183
    in
184
    let aux pkgs (id, _) =
185
      try
186
        let (pkg, _) = List.assoc id ada_supported_funs in
187
        if List.mem pkg pkgs then
188
          pkgs
189
        else
190
          pkg::pkgs
191
      with Not_found -> pkgs
192
    in
193
    let packages = List.map pp_str (List.fold_left aux [] machine.mcalls) in
194
    let pp_content fmt =
195
      fprintf fmt "%t%a"
196
        (*Define the reset procedure*)
197
        pp_reset
198
        
199
        (*Define the step procedure*)
200
        (pp_step_definition env typed_submachines) (machine, opt_spec_machine, guarantees)
201
    in
202
    fprintf fmt "%a%t%a;@."
203
      
204
      (* Include all the required packages*)
205
      (Utils.fprintf_list ~sep:";@," (pp_with AdaPrivate)) packages
206
      (Utils.pp_final_char_if_non_empty ";@,@," packages)
207
      
208
      (*Print package*)
209
      (pp_package (pp_package_name machine) [] true ) pp_content
210

    
211
end
212

    
213
(* Local Variables: *)
214
(* compile-command: "make -C ../../.." *)
215
(* End: *)
(3-3/11)