added add_obj to store objects in the environment, implementation still

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

diff --git a/helm/software/components/ng_kernel/nCicEnvironment.ml b/helm/software/components/ng_kernel/nCicEnvironment.ml
index 9f75f147c3e50b6f359eb419fec00a82d41a027d..ff54e0232fb48a89885e19604f9f905faa466cb1 100644 (file)
--- a/helm/software/components/ng_kernel/nCicEnvironment.ml
+++ b/helm/software/components/ng_kernel/nCicEnvironment.ml
@@ -1,8 +1,71 @@
+let cache = NUri.UriHash.create 313;;
 
-let get_checked_obj nuri =
-  let ouri = Obj.magic nuri in
-  let o,_ = 
-    CicEnvironment.get_cooked_obj ~trust:true CicUniv.oblivion_ugraph 
-      ouri 
-  in
-    OCic2NCic.convert_obj o
+let get_checked_obj u =
+  try NUri.UriHash.find cache u
+  with Not_found ->
+    let ouri = NUri.ouri_of_nuri u in
+    let o,_ = 
+      CicEnvironment.get_cooked_obj ~trust:false CicUniv.oblivion_ugraph ouri 
+    in
+    (* FIX: add all objects to the environment and give back the last one *)
+    let no = HExtlib.list_last (OCic2NCic.convert_obj ouri o) in
+    NUri.UriHash.add cache u no;
+    no
+;;
+
+let get_obj u =
+  try true, NUri.UriHash.find cache u
+  with Not_found ->
+    (* in th final implementation should get it from disk *)
+    let ouri = NUri.ouri_of_nuri u in
+    let o,_ = 
+      CicEnvironment.get_cooked_obj ~trust:false CicUniv.oblivion_ugraph ouri 
+    in
+    (* here we should freeze it *)
+    false, HExtlib.list_last (OCic2NCic.convert_obj ouri o)
+;;
+
+let add_obj (u,_,_,_,_ as o) =
+  (* we should unfreeze it *)
+  NUri.UriHash.add cache u o
+;;
+
+let get_checked_def = function
+  | NReference.Ref (_, uri, NReference.Def) ->
+      (match get_checked_obj uri with
+      | _,height,_,_, NCic.Constant (rlv,name,Some bo,ty,att) ->
+          rlv,name,bo,ty,att,height
+      | _,_,_,_, NCic.Constant (_,_,None,_,_) ->
+          prerr_endline "get_checked_def on an axiom"; assert false
+      | _ -> prerr_endline "get_checked_def on a non def 2"; assert false)
+  | _ -> prerr_endline "get_checked_def on a non def"; assert false
+;;
+
+let get_checked_indtys = function
+  | NReference.Ref (_, uri, NReference.Ind n) ->
+      (match get_checked_obj uri with
+      | _,_,_,_, NCic.Inductive (inductive,leftno,tys,att) ->
+        inductive,leftno,tys,att,n
+      | _ -> prerr_endline "get_checked_indtys on a non ind 2"; assert false)
+  | _ -> prerr_endline "get_checked_indtys on a non ind"; assert false
+;;
+
+let get_checked_fix_or_cofix b = function
+  | NReference.Ref (_, uri, NReference.Fix (fixno,_)) ->
+      (match get_checked_obj uri with
+      | _,height,_,_, NCic.Fixpoint (is_fix,funcs,att) when is_fix = b ->
+         let rlv, name, _, ty, bo = List.nth funcs fixno in
+         rlv, name, bo, ty, att, height
+      | _ ->prerr_endline "get_checked_(co)fix on a non (co)fix 2";assert false)
+  | _ -> prerr_endline "get_checked_(co)fix on a non (co)fix"; assert false
+;;
+let get_checked_fix r = get_checked_fix_or_cofix true r;;
+let get_checked_cofix r = get_checked_fix_or_cofix false r;;
+
+let get_indty_leftno = function 
+  | NReference.Ref (_, uri, NReference.Ind _) 
+  | NReference.Ref (_, uri, NReference.Con _) ->
+      (match get_checked_obj uri with
+      | _,_,_,_, NCic.Inductive (_,left,_,_) -> left
+      | _ ->prerr_endline "get_indty_leftno called on a non ind 2";assert false)
+  | _ -> prerr_endline "get_indty_leftno called on a non indty";assert false