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

4
let quiet = ref false
5
let toplevel = ref false
6

7
let typing_env = State.ref "Cduce.typing_env" Env.empty
8

9 10
let eval_env = State.ref "Cduce.eval_env" Env.empty

11
let enter_global_value x v t =
12
  eval_env := Env.add x v !eval_env;
13
  typing_env := Env.add x t !typing_env
14

15 16 17 18 19
let rec is_abstraction = function
  | Ast.Abstraction _ -> true
  | Ast.LocatedExpr (_,e) -> is_abstraction e
  | _ -> false

20
let print_norm ppf d = 
21
  Location.protect ppf 
22
    (fun ppf -> Types.Print.print ppf ((*Types.normalize*) d))
23

24 25 26 27
let print_sample ppf s =
  Location.protect ppf
    (fun ppf -> Sample.print ppf s)

28 29 30
let print_protect ppf s =
  Location.protect ppf (fun ppf -> Format.fprintf ppf "%s" s)

31 32
let print_value ppf v =
  Location.protect ppf (fun ppf -> Value.print ppf v)
33

34
let dump_env ppf =
35
  Format.fprintf ppf "Types:%t@." Typer.dump_global_types;
36
  Format.fprintf ppf "Namespace prefixes:@\n%t" Typer.dump_global_ns;
37
  Format.fprintf ppf "Namespace prefixes used for pretty-printing:@.%t"
38 39
    Ns.InternalPrinter.dump;
  Format.fprintf ppf "Values:@\n";
40
  Env.iter 
41
    (fun x v ->
42
       let t = Env.find x !typing_env in
43 44
       Format.fprintf ppf "@[val %a : @[%a = %a@]@]@."
	 U.print (Id.value x) print_norm t print_value v
45
    )
46
    !eval_env
47

48
let rec print_exn ppf = function
49 50
  | Location (loc, w, exn) ->
      Format.fprintf ppf "Error %a:@." Location.print_loc (loc,w);
51
      Format.fprintf ppf "%a" Location.html_hilight (loc,w); 
52
      print_exn ppf exn
53
  | Value.CDuceExn v ->
54
      Format.fprintf ppf "Uncaught CDuce exception: @[%a@]@."
55
        print_value v
56
  | Eval.MultipleDeclaration v ->
57
      Format.fprintf ppf "Multiple declaration for global value %a@." 
58
        U.print (Id.value v)
59
  | Typer.WrongLabel (t,l) ->
60
      Format.fprintf ppf "Wrong record selection; field %a " 
61
        Label.print (LabelPool.value l);
62
      Format.fprintf ppf "not present in an expression of type:@.%a@."
63
        print_norm t
64
  | Typer.ShouldHave (t,msg) ->
65
      Format.fprintf ppf "This expression should have type:@.%a@.%a@." 
66
        print_norm t
67
        print_protect msg
68
  | Typer.ShouldHave2 (t1,msg,t2) ->
69
      Format.fprintf ppf "This expression should have type:@.%a@.%a %a@." 
70
        print_norm t1
71
        print_protect msg
72
        print_norm t2
73
  | Typer.Error s ->
74
      Format.fprintf ppf "%a@." print_protect s
75
  | Typer.Constraint (s,t) ->
76
      Format.fprintf ppf "This expression should have type:@.%a@." 
77
        print_norm t;
78
      Format.fprintf ppf "but its inferred type is:@.%a@." 
79
        print_norm s;
80
      Format.fprintf ppf "which is not a subtype, as shown by the sample:@.%a@." 
81
	print_sample (Sample.get (Types.diff s t))
82
  | Typer.NonExhaustive t ->
83 84
      Format.fprintf ppf "This pattern matching is not exhaustive@.";
      Format.fprintf ppf "Residual type:@.%a@."
85
	print_norm t;
86
      Format.fprintf ppf "Sample:@.%a@." print_sample (Sample.get t)
87
  | Typer.UnboundId x ->
88
      Format.fprintf ppf "Unbound identifier %a@." U.print (Id.value x)
89
  | Wlexer.Illegal_character c ->
90
      Format.fprintf ppf "Illegal character (%a)@." print_protect (Char.escaped c)
91
  | Wlexer.Unterminated_comment ->
92
      Format.fprintf ppf "Comment not terminated@."
93
  | Wlexer.Unterminated_string ->
94
      Format.fprintf ppf "String literal not terminated@."
95
  | Wlexer.Unterminated_string_in_comment ->
96
      Format.fprintf ppf "This comment contains an unterminated string literal@."
97
  | Parser.Error s | Stream.Error s -> 
98
      Format.fprintf ppf "Parsing error: %a@." print_protect s
99
  | Location.Generic s ->
100
      Format.fprintf ppf "%a@." print_protect s
101
  | exn ->
102
(*      raise exn *)
103
      Format.fprintf ppf "%a@." print_protect (Printexc.to_string exn)
104

105
let debug ppf = function
106
  | `Subtype (t1,t2) ->
107
      Format.fprintf ppf "[DEBUG:subtype]@.";
108 109
      let t1 = Types.descr (Typer.typ t1)
      and t2 = Types.descr (Typer.typ t2) in
110 111
      let s = Types.subtype t1 t2 in
      Format.fprintf ppf "%a %a %a : %b@." print_norm t1 print_protect "<=" print_norm t2 s
112
  | `Sample t ->
113 114 115 116 117 118
      Format.fprintf ppf "[DEBUG:sample]@.";
      (try
	 let t = Types.descr (Typer.typ t) in
	 Format.fprintf ppf "%a@." print_sample (Sample.get t)
       with Not_found ->
	 Format.fprintf ppf "Empty type : no sample !@.")
119
  | `Filter (t,p) -> 
120
      Format.fprintf ppf "[DEBUG:filter]@.";
121 122
      let t = Typer.typ t
      and p = Typer.pat p in
123 124
      let f = Patterns.filter (Types.descr t) p in
      List.iter (fun (x,t) ->
125
		   Format.fprintf ppf " %a:%a@." U.print (Id.value x)
126 127
		     print_norm (Types.descr t)) f
  | `Accept p ->
128
      Format.fprintf ppf "[DEBUG:accept]@.";
129
      let p = Typer.pat p in
130
      let t = Patterns.accept p in
131
      Format.fprintf ppf " %a@." Types.Print.print (Types.descr t)
132
  | `Compile (t,pl) ->
133
      Format.fprintf ppf "[DEBUG:compile]@.";
134 135
      let t = Typer.typ t
      and pl = List.map Typer.pat pl in
136
      Patterns.Compile.debug_compile ppf t pl
137

138 139 140 141
let insert_bindings ppf =
  List.iter2 
    (fun (x,t) (y,v) ->
       assert (x = y);
142 143 144
       typing_env := Env.add x t !typing_env;
       eval_env := Env.add x v !eval_env;
       if not !quiet then
145
	 Format.fprintf ppf "val %a : @[@[%a@] =@ @[%a@]@]@." 
146
	   U.print (Id.value x) print_norm t print_value v)
147
  
148 149 150 151 152
let rec collect_funs ppf accu = function
  | { descr = Ast.FunDecl e } :: rest -> 
      let (_,e) = Typer.expr e in
      collect_funs ppf (e::accu) rest
  | rest ->
153
      let typs = Typer.type_rec_funs !typing_env accu in
154
      Typer.report_unused_branches ();
155 156
      let vals = Eval.eval_rec_funs !eval_env accu in
      insert_bindings ppf typs vals;
157 158 159 160 161 162 163 164 165 166 167 168 169 170
      rest

let rec collect_types ppf accu = function
  | { descr = Ast.TypeDecl (x,t) } :: rest -> 
      collect_types ppf ((x,t) :: accu) rest
  | rest ->
      Typer.register_global_types accu;
      rest

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)
171 172 173
  | { descr = Ast.SchemaDecl (name, schema) } :: rest ->
      Typer.register_schema name schema;
      phrases ppf rest
174
  | { descr = Ast.Namespace (pr,ns) } :: rest ->
175
      Typer.register_global_ns pr ns;
176
      phrases ppf rest
177 178 179 180 181
  | { descr = Ast.EvalStatement e } :: rest ->
      let (fv,e) = Typer.expr e in
      let t = Typer.type_check !typing_env e Types.any true in
      Typer.report_unused_branches ();
      if not !quiet then
182
	Location.dump_loc ppf (e.Typed.exp_loc,`Full);
183 184
      let v = Eval.eval !eval_env e in
      if not !quiet then
185
	Format.fprintf ppf "- : @[@[%a@] =@ @[%a@]@]@." print_norm t print_value v;
186 187 188
      phrases ppf rest
  | { descr = Ast.LetDecl (p,e) } :: rest ->
      let decl = Typer.let_decl p e in
189
      let typs = Typer.type_let_decl !typing_env decl in
190
      Typer.report_unused_branches ();
191 192
      let vals = Eval.eval_let_decl !eval_env decl in
      insert_bindings ppf typs vals;
193 194
      phrases ppf rest
  | { descr = Ast.Debug l } :: rest -> 
195
      debug ppf l; 
196
      phrases ppf rest
197 198 199 200 201 202
  | { 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
203 204 205
  | { descr = Ast.Directive `Reinit_ns } :: rest ->
      Typer.set_ns_table_for_printer ();
      phrases ppf rest
206 207 208
  | [] -> ()

let run rule ppf ppf_err input =
209
  Typer.clear_unused_branches ();
210
  try 
211
    let p = 
212
      try rule input
213
      with
214
	| Stdpp.Exc_located (_, (Location _ as e)) ->
215
	    Parser.sync (); raise e
216
	| Stdpp.Exc_located ((i,j), e) -> 
217
	    Parser.sync (); raise_loc i j e
218
    in
219
    phrases ppf p;
220
    true
221
  with 
222 223 224
    | (End_of_file | Failure _ | Not_found | Invalid_argument _ | Sys.Break) 
	as e -> 
	raise e
225 226 227 228
    | exn -> 
	print_exn ppf_err exn;
	Format.fprintf ppf_err "@.";
	false
229

230
let script = run Parser.prog
231
let topinput = run Parser.top_phrases