|
17 | 17 | *)
|
18 | 18 | module Int = Belt_MapInt
|
19 | 19 | (** specalized when key type is [string], more efficient
|
20 |
| - than the gerneic type *) |
| 20 | + than the gerneic type *) |
21 | 21 | module String = Belt_MapString
|
22 | 22 |
|
23 | 23 | (** seprate function from data, a more verbsoe but slightly
|
24 | 24 | more efficient
|
25 |
| -*) |
| 25 | +*) |
26 | 26 | module Dict = Belt_MapDict
|
27 | 27 |
|
28 | 28 | module N = Belt_MapDict
|
29 | 29 | module A = Belt_Array
|
30 | 30 |
|
31 | 31 | type ('key, 'id ) id = ('key, 'id) Belt_Id.comparable
|
32 | 32 | type ('key, 'id ) cmp = ('key, 'id) Belt_Id.cmp
|
33 |
| -module S = struct |
| 33 | +module S = struct |
34 | 34 | type ('k,'v,'id) t = {
|
35 | 35 | cmp: ('k,'id) cmp;
|
36 | 36 | data: ('k,'v, 'id) Dict.t
|
37 | 37 | }
|
38 | 38 | [@@bs.deriving abstract]
|
39 | 39 | end
|
40 | 40 |
|
41 |
| -type ('k, 'v, 'id ) t = ('k, 'v, 'id) S.t |
| 41 | +type ('k, 'v, 'id ) t = ('k, 'v, 'id) S.t |
42 | 42 |
|
43 | 43 | let fromArray (type k) (type idx) data ~(id : (k,idx) id) =
|
44 | 44 | let module M = (val id) in
|
45 |
| - let cmp = M.cmp in |
46 |
| - S.t ~cmp ~data:(Dict.fromArray ~cmp data) |
| 45 | + let cmp = M.cmp in |
| 46 | + S.t ~cmp ~data:(Dict.fromArray ~cmp data) |
47 | 47 |
|
48 |
| -let ofArray = fromArray |
| 48 | +let ofArray = fromArray |
49 | 49 |
|
50 |
| -let remove m x = |
51 |
| - let cmp, odata = S.cmp m, S.data m in |
| 50 | +let remove m x = |
| 51 | + let cmp, odata = m |. S.(cmp, data) in |
52 | 52 | let newData = Dict.remove odata x ~cmp in
|
53 | 53 | if newData == odata then m
|
54 |
| - else S.t ~cmp ~data:newData |
| 54 | + else S.t ~cmp ~data:newData |
55 | 55 |
|
56 |
| -let removeMany m x = |
57 |
| - let cmp, odata = S.cmp m, S.data m in |
| 56 | +let removeMany m x = |
| 57 | + let cmp, odata = m |. (S.cmp, S.data) in |
58 | 58 | let newData = Dict.removeMany odata x ~cmp in
|
59 | 59 | S.t ~cmp ~data:newData
|
60 | 60 |
|
61 |
| -let set m key d = |
62 |
| - let cmp = S.cmp m in |
63 |
| - S.t ~cmp ~data:(Dict.set ~cmp (S.data m) key d) |
| 61 | +let set m key d = |
| 62 | + let cmp = S.cmp m in |
| 63 | + S.t ~cmp ~data:(Dict.set ~cmp (S.data m) key d) |
64 | 64 |
|
65 |
| -let mergeMany m e = |
66 |
| - let cmp = S.cmp m in |
| 65 | +let mergeMany m e = |
| 66 | + let cmp = S.cmp m in |
67 | 67 | S.t ~cmp ~data:(Dict.mergeMany ~cmp (S.data m) e)
|
68 | 68 |
|
69 |
| -let updateU m key f = |
70 |
| - let cmp = S.cmp m in |
| 69 | +let updateU m key f = |
| 70 | + let cmp = S.cmp m in |
71 | 71 | S.t ~cmp ~data:(Dict.updateU ~cmp (S.data m) key f )
|
72 | 72 | let update m key f = updateU m key (fun [@bs] a -> f a )
|
73 |
| -let split m x = |
74 |
| - let cmp = S.cmp m in |
75 |
| - let (l,r),b = Dict.split ~cmp (S.data m) x in |
76 |
| - (S.t ~cmp ~data:l, S.t ~cmp ~data:r), b |
| 73 | +let split m x = |
| 74 | + let cmp = S.cmp m in |
| 75 | + let (l,r),b = Dict.split ~cmp (S.data m) x in |
| 76 | + (S.t ~cmp ~data:l, S.t ~cmp ~data:r), b |
77 | 77 |
|
78 |
| -let mergeU s1 s2 f = |
79 |
| - let cmp = S.cmp s1 in |
| 78 | +let mergeU s1 s2 f = |
| 79 | + let cmp = S.cmp s1 in |
80 | 80 | S.t ~cmp ~data:(Dict.mergeU ~cmp (S.data s1) (S.data s2) f)
|
81 | 81 |
|
82 |
| -let merge s1 s2 f = |
| 82 | +let merge s1 s2 f = |
83 | 83 | mergeU s1 s2 (fun [@bs] a b c -> f a b c)
|
84 | 84 |
|
85 | 85 | let make (type key) (type idx) ~(id: (key, idx) id) =
|
86 |
| - let module M = (val id) in |
| 86 | + let module M = (val id) in |
87 | 87 | S.t ~cmp:M.cmp ~data:Dict.empty
|
88 | 88 |
|
89 |
| -let isEmpty map = |
| 89 | +let isEmpty map = |
90 | 90 | Dict.isEmpty (S.data map)
|
91 | 91 |
|
92 | 92 |
|
93 | 93 | let forEachU m f = Dict.forEachU (S.data m) f
|
94 | 94 | let forEach m f = forEachU m (fun [@bs] a b -> f a b)
|
95 |
| -let reduceU m acc f = Dict.reduceU (S.data m) acc f |
| 95 | +let reduceU m acc f = Dict.reduceU (S.data m) acc f |
96 | 96 | let reduce m acc f = reduceU m acc (fun[@bs] a b c -> f a b c )
|
97 |
| -let everyU m f = Dict.everyU (S.data m) f |
| 97 | +let everyU m f = Dict.everyU (S.data m) f |
98 | 98 | let every m f = everyU m (fun [@bs] a b -> f a b)
|
99 | 99 | let someU m f = Dict.someU (S.data m) f
|
100 | 100 | let some m f = someU m (fun[@bs] a b -> f a b)
|
101 | 101 | let keepU m f =
|
102 | 102 | S.t ~cmp:(S.cmp m) ~data:(Dict.keepU (S.data m) f)
|
103 | 103 | let keep m f = keepU m (fun [@bs] a b -> f a b)
|
104 |
| - |
105 |
| -let partitionU m p = |
106 |
| - let cmp = S.cmp m in |
107 |
| - let l,r = Dict.partitionU (S.data m) p in |
108 |
| - S.t ~cmp ~data:l, S.t ~cmp ~data:r |
| 104 | + |
| 105 | +let partitionU m p = |
| 106 | + let cmp = S.cmp m in |
| 107 | + let l,r = m |. S.data |. Dict.partitionU p in |
| 108 | + S.t ~cmp ~data:l, S.t ~cmp ~data:r |
109 | 109 | let partition m p = partitionU m (fun [@bs] a b -> p a b)
|
110 | 110 |
|
111 |
| -let mapU m f = |
| 111 | +let mapU m f = |
112 | 112 | S.t ~cmp:(S.cmp m) ~data:(Dict.mapU (S.data m) f)
|
113 |
| -let map m f = mapU m (fun [@bs] a -> f a) |
114 |
| -let mapWithKeyU m f = |
| 113 | +let map m f = mapU m (fun [@bs] a -> f a) |
| 114 | +let mapWithKeyU m f = |
115 | 115 | S.t ~cmp:(S.cmp m) ~data:(Dict.mapWithKeyU (S.data m) f)
|
116 | 116 | let mapWithKey m f = mapWithKeyU m (fun [@bs] a b -> f a b)
|
117 |
| -let size map = Dict.size (S.data map) |
118 |
| -let toList map = Dict.toList (S.data map) |
| 117 | +let size map = Dict.size (S.data map) |
| 118 | +let toList map = Dict.toList (S.data map) |
119 | 119 | let toArray m = Dict.toArray (S.data m)
|
120 | 120 | let keysToArray m = Dict.keysToArray (S.data m)
|
121 | 121 | let valuesToArray m = Dict.valuesToArray (S.data m)
|
122 | 122 | let minKey m = Dict.minKey (S.data m)
|
123 | 123 | let minKeyUndefined m = Dict.minKeyUndefined (S.data m)
|
124 | 124 | let maxKey m = Dict.maxKey (S.data m)
|
125 |
| -let maxKeyUndefined m = Dict.maxKeyUndefined (S.data m) |
| 125 | +let maxKeyUndefined m = Dict.maxKeyUndefined (S.data m) |
126 | 126 | let minimum m = Dict.minimum (S.data m)
|
127 |
| -let minUndefined m = Dict.minUndefined (S.data m) |
| 127 | +let minUndefined m = Dict.minUndefined (S.data m) |
128 | 128 | let maximum m = Dict.maximum (S.data m)
|
129 | 129 | let maxUndefined m = Dict.maxUndefined (S.data m)
|
130 | 130 |
|
131 | 131 | let get map x =
|
132 |
| - Dict.get ~cmp:(S.cmp map) (S.data map) x |
| 132 | + Dict.get ~cmp:(S.cmp map) (S.data map) x |
133 | 133 |
|
134 |
| -let getUndefined map x = |
| 134 | +let getUndefined map x = |
135 | 135 | Dict.getUndefined ~cmp:(S.cmp map) (S.data map) x
|
136 | 136 |
|
137 |
| -let getWithDefault map x def = |
| 137 | +let getWithDefault map x def = |
138 | 138 | Dict.getWithDefault ~cmp:(S.cmp map) (S.data map) x def
|
139 | 139 |
|
140 |
| -let getExn map x = |
141 |
| - Dict.getExn ~cmp:(S.cmp map) (S.data map) x |
| 140 | +let getExn map x = |
| 141 | + Dict.getExn ~cmp:(S.cmp map) (S.data map) x |
142 | 142 |
|
143 |
| -let has map x = |
| 143 | +let has map x = |
144 | 144 | Dict.has ~cmp:(S.cmp map) (S.data map) x
|
145 | 145 |
|
146 | 146 | let checkInvariantInternal m =
|
147 | 147 | Dict.checkInvariantInternal (S.data m)
|
148 | 148 |
|
149 |
| -let eqU m1 m2 veq = |
| 149 | +let eqU m1 m2 veq = |
150 | 150 | Dict.eqU ~kcmp:(S.cmp m1) ~veq (S.data m1) (S.data m2)
|
151 | 151 | let eq m1 m2 veq = eqU m1 m2 (fun[@bs] a b -> veq a b)
|
152 | 152 |
|
153 | 153 | let cmpU m1 m2 vcmp =
|
154 | 154 | Dict.cmpU ~kcmp:(S.cmp m1) ~vcmp (S.data m1) (S.data m2)
|
155 | 155 | let cmp m1 m2 vcmp = cmpU m1 m2 (fun [@bs] a b -> vcmp a b)
|
156 |
| - |
| 156 | + |
157 | 157 | let getData = S.data
|
158 |
| - |
| 158 | + |
159 | 159 | let getId (type key) (type identity) (m : (key,_,identity) t) : (key, identity) id =
|
160 | 160 | let module T = struct
|
161 | 161 | type nonrec identity = identity
|
162 | 162 | type nonrec t = key
|
163 | 163 | let cmp = S.cmp m
|
164 | 164 | end in
|
165 | 165 | (module T )
|
166 |
| - |
| 166 | + |
167 | 167 | let packIdData (type key) (type idx) ~(id : (key, idx) id) ~data =
|
168 |
| - let module M = (val id) in |
| 168 | + let module M = (val id) in |
169 | 169 | S.t ~cmp:M.cmp ~data
|
170 | 170 |
|
0 commit comments