Metasenv added as parameter to eta_fixing.

[helm.git] / helm / ocaml / cic_omdoc / eta_fixing.ml

diff --git a/helm/ocaml/cic_omdoc/eta_fixing.ml b/helm/ocaml/cic_omdoc/eta_fixing.ml
index 9b20729103b37a0061d4f54a5b72cf4c9456b85a..6d5ce9833e86a6b592b91b531c2fd1c9f2da3bef 100644 (file)
--- a/helm/ocaml/cic_omdoc/eta_fixing.ml
+++ b/helm/ocaml/cic_omdoc/eta_fixing.ml
@@ -23,9 +23,7 @@
  * http://cs.unibo.it/helm/.
  *)
 
-exception ReferenceToVariable;;
-exception RferenceToCurrentProof;;
-exception ReferenceToInductiveDefinition;;
+exception ReferenceToNonVariable;;
 
 let prerr_endline _ = ();;
 
@@ -132,7 +130,11 @@ let fix_according_to_type ty hd tl =
   let rec aux n ty tl res =
     if n = 0 then
       (match tl with 
-         [] -> C.Appl res
+         [] ->
+          (match res with
+              [] -> assert false
+            | [res] -> res
+            | _ -> C.Appl res)
        | _ -> 
           match res with
             [] -> assert false
@@ -163,7 +165,7 @@ let fix_according_to_type ty hd tl =
   aux expected_arity ty tl [hd]
 ;;
 
-let eta_fix metasenv t =
+let eta_fix metasenv context t =
  let rec eta_fix' context t = 
   (* prerr_endline ("entering aux with: term=" ^ CicPp.ppterm t); 
   flush stderr ; *)
@@ -172,11 +174,8 @@ let eta_fix metasenv t =
   match t with
      C.Rel n -> C.Rel n 
    | C.Var (uri,exp_named_subst) ->
-      let exp_named_subst' =
-       List.map
-        (function i,t -> i, (eta_fix' context t)) exp_named_subst
-      in
-      C.Var (uri,exp_named_subst')
+      let exp_named_subst' = fix_exp_named_subst context exp_named_subst in
+       C.Var (uri,exp_named_subst')
    | C.Meta (n,l) ->
       let (_,canonical_context,_) =
        List.find (function (m,_,_) -> n = m) metasenv
@@ -224,22 +223,13 @@ let eta_fix metasenv t =
              constant_type (C.Const(uri,exp_named_subst)) l''
         | _ -> C.Appl l' *))
    | C.Const (uri,exp_named_subst) ->
-       let exp_named_subst' =
-       List.map
-        (function i,t -> i, (eta_fix' context t)) exp_named_subst
-       in
-       C.Const (uri,exp_named_subst')
+       let exp_named_subst' = fix_exp_named_subst context exp_named_subst in
+        C.Const (uri,exp_named_subst')
    | C.MutInd (uri,tyno,exp_named_subst) ->
-       let exp_named_subst' =
-       List.map
-        (function i,t -> i, (eta_fix' context t)) exp_named_subst
-       in
+       let exp_named_subst' = fix_exp_named_subst context exp_named_subst in
         C.MutInd (uri, tyno, exp_named_subst')
    | C.MutConstruct (uri,tyno,consno,exp_named_subst) ->
-       let exp_named_subst' =
-       List.map
-        (function i,t -> i, (eta_fix' context t)) exp_named_subst
-       in
+       let exp_named_subst' = fix_exp_named_subst context exp_named_subst in
         C.MutConstruct (uri, tyno, consno, exp_named_subst')
    | C.MutCase (uri, tyno, outty, term, patterns) as prima ->
        let outty' =  eta_fix' context outty in
@@ -298,17 +288,18 @@ let eta_fix metasenv t =
         List.map
          (fun (name, ty, bo) ->
            (name, eta_fix' context ty, eta_fix' (fun_types@context) bo)) funs)
-   in
-   eta_fix' [] t
+  and fix_exp_named_subst context exp_named_subst =
+   List.rev
+    (List.fold_left
+      (fun newsubst (uri,t) ->
+        let t' = eta_fix' context t in
+        let ty =
+         match CicEnvironment.get_obj uri with
+            Cic.Variable (_,_,ty,_) -> CicSubstitution.subst_vars newsubst ty
+          | _ ->  raise ReferenceToNonVariable in
+        let t'' = fix_according_to_type ty t' [] in
+         (uri,t'')::newsubst
+      ) [] exp_named_subst)
+  in
+   eta_fix' context t
 ;;
-
-
-
-
-
-
-
-
-
-
-