]> matita.cs.unibo.it Git - helm.git/blobdiff - components/cic_exportation/cicExportation.ml
Trivial bug fixed in type inference of LetIn source types.
[helm.git] / components / cic_exportation / cicExportation.ml
index e5f1face1e9eb854049ba089b723f8d0a8705321..813e65c298bf9c5c9281597088dadf655b7db3e6 100644 (file)
@@ -62,7 +62,8 @@ let analyze_type context t =
 let ppid =
  let reserved =
   [ "to";
-    "mod"
+    "mod";
+    "val"
   ]
  in
   function n ->
@@ -85,27 +86,25 @@ let rec get_nth l n =
   | (_,_) -> raise NotEnoughElements
 ;;
 
-let qualified_name_of_uri current_module_name ?(capitalize=false) uri =
+let qualified_name_of_uri current_module_uri ?(capitalize=false) uri =
  let name =
   if capitalize then
    String.capitalize (UriManager.name_of_uri uri)
   else
    ppid (UriManager.name_of_uri uri) in
- let buri = UriManager.buri_of_uri uri in
- let index = String.rindex buri '/' in
- let filename = String.sub buri (index + 1) (String.length buri - index - 1) in
-  if current_module_name = filename then
+  let filename =
+   let suri = UriManager.buri_of_uri uri in
+   let s = String.sub suri 5 (String.length suri - 5) in
+   let s = Pcre.replace ~pat:"/" ~templ:"_" s in
+    String.uncapitalize s in
+  if current_module_uri = UriManager.buri_of_uri uri then
    name
   else
    String.capitalize filename ^ "." ^ name
 ;;
 
-(* pp t l                                                                  *)
-(* pretty-prints a term t of cic in an environment l where l is a list of  *)
-(* identifier names used to resolve DeBrujin indexes. The head of l is the *)
-(* name associated to the greatest DeBrujin index in t                     *)
-let pp current_module_name ?metasenv =
-let rec pp t context =
+let pp current_module_uri ?metasenv ~in_type =
+let rec pp ~in_type t context =
  let module C = Cic in
    match t with
       C.Rel n ->
@@ -120,14 +119,17 @@ let rec pp t context =
          NotEnoughElements -> string_of_int (List.length context - n)
        end
     | C.Var (uri,exp_named_subst) ->
-        qualified_name_of_uri current_module_name uri ^
+        qualified_name_of_uri current_module_uri uri ^
          pp_exp_named_subst exp_named_subst context
     | C.Meta (n,l1) ->
        (match metasenv with
            None ->
             "?" ^ (string_of_int n) ^ "[" ^ 
              String.concat " ; "
-              (List.rev_map (function None -> "_" | Some t -> pp t context) l1) ^
+              (List.rev_map
+                (function
+                    None -> "_"
+                  | Some t -> pp ~in_type:false t context) l1) ^
              "]"
          | Some metasenv ->
             try
@@ -140,7 +142,7 @@ let rec pp t context =
                       match x,y with
                          _, None
                        | None, _ -> "_"
-                       | Some _, Some t -> pp t context
+                       | Some _, Some t -> pp ~in_type:false t context
                     ) context l1)) ^
                "]"
             with
@@ -148,7 +150,8 @@ let rec pp t context =
             | Invalid_argument _ ->
               "???" ^ (string_of_int n) ^ "[" ^ 
                String.concat " ; "
-                (List.rev_map (function None -> "_" | Some t -> pp t context) l1) ^
+                (List.rev_map (function None -> "_" | Some t ->
+                  pp ~in_type:false t context) l1) ^
                "]"
        )
     | C.Sort s ->
@@ -163,33 +166,48 @@ let rec pp t context =
     | C.Implicit _ -> "?"
     | C.Prod (b,s,t) ->
        (match b with
-          C.Name n -> "(\\forall " ^ n ^ ":" ^ pp s context ^ "." ^ pp t ((Some (b,Cic.Decl s))::context) ^ ")"
-        | C.Anonymous -> "(" ^ pp s context ^ "\\to " ^ pp t ((Some (b,Cic.Decl s))::context) ^ ")"
-       )
-    | C.Cast (v,t) -> pp v context
+          C.Name n ->
+           let n = "'" ^ String.uncapitalize n in
+            "(" ^ pp ~in_type:true s context ^ " -> " ^
+            pp ~in_type:true t ((Some (Cic.Name n,Cic.Decl s))::context) ^ ")"
+        | C.Anonymous ->
+           "(" ^ pp ~in_type:true s context ^ " -> " ^
+           pp ~in_type:true t ((Some (b,Cic.Decl s))::context) ^ ")")
+    | C.Cast (v,t) -> pp ~in_type v context
     | C.Lambda (b,s,t) ->
        (match analyze_type context s with
            `Sort _
-         | `Statement -> pp t ((Some (b,Cic.Decl s))::context)
-         | `Type -> "(function " ^ ppname b ^ " -> " ^ pp t ((Some (b,Cic.Decl s))::context) ^ ")")
+         | `Statement -> pp ~in_type t ((Some (b,Cic.Decl s))::context)
+         | `Type ->
+            "(function " ^ ppname b ^ " -> " ^
+             pp ~in_type t ((Some (b,Cic.Decl s))::context) ^ ")")
     | C.LetIn (b,s,t) ->
-       let ty,_ = CicTypeChecker.type_of_aux' [] context t CicUniv.oblivion_ugraph in
-       "(let " ^ ppname b ^ " = " ^ pp s context ^ " in " ^ pp t ((Some (b,Cic.Def (s,Some ty)))::context) ^ ")"
-    | C.Appl (C.MutConstruct _ as he::tl) ->
-       let hes = pp he context in
-       let stl = String.concat "," (clean_args context tl) in
+       let ty,_ = CicTypeChecker.type_of_aux' [] context s CicUniv.oblivion_ugraph in
+       "(let " ^ ppname b ^ " = " ^ pp ~in_type:false s context ^ " in " ^
+        pp ~in_type t ((Some (b,Cic.Def (s,Some ty)))::context) ^ ")"
+    | C.Appl (C.MutInd _ as he::tl) ->
+       let hes = pp ~in_type he context in
+       let stl = String.concat "," (clean_args_for_ty context tl) in
+        (if stl = "" then "" else "(" ^ stl ^ ") ") ^ hes
+    | C.Appl (C.MutConstruct (uri,n,_,_) as he::tl) ->
+       let nparams =
+        match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
+           C.InductiveDefinition (_,_,nparams,_) -> nparams
+         | _ -> assert false in
+       let hes = pp ~in_type he context in
+       let stl = String.concat "," (clean_args nparams context tl) in
         "(" ^ hes ^ (if stl = "" then "" else "(" ^ stl ^ ")") ^ ")"
     | C.Appl li ->
-       "(" ^ String.concat " " (clean_args context li) ^ ")"
+       "(" ^ String.concat " " (clean_args context li) ^ ")"
     | C.Const (uri,exp_named_subst) ->
-       qualified_name_of_uri current_module_name uri ^
+       qualified_name_of_uri current_module_uri uri ^
         pp_exp_named_subst exp_named_subst context
     | C.MutInd (uri,n,exp_named_subst) ->
        (try
          match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
             C.InductiveDefinition (dl,_,_,_) ->
              let (name,_,_,_) = get_nth dl (n+1) in
-              qualified_name_of_uri current_module_name
+              qualified_name_of_uri current_module_uri
                (UriManager.uri_of_string
                  (UriManager.buri_of_uri uri ^ "/" ^ name ^ ".con")) ^
               pp_exp_named_subst exp_named_subst context
@@ -204,7 +222,7 @@ let rec pp t context =
             C.InductiveDefinition (dl,_,_,_) ->
              let _,_,_,cons = get_nth dl (n1+1) in
               let id,_ = get_nth cons n2 in
-               qualified_name_of_uri current_module_name ~capitalize:true
+               qualified_name_of_uri current_module_uri ~capitalize:true
                 (UriManager.uri_of_string
                   (UriManager.buri_of_uri uri ^ "/" ^ id ^ ".con")) ^
                pp_exp_named_subst exp_named_subst context
@@ -216,64 +234,101 @@ let rec pp t context =
            string_of_int n2
        )
     | C.MutCase (uri,n1,ty,te,patterns) ->
-       let connames_and_argsno =
-        (match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
-            C.InductiveDefinition (dl,_,paramsno,_) ->
-             let (_,_,_,cons) = get_nth dl (n1+1) in
-              List.map
-               (fun (id,ty) ->
-                 (* this is just an approximation since we do not have
-                    reduction yet! *)
-                 let rec count_prods toskip =
-                  function
-                     C.Prod (_,_,bo) when toskip > 0 ->
-                      count_prods (toskip - 1) bo
-                   | C.Prod (_,_,bo) -> 1 + count_prods 0 bo
-                   | _ -> 0
-                 in
-                  qualified_name_of_uri current_module_name ~capitalize:true
-                   (UriManager.uri_of_string
-                     (UriManager.buri_of_uri uri ^ "/" ^ id ^ ".con")),
-                  count_prods paramsno ty
-               ) cons
-          | _ -> raise CicExportationInternalError
-        )
+       if in_type then
+        "unit (* TOO POLYMORPHIC TYPE *)"
+       else (
+       let needs_obj_magic =
+        (* BUG HERE: we should consider also the right parameters *)
+        match CicReduction.whd context ty with
+           Cic.Lambda (_,_,t) -> not (DoubleTypeInference.does_not_occur 1 t)
+         | _ -> false (* it can be a Rel, e.g. in *_rec *)
        in
-        let connames_and_argsno_and_patterns =
-         let rec combine =
-            function
-               [],[] -> []
-             | [],l -> List.map (fun x -> "???",0,Some x) l
-             | l,[] -> List.map (fun (x,no) -> x,no,None) l
-             | (x,no)::tlx,y::tly -> (x,no,Some y)::(combine (tlx,tly))
-         in
-          combine (connames_and_argsno,patterns)
-        in
-         "\n(match " ^ pp te context ^ " with \n" ^
-          (String.concat "\n | "
-           (List.map
-            (fun (x,argsno,y) ->
-              let rec aux argsno context =
-               function
-                  Cic.Lambda (name,ty,bo) when argsno > 0 ->
-                   let args,res = aux (argsno - 1) (Some (name,Cic.Decl ty)::context) bo in
-                    (match name with C.Anonymous -> "_" | C.Name s -> s)::args,
-                    res
-                | t when argsno = 0 -> [],pp t context
-                | t -> ["{" ^ string_of_int argsno ^ " args missing}"],pp t context
+       (match analyze_term context te with
+           `Type -> assert false
+         | `Proof ->
+             (match patterns with
+                 [] -> "assert false"   (* empty type elimination *)
+               | [he] ->
+                  pp ~in_type:false he context  (* singleton elimination *)
+               | _ -> assert false)
+         | `Term ->
+            if patterns = [] then "assert false"
+            else
+             (let connames_and_argsno =
+               (match fst(CicEnvironment.get_obj CicUniv.empty_ugraph uri) with
+                   C.InductiveDefinition (dl,_,paramsno,_) ->
+                    let (_,_,_,cons) = get_nth dl (n1+1) in
+                     List.map
+                      (fun (id,ty) ->
+                        (* this is just an approximation since we do not have
+                           reduction yet! *)
+                        let rec count_prods toskip =
+                         function
+                            C.Prod (_,_,bo) when toskip > 0 ->
+                             count_prods (toskip - 1) bo
+                          | C.Prod (_,_,bo) -> 1 + count_prods 0 bo
+                          | _ -> 0
+                        in
+                         qualified_name_of_uri current_module_uri
+                          ~capitalize:true
+                          (UriManager.uri_of_string
+                            (UriManager.buri_of_uri uri ^ "/" ^ id ^ ".con")),
+                         count_prods paramsno ty
+                      ) cons
+                 | _ -> raise CicExportationInternalError
+               )
               in
-               let pattern,body =
-                match y with
-                   None -> x,""
-                 | Some y when argsno = 0 -> x,pp y context
-                 | Some y ->
-                    let args,body = aux argsno context y in
-                    let sargs = String.concat "," args in
-                     x ^ (if sargs = "" then "" else "(" ^ sargs^ ")"),body
+               let connames_and_argsno_and_patterns =
+                let rec combine =
+                   function
+                      [],[] -> []
+                    | (x,no)::tlx,y::tly -> (x,no,y)::(combine (tlx,tly))
+                    | _,_ -> assert false
+                in
+                 combine (connames_and_argsno,patterns)
                in
-                pattern ^ " -> " ^ body
-            ) connames_and_argsno_and_patterns)) ^
-          ")\n"
+                "\n(match " ^ pp ~in_type:false te context ^ " with \n   " ^
+                 (String.concat "\n | "
+                  (List.map
+                   (fun (x,argsno,y) ->
+                     let rec aux argsno context =
+                      function
+                         Cic.Lambda (name,ty,bo) when argsno > 0 ->
+                          let name =
+                           match name with
+                              Cic.Anonymous -> Cic.Anonymous
+                            | Cic.Name n -> Cic.Name (ppid n) in
+                          let args,res =
+                           aux (argsno - 1) (Some (name,Cic.Decl ty)::context)
+                            bo
+                          in
+                           (match analyze_type context ty with
+                               `Statement
+                             | `Sort _ -> args,res
+                             | `Type ->
+                                 (match name with
+                                     C.Anonymous -> "_"
+                                   | C.Name s -> s)::args,res)
+                       | t when argsno = 0 -> [],pp ~in_type:false t context
+                       | t ->
+                          ["{" ^ string_of_int argsno ^ " args missing}"],
+                           pp ~in_type:false t context
+                     in
+                      let pattern,body =
+                       if argsno = 0 then x,pp ~in_type:false y context
+                       else
+                        let args,body = aux argsno context y in
+                        let sargs = String.concat "," args in
+                         x ^ (if sargs = "" then "" else "(" ^ sargs^ ")"),
+                          body
+                      in
+                       pattern ^ " -> " ^
+                        if needs_obj_magic then
+                         "Obj.magic (" ^ body ^ ")"
+                        else
+                         body
+                   ) connames_and_argsno_and_patterns)) ^
+                 ")\n")))
     | C.Fix (no, funs) ->
        let names =
         List.rev
@@ -281,15 +336,15 @@ let rec pp t context =
            (function (name,_,ty,_) ->
              Some (C.Name name,Cic.Decl ty)) funs)
        in
-         "\nlet rec " ^
+         "let rec " ^
          List.fold_right
           (fun (name,ind,ty,bo) i -> name ^ " = \n" ^
-            pp bo (names@context) ^ i)
+            pp ~in_type:false bo (names@context) ^ i)
           funs "" ^
          " in " ^
          (match get_nth names (no + 1) with
              Some (Cic.Name n,_) -> n
-           | _ -> assert false) ^ "\n"
+           | _ -> assert false)
     | C.CoFix (no,funs) ->
        let names =
         List.rev
@@ -300,8 +355,8 @@ let rec pp t context =
          "\nCoFix " ^ " {" ^
          List.fold_right
           (fun (name,ty,bo) i -> "\n" ^ name ^ 
-            " : " ^ pp ty context ^ " := \n" ^
-            pp bo (names@context) ^ i)
+            " : " ^ pp ~in_type:true ty context ^ " := \n" ^
+            pp ~in_type:false bo (names@context) ^ i)
           funs "" ^
          "}\n"
 and pp_exp_named_subst exp_named_subst context =
@@ -309,40 +364,63 @@ and pp_exp_named_subst exp_named_subst context =
   "\\subst[" ^
    String.concat " ; " (
     List.map
-     (function (uri,t) -> UriManager.name_of_uri uri ^ " \\Assign " ^ pp t context)
+     (function (uri,t) -> UriManager.name_of_uri uri ^ " \\Assign " ^ pp ~in_type:false t context)
      exp_named_subst
    ) ^ "]"
-and clean_args context =
+and clean_args nparams context =
+ let nparams = ref nparams in
  HExtlib.filter_map
   (function t ->
+    decr nparams;
     match analyze_term context t with
-       `Type -> None
+       `Term when !nparams < 0 -> Some (pp ~in_type:false t context)
+     | `Term
+     | `Type
+     | `Proof -> None)
+and clean_args_for_ty context =
+ HExtlib.filter_map
+  (function t ->
+    match analyze_term context t with
+       `Type -> Some (pp ~in_type:true t context)
      | `Proof -> None
-     | `Term -> Some pp t context)
+     | `Term -> None)
 in
- pp
+ pp ~in_type
 ;;
 
-let ppty current_module_name =
- let rec args context =
+let ppty current_module_uri =
+ (* nparams is the number of left arguments
+    left arguments should either become parameters or be skipped altogether *)
+ let rec args nparams context =
   function
      Cic.Prod (n,s,t) ->
+      let n =
+       match n with
+          Cic.Anonymous -> Cic.Anonymous
+        | Cic.Name n -> Cic.Name (String.uncapitalize n)
+      in
        (match analyze_type context s with
-           `Sort Cic.Prop -> args ((Some (n,Cic.Decl s))::context) t
-         | `Statement
+           `Statement
+         | `Sort Cic.Prop ->
+             args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
+         | `Type when nparams > 0 ->
+             args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
+         | `Type ->
+             let abstr,args =
+              args (nparams - 1) ((Some (n,Cic.Decl s))::context) t in
+               abstr,pp ~in_type:true current_module_uri s context::args
          | `Sort _ ->
              let n =
               match n with
                  Cic.Anonymous -> Cic.Anonymous
                | Cic.Name name -> Cic.Name ("'" ^ name) in
-             let abstr,args = args ((Some (n,Cic.Decl s))::context) t in
+             let abstr,args =
+              args (nparams - 1) ((Some (n,Cic.Decl s))::context) t
+             in
               (match n with
                   Cic.Anonymous -> abstr
                 | Cic.Name name -> name::abstr),
-              args
-         | `Type ->
-             let abstr,args = args ((Some (n,Cic.Decl s))::context) t in
-              abstr,pp current_module_name s context::args)
+              args)
    | _ -> [],[]
  in
   args
@@ -351,45 +429,49 @@ let ppty current_module_name =
 (* ppinductiveType (typename, inductive, arity, cons)                       *)
 (* pretty-prints a single inductive definition                              *)
 (* (typename, inductive, arity, cons)                                       *)
-let ppinductiveType current_module_name (typename, inductive, arity, cons) =
+let ppinductiveType current_module_uri nparams (typename, inductive, arity, cons)
+=
  match analyze_type [] arity with
     `Sort Cic.Prop -> ""
   | `Statement
   | `Type -> assert false
   | `Sort _ ->
-    let abstr,scons =
-     List.fold_right
-      (fun (id,ty) (abstr,i) ->
-         let abstr',sargs = ppty current_module_name [] ty in
-         let sargs = String.concat " * " sargs in
-          abstr'@abstr,
-          String.capitalize id ^
-           (if sargs = "" then "" else " of " ^ sargs) ^
-           (if i = "" then "\n" else "\n | ") ^ i)
-      cons ([],"")
-     in
-      let abstr =
-       let s = String.concat "," abstr in
-       if s = "" then "" else "(" ^ s ^ ") "
+    if cons = [] then
+     "type " ^ String.uncapitalize typename ^ " = unit (* empty type *)\n"
+    else (
+     let abstr,scons =
+      List.fold_right
+       (fun (id,ty) (_abstr,i) -> (* we should verify _abstr = abstr' *)
+          let abstr',sargs = ppty current_module_uri nparams [] ty in
+          let sargs = String.concat " * " sargs in
+           abstr',
+           String.capitalize id ^
+            (if sargs = "" then "" else " of " ^ sargs) ^
+            (if i = "" then "" else "\n | ") ^ i)
+       cons ([],"")
       in
-      "type " ^ abstr ^ String.uncapitalize typename ^ " =\n" ^ scons ^ "\n"
+       let abstr =
+        let s = String.concat "," abstr in
+        if s = "" then "" else "(" ^ s ^ ") "
+       in
+       "type " ^ abstr ^ String.uncapitalize typename ^ " =\n" ^ scons ^ "\n")
 ;;
 
-let ppobj current_module_name obj =
+let ppobj current_module_uri obj =
  let module C = Cic in
  let module U = UriManager in
-  let pp = pp current_module_name in
+  let pp ~in_type = pp ~in_type current_module_uri in
   match obj with
     C.Constant (name, Some t1, t2, params, _) ->
       (match analyze_type [] t2 with
           `Sort Cic.Prop
         | `Statement -> ""
-        | `Type -> "let " ^ ppid name ^ " =\n" ^ pp t1 [] ^ "\n"
+        | `Type -> "let " ^ ppid name ^ " =\n" ^ pp ~in_type:false t1 [] ^ "\n"
         | `Sort _ ->
             match analyze_type [] t1 with
                `Sort Cic.Prop -> ""
              | _ ->
-               let abstr,args = ppty current_module_name [] t1 in
+               let abstr,args = ppty current_module_uri 0 [] t1 in
                let abstr =
                 let s = String.concat "," abstr in
                 if s = "" then "" else "(" ^ s ^ ") "
@@ -406,8 +488,8 @@ let ppobj current_module_name obj =
       "Variable " ^ name ^
        "(" ^ String.concat ";" (List.map UriManager.string_of_uri params) ^
        ")" ^ ":\n" ^
-       pp ty [] ^ "\n" ^
-       (match bo with None -> "" | Some bo -> ":= " ^ pp bo [])
+       pp ~in_type:true ty [] ^ "\n" ^
+       (match bo with None -> "" | Some bo -> ":= " ^ pp ~in_type:false bo [])
    | C.CurrentProof (name, conjectures, value, ty, params, _) ->
       "Current Proof of " ^ name ^
        "(" ^ String.concat ";" (List.map UriManager.string_of_uri params) ^
@@ -422,12 +504,13 @@ let ppobj current_module_name obj =
                        Some (n,C.Decl at) ->
                          (separate i) ^
                            ppname n ^ ":" ^
-                            pp ~metasenv:conjectures at name_context ^ " ",
+                            pp ~in_type:true ~metasenv:conjectures
+                            at name_context ^ " ",
                             context_entry::name_context
                      | Some (n,C.Def (at,None)) ->
                          (separate i) ^
-                           ppname n ^ ":= " ^ pp ~metasenv:conjectures
-                            at name_context ^ " ",
+                           ppname n ^ ":= " ^ pp ~in_type:false
+                            ~metasenv:conjectures at name_context ^ " ",
                             context_entry::name_context
                       | None ->
                          (separate i) ^ "_ :? _ ", context_entry::name_context
@@ -435,16 +518,16 @@ let ppobj current_module_name obj =
             ) context ("",[])
           in
            conjectures' ^ " |- " ^ "?" ^ (string_of_int n) ^ ": " ^
-            pp ~metasenv:conjectures t name_context ^ "\n" ^ i
+            pp ~in_type:true ~metasenv:conjectures t name_context ^ "\n" ^ i
         ) conjectures "" ^
-        "\n" ^ pp ~metasenv:conjectures value [] ^ " : " ^
-          pp ~metasenv:conjectures ty [] 
+        "\n" ^ pp ~in_type:false ~metasenv:conjectures value [] ^ " : " ^
+          pp ~in_type:true ~metasenv:conjectures ty [] 
    | C.InductiveDefinition (l, params, nparams, _) ->
       List.fold_right
-       (fun x i -> ppinductiveType current_module_name x ^ i) l ""
+       (fun x i -> ppinductiveType current_module_uri nparams x ^ i) l ""
 ;;
 
-let ppobj current_module_name obj =
- let res = ppobj current_module_name obj in
-  if res = "" then "" else res ^ ";;\n"
+let ppobj current_module_uri obj =
+ let res = ppobj current_module_uri obj in
+  if res = "" then "" else res ^ ";;\n\n"
 ;;