let ppterm =
- ref (fun ~context ~subst ~metasenv t -> "Please, set a pp callback")
+ ref (fun ~context ~subst ~metasenv ?inside_fix t -> "Please, set a pp callback")
;;
let set_ppterm f = ppterm := f;;
-let ppterm ?(context=[]) ?(subst=[]) ?(metasenv=[]) t =
- !ppterm ~context ~subst ~metasenv t
+let ppterm ~context ~subst ~metasenv ?inside_fix t =
+ !ppterm ~context ~subst ~metasenv ?inside_fix t
+;;
+
+module C = NCic
+module R = NReference
+
+let r2s pp_fix_name r =
+ try
+ match r with
+ | R.Ref (_,u,R.Ind i) ->
+ (match snd(NCicEnvironment.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 snd(NCicEnvironment.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 snd(NCicEnvironment.get_obj u) with
+ | _,_,_,_, C.Constant (_,name,_,_,_) -> name
+ | _ -> assert false)
+ | R.Ref (_,u,(R.Fix (i,_)|R.CoFix i)) ->
+ (match snd(NCicEnvironment.get_obj u) with
+ | _,_,_,_, C.Fixpoint (_,fl,_) ->
+ if pp_fix_name then
+ let _,name,_,_,_ = List.nth fl i in name
+ else
+ let name = NUri.string_of_uri u in
+ let name = Filename.basename name in
+ let name = Filename.chop_extension name in
+ name ^"("^ string_of_int i ^ ")"
+ | _ -> assert false)
+ with exn -> R.string_of_reference r
+;;
+
+let trivial_pp_term ~context ~subst ~metasenv ?(inside_fix=false) t =
+ let buff = Buffer.create 100 in
+ let f = Format.formatter_of_buffer buff in
+ let module F = Format in
+ 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 _ -> F.fprintf f " -%d" (m - List.length context))
+ | 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>";
+ 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 ⇒@;" (r2s inside_fix (R.mk_constructor 1 r));
+ aux ~toplevel:true ctx (List.hd pl);
+ F.fprintf f "@]";
+ ignore(List.fold_left
+ (fun i t ->
+ F.fprintf f "@;| @[<hov 2>%s ⇒@;" (r2s inside_fix (R.mk_constructor i r));
+ aux ~toplevel:true ctx t;
+ F.fprintf f "@]";
+ i+1)
+ 2 (List.tl pl));
+ end;
+ F.fprintf f "]@] @]";
+ | 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 _ -> F.fprintf f "?"
+ | C.Meta (n,_) -> F.fprintf f "?%d" n
+ | C.Sort C.Prop -> F.fprintf f "Prop"
+ | C.Sort C.CProp -> F.fprintf f "CProp"
+ | C.Sort (C.Type n) -> F.fprintf f "Type%d" n
+ in
+ aux ~toplevel:true (List.map fst context) t;
+ F.fprintf f "@?";
+ Buffer.contents buff
+;;
+
+let ppobj = function
+ | (u,_,metasenv,subst,NCic.Fixpoint (b, fl, _)) ->
+ "{"^NUri.string_of_uri u^"}\n"^
+ "let rec "^
+ 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:[] ~inside_fix:true bo) fl)
+ | (u,_,metasenv,subst,NCic.Inductive (b, leftno,tyl, _)) ->
+ "{"^NUri.string_of_uri u^"} with "^string_of_int leftno^" fixed params\n"^
+ "inductive "^
+ String.concat "\nand "
+ (List.map (fun (_,name,ty,cl) ->
+ name^": "^ppterm ~metasenv ~subst ~context:[] ty^ " :=\n"^
+ String.concat "\n"
+ (List.map (fun (_,name,ty) ->
+ " | "^name^": "^ppterm ~metasenv ~subst ~context:[] ty)
+ cl)) tyl) ^ "."
+ | (u,_,metasenv,subst,NCic.Constant (_,name,None,ty, _)) ->
+ "{"^NUri.string_of_uri u^"}\n"^
+ "axiom " ^ name ^ " : " ^
+ ppterm ~metasenv ~subst ~context:[] ty ^ "\n"
+ | (u,_,metasenv,subst,NCic.Constant (_,name,Some bo,ty, _)) ->
+ "{"^NUri.string_of_uri u^"}\n"^
+ "definition " ^ name ^ " : " ^
+ ppterm ~metasenv ~subst ~context:[] ty ^ " := \n"^
+ ppterm ~metasenv ~subst ~context:[] bo ^ "\n"
;;