-let ppterm =
- ref (fun ~context ~subst ~metasenv t -> "Please, set a pp callback")
+(*
+ ||M|| This file is part of HELM, an Hypertextual, Electronic
+ ||A|| Library of Mathematics, developed at the Computer Science
+ ||T|| Department, University of Bologna, Italy.
+ ||I||
+ ||T|| HELM is free software; you can redistribute it and/or
+ ||A|| modify it under the terms of the GNU General Public License
+ \ / version 2 or (at your option) any later version.
+ \ / This software is distributed as is, NO WARRANTY.
+ V_______________________________________________________________ *)
+
+(* $Id$ *)
+
+module C = NCic
+module R = NReference
+module F = Format
+
+let head_beta_reduce = ref (fun ~upto:_ _ -> assert false);;
+let set_head_beta_reduce = (:=) head_beta_reduce;;
+
+let get_obj = ref (fun _ -> assert false);;
+let set_get_obj f = get_obj := f;;
+
+let r2s pp_fix_name r =
+ try
+ match r with
+ | R.Ref (u,R.Ind (_,i,_)) ->
+ (match !get_obj u with
+ | _,_,_,_, C.Inductive (_,_,itl,_) ->
+ let _,name,_,_ = List.nth itl i in name
+ | _ -> assert false)
+ | R.Ref (u,R.Con (i,j,_)) ->
+ (match !get_obj u with
+ | _,_,_,_, C.Inductive (_,_,itl,_) ->
+ let _,_,_,cl = List.nth itl i in
+ let _,name,_ = List.nth cl (j-1) in name
+ | _ -> assert false)
+ | R.Ref (u,(R.Decl | R.Def _)) ->
+ (match !get_obj u with
+ | _,_,_,_, C.Constant (_,name,_,_,_) -> name
+ | _ -> assert false)
+ | R.Ref (u,(R.Fix (i,_,_)|R.CoFix i)) ->
+ (match !get_obj u with
+ | _,_,_,_, C.Fixpoint (_,fl,_) ->
+ if pp_fix_name then
+ let _,name,_,_,_ = List.nth fl i in name
+ else
+ NUri.name_of_uri u (*^"("^ string_of_int i ^ ")"*)
+ | _ -> assert false)
+ with
+ | NCicEnvironment.ObjectNotFound _
+ | Failure "nth"
+ | Invalid_argument "List.nth" -> R.string_of_reference r
+
;;
-let set_ppterm f = ppterm := f;;
+let string_of_implicit_annotation = function
+ | `Closed -> "▪"
+ | `Type -> ""
+ | `Hole -> "□"
+ | `Term -> "Term"
+ | `Typeof x -> "Ty("^string_of_int x^")"
+ | `Vector -> "…"
+;;
-let ppterm ~context ~subst ~metasenv t =
- !ppterm ~context ~subst ~metasenv t
+let ppterm ~formatter:f ~context ~subst ~metasenv:_ ?(inside_fix=false) t =
+ let rec aux ?(toplevel=false) ctx = function
+ | C.Rel m ->
+ (try
+ let name = List.nth ctx (m-1) in
+ F.fprintf f "%s" (if name = "_" then "__"^string_of_int m else name)
+ with Failure "nth" | Invalid_argument "List.nth" ->
+ F.fprintf f " -%d" (m - List.length ctx))
+ | C.Const r -> F.fprintf f "%s" (r2s inside_fix r)
+ | C.Prod ("_",s,t) ->
+ if not toplevel then F.fprintf f "(";
+ F.fprintf f "@[<hov 1>";
+ (match s with
+ | C.Prod ("_",_,_) -> aux ~toplevel:false ctx s
+ | _ -> aux ~toplevel:true ctx s);
+ F.fprintf f "@;→ ";
+ aux ~toplevel:true ("_"::ctx) t;
+ F.fprintf f "@]";
+ if not toplevel then F.fprintf f ")";
+ | C.Prod (name,s,t) ->
+ if not toplevel then F.fprintf f "(";
+ F.fprintf f "@[<hov 1>";
+ F.fprintf f "@[<hov 2>∀%s:@;" name;
+ aux ~toplevel:true ctx s;
+ F.fprintf f "@].@;";
+ aux ~toplevel:true (name::ctx) t;
+ F.fprintf f "@]";
+ if not toplevel then F.fprintf f ")";
+ | C.Lambda (name,s,t) ->
+ if not toplevel then F.fprintf f "(";
+ F.fprintf f "@[<hov 1>";
+ F.fprintf f "λ%s:" name;
+ aux ~toplevel:true ctx s;
+ F.fprintf f ".@;";
+ aux ~toplevel:true (name::ctx) t;
+ F.fprintf f "@]";
+ if not toplevel then F.fprintf f ")";
+ | C.LetIn (name,ty,t,b) ->
+ if not toplevel then F.fprintf f "(";
+ F.fprintf f "@[<hov 1>";
+ F.fprintf f "let %s:@;" name;
+ aux ~toplevel:true ctx ty;
+ F.fprintf f "@;≝ ";
+ aux ~toplevel:true ctx t;
+ F.fprintf f "@;in@;";
+ (aux ~toplevel:true (name::ctx) b);
+ F.fprintf f "@]";
+ if not toplevel then F.fprintf f ")";
+ | C.Match (r,oty,t,pl) ->
+ F.fprintf f "@[<hov>match ";
+ aux ~toplevel:true ctx t;
+ F.fprintf f "@;return ";
+ aux ~toplevel:true ctx oty;
+ F.fprintf f "@; @[<v>[ ";
+ if pl <> [] then
+ begin
+ F.fprintf f "@[<hov 2>%s ⇒@;"
+ (try r2s inside_fix (R.mk_constructor 1 r)
+ with R.IllFormedReference _ -> "#ERROR#");
+ aux ~toplevel:true ctx (List.hd pl);
+ F.fprintf f "@]";
+ ignore(List.fold_left
+ (fun i t ->
+ F.fprintf f "@;| @[<hov 2>%s ⇒@;"
+ (try r2s inside_fix (R.mk_constructor i r)
+ with R.IllFormedReference _ -> "#ERROR#");
+ aux ~toplevel:true ctx t;
+ F.fprintf f "@]";
+ i+1)
+ 2 (List.tl pl));
+ end;
+ F.fprintf f "]@] @]";
+ | C.Appl [] | C.Appl [_] | C.Appl (C.Appl _::_) ->
+ F.fprintf f "BAD APPLICATION"
+ | C.Appl (C.Meta (n,lc) :: args) when List.mem_assoc n subst ->
+ let _,_,t,_ = List.assoc n subst in
+ let hd = NCicSubstitution.subst_meta lc t in
+ aux ctx
+ (!head_beta_reduce ~upto:(List.length args)
+ (match hd with
+ | NCic.Appl l -> NCic.Appl (l@args)
+ | _ -> NCic.Appl (hd :: args)))
+ | C.Appl l ->
+ F.fprintf f "@[<hov 2>";
+ if not toplevel then F.fprintf f "(";
+ aux ctx (List.hd l);
+ List.iter (fun x -> F.fprintf f "@;";aux ctx x) (List.tl l);
+ if not toplevel then F.fprintf f ")";
+ F.fprintf f "@]"
+ | C.Implicit annot ->
+ F.fprintf f "?%s" (string_of_implicit_annotation annot)
+ | C.Meta (n,lc) when List.mem_assoc n subst ->
+ let _,_,t,_ = List.assoc n subst in
+ aux ctx (NCicSubstitution.subst_meta lc t)
+ | C.Meta (n,(shift,C.Irl len)) ->
+ F.fprintf f "?%d(%d,%d)" n shift len
+ | C.Meta (n,(shift,C.Ctx l)) ->
+ F.fprintf f "?%d(%d,[" n shift;
+ if List.length l > 0 then
+ begin
+ aux ctx (NCicSubstitution.lift shift (List.hd l));
+ List.iter (fun x -> F.fprintf f ",";aux ctx x)
+ (List.map (NCicSubstitution.lift shift) (List.tl l));
+ end;
+ F.fprintf f "])"
+ | C.Sort C.Prop -> F.fprintf f "Prop"
+ | C.Sort (C.Type []) -> F.fprintf f "Type0"
+ | C.Sort (C.Type [false, u]) -> F.fprintf f "%s" (NUri.name_of_uri u)
+ | C.Sort (C.Type [true, u]) -> F.fprintf f "S(%s)" (NUri.name_of_uri u)
+ | C.Sort (C.Type l) ->
+ F.fprintf f "Max(";
+ aux ctx (C.Sort (C.Type [List.hd l]));
+ List.iter (fun x -> F.fprintf f ",";aux ctx (C.Sort (C.Type [x])))
+ (List.tl l);
+ F.fprintf f ")"
+ in
+ aux ~toplevel:true (List.map fst context) t
;;
-module C = NCic
-module R = NReference
+let on_buffer f t =
+ let buff = Buffer.create 100 in
+ let formatter = F.formatter_of_buffer buff in
+ f ~formatter:formatter t;
+ F.fprintf formatter "@?";
+ Buffer.contents buff
+;;
-let trivial_pp_term ~context ~subst ~metasenv =
- let rec aux ctx = function
- | C.Rel m ->
- (try List.nth ctx (m-1)
- with Failure _ -> " -" ^ string_of_int (m - List.length context))
- | C.Const r -> R.string_of_reference r
- | C.Prod (name,s,t) -> "(\Pi "^name^":"^aux ctx s^"."^aux (name::ctx) t^")"
- | C.Lambda (name,s,t) ->
- "(\lambda "^name^":"^aux ctx s^"."^aux (name::ctx) t^")"
- | C.LetIn (name,ty,t,b) ->
- "(let "^name^":"^aux ctx ty^":="^aux ctx t^" in "^aux (name::ctx) b^")"
- | C.Match (_,oty,t,pl) ->
- " match " ^ aux ctx t ^ " return " ^ aux ctx oty ^ "[" ^
- String.concat "|" (List.map (aux ctx) pl) ^ "]"
- | C.Appl l -> "("^String.concat " " (List.map (aux ctx) l) ^")"
- | C.Implicit _ -> "?"
- | C.Meta (n,_) -> "?"^string_of_int n
- | C.Sort C.Prop -> "Prop"
- | C.Sort C.CProp -> "CProp"
- | C.Sort (C.Type n) -> "Type"^string_of_int n
- in
- aux (List.map fst context)
-;;
-
-let ppobj = function
- | (u,_,metasenv,subst,NCic.Fixpoint (b, fl, _)) ->
- "let rec "^NUri.string_of_uri u^"\n"^
- String.concat "\nand "
- (List.map (fun (_,name,n,ty,bo) ->
- name ^ " on " ^ string_of_int n ^ " : " ^
- ppterm ~metasenv ~subst ~context:[] ty ^ " :=\n"^
- ppterm ~metasenv ~subst ~context:[] bo) fl)
- | (u,_,_,_,NCic.Inductive (b, _,tl, _)) -> "inductive"
- | (u,_,_,_,NCic.Constant (_,_,_, _, _)) -> "constant"
+let ppterm ~formatter ~context ~subst ~metasenv ?(margin=80) ?inside_fix t =
+ F.set_margin margin;
+ ppterm ~formatter ~context ~subst ~metasenv ?inside_fix t
;;
+let rec ppcontext ~formatter ?(sep="\n") ~subst ~metasenv = function
+ | [] -> ()
+ | (name, NCic.Decl t) :: tl ->
+ ppcontext ~formatter ~sep ~subst ~metasenv tl;
+ F.fprintf formatter "%s: " name;
+ ppterm ~formatter ~subst ~metasenv ~context:tl t;
+ F.fprintf formatter "%s" sep
+ | (name, NCic.Def (bo,ty)) :: tl->
+ ppcontext ~formatter ~sep ~subst ~metasenv tl;
+ F.fprintf formatter "%s: " name;
+ ppterm ~formatter ~subst ~metasenv ~context:tl ty;
+ F.fprintf formatter " := ";
+ ppterm ~formatter ~subst ~metasenv ~context:tl bo;
+ F.fprintf formatter "%s" sep
+;;
+
+let rec ppmetasenv ~formatter ~subst metasenv = function
+ | [] -> ()
+ | (i,(name, ctx, ty)) :: tl ->
+ F.fprintf formatter "@[<hov 2>";
+ let name = match name with Some n -> "("^n^")" | _ -> "" in
+ ppcontext ~formatter ~sep:"; " ~subst ~metasenv ctx;
+ F.fprintf formatter "@;⊢@;?%d%s :@;" i name;
+ ppterm ~formatter ~metasenv ~subst ~context:ctx ty;
+ F.fprintf formatter "@]@\n";
+ ppmetasenv ~formatter ~subst metasenv tl
+;;
+
+let ppmetasenv ~formatter ~subst metasenv =
+ ppmetasenv ~formatter ~subst metasenv metasenv
+;;
+
+let rec ppsubst ~formatter ~subst ~metasenv = function
+ | [] -> ()
+ | (i,(name, ctx, t, ty)) :: tl ->
+ let name = match name with Some n -> "("^n^")" | _ -> "" in
+ ppcontext ~formatter ~sep:"; " ~subst ~metasenv ctx;
+ F.fprintf formatter " ⊢ ?%d%s := " i name;
+ ppterm ~formatter ~metasenv ~subst ~context:ctx t;
+ F.fprintf formatter " : ";
+ ppterm ~formatter ~metasenv ~subst ~context:ctx ty;
+ F.fprintf formatter "\n";
+ ppsubst ~formatter ~subst ~metasenv tl
+;;
+
+let ppsubst ~formatter ~metasenv subst =
+ ppsubst ~formatter ~metasenv ~subst subst
+;;
+
+let ppobj ~formatter (u,_,metasenv, subst, o) =
+ F.fprintf formatter "metasenv:\n";
+ ppmetasenv ~formatter ~subst metasenv;
+ F.fprintf formatter "\n";
+ F.fprintf formatter "subst:\n";
+ (*ppsubst subst ~formatter ~metasenv;*) F.fprintf formatter "...";
+ F.fprintf formatter "\n";
+ match o with
+ | NCic.Fixpoint (b, fl, _) ->
+ F.fprintf formatter "{%s}@\n@[<hov 0>" (NUri.string_of_uri u);
+ F.fprintf formatter "@[<hov 2>%s"(if b then "let rec " else "let corec ");
+ HExtlib.list_iter_sep
+ ~sep:(fun () -> F.fprintf formatter "@\n@[<hov 2>and ")
+ (fun (_,name,n,ty,bo) ->
+ F.fprintf formatter "%s on %d :@;" name n;
+ ppterm ~formatter ~metasenv ~subst ~context:[] ty;
+ F.fprintf formatter "@]@;@[<hov 2>:=@;";
+ ppterm ~formatter ~metasenv ~subst ~context:[] ~inside_fix:true bo;
+ F.fprintf formatter "@]") fl;
+ F.fprintf formatter "@]"
+ | NCic.Inductive (b, leftno,tyl, _) ->
+ F.fprintf formatter "{%s} with %d fixed params@\n@[<hov 0>@[<hov 2>%s"
+ (NUri.string_of_uri u) leftno
+ (if b then "inductive " else "coinductive ");
+ HExtlib.list_iter_sep
+ ~sep:(fun () -> F.fprintf formatter "@\n@[<hov 2>and ")
+ (fun (_,name,ty,cl) ->
+ F.fprintf formatter "%s:@;" name;
+ ppterm ~formatter ~metasenv ~subst ~context:[] ty;
+ F.fprintf formatter "@]@;@[<hov 3>:=@;";
+ HExtlib.list_iter_sep ~sep:(fun () -> F.fprintf formatter "@;")
+ (fun (_,name,ty) ->
+ F.fprintf formatter "| %s: " name;
+ ppterm ~formatter ~metasenv ~subst ~context:[] ty;)
+ cl;
+ F.fprintf formatter "@]"
+ ) tyl ;
+ F.fprintf formatter "@]"
+ | NCic.Constant (_,name,None,ty, _) ->
+ F.fprintf formatter "{%s}@\n@[<hov 2>axiom %s :@;" (NUri.string_of_uri u) name;
+ ppterm ~formatter ~metasenv ~subst ~context:[] ty;
+ F.fprintf formatter "@]@\n"
+ | NCic.Constant (_,name,Some bo,ty, _) ->
+ F.fprintf formatter "{%s}@\n@[<hov 0>@[<hov 2>definition %s :@;" (NUri.string_of_uri u) name;
+ ppterm ~formatter ~metasenv ~subst ~context:[] ty;
+ F.fprintf formatter "@]@;@[<hov 2>:=@;";
+ ppterm ~formatter ~metasenv ~subst ~context:[] bo;
+ F.fprintf formatter "@]@\n@]"
+;;
+
+let ppterm ~context ~subst ~metasenv ?margin ?inside_fix t =
+ on_buffer (ppterm ~context ~subst ~metasenv ?margin ?inside_fix) t
+;;
+
+let ppcontext ?sep ~subst ~metasenv ctx =
+ on_buffer (ppcontext ?sep ~subst ~metasenv) ctx
+;;
+
+let ppmetasenv ~subst metasenv = on_buffer (ppmetasenv ~subst) metasenv;;
+
+let ppsubst ~metasenv subst = on_buffer (ppsubst ~metasenv) subst;;
+let ppobj obj = on_buffer ppobj obj;;
+let _ = NCicSubstitution.set_ppterm (ppterm ~margin:80);;