cduce.ml 12.8 KB
Newer Older
1
open Location
2
open Ident
3

4
exception Escape of exn
5
6
7
exception InvalidInputFilename of string
exception InvalidObjectFilename of string

8
9
let extra_specs = ref []

10
11
(* if set to false toplevel exception aren't cought. 
 * Useful for debugging with OCAMLRUNPARAM="b" *)
12
13
let catch_exceptions = true

14
15
16
17
18
19
20
21
(* retuns a filename without the suffix suff if any *) 
let prefix filename suff =
  if Filename.check_suffix filename suff then
    try
      Filename.chop_extension filename
    with Invalid_argument filename -> failwith "Not a point in the suffix?"
  else filename

22
let toplevel = ref false
23
let verbose = ref false
24
let silent = ref false
25

26
let typing_env = State.ref "Cduce.typing_env" Builtin.env
27
let compile_env = State.ref "Cduce.compile_env" Compile.empty_toplevel
28

29
let get_global_value cenv v =
30
  Eval.var (Compile.find v !compile_env)
31
32
33

let get_global_type v =
  Typer.find_value v !typing_env
34

35
36
37
38
39
let rec is_abstraction = function
  | Ast.Abstraction _ -> true
  | Ast.LocatedExpr (_,e) -> is_abstraction e
  | _ -> false

40
let print_norm ppf d = 
41
  Types.Print.print ppf ((*Types.normalize*) d)
42

43
let print_sample ppf s =
44
  Sample.print ppf s
45

46
let print_protect ppf s =
47
  Format.fprintf ppf "%s" s
48

49
let print_value ppf v =
50
  Value.print ppf v
51

52
53
54
55
let dump_value ppf x t v =
  Format.fprintf ppf "@[val %a : @[%a = %a@]@]@."
    U.print (Id.value x) print_norm t print_value v

56
57
58
let dump_env ppf tenv cenv =
  Format.fprintf ppf "Types:%a@." Typer.dump_types tenv;
  Format.fprintf ppf "Namespace prefixes:@\n%a" Typer.dump_ns tenv;
59
  Format.fprintf ppf "Namespace prefixes used for pretty-printing:@.%t"
60
    Ns.InternalPrinter.dump;
61
  Format.fprintf ppf "Schemas: %s@."
62
    (String.concat " " (List.map U.get_str (Typer.get_schema_names tenv)));
63
  Format.fprintf ppf "Values:@.";
64
65
  Typer.iter_values tenv
    (fun x t -> dump_value ppf x t (get_global_value cenv x))
66

67
68
69
let directive_help ppf =
  Format.fprintf ppf
"Toplevel directives:
70
71
72
73
74
75
76
77
  #quit;;                 quit the interpreter
  #env;;                  dump current environment
  #reinit_ns;;            reinitialize namespace processing
  #help;;                 shows this help message
  #dump_value <expr>;;    dump an XML-ish representation of the resulting
                          value of a given expression
  #print_schema <name>;;
  #print_type <name>;;
78
79
  #silent;;               turn off outputs from the toplevel
  #verbose;;              turn on outputs from the toplevel
80
81
"

82
let rec print_exn ppf = function
83
84
  | Location (loc, w, exn) ->
      Format.fprintf ppf "Error %a:@." Location.print_loc (loc,w);
85
      Location.html_hilight (loc,w); 
86
      print_exn ppf exn
87
  | Value.CDuceExn v ->
88
      Format.fprintf ppf "Uncaught CDuce exception: @[%a@]@."
89
        print_value v
90
  | Typer.WrongLabel (t,l) ->
91
      Format.fprintf ppf "Wrong record selection; field %a " 
92
        Label.print (LabelPool.value l);
93
      Format.fprintf ppf "not present in an expression of type:@.%a@."
94
        print_norm t
95
  | Typer.ShouldHave (t,msg) ->
96
      Format.fprintf ppf "This expression should have type:@.%a@.%a@." 
97
        print_norm t
98
        print_protect msg
99
  | Typer.ShouldHave2 (t1,msg,t2) ->
100
      Format.fprintf ppf "This expression should have type:@.%a@.%a %a@." 
101
        print_norm t1
102
        print_protect msg
103
        print_norm t2
104
  | Typer.Error s ->
105
      Format.fprintf ppf "%a@." print_protect s
106
  | Typer.Constraint (s,t) ->
107
      Format.fprintf ppf "This expression should have type:@.%a@." 
108
        print_norm t;
109
      Format.fprintf ppf "but its inferred type is:@.%a@." 
110
        print_norm s;
111
      Format.fprintf ppf "which is not a subtype, as shown by the sample:@.%a@." 
112
	print_sample (Sample.get (Types.diff s t))
113
  | Typer.NonExhaustive t ->
114
115
      Format.fprintf ppf "This pattern matching is not exhaustive@.";
      Format.fprintf ppf "Residual type:@.%a@."
116
	print_norm t;
117
      Format.fprintf ppf "Sample:@.%a@." print_sample (Sample.get t)
118
119
120
  | Typer.UnboundId (x,tn) ->
      Format.fprintf ppf "Unbound identifier %a%s@." U.print (Id.value x)
        (if tn then " (it is a type name)" else "")
121
122
123
124
  | Typer.UnboundExtId (cu,x) ->
      Format.fprintf ppf "Unbound external identifier %a:%a@." 
	U.print (Types.CompUnit.value cu)
	U.print (Id.value x)
125
126
127
  | Ulexer.Error (i,j,s) ->
      let loc = Location.loc_of_pos (i,j), `Full in
      Format.fprintf ppf "Error %a:@." Location.print_loc loc;
128
129
      Location.html_hilight loc;
      Format.fprintf ppf "%s" s
130
  | Parser.Error s | Stream.Error s -> 
131
      Format.fprintf ppf "Parsing error: %a@." print_protect s
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
  | Librarian.InconsistentCrc id ->
      Format.fprintf ppf "Link error:@.";
      let name = Encodings.Utf8.to_string (Types.CompUnit.value id) in
      Format.fprintf ppf "Inconsistent checksum (compilation unit: %s)@."
	name
  | Librarian.NoImplementation id ->
      Format.fprintf ppf "Link error:@.";
      let name = Encodings.Utf8.to_string (Types.CompUnit.value id) in
      Format.fprintf ppf "No implementation found for compilation unit: %s@."
	name
  | Librarian.Loop id ->
      Format.fprintf ppf "Compilation error:@.";
      let name = Encodings.Utf8.to_string (Types.CompUnit.value id) in
      Format.fprintf ppf "Loop between compilation unit (compilation unit: %s)@."
	name
  | InvalidInputFilename f ->
      Format.fprintf ppf "Compilation error:@.";
      Format.fprintf ppf "Source filename must have extension .cd@.";
  | InvalidObjectFilename f ->
      Format.fprintf ppf "Compilation error:@.";
152
      Format.fprintf ppf "Object filename must have extension .cdo and no path@.";
153
154
155
156
  | Librarian.InvalidObject f ->
      Format.fprintf ppf "Invalid object file %s@." f
  | Librarian.CannotOpen f ->
      Format.fprintf ppf "Cannot open file %s@." f
157
  | Location.Generic s ->
158
      Format.fprintf ppf "%a@." print_protect s
159
  | exn ->
160
(*      raise exn *)
161
      Format.fprintf ppf "%a@." print_protect (Printexc.to_string exn)
162

163

164
165
let eval_quiet tenv cenv e = 
  let (e,_) = Typer.type_expr tenv e in
166
  let e = Compile.compile_expr cenv e in 
167
168
  Eval.expr e

169
let debug ppf tenv cenv = function
170
  | `Subtype (t1,t2) ->
171
      Format.fprintf ppf "[DEBUG:subtype]@.";
172
173
      let t1 = Types.descr (Typer.typ tenv t1)
      and t2 = Types.descr (Typer.typ tenv t2) in
174
175
      let s = Types.subtype t1 t2 in
      Format.fprintf ppf "%a %a %a : %b@." print_norm t1 print_protect "<=" print_norm t2 s
176
  | `Sample t ->
177
178
      Format.fprintf ppf "[DEBUG:sample]@.";
      (try
179
	 let t = Types.descr (Typer.typ tenv t) in
180
181
182
	 Format.fprintf ppf "%a@." print_sample (Sample.get t)
       with Not_found ->
	 Format.fprintf ppf "Empty type : no sample !@.")
183
  | `Filter (t,p) -> 
184
185
      let t = Typer.typ tenv t
      and p = Typer.pat tenv p in
186
187
188
      Format.fprintf ppf "[DEBUG:filter t=%a p=%a]@."
	Types.Print.print (Types.descr t)
	Patterns.Print.print (Patterns.descr p);
189
      let f = Patterns.filter (Types.descr t) p in
190
      IdMap.iteri (fun x t ->
191
		   Format.fprintf ppf " %a:%a@." U.print (Id.value x)
192
193
		     print_norm (Types.descr t)) f
  | `Accept p ->
194
      Format.fprintf ppf "[DEBUG:accept]@.";
195
      let p = Typer.pat tenv p in
196
      let t = Patterns.accept p in
197
      Format.fprintf ppf " %a@." Types.Print.print (Types.descr t)
198
  | `Compile (t,pl) ->
199
      Format.fprintf ppf "[DEBUG:compile]@.";
200
      let t = Typer.typ tenv t
201
      and pl = List.map (Typer.pat tenv) pl in  
202
203
204
      Patterns.Compile.debug_compile ppf t pl;
      Format.fprintf ppf "@.";

205
(*
206
       Patterns.demo_compile ppf (Types.descr t) (List.map Patterns.descr pl)  
207
*)
208
  | `Explain (t0,t,e) ->
209
      Format.fprintf ppf "[DEBUG:explain]@.";
210
211
212
      let t = Types.descr (Typer.typ tenv t) in
      let t0 = Types.descr (Typer.typ tenv t0) in
      (match Explain.explain t0 t (eval_quiet tenv cenv e) with
213
	 | Some p ->
214
	     Format.fprintf ppf "%a@."
215
	       Explain.print p
216
	 | None ->
217
	     Format.fprintf ppf "Value has given type@.")
218
219
220
221
222
223
224
225
226
227
228
229
230
  | `Single t ->
      Format.fprintf ppf "[DEBUG:single]@.";
      let t = Typer.typ tenv t in
      (try 
	 let c = Sample.single (Types.descr t) in
	 Format.fprintf ppf "Constant:%a@." Types.Print.print_const c
       with
	 | Exit -> Format.fprintf ppf "Non constant@." 
	 | Not_found ->  Format.fprintf ppf "Empty@.")
  | `Approx (p,t) ->
      Format.fprintf ppf "[DEBUG:approx]@.";
      let t = Typer.typ tenv t in
      let p = Typer.pat tenv p in
231
      Patterns.demo ppf (Patterns.descr p) (Types.descr t); 
232
(*
233
234
235
      let (x,c) = Patterns.approx (Patterns.descr p) (Types.descr t) in
      List.iter (fun x -> Format.fprintf ppf "%a=* " U.print (Id.value x)) x;
      List.iter
236
	(fun (x,c) ->
237
238
239
	   Format.fprintf ppf "%a=%a "
	     U.print (Id.value x) 
	     Types.Print.print_const c
240
	) c;  *)
241
      Format.fprintf ppf "@."
242

243
244
245
246
247
248
249
250
251
252
let flush_ppf ppf = Format.fprintf ppf "@."

let directive ppf tenv cenv = function
  | `Debug d ->
      debug ppf tenv cenv d
  | `Quit ->
      (if !toplevel then raise End_of_file)
  | `Env ->
      dump_env ppf tenv cenv
  | `Print_schema schema ->
253
254
255
      let uri = Typer.find_schema schema tenv in
      let sch = Typer.get_schema uri in
      Schema_common.print_schema ppf sch;
256
257
258
259
260
      flush_ppf ppf
  | `Print_type name ->
      Typer.dump_type ppf tenv name;
      flush_ppf ppf
  | `Print_schema_type schema_ref ->
261
262
      Typer.dump_schema_type ppf tenv schema_ref;
      flush_ppf ppf
263
264
265
266
267
268
269
  | `Reinit_ns ->
      Typer.set_ns_table_for_printer tenv
  | `Help ->
      directive_help ppf
  | `Dump pexpr ->
      Value.dump_xml ppf (eval_quiet tenv cenv pexpr);
      flush_ppf ppf
270
271
272
273
  | `Silent ->
      silent := true
  | `Verbose ->
      silent := false
274
275
276
277
278
279
280
281
282
283

let print_id_opt ppf = function
  | None -> Format.fprintf ppf "-"
  | Some id -> Format.fprintf ppf "val %a" U.print (Id.value id)

let print_value_opt ppf = function
  | None -> ()
  | Some v -> Format.fprintf ppf " = %a" print_value v

let show ppf id t v =
284
285
  if !silent then ()
  else Format.fprintf ppf "@[%a : @[%a%a@]@]@."
286
287
288
289
290
291
292
293
    print_id_opt id 
    print_norm t 
    print_value_opt v

let phrases ppf phs =
  let (tenv,cenv,_) = 
    Compile.comp_unit 
      ~run:true ~show:(show ppf) 
294
      ~loading:Librarian.import_and_run
295
296
297
298
      ~directive:(directive ppf)
      !typing_env !compile_env phs in
  typing_env := tenv;
  compile_env := cenv
299

300
301
302
let catch_exn ppf_err exn =
  if not catch_exceptions then raise exn;
  match exn with
303
304
305
306
307
308
309
310
  | (End_of_file | Failure _ | Not_found | Invalid_argument _ | Sys.Break) 
      as e -> 
      raise e
  | exn -> 
      print_exn ppf_err exn;
      Format.fprintf ppf_err "@."

let parse rule input =
311
312
  try Parser.localize_exn (fun () -> rule input)
  with e -> Parser.sync (); raise e
313

314
let run rule ppf ppf_err input =
315
  try phrases ppf (parse rule input); true
316
  with Escape exn -> raise exn | exn -> catch_exn ppf_err exn; false
317

318
let topinput = run Parser.top_phrases
319
let script = run Parser.prog
320

321

322
let compile src out_dir =
323
  try
324
325
326
327
328
329
330
331
332
    if not (Filename.check_suffix src ".cd")
    then raise (InvalidInputFilename src);
    let cu = Filename.chop_suffix (Filename.basename src) ".cd" in
    let out_dir = 
      match out_dir with
	| None -> Filename.dirname src
	| Some x -> x in
    let out = Filename.concat out_dir (cu ^ ".cdo") in
    let id = Types.CompUnit.mk (U.mk_latin1 cu) in
333
    Librarian.compile !verbose cu id src;
334
    Librarian.save cu id out;
335
    exit 0
336
337
  with exn -> catch_exn Format.err_formatter exn; exit 1
  
338
let compile_run src =
339
  try
340
341
342
343
344
345
    let cu = 
      if src = "" then "<stdin>"
      else
	if not (Filename.check_suffix src ".cd")
	then raise (InvalidInputFilename src)
	else Filename.chop_suffix (Filename.basename src) ".cd" in
346
    let id = Types.CompUnit.mk (U.mk_latin1 cu) in
347
    Librarian.compile !verbose cu id src;
348
    Librarian.run id
349
  with exn -> catch_exn Format.err_formatter exn; exit 1
350

351
let run obj =  
352
  try
353
    if not (Filename.check_suffix obj ".cdo") || (Filename.basename obj <> obj)
354
355
356
    then raise (InvalidObjectFilename obj);
    let cu = Filename.chop_suffix (Filename.basename obj) ".cdo" in
    let id = Types.CompUnit.mk (U.mk_latin1 cu) in
357
    Librarian.import_and_run id
358
  with exn -> catch_exn Format.err_formatter exn; exit 1
359

360
361

let dump_env ppf = dump_env ppf !typing_env !compile_env
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388

let eval s =
  let st = Stream.of_string s in
  let phs = parse Parser.prog st in
  let vals = ref [] in
  let show id t v =
    match id,v with
      | Some id, Some v -> 
	  let id = Id.value id in
	  vals := (Some id,v) :: !vals
      | None, Some v ->
	  vals := (None,v) :: !vals
      | _ -> assert false
  in
  let r () = 
    ignore (Compile.comp_unit 
	      ~run:true ~show Builtin.env Compile.empty_toplevel phs) in
  Eval.new_stack r ();
  List.rev !vals
  
let eval s =
  try eval s
  with exn -> 
    let b = Buffer.create 1024 in
    let ppf = Format.formatter_of_buffer b in
    print_exn ppf exn;
    Format.fprintf ppf "@.";
389
    Value.failwith' (Buffer.contents b)
390
	   
391

392
let () =        
393
  Operators.register_fun "eval_expr" Builtin_defs.string_latin1 Types.any
394
395
396
397
398
  (fun v ->
     match eval (Value.cduce2ocaml_string v) with
       | [ (None,v) ] -> v
       | _ -> Value.failwith' "eval: the string must evaluate to a single value"
  )
399