]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/software/components/ng_tactics/nCicElim.ml
Introduction of vectors of implicit (only for NG).
[helm.git] / helm / software / components / ng_tactics / nCicElim.ml
index fba21753a53ce6ef8b749839b716020ca7b702e8..0abff3a6c694727f79cc2e07eded3fd6ce48cc82 100644 (file)
@@ -34,89 +34,262 @@ let rec my_split_prods ~subst context n te =
    | (_, _) -> raise (Failure "my_split_prods")
 ;;
 
+let mk_appl =
+ function
+    [] -> assert false
+  | [x] -> x
+  | l -> CicNotationPt.Appl l
+;;
+
+let mk_elim uri leftno [it] (outsort,suffix) =
+ let _,ind_name,ty,cl = it in
+ let srec_name = ind_name ^ "_" ^ suffix in
+ let rec_name = mk_id srec_name in
+ let name_of_k id = mk_id ("H_" ^ id) in
+ let p_name = mk_id "Q_" in
+ let params,ty = NCicReduction.split_prods ~subst:[] [] leftno ty in
+ let params = List.rev_map (function name,_ -> mk_id name) params in
+ let args,sort = NCicReduction.split_prods ~subst:[] [] (-1) ty in
+ let args = List.rev_map (function name,_ -> mk_id name) args in
+ let rec_arg = mk_id (fresh_name ()) in
+ let p_ty =
+  List.fold_right
+   (fun name res -> CicNotationPt.Binder (`Forall,(name,None),res)) args
+   (CicNotationPt.Binder
+    (`Forall,
+     (rec_arg,Some (mk_appl (mk_id ind_name :: params @ args))),
+     CicNotationPt.Sort outsort)) in
+ let args = args @ [rec_arg] in
+ let k_names = List.map (function _,name,_ -> name_of_k name) cl in
+ let final_params =
+  List.map (function name -> name, None) params @
+  [p_name,Some p_ty] @
+  List.map (function name -> name, None) k_names @
+  List.map (function name -> name, None) args in
+ let cty = mk_appl (p_name :: args) in
+ let ty = Some cty in
+ let branches =
+  List.map
+   (function (_,name,ty) ->
+     let _,ty = NCicReduction.split_prods ~subst:[] [] leftno ty in
+     let cargs,ty= my_split_prods ~subst:[] [] (-1) ty in
+     let cargs_and_recursive_args =
+      List.rev_map
+       (function
+           _,NCic.Def _ -> assert false
+         | name,NCic.Decl ty ->
+            let context,ty = my_split_prods ~subst:[] [] (-1) ty in
+             match ty with
+              | NCic.Const nref
+              | NCic.Appl (NCic.Const nref::_)
+                 when
+                  let NReference.Ref (uri',_) = nref in
+                   NUri.eq uri uri'
+                 ->
+                  let abs = List.rev_map (fun id,_ -> mk_id id) context in
+                  let name = mk_id name in
+                   name, Some (
+                   List.fold_right
+                    (fun id res ->
+                      CicNotationPt.Binder (`Lambda,(id,None),res))
+                    abs
+                    (CicNotationPt.Appl
+                     (rec_name ::
+                      params @
+                      [p_name] @
+                      k_names @
+                      List.map (fun _ -> CicNotationPt.Implicit `JustOne)
+                       (List.tl args) @
+                      [mk_appl (name::abs)])))
+              | _ -> mk_id name,None
+       ) cargs in
+     let cargs,recursive_args = List.split cargs_and_recursive_args in
+     let recursive_args = HExtlib.filter_map (fun x -> x) recursive_args in
+      CicNotationPt.Pattern (name,None,List.map (fun x -> x,None) cargs),
+       mk_appl (name_of_k name :: cargs @ recursive_args)
+   ) cl
+ in
+ let bo = CicNotationPt.Case (rec_arg,Some (ind_name,None),None,branches) in
+ let recno = List.length final_params in
+ let where = recno - 1 in
+ let res =
+  CicNotationPt.LetRec (`Inductive,
+   [final_params, (rec_name,ty), bo, where], rec_name)
+ in
+(*
+  prerr_endline
+   (BoxPp.render_to_string
+     ~map_unicode_to_tex:false
+     (function x::_ -> x | _ -> assert false)
+     80 (CicNotationPres.render (fun _ -> None)
+     (TermContentPres.pp_ast res)));
+  prerr_endline "#####";
+  let cobj = ("xxx", [], None, `Joint {
+      Content.joint_id = "yyy";
+      joint_kind = `Recursive [recno];
+      joint_defs =
+       [ `Definition {
+            Content.def_name = Some srec_name;
+            def_id = "zzz";
+            def_aref = "www";
+            def_term = bo;
+            def_type = 
+              List.fold_right 
+                (fun x t -> CicNotationPt.Binder(`Forall,x,t))
+                final_params cty
+          }
+       ];
+    })
+  in
+  let ids_to_nrefs = Hashtbl.create 1 in
+  let boxml = Content2pres.ncontent2pres ~ids_to_nrefs cobj in
+  prerr_endline (
+   (BoxPp.render_to_string ~map_unicode_to_tex:false
+     (function x::_ -> x | _ -> assert false) 80 
+     (CicNotationPres.mpres_of_box boxml)));
+*)
+  CicNotationPt.Theorem
+   (`Definition,srec_name,CicNotationPt.Implicit `JustOne,Some res)
+;;
+
+let ast_of_sort s =
+  match s with
+     NCic.Prop -> `Prop,"ind"
+   | NCic.Type u ->
+      let u = NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] (NCic.Sort s) in
+      (try
+        if String.sub u 0 4 = "Type" then
+         `NType (String.sub u 4 (String.length u - 4)), "rect_" ^ u
+        else if String.sub u 0 5 = "CProp" then
+         `NCProp (String.sub u 5 (String.length u - 5)), "rect_" ^ u
+        else
+         (prerr_endline u;
+         assert false)
+       with Failure _ -> assert false)
+;;
+
 let mk_elims (uri,_,_,_,obj) =
- match obj with
-    NCic.Inductive (true,leftno,[it],_) ->
-     let _,ind_name,ty,cl = it in
-     let rec_name = ind_name ^ "_rect" in
-     let rec_name = mk_id rec_name in
-     let name_of_k id = mk_id ("H_" ^ id) in
-     let p_name = mk_id "Q_" in
-     let params,ty = NCicReduction.split_prods ~subst:[] [] leftno ty in
-     let params = List.rev_map (function name,_ -> mk_id name) params in
-     let args,sort = NCicReduction.split_prods ~subst:[] [] (-1) ty in
-     let args = List.rev_map (function name,_ -> mk_id name) args in
-     let rec_arg = mk_id (fresh_name ()) in
-     let p_ty =
-      List.fold_right
-       (fun name res -> CicNotationPt.Binder (`Forall,(name,None),res)) args
-       (CicNotationPt.Binder
-        (`Forall,
-         (rec_arg,Some (CicNotationPt.Appl (mk_id ind_name :: params @ args))),
-         CicNotationPt.Sort (`Type (CicUniv.fresh ())))) in
-     let args = args @ [rec_arg] in
-     let k_names = List.map (function _,name,_ -> name_of_k name) cl in
-     let final_params =
-      List.map (function name -> name, None) params @
-      [p_name,Some p_ty] @
-      List.map (function name -> name, None) k_names @
-      List.map (function name -> name, None) args in
-     let ty = Some (CicNotationPt.Appl (p_name :: args)) in
-     let branches =
+  match obj with
+     NCic.Inductive (true,leftno,itl,_) ->
+      List.map (fun s -> mk_elim uri leftno itl (ast_of_sort s))
+       (NCic.Prop::
+         List.map (fun s -> NCic.Type s) (NCicEnvironment.get_universes ()))
+   | _ -> []
+;;
+
+(********************* Projections **********************)
+
+let mk_lambda =
+ function
+    [] -> assert false 
+  | [t] -> t
+  | l -> CicNotationPt.Appl l
+;;
+
+let rec count_prods = function NCic.Prod (_,_,t) -> 1 + count_prods t | _ -> 0;;
+
+let rec nth_prod projs n ty =
+ match ty with
+    NCic.Prod (_,s,_) when n=0 -> projs, s
+  | NCic.Prod (name,_,t) -> nth_prod (name::projs) (n-1) t
+  | _ -> assert false
+;;
+
+(* this code should be unified with NTermCicContent.nast_of_cic0,
+   but the two contexts have different types *)
+let rec pp rels =
+ function
+    NCic.Rel i -> List.nth rels (i - 1)
+  | NCic.Const _ as t ->
+     CicNotationPt.Ident
+      (NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] t,None)
+  | NCic.Sort s -> CicNotationPt.Sort (fst (ast_of_sort s))
+  | NCic.Meta _
+  | NCic.Implicit _ -> assert false
+  | NCic.Appl l -> CicNotationPt.Appl (List.map (pp rels) l)
+  | NCic.Prod (n,s,t) ->
+     let n = mk_id n in
+      CicNotationPt.Binder (`Pi, (n,Some (pp rels s)), pp (n::rels) t)
+  | NCic.Lambda (n,s,t) ->
+     let n = mk_id n in
+      CicNotationPt.Binder (`Lambda, (n,Some (pp rels s)), pp (n::rels) t)
+  | NCic.LetIn (n,s,ty,t) ->
+     let n = mk_id n in
+      CicNotationPt.LetIn ((n, Some (pp rels ty)), pp rels s, pp (n::rels) t)
+  | NCic.Match (NReference.Ref (uri,_) as r,outty,te,patterns) ->
+    let name = NUri.name_of_uri uri in
+    let case_indty = Some (name, None) in
+    let constructors, leftno =
+     let _,leftno,tys,_,n = NCicEnvironment.get_checked_indtys r in
+     let _,_,_,cl  = List.nth tys n in
+      cl,leftno
+    in
+    let rec eat_branch n rels ty pat =
+      match (ty, pat) with
+      | NCic.Prod (name, s, t), _ when n > 0 ->
+         eat_branch (pred n) rels t pat
+      | NCic.Prod (_, _, t), NCic.Lambda (name, s, t') ->
+          let cv, rhs = eat_branch 0 ((mk_id name)::rels) t t' in
+           (mk_id name, Some (pp rels s)) :: cv, rhs
+      | _, _ -> [], pp rels pat
+    in
+    let patterns =
+      try
+        List.map2
+          (fun (_, name, ty) pat ->
+            let capture_variables,rhs = eat_branch leftno rels ty pat in
+             CicNotationPt.Pattern (name, None, capture_variables), rhs
+          ) constructors patterns
+      with Invalid_argument _ -> assert false
+    in
+     CicNotationPt.Case (pp rels te, case_indty, Some (pp rels outty), patterns)
+;;
+
+let mk_projection leftno tyname consname consty (projname,_,_) i =
+ let argsno = count_prods consty - leftno in
+ let rec aux names ty leftno =
+  match leftno,ty with
+   | 0,_ ->
+     let arg = mk_id "xxx" in
+     let arg_ty = mk_appl (mk_id tyname :: List.rev names) in
+     let bvar = mk_id "yyy" in
+     let underscore = CicNotationPt.Ident ("_",None),None in
+     let bvars =
+      HExtlib.mk_list underscore i @ [bvar,None] @
+       HExtlib.mk_list underscore (argsno - i -1) in
+     let branch = CicNotationPt.Pattern (consname,None,bvars), bvar in
+     let projs,outtype = nth_prod [] i ty in
+     let rels =
       List.map
-       (function (_,name,ty) ->
-         let _,ty = NCicReduction.split_prods ~subst:[] [] leftno ty in
-         let cargs,ty= my_split_prods ~subst:[] [] (-1) ty in
-         let cargs_and_recursive_args =
-          List.rev_map
-           (function
-               _,NCic.Def _ -> assert false
-             | name,NCic.Decl ty ->
-                let context,ty = my_split_prods ~subst:[] [] (-1) ty in
-                 match ty with
-                  | NCic.Const nref
-                  | NCic.Appl (NCic.Const nref::_)
-                     when
-                      let NReference.Ref (uri',_) = nref in
-                       NUri.eq uri uri'
-                     ->
-                      let abs = List.rev_map (fun id,_ -> mk_id id) context in
-                      let name = mk_id name in
-                       name, Some (
-                       List.fold_right
-                        (fun id res ->
-                          CicNotationPt.Binder (`Lambda,(id,None),res))
-                        abs
-                        (CicNotationPt.Appl
-                         (rec_name ::
-                          params @
-                          [p_name] @
-                          k_names @
-                          List.map (fun _ -> CicNotationPt.Implicit)
-                           (List.tl args) @
-                          [CicNotationPt.Appl (name::abs)])))
-                  | _ -> mk_id name,None
-           ) cargs in
-         let cargs,recursive_args = List.split cargs_and_recursive_args in
-         let recursive_args = HExtlib.filter_map (fun x -> x) recursive_args in
-          CicNotationPt.Pattern (name,None,List.map (fun x -> x,None) cargs),
-           CicNotationPt.Appl (name_of_k name :: cargs @ recursive_args)
-       ) cl
-     in
-     let bo = CicNotationPt.Case (rec_arg,None,None,branches) in
-     let where = List.length final_params - 1 in
-     let res =
-      CicNotationPt.LetRec (`Inductive,
-       [final_params, (rec_name,ty), bo, where], rec_name)
-     in
-      prerr_endline (CicNotationPp.pp_term res);
-      prerr_endline "#####";
-      prerr_endline
-       (BoxPp.render_to_string
-         ~map_unicode_to_tex:false
-         (function x::_ -> x | _ -> assert false)
-         80 (CicNotationPres.render (fun _ -> None)
-         (TermContentPres.pp_ast res)));
-      []
+       (fun name -> mk_appl (mk_id name :: List.rev names @ [arg])) projs
+      @ names in
+     let outtype = pp rels outtype in
+     let outtype= CicNotationPt.Binder (`Lambda, (arg, Some arg_ty), outtype) in
+      CicNotationPt.Binder
+       (`Lambda, (arg,Some arg_ty),
+         CicNotationPt.Case (arg,None,Some outtype,[branch]))
+   | _,NCic.Prod (name,_,t) ->
+      let name = mk_id name in
+       CicNotationPt.Binder
+        (`Lambda, (name,None), aux (name::names) t (leftno - 1))
+   | _,_ -> assert false
+ in
+ let res = aux [] consty leftno in
+(* prerr_endline
+   (BoxPp.render_to_string
+     ~map_unicode_to_tex:false
+     (function x::_ -> x | _ -> assert false)
+     80 (CicNotationPres.render (fun _ -> None)
+     (TermContentPres.pp_ast res)));*)
+  CicNotationPt.Theorem
+   (`Definition,projname,CicNotationPt.Implicit `JustOne,Some res)
+;;
+
+let mk_projections (_,_,_,_,obj) =
+ match obj with
+    NCic.Inductive
+     (true,leftno,[_,tyname,_,[_,consname,consty]],(_,`Record fields))
+    ->
+     HExtlib.list_mapi (mk_projection leftno tyname consname consty) fields
   | _ -> []
 ;;