Project

General

Profile

Statistics
| Branch: | Tag: | Revision:

lustrec / src / main_lustre_compiler.ml @ 13507742

History | View | Annotate | Download (15.1 KB)

1
(********************************************************************)
2
(*                                                                  *)
3
(*  The LustreC compiler toolset   /  The LustreC Development Team  *)
4
(*  Copyright 2012 -    --   ONERA - CNRS - INPT                    *)
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
open Log
14

    
15
open Utils
16
open LustreSpec
17
open Compiler_common
18
 
19
exception StopPhase1 of program
20

    
21
let usage = "Usage: lustrec [options] \x1b[4msource file\x1b[0m"
22

    
23
let extensions = [".ec"; ".lus"; ".lusi"]
24

    
25
(* print a .lusi header file from a source prog *)
26
let print_lusi prog dirname basename extension =
27
  let header = Lusic.extract_header dirname basename prog in
28
  let header_name = dirname ^ "/" ^ basename ^ extension in
29
  let h_out = open_out header_name in
30
  let h_fmt = formatter_of_out_channel h_out in
31
  begin
32
    Typing.uneval_prog_generics header;
33
    Clock_calculus.uneval_prog_generics header;
34
    Printers.pp_lusi_header h_fmt basename header;
35
    close_out h_out
36
  end
37

    
38
(* compile a .lusi header file *)
39
let compile_header dirname  basename extension =
40
  let destname = !Options.dest_dir ^ "/" ^ basename in
41
  let header_name = basename ^ extension in
42
  let lusic_ext = extension ^ "c" in
43
  begin
44
    Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v>");
45
    let header = parse_header true (dirname ^ "/" ^ header_name) in
46
    ignore (Modules.load_header ISet.empty header);
47
    ignore (check_top_decls header);
48
    create_dest_dir ();
49
    Log.report ~level:1
50
      (fun fmt -> fprintf fmt ".. generating compiled header file %sc@," (destname ^ extension));
51
    Lusic.write_lusic true header destname lusic_ext;
52
    generate_lusic_header destname lusic_ext;
53
    Log.report ~level:1 (fun fmt -> fprintf fmt ".. done !@ @]@ ")
54
  end
55

    
56
(* check whether a source file has a compiled header,
57
   if not, generate the compiled header *)
58
let compile_source_to_header prog computed_types_env computed_clocks_env dirname basename extension =
59
  let destname = !Options.dest_dir ^ "/" ^ basename in
60
  let lusic_ext = extension ^ "c" in
61
  let header_name = destname ^ lusic_ext in
62
  begin
63
    if not (Sys.file_exists header_name) then
64
      begin
65
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. generating compiled header file %s@," header_name);
66
	Lusic.write_lusic false (Lusic.extract_header dirname basename prog) destname lusic_ext;
67
	Lusic.print_lusic_to_h destname lusic_ext
68
      end
69
    else
70
      let lusic = Lusic.read_lusic destname lusic_ext in
71
      if not lusic.Lusic.from_lusi then
72
	begin
73
	  Log.report ~level:1 (fun fmt -> fprintf fmt ".. generating compiled header file %s@," header_name);
74
       	  Lusic.write_lusic false (Lusic.extract_header dirname basename prog) destname lusic_ext;
75
	  (*List.iter (fun top_decl -> Format.eprintf "lusic: %a@." Printers.pp_decl top_decl) lusic.Lusic.contents;*)
76
	  Lusic.print_lusic_to_h destname lusic_ext
77
	end
78
      else
79
	begin
80
	  Log.report ~level:1 (fun fmt -> fprintf fmt ".. loading compiled header file %s@," header_name);
81
	  Modules.check_dependency lusic destname;
82
	  let header = lusic.Lusic.contents in
83
	  let (declared_types_env, declared_clocks_env) = get_envs_from_top_decls header in
84
	  check_compatibility
85
	    (prog, computed_types_env, computed_clocks_env)
86
	    (header, declared_types_env, declared_clocks_env)
87
	end
88
  end
89

    
90

    
91

    
92
let dynamic_checks () =
93
  match !Options.output, !Options.spec with
94
  | "C", "C" -> true
95
  | _ -> false
96
     
97
(* From prog to prog *)
98
let stage1 prog dirname basename =
99
  (* Removing automata *)
100
  let prog = expand_automata prog in
101
  Log.report ~level:4 (fun fmt -> fprintf fmt ".. after automata expansion:@,  @[<v 2>@,%a@]@ " Printers.pp_prog prog);
102

    
103
  (* Importing source *)
104
  let _ = Modules.load_program ISet.empty prog in
105

    
106
  (* Extracting dependencies *)
107
  let dependencies, type_env, clock_env = import_dependencies prog in
108

    
109
  (* Sorting nodes *)
110
  let prog = SortProg.sort prog in
111

    
112
  (* Perform inlining before any analysis *)
113
  let orig, prog =
114
    if !Options.global_inline && !Options.main_node <> "" then
115
      (if !Options.witnesses then prog else []),
116
      Inliner.global_inline basename prog type_env clock_env
117
    else (* if !Option.has_local_inline *)
118
      [],
119
      Inliner.local_inline basename prog type_env clock_env
120
  in
121

    
122
  (* Checking stateless/stateful status *)
123
  if Plugins.check_force_stateful () then
124
    force_stateful_decls prog
125
  else
126
    check_stateless_decls prog;
127

    
128
  (* Typing *)
129
  let computed_types_env = type_decls type_env prog in
130

    
131
  (* Clock calculus *)
132
  let computed_clocks_env = clock_decls clock_env prog in
133

    
134
  (* Generating a .lusi header file only *)
135
  if !Options.lusi then
136
    (* We stop here the processing and produce the current prog. It will be
137
       exported as a lusi *)
138
    raise (StopPhase1 prog);
139

    
140
 (* Optimization of prog: 
141
     - Unfold consts 
142
     - eliminate trivial expressions
143
 *)
144
  (*
145
  let prog = 
146
    if !Options.const_unfold || !Options.optimization >= 5 then
147
      begin
148
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. eliminating constants and aliases@,");
149
	Optimize_prog.prog_unfold_consts prog
150
      end
151
    else
152
      prog
153
  in
154
  *)
155
  (* Delay calculus *)
156
  (* TO BE DONE LATER (Xavier)
157
    if(!Options.delay_calculus)
158
    then
159
    begin
160
    Log.report ~level:1 (fun fmt -> fprintf fmt ".. initialisation analysis@?");
161
    try
162
    Delay_calculus.delay_prog Basic_library.delay_env prog
163
    with (Delay.Error (loc,err)) as exc ->
164
    Location.print loc;
165
    eprintf "%a" Delay.pp_error err;
166
    Utils.track_exception ();
167
    raise exc
168
    end;
169
  *)
170

    
171
  (* Creating destination directory if needed *)
172
  create_dest_dir ();
173

    
174
  (* Compatibility with Lusi *)
175
  (* Checking the existence of a lusi (Lustre Interface file) *)
176
  let extension = ".lusi" in
177
  compile_source_to_header prog computed_types_env computed_clocks_env dirname basename extension;
178

    
179
  Typing.uneval_prog_generics prog;
180
  Clock_calculus.uneval_prog_generics prog;
181

    
182
  if !Options.global_inline && !Options.main_node <> "" && !Options.witnesses then
183
    begin
184
      let orig = Corelang.copy_prog orig in
185
      Log.report ~level:1 (fun fmt -> fprintf fmt ".. generating witness file@,");
186
      check_stateless_decls orig;
187
      let _ = Typing.type_prog type_env orig in
188
      let _ = Clock_calculus.clock_prog clock_env orig in
189
      Typing.uneval_prog_generics orig;
190
      Clock_calculus.uneval_prog_generics orig;
191
      Inliner.witness
192
	basename
193
	!Options.main_node
194
	orig prog type_env clock_env
195
    end;
196
  
197
  (* Computes and stores generic calls for each node,
198
     only useful for ANSI C90 compliant generic node compilation *)
199
  if !Options.ansi then Causality.NodeDep.compute_generic_calls prog;
200
  (*Hashtbl.iter (fun id td -> match td.Corelang.top_decl_desc with
201
    Corelang.Node nd -> Format.eprintf "%s calls %a" id
202
    Causality.NodeDep.pp_generic_calls nd | _ -> ()) Corelang.node_table;*)
203

    
204
  (* If some backend involving dynamic checks are active, then node annotations become runtime checks *)
205
  let prog =
206
    if dynamic_checks () then
207
      Spec.enforce_spec_prog prog
208
    else
209
      prog
210
  in
211
  
212
  (* Normalization phase *)
213
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. normalization@,");
214
  let prog = Normalization.normalize_prog ~backend:!Options.output prog in
215
  Log.report ~level:2 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Printers.pp_prog prog);
216

    
217
  let prog =
218
    if !Options.mpfr
219
    then
220
      begin
221
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. targetting MPFR library@,");
222
	Mpfr.inject_prog prog
223
      end
224
    else
225
      begin
226
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. keeping floating-point numbers@,");
227
	prog
228
      end in
229
  Log.report ~level:2 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Printers.pp_prog prog);
230

    
231
  (* Checking array accesses *)
232
  if !Options.check then
233
    begin
234
      Log.report ~level:1 (fun fmt -> fprintf fmt ".. checking array accesses@,");
235
      Access.check_prog prog;
236
    end;
237

    
238
  prog, dependencies
239

    
240
(* from source to machine code, with optimization *)
241
let stage2 prog =    
242
  (* Computation of node equation scheduling. It also breaks dependency cycles
243
     and warns about unused input or memory variables *)
244
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. scheduling@,");
245
  let prog, node_schs = Scheduling.schedule_prog prog in
246
  Log.report ~level:1 (fun fmt -> fprintf fmt "%a"              Scheduling.pp_warning_unused node_schs);
247
  Log.report ~level:3 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Scheduling.pp_schedule node_schs);
248
  Log.report ~level:3 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Scheduling.pp_fanin_table node_schs);
249
  Log.report ~level:5 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Scheduling.pp_dep_graph node_schs);
250
  Log.report ~level:3 (fun fmt -> fprintf fmt "@[<v 2>@ %a@]@," Printers.pp_prog prog);
251

    
252

    
253
  (* TODO Salsa optimize prog: 
254
     - emits warning for programs with pre inside expressions
255
     - make sure each node arguments and memory is bounded by a local annotation
256
     - introduce fresh local variables for each real pure subexpression
257
  *)
258
  (* DFS with modular code generation *)
259
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. machines generation@,");
260
  let machine_code = Machine_code.translate_prog prog node_schs in
261

    
262
  Log.report ~level:3 (fun fmt -> fprintf fmt ".. generated machines (unoptimized):@ %a@ "Machine_code.pp_machines machine_code);
263

    
264
  (* Optimize machine code *)
265
  Optimize_machine.optimize prog node_schs machine_code
266

    
267

    
268
(* printing code *)
269
let stage3 prog machine_code dependencies basename =
270
  let basename    =  Filename.basename basename in
271
  match !Options.output with
272
    "C" -> 
273
      begin
274
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. C code generation@,");
275
	C_backend.translate_to_c
276
	  (* alloc_header_file source_lib_file source_main_file makefile_file *)
277
	  basename prog machine_code dependencies
278
      end
279
  | "java" ->
280
     begin
281
       (Format.eprintf "internal error: sorry, but not yet supported !"; assert false)
282
     (*let source_file = basename ^ ".java" in
283
       Log.report ~level:1 (fun fmt -> fprintf fmt ".. opening file %s@,@?" source_file);
284
       let source_out = open_out source_file in
285
       let source_fmt = formatter_of_out_channel source_out in
286
       Log.report ~level:1 (fun fmt -> fprintf fmt ".. java code generation@,@?");
287
       Java_backend.translate_to_java source_fmt basename normalized_prog machine_code;*)
288
     end
289
  | "horn" ->
290
     begin
291
       let destname = !Options.dest_dir ^ "/" ^ basename in
292
       let source_file = destname ^ ".smt2" in (* Could be changed *)
293
       let source_out = open_out source_file in
294
       let fmt = formatter_of_out_channel source_out in
295
       Log.report ~level:1 (fun fmt -> fprintf fmt ".. hornification@,");
296
       Horn_backend.translate fmt basename prog (Machine_code.arrow_machine::machine_code);
297
       (* Tracability file if option is activated *)
298
       if !Options.traces then (
299
	 let traces_file = destname ^ ".traces.xml" in (* Could be changed *)
300
	 let traces_out = open_out traces_file in
301
	 let fmt = formatter_of_out_channel traces_out in
302
         Log.report ~level:1 (fun fmt -> fprintf fmt ".. tracing info@,");
303
	 Horn_backend_traces.traces_file fmt basename prog machine_code;
304
       )
305
     end
306
  | "lustre" ->
307
     begin
308
       let destname = !Options.dest_dir ^ "/" ^ basename in
309
       let source_file = destname ^ ".lustrec.lus" in (* Could be changed *)
310
       let source_out = open_out source_file in
311
       let fmt = formatter_of_out_channel source_out in
312
       Printers.pp_prog fmt prog;
313
       (*	Lustre_backend.translate fmt basename normalized_prog machine_code *)
314
       ()
315
     end
316
  | "emf" ->
317
     begin
318
       let destname = !Options.dest_dir ^ "/" ^ basename in
319
       let source_file = destname ^ ".emf" in (* Could be changed *)
320
       let source_out = open_out source_file in
321
       let fmt = formatter_of_out_channel source_out in
322
       EMF_backend.translate fmt basename prog machine_code;
323
       ()
324
     end
325

    
326
  | _ -> assert false
327

    
328
(* compile a .lus source file *)
329
let rec compile_source dirname basename extension =
330
  let source_name = dirname ^ "/" ^ basename ^ extension in
331

    
332
  Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 0>");
333

    
334
  (* Parsing source *)
335
  let prog = parse_source source_name in
336

    
337
  let prog =
338
    if !Options.mpfr then
339
      Mpfr.mpfr_module::prog
340
    else
341
      prog
342
  in
343
  let prog, dependencies = 
344
    Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 2>.. Phase 1 : Normalisation@,");
345
    try 
346
      stage1 prog dirname basename
347
    with StopPhase1 prog -> (
348
      if !Options.lusi then
349
	begin
350
	  let lusi_ext = extension ^ "i" in
351
	  Log.report ~level:1 (fun fmt -> fprintf fmt ".. generating interface file %s@ " (basename ^ lusi_ext));
352
	  print_lusi prog dirname basename lusi_ext;
353
	  Log.report ~level:1 (fun fmt -> fprintf fmt ".. done !@ @]@.");
354
	  exit 0
355
	end
356
      else
357
        assert false
358
    )
359
  in
360
  Log.report ~level:1 (fun fmt -> fprintf fmt "@]@,");
361
  Log.report ~level:3 (fun fmt -> fprintf fmt ".. Normalized program:@ %a@ "Printers.pp_prog prog);
362

    
363
  Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 2>.. Phase 2 : Machines generation@,");
364

    
365
  let machine_code = 
366
    stage2 prog 
367
  in
368

    
369
  Log.report ~level:1 (fun fmt -> fprintf fmt "@]@ ");
370
  Log.report ~level:3 (fun fmt -> fprintf fmt ".. Generated machines:@ %a@ "Machine_code.pp_machines machine_code);
371

    
372
  if Scopes.Plugin.show_scopes () then
373
    begin
374
      let all_scopes = Scopes.compute_scopes prog !Options.main_node in
375
      (* Printing scopes *)
376
      if !Options.verbose_level >= 1 then
377
	Format.printf "Possible scopes are:@   ";
378
      Format.printf "@[<v>%a@ @]@ @?" Scopes.print_scopes all_scopes;
379
      exit 0
380
	
381
    end;
382

    
383
  let machine_code = Plugins.refine_machine_code prog machine_code in
384
  
385
  stage3 prog machine_code dependencies basename;
386
  begin
387
    Log.report ~level:1 (fun fmt -> fprintf fmt ".. done !@ @]@.");
388
    (* We stop the process here *)
389
    exit 0
390
  end
391

    
392
let compile dirname basename extension =
393
  match extension with
394
  | ".lusi"  -> compile_header dirname basename extension
395
  | ".lus"   -> compile_source dirname basename extension
396
  | _        -> assert false
397

    
398
let anonymous filename =
399
  let ok_ext, ext = List.fold_left
400
    (fun (ok, ext) ext' ->
401
      if not ok && Filename.check_suffix filename ext' then
402
	true, ext'
403
      else
404
	ok, ext)
405
    (false, "") extensions in
406
  if ok_ext then
407
    let dirname = Filename.dirname filename in
408
    let basename = Filename.chop_suffix (Filename.basename filename) ext in
409
    compile dirname basename ext
410
  else
411
    raise (Arg.Bad ("Can only compile *.lusi, *.lus or *.ec files"))
412

    
413
let _ =
414
  Global.initialize ();
415
  Corelang.add_internal_funs ();
416
  try
417
    Printexc.record_backtrace true;
418

    
419
    let options = Options_management.lustrec_options @ (Plugins.options ()) in
420
    
421
    Arg.parse options anonymous usage
422
  with
423
  | Parse.Error _
424
  | Types.Error (_,_) | Clocks.Error (_,_)
425
  | Corelang.Error _ (*| Task_set.Error _*)
426
  | Causality.Error _ -> exit 1
427
  | Sys_error msg -> (eprintf "Failure: %s@." msg)
428
  | exc -> (track_exception (); raise exc)
429

    
430
(* Local Variables: *)
431
(* compile-command:"make -C .." *)
432
(* End: *)