pretty printer on steroids

[helm.git] / helm / software / components / ng_kernel / nCicPp.ml

diff --git a/helm/software/components/ng_kernel/nCicPp.ml b/helm/software/components/ng_kernel/nCicPp.ml
index a792e865d060dfdca75eda3b28b01d09427e1930..7573ad542ed2a6f15e0f84e1e162f431850fe1e4 100644 (file)
--- a/helm/software/components/ng_kernel/nCicPp.ml
+++ b/helm/software/components/ng_kernel/nCicPp.ml
@@ -1 +1,128 @@
-let ppterm t = "TODO";;
+let ppterm = 
+  ref (fun ~context ~subst ~metasenv t -> "Please, set a pp callback")
+;;
+
+let set_ppterm f = ppterm := f;;
+
+let ppterm ~context ~subst ~metasenv t = 
+  !ppterm ~context ~subst ~metasenv t
+;;
+
+module C = NCic
+module R = NReference
+
+let r2s = function
+  | 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,_) -> 
+          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)
+;;
+
+let trivial_pp_term ~context ~subst ~metasenv t = 
+  let buff = Buffer.create 100 in
+  let f = Format.formatter_of_buffer buff in
+  let module F = Format in
+  let rec aux ctx = function
+   | C.Rel m ->
+       (try F.fprintf f "%s" (List.nth ctx (m-1))
+       with Failure _ -> F.fprintf f " -%d" (m - List.length context))
+   | C.Const r -> F.fprintf f "%s" (r2s r)
+   | C.Prod ("_",s,t) -> 
+       F.fprintf f "@[<hov 1>";
+       aux ctx s; 
+       F.fprintf f "@;→ ";
+       aux ("_"::ctx) t;
+       F.fprintf f "@]";
+   | C.Prod (name,s,t) -> 
+       F.fprintf f "@[<hov 1>";
+       F.fprintf f "∀%s:" name;
+       aux ctx s; 
+       F.fprintf f ".@;";
+       aux (name::ctx) t;
+       F.fprintf f "@]";
+   | C.Lambda (name,s,t) -> 
+       F.fprintf f "@[<hov 1>";
+       F.fprintf f "λ%s:" name;
+       aux ctx s; 
+       F.fprintf f ".@;";
+       aux (name::ctx) t;
+       F.fprintf f "@]";
+   | C.LetIn (name,ty,t,b) -> 
+       F.fprintf f "let %s:" name;
+       aux ctx ty;
+       F.fprintf f "≝@;";
+       aux ctx t;
+       F.fprintf f "in@;";
+       (aux (name::ctx) b)
+   | C.Match (r,oty,t,pl) ->
+       F.fprintf f "@[<hov>match ";
+       aux ctx t;
+       F.fprintf f "@;return ";
+       aux ctx oty;
+       F.fprintf f "@; @[<v>[ ";
+       if pl <> [] then
+         begin
+           F.fprintf f "%s ⇒ " (r2s (R.mk_constructor 1 r));
+           aux ctx (List.hd pl);
+           ignore(List.fold_left 
+             (fun i t -> 
+              F.fprintf f "@;| %s ⇒ " (r2s (R.mk_constructor i r));
+              aux ctx t; i+1)
+             2 (List.tl pl));
+         end;
+      F.fprintf f "]@] @]";
+   | C.Appl l -> 
+       F.fprintf f "@[<h 2>(";
+       aux ctx (List.hd l);
+       List.iter (fun x -> F.fprintf f "@;";aux ctx x) (List.tl l);
+       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 (List.map fst context) t;
+  F.fprintf f "@?";
+  Buffer.contents buff
+;;
+
+let ppobj = function
+  | (u,_,metasenv,subst,NCic.Fixpoint (b, fl, _)) -> 
+       "let rec "^
+       String.concat "\nand " (fst
+        (List.fold_right (fun (_,name,n,ty,bo) (l,i) ->
+          ((Filename.chop_extension (Filename.basename (NUri.string_of_uri u))^
+          "("^string_of_int (i-1) ^") aka "^name^
+          " on " ^ string_of_int n ^ " : " ^ 
+          ppterm ~metasenv ~subst ~context:[] ty ^ " :=\n"^
+          ppterm ~metasenv ~subst ~context:[] bo) :: l),i-1) fl 
+           ([],List.length fl)))
+  | (u,_,_,_,NCic.Inductive (b, _,tyl, _)) -> "inductive"
+  | (u,_,metasenv,subst,NCic.Constant (_,name,None,ty, _)) -> 
+        "axiom " ^ name ^ " : " ^ 
+          ppterm ~metasenv ~subst ~context:[] ty ^ "\n"
+  | (u,_,metasenv,subst,NCic.Constant (_,name,Some bo,ty, _)) ->
+        "definition " ^ name ^ " : " ^ 
+          ppterm ~metasenv ~subst ~context:[] ty ^ " := \n"^
+          ppterm ~metasenv ~subst ~context:[] bo ^ "\n"
+;;