Project

General

Profile

Statistics
| Branch: | Tag: | Revision:

lustrec / src / compiler_common.ml @ 3b2bd83d

History | View | Annotate | Download (8.97 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 Utils
13
open Format 
14
open LustreSpec
15
open Corelang
16

    
17
let check_main () =
18
  if !Options.main_node = "" then
19
    begin
20
      eprintf "Code generation error: %a@." pp_error No_main_specified;
21
      raise (Error (Location.dummy_loc, No_main_specified))
22
    end
23

    
24
let create_dest_dir () =
25
  begin
26
    if not (Sys.file_exists !Options.dest_dir) then
27
      begin
28
	Log.report ~level:1 (fun fmt -> fprintf fmt ".. creating destination directory@,");
29
	Unix.mkdir !Options.dest_dir (Unix.stat ".").Unix.st_perm
30
      end;
31
    if (Unix.stat !Options.dest_dir).Unix.st_kind <> Unix.S_DIR then
32
      begin
33
	eprintf "Failure: destination %s is not a directory.@.@." !Options.dest_dir;
34
	exit 1
35
      end
36
  end
37

    
38
(* Loading Lusi file and filling type tables with parsed
39
   functions/nodes *)
40
let parse_header own filename =
41
  Location.set_input filename;
42
  let h_in = open_in filename in
43
  let lexbuf = Lexing.from_channel h_in in
44
  Location.init lexbuf filename;
45
  (* Parsing *)
46
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. parsing header file %s@ " filename);
47
    try
48
      let header = Parse.header Parser_lustre.header Lexer_lustre.token lexbuf in
49
      (*ignore (Modules.load_header ISet.empty header);*)
50
      close_in h_in;
51
      header
52
    with
53
    | (Parse.Error err) as exc -> 
54
      Parse.report_error err;
55
      raise exc
56
    | Corelang.Error (loc, err) as exc -> (
57
      eprintf "Parsing error: %a%a@."
58
	Corelang.pp_error err
59
	Location.pp_loc loc;
60
      raise exc
61
    )
62

    
63
let parse_source source_name =
64
  (* Loading the input file *)
65
  Location.set_input source_name;
66
  let s_in = open_in source_name in
67
  let lexbuf = Lexing.from_channel s_in in
68
  Location.init lexbuf source_name;
69

    
70
  (* Parsing *)
71
  Log.report ~level:1 
72
    (fun fmt -> fprintf fmt ".. parsing source file %s@," source_name);
73
  try
74
    let prog = Parse.prog Parser_lustre.prog Lexer_lustre.token lexbuf in
75
    (*ignore (Modules.load_program ISet.empty prog);*)
76
    close_in s_in;
77
    prog
78
  with
79
  | (Parse.Error err) as exc -> 
80
    Parse.report_error err;
81
    raise exc
82
  | Corelang.Error (loc, err) as exc ->
83
    eprintf "Parsing error: %a%a@."
84
      Corelang.pp_error err
85
      Location.pp_loc loc;
86
    raise exc
87

    
88
let expand_automata decls =
89
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. expanding automata@ ");
90
  try
91
    Automata.expand_decls decls
92
  with (Corelang.Error (loc, err)) as exc ->
93
    eprintf "Automata error: %a%a@."
94
      Corelang.pp_error err
95
      Location.pp_loc loc;
96
    raise exc
97

    
98
let check_stateless_decls decls =
99
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. checking stateless/stateful status@ ");
100
  try
101
    Stateless.check_prog decls
102
  with (Stateless.Error (loc, err)) as exc ->
103
    eprintf "Stateless status error: %a%a@."
104
      Stateless.pp_error err
105
      Location.pp_loc loc;
106
    raise exc
107

    
108
let force_stateful_decls decls =
109
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. forcing stateful status@ ");
110
  try
111
    Stateless.force_prog decls
112
  with (Stateless.Error (loc, err)) as exc ->
113
    eprintf "Stateless status error: %a%a@."
114
      Stateless.pp_error err
115
      Location.pp_loc loc;
116
    raise exc
117

    
118
let type_decls env decls =  
119
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. typing@ ");
120
  let new_env = 
121
    begin
122
      try
123
	Typing.type_prog env decls
124
      with (Types.Error (loc,err)) as exc ->
125
	eprintf "Typing error: %a%a@."
126
	  Types.pp_error err
127
	  Location.pp_loc loc;
128
	raise exc
129
    end 
130
  in
131
  if !Options.print_types then
132
    Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 2>  %a@]@ " Corelang.pp_prog_type decls);
133
  new_env
134
      
135
let clock_decls env decls = 
136
  Log.report ~level:1 (fun fmt -> fprintf fmt ".. clock calculus@ ");
137
  let new_env =
138
    begin
139
      try
140
	Clock_calculus.clock_prog env decls
141
      with (Clocks.Error (loc,err)) as exc ->
142
	eprintf "Clock calculus error: %a%a@." Clocks.pp_error err Location.pp_loc loc;
143
	raise exc
144
    end
145
  in
146
  if !Options.print_clocks then
147
    Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 2>  %a@]@ " Corelang.pp_prog_clock decls);
148
  new_env
149

    
150
let check_top_decls header =
151
  let new_tenv = type_decls Basic_library.type_env header in   (* Typing *)
152
  let new_cenv = clock_decls Basic_library.clock_env header in   (* Clock calculus *)
153
  header, new_tenv, new_cenv
154

    
155
let get_envs_from_const const_decl (ty_env, ck_env) =
156
  (Env.add_value ty_env const_decl.const_id const_decl.const_type,
157
   Env.add_value ck_env const_decl.const_id (Clocks.new_var true))
158

    
159
let get_envs_from_consts const_decls (ty_env, ck_env) =
160
  List.fold_right get_envs_from_const const_decls (ty_env, ck_env)
161

    
162
let rec get_envs_from_top_decl (ty_env, ck_env) top_decl =
163
 match top_decl.top_decl_desc with
164
 | Node nd          -> (Env.add_value ty_env nd.node_id nd.node_type,
165
			Env.add_value ck_env nd.node_id nd.node_clock)
166
 | ImportedNode ind -> (Env.add_value ty_env ind.nodei_id ind.nodei_type,
167
			Env.add_value ck_env ind.nodei_id ind.nodei_clock)
168
 | Const c          -> get_envs_from_const c (ty_env, ck_env)
169
 | TypeDef _        -> List.fold_left get_envs_from_top_decl (ty_env, ck_env) (consts_of_enum_type top_decl)
170
 | Open _           -> (ty_env, ck_env)
171

    
172
(* get type and clock environments from a header *)
173
let get_envs_from_top_decls header =
174
  List.fold_left get_envs_from_top_decl (Env.initial, Env.initial) header
175

    
176
(*
177
 List.fold_right
178
   (fun top_decl (ty_env, ck_env) ->
179
     match top_decl.top_decl_desc with
180
     | Node nd          -> (Env.add_value ty_env nd.node_id nd.node_type,
181
			    Env.add_value ck_env nd.node_id nd.node_clock)
182
     | ImportedNode ind -> (Env.add_value ty_env ind.nodei_id ind.nodei_type,
183
			    Env.add_value ck_env ind.nodei_id ind.nodei_clock)
184
     | Const c          -> get_envs_from_const c (ty_env, ck_env)
185
     | TypeDef _        -> List.fold_left (fun envs top -> consts_of_enum_type top_decl
186
     | Open _           -> (ty_env, ck_env))
187
   header
188
   (Env.initial, Env.initial)
189
 *)
190

    
191
let check_compatibility (prog, computed_types_env, computed_clocks_env) (header, declared_types_env, declared_clocks_env) =
192
  try
193
    (* checking defined types are compatible with declared types*)
194
    Typing.check_typedef_compat header;
195

    
196
    (* checking type compatibility with computed types*)
197
    Typing.check_env_compat header declared_types_env computed_types_env;
198

    
199
    (* checking clocks compatibility with computed clocks*)
200
    Clock_calculus.check_env_compat header declared_clocks_env computed_clocks_env;
201

    
202
    (* checking stateless status compatibility *)
203
    Stateless.check_compat header
204
  with
205
  | (Types.Error (loc,err)) as exc ->
206
    eprintf "Type mismatch between computed type and declared type in lustre interface file: %a%a@."
207
      Types.pp_error err
208
      Location.pp_loc loc;
209
    raise exc
210
  | Clocks.Error (loc, err) as exc ->
211
    eprintf "Clock mismatch between computed clock and declared clock in lustre interface file: %a%a@."
212
      Clocks.pp_error err
213
      Location.pp_loc loc;
214
    raise exc
215
  | Stateless.Error (loc, err) as exc ->
216
    eprintf "Stateless status mismatch between defined status and declared status in lustre interface file: %a%a@."
217
      Stateless.pp_error err
218
      Location.pp_loc loc;
219
    raise exc
220

    
221
let is_stateful topdecl =
222
  match topdecl.top_decl_desc with
223
  | Node nd -> (match nd.node_stateless with Some b -> not b | None -> not nd.node_dec_stateless)
224
  | ImportedNode nd -> not nd.nodei_stateless 
225
  | _ -> false
226

    
227

    
228
let import_dependencies prog =
229
  Log.report ~level:1 (fun fmt -> fprintf fmt "@[<v 2>.. extracting dependencies@,");
230
  let dependencies = Corelang.get_dependencies prog in
231
  let deps =
232
  List.fold_left
233
    (fun (compilation_dep, type_env, clock_env) dep ->
234
      let (local, s) = Corelang.dependency_of_top dep in
235
      let basename = Modules.name_dependency (local, s) in
236
      Log.report ~level:1 (fun fmt -> Format.fprintf fmt "@[<v 0>Library %s@," basename);
237
      let lusic = Modules.import_dependency dep.top_decl_loc (local, s) in
238
      Log.report ~level:1 (fun fmt -> Format.fprintf fmt "@]@ ");
239
      let (lusi_type_env, lusi_clock_env) = get_envs_from_top_decls lusic.Lusic.contents in
240
      let is_stateful = List.exists is_stateful lusic.Lusic.contents in
241
      let new_dep = Dep (local, s, lusic.Lusic.contents, is_stateful ) in
242
      new_dep::compilation_dep,
243
      Env.overwrite type_env lusi_type_env,
244
      Env.overwrite clock_env lusi_clock_env)
245
    ([], Basic_library.type_env, Basic_library.clock_env)
246
    dependencies in
247
  begin
248
    Log.report ~level:1 (fun fmt -> fprintf fmt "@]@ ");
249
    deps
250
  end
251