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

4 5 6
exception InvalidInputFilename of string
exception InvalidObjectFilename of string

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

11 12 13 14 15 16 17 18
(* 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

19
let quiet = ref false
20
let toplevel = ref false
21

22

23
let typing_env = State.ref "Cduce.typing_env" Builtin.env
24
let eval_env = State.ref "Cduce.eval_env" Eval.empty
25 26 27 28 29 30
let compile_env = State.ref "Cduce.compile_env" Compile.empty

let do_compile = ref false

let get_global_value v =
  if !do_compile 
31
  then Eval.L.var (Compile.find v !compile_env)
32
  else Eval.find_value v !eval_env
33 34 35

let get_global_type v =
  Typer.find_value v !typing_env
36

37
let enter_global_value x v t =
38
  typing_env := Typer.enter_value x t !typing_env;
39

40 41
  if !do_compile 
  then (compile_env := Compile.enter_global !compile_env x; Eval.L.push v)
42
  else eval_env := Eval.enter_value x v !eval_env
43
  
44 45 46 47 48
let rec is_abstraction = function
  | Ast.Abstraction _ -> true
  | Ast.LocatedExpr (_,e) -> is_abstraction e
  | _ -> false

49
let print_norm ppf d = 
50
  Location.protect ppf 
51
    (fun ppf -> Types.Print.print ppf ((*Types.normalize*) d))
52

53 54 55 56
let print_sample ppf s =
  Location.protect ppf
    (fun ppf -> Sample.print ppf s)

57 58 59
let print_protect ppf s =
  Location.protect ppf (fun ppf -> Format.fprintf ppf "%s" s)

60 61
let print_value ppf v =
  Location.protect ppf (fun ppf -> Value.print ppf v)
62

63 64 65 66
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

67
let dump_env ppf =
68 69
  Format.fprintf ppf "Types:%a@." Typer.dump_types !typing_env;
  Format.fprintf ppf "Namespace prefixes:@\n%a" Typer.dump_ns !typing_env;
70
  Format.fprintf ppf "Namespace prefixes used for pretty-printing:@.%t"
71
    Ns.InternalPrinter.dump;
72
  Format.fprintf ppf "Schemas: %s@."
73
    (String.concat " " (List.map U.get_str (Typer.get_schema_names ())));
74
  Format.fprintf ppf "Values:@.";
75 76
  Typer.iter_values !typing_env
    (fun x t -> dump_value ppf x t (get_global_value x))
77

78 79 80
let directive_help ppf =
  Format.fprintf ppf
"Toplevel directives:
81 82 83 84 85 86 87 88
  #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>;;
89 90
"

91
let rec print_exn ppf = function
92 93
  | Location (loc, w, exn) ->
      Format.fprintf ppf "Error %a:@." Location.print_loc (loc,w);
94
      Format.fprintf ppf "%a" Location.html_hilight (loc,w); 
95
      print_exn ppf exn
96
  | Value.CDuceExn v ->
97
      Format.fprintf ppf "Uncaught CDuce exception: @[%a@]@."
98
        print_value v
99
  | Typer.WrongLabel (t,l) ->
100
      Format.fprintf ppf "Wrong record selection; field %a " 
101
        Label.print (LabelPool.value l);
102
      Format.fprintf ppf "not present in an expression of type:@.%a@."
103
        print_norm t
104
  | Typer.ShouldHave (t,msg) ->
105
      Format.fprintf ppf "This expression should have type:@.%a@.%a@." 
106
        print_norm t
107
        print_protect msg
108
  | Typer.ShouldHave2 (t1,msg,t2) ->
109
      Format.fprintf ppf "This expression should have type:@.%a@.%a %a@." 
110
        print_norm t1
111
        print_protect msg
112
        print_norm t2
113
  | Typer.Error s ->
114
      Format.fprintf ppf "%a@." print_protect s
115
  | Typer.Constraint (s,t) ->
116
      Format.fprintf ppf "This expression should have type:@.%a@." 
117
        print_norm t;
118
      Format.fprintf ppf "but its inferred type is:@.%a@." 
119
        print_norm s;
120
      Format.fprintf ppf "which is not a subtype, as shown by the sample:@.%a@." 
121
	print_sample (Sample.get (Types.diff s t))
122
  | Typer.NonExhaustive t ->
123 124
      Format.fprintf ppf "This pattern matching is not exhaustive@.";
      Format.fprintf ppf "Residual type:@.%a@."
125
	print_norm t;
126
      Format.fprintf ppf "Sample:@.%a@." print_sample (Sample.get t)
127 128 129
  | 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 "")
130 131 132 133
  | Ulexer.Error (i,j,s) ->
      let loc = Location.loc_of_pos (i,j), `Full in
      Format.fprintf ppf "Error %a:@." Location.print_loc loc;
      Format.fprintf ppf "%a%s" Location.html_hilight loc s
134
  | Parser.Error s | Stream.Error s -> 
135
      Format.fprintf ppf "Parsing error: %a@." print_protect s
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
  | 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:@.";
156
      Format.fprintf ppf "Object filename must have extension .cdo and no path@.";
157 158 159 160
  | Librarian.InvalidObject f ->
      Format.fprintf ppf "Invalid object file %s@." f
  | Librarian.CannotOpen f ->
      Format.fprintf ppf "Cannot open file %s@." f
161
  | Location.Generic s ->
162
      Format.fprintf ppf "%a@." print_protect s
163
  | exn ->
164
(*      raise exn *)
165
      Format.fprintf ppf "%a@." print_protect (Printexc.to_string exn)
166

167

168 169 170 171 172
let display ppf l =
  if not !quiet then
    List.iter
      (fun (x,t) -> dump_value ppf x t (get_global_value x))
      l
173 174

let eval ppf e =
175
  let (e,t) = Typer.type_expr !typing_env e in
176 177 178
  
  if not !quiet then
    Location.dump_loc ppf (e.Typed.exp_loc,`Full);
179 180 181

  let v =   
    if !do_compile then
182 183
      let e = Compile.compile_eval !compile_env e in 
      Eval.L.expr e
184 185 186 187 188 189 190
    else
      Eval.eval !eval_env e
  in
  if not !quiet then
    Format.fprintf ppf "- : @[@[%a@] =@ @[%a@]@]@." 
      print_norm t print_value v; 
  v
191
  
192
let let_decl ppf p e =
193
  let (tenv,decl,typs) = Typer.type_let_decl !typing_env p e in
194 195 196 197
  
  let () =
    if !do_compile then
      let (env,decl) = Compile.compile_let_decl !compile_env decl in
198
      Eval.L.eval decl;
199 200 201 202
      compile_env := env
    else
      eval_env := Eval.eval_let_decl !eval_env decl
  in
203
  typing_env := tenv;
204 205 206 207
  display ppf typs
  

let let_funs ppf funs =
208
  let (tenv,funs,typs) = Typer.type_let_funs !typing_env funs in
209
  
210 211 212
  let () =
    if !do_compile then
      let (env,funs) = Compile.compile_rec_funs !compile_env funs in
213
      Eval.L.eval funs;
214 215 216 217
      compile_env := env;
    else 
      eval_env := Eval.eval_rec_funs !eval_env funs 
  in
218
  typing_env := tenv;
219 220
  display ppf typs

221

222
let debug ppf = function
223
  | `Subtype (t1,t2) ->
224
      Format.fprintf ppf "[DEBUG:subtype]@.";
225 226
      let t1 = Types.descr (Typer.typ !typing_env t1)
      and t2 = Types.descr (Typer.typ !typing_env t2) in
227 228
      let s = Types.subtype t1 t2 in
      Format.fprintf ppf "%a %a %a : %b@." print_norm t1 print_protect "<=" print_norm t2 s
229
  | `Sample t ->
230 231
      Format.fprintf ppf "[DEBUG:sample]@.";
      (try
232
	 let t = Types.descr (Typer.typ !typing_env t) in
233 234 235
	 Format.fprintf ppf "%a@." print_sample (Sample.get t)
       with Not_found ->
	 Format.fprintf ppf "Empty type : no sample !@.")
236
  | `Filter (t,p) -> 
237
      Format.fprintf ppf "[DEBUG:filter]@.";
238 239
      let t = Typer.typ !typing_env t
      and p = Typer.pat !typing_env p in
240 241
      let f = Patterns.filter (Types.descr t) p in
      List.iter (fun (x,t) ->
242
		   Format.fprintf ppf " %a:%a@." U.print (Id.value x)
243 244
		     print_norm (Types.descr t)) f
  | `Accept p ->
245
      Format.fprintf ppf "[DEBUG:accept]@.";
246
      let p = Typer.pat !typing_env p in
247
      let t = Patterns.accept p in
248
      Format.fprintf ppf " %a@." Types.Print.print (Types.descr t)
249
  | `Compile (t,pl) ->
250
      Format.fprintf ppf "[DEBUG:compile]@.";
251 252
      let t = Typer.typ !typing_env t
      and pl = List.map (Typer.pat !typing_env) pl in
253
      Patterns.Compile.debug_compile ppf t pl
254 255 256 257 258 259 260 261 262
  | `Explain (t,e) ->
      Format.fprintf ppf "[DEBUG:explain]@.";
      let t = Typer.typ !typing_env t in
      (match Explain.explain (Types.descr t) (eval ppf e) with
	 | Some p ->
	     Format.fprintf ppf "Explanation: @[%a@]@." 
	       Explain.print_path p
	 | None ->
	     Format.fprintf ppf "Explanation: value has given type@.")
263

264

265 266 267
let rec collect_funs ppf accu = function
  | { descr = Ast.FunDecl e } :: rest -> collect_funs ppf (e::accu) rest
  | rest -> let_funs ppf accu; rest
268 269 270 271 272

let rec collect_types ppf accu = function
  | { descr = Ast.TypeDecl (x,t) } :: rest -> 
      collect_types ppf ((x,t) :: accu) rest
  | rest ->
273 274
      typing_env := 
        Typer.enter_types (Typer.type_defs !typing_env accu) !typing_env;
275 276
      rest

277
let flush_stdout () = Format.fprintf Format.std_formatter "@."
278

279 280 281 282 283
let rec phrases ppf phs = match phs with
  | { descr = Ast.FunDecl _ } :: _ -> 
      phrases ppf (collect_funs ppf [] phs)
  | { descr = Ast.TypeDecl (_,_) } :: _ ->
      phrases ppf (collect_types ppf [] phs)
284 285 286
  | { descr = Ast.SchemaDecl (name, schema) } :: rest ->
      Typer.register_schema name schema;
      phrases ppf rest
287
  | { descr = Ast.Namespace (pr,ns) } :: rest ->
288
      typing_env := Typer.enter_ns pr ns !typing_env;
289
      phrases ppf rest
290
  | { descr = Ast.Using (x,cu) } :: rest ->
291 292
      Librarian.import cu;
      Librarian.run Value.nil cu;
293 294
      typing_env := Typer.enter_cu x cu !typing_env;
      phrases ppf rest
295
  | { descr = Ast.EvalStatement e } :: rest ->
296
      ignore (eval ppf e);
297 298
      phrases ppf rest
  | { descr = Ast.LetDecl (p,e) } :: rest ->
299
      let_decl ppf p e;
300 301
      phrases ppf rest
  | { descr = Ast.Debug l } :: rest -> 
302
      debug ppf l; 
303
      phrases ppf rest
304 305 306 307 308 309
  | { descr = Ast.Directive `Quit } :: rest ->
      if !toplevel then raise End_of_file;
      phrases ppf rest
  | { descr = Ast.Directive `Env } :: rest ->
      dump_env ppf;
      phrases ppf rest
310 311 312 313 314 315 316 317 318 319 320 321
  | { descr = Ast.Directive (`Print_schema schema) } :: rest ->
      Schema_common.print_schema ppf (Typer.get_schema schema);
      flush_stdout ();
      phrases ppf rest
  | { descr = Ast.Directive (`Print_type name) } :: rest ->
      Typer.dump_type Format.std_formatter !typing_env name;
      flush_stdout ();
      phrases ppf rest
  | { descr = Ast.Directive (`Print_schema_type schema_ref) } :: rest ->
      Typer.dump_schema_type Format.std_formatter schema_ref;
      flush_stdout ();
      phrases ppf rest
322
  | { descr = Ast.Directive `Reinit_ns } :: rest ->
323
      Typer.set_ns_table_for_printer !typing_env;
324
      phrases ppf rest
325 326 327
  | { descr = Ast.Directive `Help } :: rest ->
      directive_help ppf;
      phrases ppf rest
328 329 330 331 332
  | { descr = Ast.Directive (`Dump pexpr) } :: rest ->
      Format.fprintf ppf "%a@."
        Value.dump_xml (Eval.eval !eval_env
          (fst (Typer.type_expr !typing_env pexpr)));
      phrases ppf rest
333 334
  | [] -> ()

335 336 337
let catch_exn ppf_err exn =
  if not catch_exceptions then raise exn;
  match exn with
338 339 340 341 342 343 344 345
  | (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 =
346 347
  try Parser.localize_exn (fun () -> rule input)
  with e -> Parser.sync (); raise e
348

349
let run rule ppf ppf_err input =
350
  try phrases ppf (parse rule input); true
351
  with exn -> catch_exn ppf_err exn; false
352

353
let script = run Parser.prog
354
let topinput = run Parser.top_phrases
355

356
let compile src out_dir =
357
  try
358 359 360 361 362 363 364 365 366 367 368
    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
    Librarian.compile id src;
    Librarian.save id out;
369
    exit 0
370 371 372
  with exn -> catch_exn Format.err_formatter exn; exit 1
  
let compile_run src argv =
373
  try
374 375 376 377 378
    if not (Filename.check_suffix src ".cd")
    then raise (InvalidInputFilename src);
    let cu = Filename.chop_suffix (Filename.basename src) ".cd" in
    let id = Types.CompUnit.mk (U.mk_latin1 cu) in
    Librarian.compile id src;
379 380
    Librarian.run argv id
  with exn -> catch_exn Format.err_formatter exn; exit 1
381 382

let run obj argv =  
383
  try
384
    if not (Filename.check_suffix obj ".cdo") || (Filename.basename obj <> obj)
385 386 387
    then raise (InvalidObjectFilename obj);
    let cu = Filename.chop_suffix (Filename.basename obj) ".cdo" in
    let id = Types.CompUnit.mk (U.mk_latin1 cu) in
388 389 390
    Librarian.import id;
    Librarian.run argv id
  with exn -> catch_exn Format.err_formatter exn; exit 1
391