Clarifying the implementation of universe hashconsing.

1 files changed, 16 insertions, 17 deletions

diff --git a/checker/univ.ml b/checker/univ.ml
index 5fed6dcd5..8d0ae8f1e 100644
--- a/checker/univ.ml
+++ b/checker/univ.ml
@@ -211,27 +211,26 @@ module Level = struct
 
   let hash x = x.hash
 
-  let hcons x = 
-    let data' = RawLevel.hcons x.data in
-      if data' == x.data then x
-      else { x with data = data' }
-
   let data x = x.data
 
   (** Hashcons on levels + their hash *)
 
-  let make =
-    let module Self = struct
-      type _t = t
-      type t = _t
-      let equal = equal
-      let hash = hash
-    end in
-    let module WH = Weak.Make(Self) in
-    let pool = WH.create 4910 in fun x ->
-    let x = { hash = RawLevel.hash x; data = x } in
-    try WH.find pool x
-    with Not_found -> WH.add pool x; x
+  module Self = struct
+    type _t = t
+    type t = _t
+    type u = raw_level -> raw_level
+    let equal x y = x.hash == y.hash && x.data == y.data
+    let hash x = x.hash
+    let hashcons hraw x =
+      let data' = hraw x.data in
+      if x.data == data' then x else { x with data = data' }
+  end
+
+  let hcons =
+    let module H = Hashcons.Make(Self) in
+    Hashcons.simple_hcons H.generate H.hcons RawLevel.hcons
+
+  let make l = hcons { hash = RawLevel.hash l; data = l }
 
   let set = make Set
   let prop = make Prop