1 (* Copyright (C) 2000, HELM Team.
3 * This file is part of HELM, an Hypertextual, Electronic
4 * Library of Mathematics, developed at the Computer Science
5 * Department, University of Bologna, Italy.
7 * HELM is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * HELM is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with HELM; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
22 * For details, see the HELM World-Wide-Web page,
23 * http://cs.unibo.it/helm/.
26 (* $Id: cicPp.ml 7413 2007-05-29 15:30:53Z tassi $ *)
28 exception CicExportationInternalError;;
29 exception NotEnoughElements;;
31 (* Utility functions *)
33 let analyze_term context t =
34 match fst(CicTypeChecker.type_of_aux' [] context t CicUniv.oblivion_ugraph)with
38 fst (CicTypeChecker.type_of_aux' [] context ty CicUniv.oblivion_ugraph)
40 Cic.Sort Cic.Prop -> `Proof
44 let analyze_type context t =
48 | Cic.Prod (_,_,t) -> aux t
55 fst(CicTypeChecker.type_of_aux' [] context t CicUniv.oblivion_ugraph)
57 Cic.Sort Cic.Prop -> `Statement
71 let n = String.uncapitalize n in
72 if List.mem n reserved then n ^ "_" else n
78 | Cic.Anonymous -> "_"
81 (* get_nth l n returns the nth element of the list l if it exists or *)
82 (* raises NotEnoughElements if l has less than n elements *)
86 | (n, he::tail) when n > 1 -> get_nth tail (n-1)
87 | (_,_) -> raise NotEnoughElements
90 let qualified_name_of_uri current_module_uri ?(capitalize=false) uri =
93 String.capitalize (UriManager.name_of_uri uri)
95 ppid (UriManager.name_of_uri uri) in
97 let suri = UriManager.buri_of_uri uri in
98 let s = String.sub suri 5 (String.length suri - 5) in
99 let s = Pcre.replace ~pat:"/" ~templ:"_" s in
100 String.uncapitalize s in
101 if current_module_uri = UriManager.buri_of_uri uri then
104 String.capitalize filename ^ "." ^ name
107 let pp current_module_uri ?metasenv ~in_type =
108 let rec pp ~in_type t context =
109 let module C = Cic in
114 (match get_nth context n with
115 Some (C.Name s,_) -> ppid s
116 | Some (C.Anonymous,_) -> "__" ^ string_of_int n
117 | None -> "_hidden_" ^ string_of_int n
120 NotEnoughElements -> string_of_int (List.length context - n)
122 | C.Var (uri,exp_named_subst) ->
123 qualified_name_of_uri current_module_uri uri ^
124 pp_exp_named_subst exp_named_subst context
128 "?" ^ (string_of_int n) ^ "[" ^
133 | Some t -> pp ~in_type:false t context) l1) ^
137 let _,context,_ = CicUtil.lookup_meta n metasenv in
138 "?" ^ (string_of_int n) ^ "[" ^
146 | Some _, Some t -> pp ~in_type:false t context
150 CicUtil.Meta_not_found _
151 | Invalid_argument _ ->
152 "???" ^ (string_of_int n) ^ "[" ^
154 (List.rev_map (function None -> "_" | Some t ->
155 pp ~in_type:false t context) l1) ^
163 (*| C.Type u -> ("Type" ^ CicUniv.string_of_universe u)*)
166 | C.Implicit (Some `Hole) -> "%"
167 | C.Implicit _ -> "?"
171 let n = "'" ^ String.uncapitalize n in
172 "(" ^ pp ~in_type:true s context ^ " -> " ^
173 pp ~in_type:true t ((Some (Cic.Name n,Cic.Decl s))::context) ^ ")"
175 "(" ^ pp ~in_type:true s context ^ " -> " ^
176 pp ~in_type:true t ((Some (b,Cic.Decl s))::context) ^ ")")
177 | C.Cast (v,t) -> pp ~in_type v context
178 | C.Lambda (b,s,t) ->
179 (match analyze_type context s with
181 | `Statement -> pp ~in_type t ((Some (b,Cic.Decl s))::context)
183 "(function " ^ ppname b ^ " -> " ^
184 pp ~in_type t ((Some (b,Cic.Decl s))::context) ^ ")")
186 (match analyze_term context s with
190 CicTypeChecker.type_of_aux' [] context s CicUniv.oblivion_ugraph
192 pp ~in_type t ((Some (b,Cic.Def (s,Some ty)))::context)
195 CicTypeChecker.type_of_aux' [] context s CicUniv.oblivion_ugraph
197 "(let " ^ ppname b ^ " = " ^ pp ~in_type:false s context ^ " in " ^
198 pp ~in_type t ((Some (b,Cic.Def (s,Some ty)))::context) ^ ")")
199 | C.Appl (he::tl) when in_type ->
200 let hes = pp ~in_type he context in
201 let stl = String.concat "," (clean_args_for_ty context tl) in
202 (if stl = "" then "" else "(" ^ stl ^ ") ") ^ hes
203 | C.Appl (C.MutInd _ as he::tl) ->
204 let hes = pp ~in_type he context in
205 let stl = String.concat "," (clean_args_for_ty context tl) in
206 (if stl = "" then "" else "(" ^ stl ^ ") ") ^ hes
207 | C.Appl (C.MutConstruct (uri,n,_,_) as he::tl) ->
209 match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
210 C.InductiveDefinition (_,_,nparams,_) -> nparams
211 | _ -> assert false in
212 let hes = pp ~in_type he context in
213 let stl = String.concat "," (clean_args nparams context tl) in
214 "(" ^ hes ^ (if stl = "" then "" else "(" ^ stl ^ ")") ^ ")"
216 "(" ^ String.concat " " (clean_args 0 context li) ^ ")"
217 | C.Const (uri,exp_named_subst) ->
218 qualified_name_of_uri current_module_uri uri ^
219 pp_exp_named_subst exp_named_subst context
220 | C.MutInd (uri,n,exp_named_subst) ->
222 match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
223 C.InductiveDefinition (dl,_,_,_) ->
224 let (name,_,_,_) = get_nth dl (n+1) in
225 qualified_name_of_uri current_module_uri
226 (UriManager.uri_of_string
227 (UriManager.buri_of_uri uri ^ "/" ^ name ^ ".con")) ^
228 pp_exp_named_subst exp_named_subst context
229 | _ -> raise CicExportationInternalError
231 Sys.Break as exn -> raise exn
232 | _ -> UriManager.string_of_uri uri ^ "#1/" ^ string_of_int (n + 1)
234 | C.MutConstruct (uri,n1,n2,exp_named_subst) ->
236 match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
237 C.InductiveDefinition (dl,_,_,_) ->
238 let _,_,_,cons = get_nth dl (n1+1) in
239 let id,_ = get_nth cons n2 in
240 qualified_name_of_uri current_module_uri ~capitalize:true
241 (UriManager.uri_of_string
242 (UriManager.buri_of_uri uri ^ "/" ^ id ^ ".con")) ^
243 pp_exp_named_subst exp_named_subst context
244 | _ -> raise CicExportationInternalError
246 Sys.Break as exn -> raise exn
248 UriManager.string_of_uri uri ^ "#1/" ^ string_of_int (n1 + 1) ^ "/" ^
251 | C.MutCase (uri,n1,ty,te,patterns) ->
253 "unit (* TOO POLYMORPHIC TYPE *)"
255 let needs_obj_magic =
256 (* BUG HERE: we should consider also the right parameters *)
257 match CicReduction.whd context ty with
258 Cic.Lambda (_,_,t) -> not (DoubleTypeInference.does_not_occur 1 t)
259 | _ -> false (* it can be a Rel, e.g. in *_rec *)
261 (match analyze_term context te with
262 `Type -> assert false
265 [] -> "assert false" (* empty type elimination *)
267 pp ~in_type:false he context (* singleton elimination *)
270 if patterns = [] then "assert false"
272 (let connames_and_argsno =
273 (match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
274 C.InductiveDefinition (dl,_,paramsno,_) ->
275 let (_,_,_,cons) = get_nth dl (n1+1) in
278 (* this is just an approximation since we do not have
280 let rec count_prods toskip =
282 C.Prod (_,_,bo) when toskip > 0 ->
283 count_prods (toskip - 1) bo
284 | C.Prod (_,_,bo) -> 1 + count_prods 0 bo
287 qualified_name_of_uri current_module_uri
289 (UriManager.uri_of_string
290 (UriManager.buri_of_uri uri ^ "/" ^ id ^ ".con")),
291 count_prods paramsno ty
293 | _ -> raise CicExportationInternalError
296 let connames_and_argsno_and_patterns =
300 | (x,no)::tlx,y::tly -> (x,no,y)::(combine (tlx,tly))
301 | _,_ -> assert false
303 combine (connames_and_argsno,patterns)
305 "\n(match " ^ pp ~in_type:false te context ^ " with \n " ^
306 (String.concat "\n | "
309 let rec aux argsno context =
311 Cic.Lambda (name,ty,bo) when argsno > 0 ->
314 Cic.Anonymous -> Cic.Anonymous
315 | Cic.Name n -> Cic.Name (ppid n) in
317 aux (argsno - 1) (Some (name,Cic.Decl ty)::context)
320 (match analyze_type context ty with
322 | `Sort _ -> args,res
326 | C.Name s -> s)::args,res)
327 | t when argsno = 0 -> [],pp ~in_type:false t context
329 ["{" ^ string_of_int argsno ^ " args missing}"],
330 pp ~in_type:false t context
333 if argsno = 0 then x,pp ~in_type:false y context
335 let args,body = aux argsno context y in
336 let sargs = String.concat "," args in
337 x ^ (if sargs = "" then "" else "(" ^ sargs^ ")"),
341 if needs_obj_magic then
342 "Obj.magic (" ^ body ^ ")"
345 ) connames_and_argsno_and_patterns)) ^
347 | C.Fix (no, funs) ->
350 (fun (types,len) (n,_,ty,_) ->
351 (Some (C.Name n,(C.Decl (CicSubstitution.lift len ty)))::types,
357 (fun (name,ind,ty,bo) i -> name ^ " = \n" ^
358 pp ~in_type:false bo (names@context) ^ i)
361 (match get_nth names (no + 1) with
362 Some (Cic.Name n,_) -> n
364 | C.CoFix (no,funs) ->
367 (fun (types,len) (n,ty,_) ->
368 (Some (C.Name n,(C.Decl (CicSubstitution.lift len ty)))::types,
374 (fun (name,ty,bo) i -> "\n" ^ name ^
375 " : " ^ pp ~in_type:true ty context ^ " := \n" ^
376 pp ~in_type:false bo (names@context) ^ i)
379 and pp_exp_named_subst exp_named_subst context =
380 if exp_named_subst = [] then "" else
382 String.concat " ; " (
384 (function (uri,t) -> UriManager.name_of_uri uri ^ " \\Assign " ^ pp ~in_type:false t context)
387 and clean_args nparams context =
388 let nparams = ref nparams in
392 match analyze_term context t with
393 `Term when !nparams < 0 -> Some (pp ~in_type:false t context)
397 and clean_args_for_ty context =
400 match analyze_term context t with
401 `Type -> Some (pp ~in_type:true t context)
408 let ppty current_module_uri =
409 (* nparams is the number of left arguments
410 left arguments should either become parameters or be skipped altogether *)
411 let rec args nparams context =
416 Cic.Anonymous -> Cic.Anonymous
417 | Cic.Name n -> Cic.Name (String.uncapitalize n)
419 (match analyze_type context s with
422 args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
423 | `Type when nparams > 0 ->
424 args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
427 args (nparams - 1) ((Some (n,Cic.Decl s))::context) t in
428 abstr,pp ~in_type:true current_module_uri s context::args
429 | `Sort _ when nparams <= 0 ->
430 let n = Cic.Name "unit (* EXISTENTIAL TYPE *)" in
431 args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
435 Cic.Anonymous -> Cic.Anonymous
436 | Cic.Name name -> Cic.Name ("'" ^ name) in
438 args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
441 Cic.Anonymous -> abstr
442 | Cic.Name name -> name::abstr),
449 exception DoNotExtract;;
451 let pp_abstracted_ty current_module_uri =
452 let rec args context =
454 Cic.Lambda (n,s,t) ->
457 Cic.Anonymous -> Cic.Anonymous
458 | Cic.Name n -> Cic.Name (String.uncapitalize n)
460 (match analyze_type context s with
464 args ((Some (n,Cic.Decl s))::context) t
468 Cic.Anonymous -> Cic.Anonymous
469 | Cic.Name name -> Cic.Name ("'" ^ name) in
471 args ((Some (n,Cic.Decl s))::context) t
474 Cic.Anonymous -> abstr
475 | Cic.Name name -> name::abstr),
478 match analyze_type context ty with
480 | `Statement -> raise DoNotExtract
482 (* BUG HERE: this can be a real System F type *)
483 let head = pp ~in_type:true current_module_uri ty context in
490 (* ppinductiveType (typename, inductive, arity, cons) *)
491 (* pretty-prints a single inductive definition *)
492 (* (typename, inductive, arity, cons) *)
493 let ppinductiveType current_module_uri nparams (typename, inductive, arity, cons)
495 match analyze_type [] arity with
498 | `Type -> assert false
501 "type " ^ String.uncapitalize typename ^ " = unit (* empty type *)\n"
505 (fun (id,ty) (_abstr,i) -> (* we should verify _abstr = abstr' *)
506 let abstr',sargs = ppty current_module_uri nparams [] ty in
507 let sargs = String.concat " * " sargs in
509 String.capitalize id ^
510 (if sargs = "" then "" else " of " ^ sargs) ^
511 (if i = "" then "" else "\n | ") ^ i)
515 let s = String.concat "," abstr in
516 if s = "" then "" else "(" ^ s ^ ") "
518 "type " ^ abstr ^ String.uncapitalize typename ^ " =\n" ^ scons ^ "\n")
521 let ppobj current_module_uri obj =
522 let module C = Cic in
523 let module U = UriManager in
524 let pp ~in_type = pp ~in_type current_module_uri in
526 C.Constant (name, Some t1, t2, params, _) ->
527 (match analyze_type [] t2 with
530 | `Type -> "let " ^ ppid name ^ " =\n" ^ pp ~in_type:false t1 [] ^ "\n"
532 match analyze_type [] t1 with
536 let abstr,res = pp_abstracted_ty current_module_uri [] t1 in
538 let s = String.concat "," abstr in
539 if s = "" then "" else "(" ^ s ^ ") "
541 "type " ^ abstr ^ ppid name ^ " = " ^ res ^ "\n"
544 | C.Constant (name, None, ty, params, _) ->
545 (match analyze_type [] ty with
548 | `Sort _ -> "type " ^ ppid name ^ "\n"
549 | `Type -> "let " ^ ppid name ^ " = assert false\n")
550 | C.Variable (name, bo, ty, params, _) ->
552 "(" ^ String.concat ";" (List.map UriManager.string_of_uri params) ^
554 pp ~in_type:true ty [] ^ "\n" ^
555 (match bo with None -> "" | Some bo -> ":= " ^ pp ~in_type:false bo [])
556 | C.CurrentProof (name, conjectures, value, ty, params, _) ->
557 "Current Proof of " ^ name ^
558 "(" ^ String.concat ";" (List.map UriManager.string_of_uri params) ^
560 let separate s = if s = "" then "" else s ^ " ; " in
562 (fun (n, context, t) i ->
563 let conjectures',name_context =
565 (fun context_entry (i,name_context) ->
566 (match context_entry with
567 Some (n,C.Decl at) ->
570 pp ~in_type:true ~metasenv:conjectures
571 at name_context ^ " ",
572 context_entry::name_context
573 | Some (n,C.Def (at,None)) ->
575 ppname n ^ ":= " ^ pp ~in_type:false
576 ~metasenv:conjectures at name_context ^ " ",
577 context_entry::name_context
579 (separate i) ^ "_ :? _ ", context_entry::name_context
583 conjectures' ^ " |- " ^ "?" ^ (string_of_int n) ^ ": " ^
584 pp ~in_type:true ~metasenv:conjectures t name_context ^ "\n" ^ i
586 "\n" ^ pp ~in_type:false ~metasenv:conjectures value [] ^ " : " ^
587 pp ~in_type:true ~metasenv:conjectures ty []
588 | C.InductiveDefinition (l, params, nparams, _) ->
590 (fun x i -> ppinductiveType current_module_uri nparams x ^ i) l ""
593 let ppobj current_module_uri obj =
594 let res = ppobj current_module_uri obj in
595 if res = "" then "" else res ^ ";;\n\n"