-
Notifications
You must be signed in to change notification settings - Fork 125
/
baseHashtbl.ml
100 lines (87 loc) · 2.83 KB
/
baseHashtbl.ml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
(*
Copyright © 2011 MLstate
This file is part of OPA.
OPA is free software: you can redistribute it and/or modify it under the
terms of the GNU Affero General Public License, version 3, as published by
the Free Software Foundation.
OPA is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
more details.
You should have received a copy of the GNU Affero General Public License
along with OPA. If not, see <http://www.gnu.org/licenses/>.
*)
type ('a,'b) t = ('a,'b) Hashtbl.t
let create = Hashtbl.create
let clear = Hashtbl.clear
let add = Hashtbl.add
let copy = Hashtbl.copy
let find = Hashtbl.find
let find_all = Hashtbl.find_all
let mem = Hashtbl.mem
let remove = Hashtbl.remove
let replace = Hashtbl.replace
let iter = Hashtbl.iter
let fold = Hashtbl.fold
let length = Hashtbl.length
let hash = Hashtbl.hash
external hash_param : int -> int -> 'a -> int = "caml_hash_univ_param" "noalloc"
module type HashedType = Hashtbl.HashedType
(* could be done (with magic) more efficiently
* by just assigning the two fields of the hashtbls
* But then, we become dependent on the stdlib and if it changes,
* since not everybody uses the same version of ocaml, it won't be
* possible *)
let replace_content h1 h2 =
clear h1;
iter (fun k v -> add h1 k v) h2
let find_opt h v =
try Some (find h v) with Not_found -> None
module type S =
sig
include Hashtbl.S
val replace_content :'a t -> 'a t -> unit
val find_opt : 'a t -> key -> 'a option
end
module Make(H:HashedType) : S with type key = H.t =
struct
include Hashtbl.Make(H)
let replace_content h1 h2 =
clear h1;
iter (fun k v -> add h1 k v) h2
let find_opt h v =
try Some (find h v) with Not_found -> None
end
let combine a b = a * 19 + b
(*
To extending the interface with function needing to access the implementation
of the type [t], we use dark magie.
This MUST be exactly the same type than [Hashtbl.t]
*)
type ('a, 'b) public_t = {
mutable size: int ;
mutable data: ('a, 'b) public_bucketlist array ;
}
and ('a, 'b) public_bucketlist =
| Empty
| Cons of 'a * 'b * ('a, 'b) public_bucketlist
external public_of_t : ('a, 'b) t -> ('a, 'b) public_t = "%identity"
external t_of_public : ('a, 'b) public_t -> ('a, 'b) t = "%identity"
let pick_remove t =
let t = public_of_t t in
let size = t.size in
if size <> 0
then (
let data = t.data in
let length = Array.length data in
let rec aux index =
if index = length then () else (
match Array.unsafe_get data index with
| Empty -> aux (succ index)
| Cons (_, _, bucket) ->
Array.unsafe_set data index bucket ;
t.size <- pred size ;
)
in
aux 0 ;
)