line |
stmt |
bran |
cond |
sub |
pod |
time |
code |
1
|
|
|
|
|
|
|
/* The MIT License |
2
|
|
|
|
|
|
|
|
3
|
|
|
|
|
|
|
Copyright (c) 2008, 2009, 2011 by Attractive Chaos |
4
|
|
|
|
|
|
|
|
5
|
|
|
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining |
6
|
|
|
|
|
|
|
a copy of this software and associated documentation files (the |
7
|
|
|
|
|
|
|
"Software"), to deal in the Software without restriction, including |
8
|
|
|
|
|
|
|
without limitation the rights to use, copy, modify, merge, publish, |
9
|
|
|
|
|
|
|
distribute, sublicense, and/or sell copies of the Software, and to |
10
|
|
|
|
|
|
|
permit persons to whom the Software is furnished to do so, subject to |
11
|
|
|
|
|
|
|
the following conditions: |
12
|
|
|
|
|
|
|
|
13
|
|
|
|
|
|
|
The above copyright notice and this permission notice shall be |
14
|
|
|
|
|
|
|
included in all copies or substantial portions of the Software. |
15
|
|
|
|
|
|
|
|
16
|
|
|
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
17
|
|
|
|
|
|
|
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
18
|
|
|
|
|
|
|
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
19
|
|
|
|
|
|
|
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS |
20
|
|
|
|
|
|
|
BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN |
21
|
|
|
|
|
|
|
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN |
22
|
|
|
|
|
|
|
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
23
|
|
|
|
|
|
|
SOFTWARE. |
24
|
|
|
|
|
|
|
*/ |
25
|
|
|
|
|
|
|
|
26
|
|
|
|
|
|
|
/* |
27
|
|
|
|
|
|
|
An example: |
28
|
|
|
|
|
|
|
|
29
|
|
|
|
|
|
|
#include "khash.h" |
30
|
|
|
|
|
|
|
KHASH_MAP_INIT_INT(32, char) |
31
|
|
|
|
|
|
|
int main() { |
32
|
|
|
|
|
|
|
int ret, is_missing; |
33
|
|
|
|
|
|
|
khiter_t k; |
34
|
|
|
|
|
|
|
khash_t(32) *h = kh_init(32); |
35
|
|
|
|
|
|
|
k = kh_put(32, h, 5, &ret); |
36
|
|
|
|
|
|
|
kh_value(h, k) = 10; |
37
|
|
|
|
|
|
|
k = kh_get(32, h, 10); |
38
|
|
|
|
|
|
|
is_missing = (k == kh_end(h)); |
39
|
|
|
|
|
|
|
k = kh_get(32, h, 5); |
40
|
|
|
|
|
|
|
kh_del(32, h, k); |
41
|
|
|
|
|
|
|
for (k = kh_begin(h); k != kh_end(h); ++k) |
42
|
|
|
|
|
|
|
if (kh_exist(h, k)) kh_value(h, k) = 1; |
43
|
|
|
|
|
|
|
kh_destroy(32, h); |
44
|
|
|
|
|
|
|
return 0; |
45
|
|
|
|
|
|
|
} |
46
|
|
|
|
|
|
|
*/ |
47
|
|
|
|
|
|
|
|
48
|
|
|
|
|
|
|
/* |
49
|
|
|
|
|
|
|
2013-05-02 (0.2.8): |
50
|
|
|
|
|
|
|
|
51
|
|
|
|
|
|
|
* Use quadratic probing. When the capacity is power of 2, stepping function |
52
|
|
|
|
|
|
|
i*(i+1)/2 guarantees to traverse each bucket. It is better than double |
53
|
|
|
|
|
|
|
hashing on cache performance and is more robust than linear probing. |
54
|
|
|
|
|
|
|
|
55
|
|
|
|
|
|
|
In theory, double hashing should be more robust than quadratic probing. |
56
|
|
|
|
|
|
|
However, my implementation is probably not for large hash tables, because |
57
|
|
|
|
|
|
|
the second hash function is closely tied to the first hash function, |
58
|
|
|
|
|
|
|
which reduce the effectiveness of double hashing. |
59
|
|
|
|
|
|
|
|
60
|
|
|
|
|
|
|
Reference: http://research.cs.vt.edu/AVresearch/hashing/quadratic.php |
61
|
|
|
|
|
|
|
|
62
|
|
|
|
|
|
|
2011-12-29 (0.2.7): |
63
|
|
|
|
|
|
|
|
64
|
|
|
|
|
|
|
* Minor code clean up; no actual effect. |
65
|
|
|
|
|
|
|
|
66
|
|
|
|
|
|
|
2011-09-16 (0.2.6): |
67
|
|
|
|
|
|
|
|
68
|
|
|
|
|
|
|
* The capacity is a power of 2. This seems to dramatically improve the |
69
|
|
|
|
|
|
|
speed for simple keys. Thank Zilong Tan for the suggestion. Reference: |
70
|
|
|
|
|
|
|
|
71
|
|
|
|
|
|
|
- http://code.google.com/p/ulib/ |
72
|
|
|
|
|
|
|
- http://nothings.org/computer/judy/ |
73
|
|
|
|
|
|
|
|
74
|
|
|
|
|
|
|
* Allow to optionally use linear probing which usually has better |
75
|
|
|
|
|
|
|
performance for random input. Double hashing is still the default as it |
76
|
|
|
|
|
|
|
is more robust to certain non-random input. |
77
|
|
|
|
|
|
|
|
78
|
|
|
|
|
|
|
* Added Wang's integer hash function (not used by default). This hash |
79
|
|
|
|
|
|
|
function is more robust to certain non-random input. |
80
|
|
|
|
|
|
|
|
81
|
|
|
|
|
|
|
2011-02-14 (0.2.5): |
82
|
|
|
|
|
|
|
|
83
|
|
|
|
|
|
|
* Allow to declare global functions. |
84
|
|
|
|
|
|
|
|
85
|
|
|
|
|
|
|
2009-09-26 (0.2.4): |
86
|
|
|
|
|
|
|
|
87
|
|
|
|
|
|
|
* Improve portability |
88
|
|
|
|
|
|
|
|
89
|
|
|
|
|
|
|
2008-09-19 (0.2.3): |
90
|
|
|
|
|
|
|
|
91
|
|
|
|
|
|
|
* Corrected the example |
92
|
|
|
|
|
|
|
* Improved interfaces |
93
|
|
|
|
|
|
|
|
94
|
|
|
|
|
|
|
2008-09-11 (0.2.2): |
95
|
|
|
|
|
|
|
|
96
|
|
|
|
|
|
|
* Improved speed a little in kh_put() |
97
|
|
|
|
|
|
|
|
98
|
|
|
|
|
|
|
2008-09-10 (0.2.1): |
99
|
|
|
|
|
|
|
|
100
|
|
|
|
|
|
|
* Added kh_clear() |
101
|
|
|
|
|
|
|
* Fixed a compiling error |
102
|
|
|
|
|
|
|
|
103
|
|
|
|
|
|
|
2008-09-02 (0.2.0): |
104
|
|
|
|
|
|
|
|
105
|
|
|
|
|
|
|
* Changed to token concatenation which increases flexibility. |
106
|
|
|
|
|
|
|
|
107
|
|
|
|
|
|
|
2008-08-31 (0.1.2): |
108
|
|
|
|
|
|
|
|
109
|
|
|
|
|
|
|
* Fixed a bug in kh_get(), which has not been tested previously. |
110
|
|
|
|
|
|
|
|
111
|
|
|
|
|
|
|
2008-08-31 (0.1.1): |
112
|
|
|
|
|
|
|
|
113
|
|
|
|
|
|
|
* Added destructor |
114
|
|
|
|
|
|
|
*/ |
115
|
|
|
|
|
|
|
|
116
|
|
|
|
|
|
|
|
117
|
|
|
|
|
|
|
#ifndef __AC_KHASH_H |
118
|
|
|
|
|
|
|
#define __AC_KHASH_H |
119
|
|
|
|
|
|
|
|
120
|
|
|
|
|
|
|
/*! |
121
|
|
|
|
|
|
|
@header |
122
|
|
|
|
|
|
|
|
123
|
|
|
|
|
|
|
Generic hash table library. |
124
|
|
|
|
|
|
|
*/ |
125
|
|
|
|
|
|
|
|
126
|
|
|
|
|
|
|
#define AC_VERSION_KHASH_H "0.2.8" |
127
|
|
|
|
|
|
|
|
128
|
|
|
|
|
|
|
#include |
129
|
|
|
|
|
|
|
#include |
130
|
|
|
|
|
|
|
#include |
131
|
|
|
|
|
|
|
|
132
|
|
|
|
|
|
|
/* compiler specific configuration */ |
133
|
|
|
|
|
|
|
|
134
|
|
|
|
|
|
|
#if UINT_MAX == 0xffffffffu |
135
|
|
|
|
|
|
|
typedef unsigned int khint32_t; |
136
|
|
|
|
|
|
|
#elif ULONG_MAX == 0xffffffffu |
137
|
|
|
|
|
|
|
typedef unsigned long khint32_t; |
138
|
|
|
|
|
|
|
#endif |
139
|
|
|
|
|
|
|
|
140
|
|
|
|
|
|
|
#if ULONG_MAX == ULLONG_MAX |
141
|
|
|
|
|
|
|
typedef unsigned long khint64_t; |
142
|
|
|
|
|
|
|
#else |
143
|
|
|
|
|
|
|
typedef unsigned long long khint64_t; |
144
|
|
|
|
|
|
|
#endif |
145
|
|
|
|
|
|
|
|
146
|
|
|
|
|
|
|
#ifndef kh_inline |
147
|
|
|
|
|
|
|
#ifdef _MSC_VER |
148
|
|
|
|
|
|
|
#define kh_inline __inline |
149
|
|
|
|
|
|
|
#elif defined(__GNUC__) |
150
|
|
|
|
|
|
|
#define kh_inline __inline__ |
151
|
|
|
|
|
|
|
#else |
152
|
|
|
|
|
|
|
#define kh_inline |
153
|
|
|
|
|
|
|
#endif |
154
|
|
|
|
|
|
|
#endif /* kh_inline */ |
155
|
|
|
|
|
|
|
|
156
|
|
|
|
|
|
|
typedef khint32_t khint_t; |
157
|
|
|
|
|
|
|
typedef khint_t khiter_t; |
158
|
|
|
|
|
|
|
|
159
|
|
|
|
|
|
|
#define __ac_isempty(flag, i) ((flag[i>>4]>>((i&0xfU)<<1))&2) |
160
|
|
|
|
|
|
|
#define __ac_isdel(flag, i) ((flag[i>>4]>>((i&0xfU)<<1))&1) |
161
|
|
|
|
|
|
|
#define __ac_iseither(flag, i) ((flag[i>>4]>>((i&0xfU)<<1))&3) |
162
|
|
|
|
|
|
|
#define __ac_set_isdel_false(flag, i) (flag[i>>4]&=~(1ul<<((i&0xfU)<<1))) |
163
|
|
|
|
|
|
|
#define __ac_set_isempty_false(flag, i) (flag[i>>4]&=~(2ul<<((i&0xfU)<<1))) |
164
|
|
|
|
|
|
|
#define __ac_set_isboth_false(flag, i) (flag[i>>4]&=~(3ul<<((i&0xfU)<<1))) |
165
|
|
|
|
|
|
|
#define __ac_set_isdel_true(flag, i) (flag[i>>4]|=1ul<<((i&0xfU)<<1)) |
166
|
|
|
|
|
|
|
|
167
|
|
|
|
|
|
|
#define __ac_fsize(m) ((m) < 16? 1 : (m)>>4) |
168
|
|
|
|
|
|
|
|
169
|
|
|
|
|
|
|
#ifndef kroundup32 |
170
|
|
|
|
|
|
|
#define kroundup32(x) (--(x), (x)|=(x)>>1, (x)|=(x)>>2, (x)|=(x)>>4, (x)|=(x)>>8, (x)|=(x)>>16, ++(x)) |
171
|
|
|
|
|
|
|
#endif |
172
|
|
|
|
|
|
|
|
173
|
|
|
|
|
|
|
#ifndef kcalloc |
174
|
|
|
|
|
|
|
#define kcalloc(N,Z) calloc(N,Z) |
175
|
|
|
|
|
|
|
#endif |
176
|
|
|
|
|
|
|
#ifndef kmalloc |
177
|
|
|
|
|
|
|
#define kmalloc(Z) malloc(Z) |
178
|
|
|
|
|
|
|
#endif |
179
|
|
|
|
|
|
|
#ifndef krealloc |
180
|
|
|
|
|
|
|
#define krealloc(P,Z) realloc(P,Z) |
181
|
|
|
|
|
|
|
#endif |
182
|
|
|
|
|
|
|
#ifndef kreallocarray |
183
|
|
|
|
|
|
|
#define kreallocarray(P,N,Z) ((SIZE_MAX - N < Z) ? NULL : krealloc(P, (N*Z))) |
184
|
|
|
|
|
|
|
#endif |
185
|
|
|
|
|
|
|
#ifndef kfree |
186
|
|
|
|
|
|
|
#define kfree(P) free(P) |
187
|
|
|
|
|
|
|
#endif |
188
|
|
|
|
|
|
|
|
189
|
|
|
|
|
|
|
static const double __ac_HASH_UPPER = 0.77; |
190
|
|
|
|
|
|
|
|
191
|
|
|
|
|
|
|
#define __KHASH_TYPE(name, khkey_t, khval_t) \ |
192
|
|
|
|
|
|
|
typedef struct kh_##name##_s { \ |
193
|
|
|
|
|
|
|
khint_t n_buckets, size, n_occupied, upper_bound; \ |
194
|
|
|
|
|
|
|
khint32_t *flags; \ |
195
|
|
|
|
|
|
|
khkey_t *keys; \ |
196
|
|
|
|
|
|
|
khval_t *vals; \ |
197
|
|
|
|
|
|
|
} kh_##name##_t; |
198
|
|
|
|
|
|
|
|
199
|
|
|
|
|
|
|
#define __KHASH_PROTOTYPES(name, khkey_t, khval_t) \ |
200
|
|
|
|
|
|
|
extern kh_##name##_t *kh_init_##name(void); \ |
201
|
|
|
|
|
|
|
extern void kh_destroy_##name(kh_##name##_t *h); \ |
202
|
|
|
|
|
|
|
extern void kh_clear_##name(kh_##name##_t *h); \ |
203
|
|
|
|
|
|
|
extern khint_t kh_get_##name(const kh_##name##_t *h, khkey_t key); \ |
204
|
|
|
|
|
|
|
extern int kh_resize_##name(kh_##name##_t *h, khint_t new_n_buckets); \ |
205
|
|
|
|
|
|
|
extern khint_t kh_put_##name(kh_##name##_t *h, khkey_t key, int *ret); \ |
206
|
|
|
|
|
|
|
extern void kh_del_##name(kh_##name##_t *h, khint_t x); |
207
|
|
|
|
|
|
|
|
208
|
|
|
|
|
|
|
#define __KHASH_IMPL(name, SCOPE, khkey_t, khval_t, kh_is_map, __hash_func, __hash_equal) \ |
209
|
|
|
|
|
|
|
SCOPE kh_##name##_t *kh_init_##name(void) { \ |
210
|
|
|
|
|
|
|
return (kh_##name##_t*)kcalloc(1, sizeof(kh_##name##_t)); \ |
211
|
|
|
|
|
|
|
} \ |
212
|
|
|
|
|
|
|
SCOPE void kh_destroy_##name(kh_##name##_t *h) \ |
213
|
|
|
|
|
|
|
{ \ |
214
|
|
|
|
|
|
|
if (h) { \ |
215
|
|
|
|
|
|
|
kfree((void *)h->keys); kfree(h->flags); \ |
216
|
|
|
|
|
|
|
kfree((void *)h->vals); \ |
217
|
|
|
|
|
|
|
kfree(h); \ |
218
|
|
|
|
|
|
|
} \ |
219
|
|
|
|
|
|
|
} \ |
220
|
|
|
|
|
|
|
SCOPE void kh_clear_##name(kh_##name##_t *h) \ |
221
|
|
|
|
|
|
|
{ \ |
222
|
|
|
|
|
|
|
if (h && h->flags) { \ |
223
|
|
|
|
|
|
|
memset(h->flags, 0xaa, __ac_fsize(h->n_buckets) * sizeof(khint32_t)); \ |
224
|
|
|
|
|
|
|
h->size = h->n_occupied = 0; \ |
225
|
|
|
|
|
|
|
} \ |
226
|
|
|
|
|
|
|
} \ |
227
|
|
|
|
|
|
|
SCOPE khint_t kh_get_##name(const kh_##name##_t *h, khkey_t key) \ |
228
|
|
|
|
|
|
|
{ \ |
229
|
|
|
|
|
|
|
if (h->n_buckets) { \ |
230
|
|
|
|
|
|
|
khint_t k, i, last, mask, step = 0; \ |
231
|
|
|
|
|
|
|
mask = h->n_buckets - 1; \ |
232
|
|
|
|
|
|
|
k = __hash_func(key); i = k & mask; \ |
233
|
|
|
|
|
|
|
last = i; \ |
234
|
|
|
|
|
|
|
while (!__ac_isempty(h->flags, i) && (__ac_isdel(h->flags, i) || !__hash_equal(h->keys[i], key))) { \ |
235
|
|
|
|
|
|
|
i = (i + (++step)) & mask; \ |
236
|
|
|
|
|
|
|
if (i == last) return h->n_buckets; \ |
237
|
|
|
|
|
|
|
} \ |
238
|
|
|
|
|
|
|
return __ac_iseither(h->flags, i)? h->n_buckets : i; \ |
239
|
|
|
|
|
|
|
} else return 0; \ |
240
|
|
|
|
|
|
|
} \ |
241
|
|
|
|
|
|
|
SCOPE int kh_resize_##name(kh_##name##_t *h, khint_t new_n_buckets) \ |
242
|
|
|
|
|
|
|
{ /* This function uses 0.25*n_buckets bytes of working space instead of [sizeof(key_t+val_t)+.25]*n_buckets. */ \ |
243
|
|
|
|
|
|
|
khint32_t *new_flags = 0; \ |
244
|
|
|
|
|
|
|
khint_t j = 1; \ |
245
|
|
|
|
|
|
|
{ \ |
246
|
|
|
|
|
|
|
kroundup32(new_n_buckets); \ |
247
|
|
|
|
|
|
|
if (new_n_buckets < 4) new_n_buckets = 4; \ |
248
|
|
|
|
|
|
|
if (h->size >= (khint_t)(new_n_buckets * __ac_HASH_UPPER + 0.5)) j = 0; /* requested size is too small */ \ |
249
|
|
|
|
|
|
|
else { /* hash table size to be changed (shrink or expand); rehash */ \ |
250
|
|
|
|
|
|
|
new_flags = (khint32_t*)kreallocarray(NULL, __ac_fsize(new_n_buckets), sizeof(khint32_t)); \ |
251
|
|
|
|
|
|
|
if (!new_flags) return -1; \ |
252
|
|
|
|
|
|
|
memset(new_flags, 0xaa, __ac_fsize(new_n_buckets) * sizeof(khint32_t)); \ |
253
|
|
|
|
|
|
|
if (h->n_buckets < new_n_buckets) { /* expand */ \ |
254
|
|
|
|
|
|
|
khkey_t *new_keys = (khkey_t*)kreallocarray((void *)h->keys, new_n_buckets, sizeof(khkey_t)); \ |
255
|
|
|
|
|
|
|
if (!new_keys) { kfree(new_flags); return -1; } \ |
256
|
|
|
|
|
|
|
h->keys = new_keys; \ |
257
|
|
|
|
|
|
|
if (kh_is_map) { \ |
258
|
|
|
|
|
|
|
khval_t *new_vals = (khval_t*)kreallocarray((void *)h->vals, new_n_buckets, sizeof(khval_t)); \ |
259
|
|
|
|
|
|
|
if (!new_vals) { kfree(new_flags); return -1; } \ |
260
|
|
|
|
|
|
|
h->vals = new_vals; \ |
261
|
|
|
|
|
|
|
} \ |
262
|
|
|
|
|
|
|
} /* otherwise shrink */ \ |
263
|
|
|
|
|
|
|
} \ |
264
|
|
|
|
|
|
|
} \ |
265
|
|
|
|
|
|
|
if (j) { /* rehashing is needed */ \ |
266
|
|
|
|
|
|
|
for (j = 0; j != h->n_buckets; ++j) { \ |
267
|
|
|
|
|
|
|
if (__ac_iseither(h->flags, j) == 0) { \ |
268
|
|
|
|
|
|
|
khkey_t key = h->keys[j]; \ |
269
|
|
|
|
|
|
|
khval_t val; \ |
270
|
|
|
|
|
|
|
khint_t new_mask; \ |
271
|
|
|
|
|
|
|
new_mask = new_n_buckets - 1; \ |
272
|
|
|
|
|
|
|
if (kh_is_map) val = h->vals[j]; \ |
273
|
|
|
|
|
|
|
__ac_set_isdel_true(h->flags, j); \ |
274
|
|
|
|
|
|
|
while (1) { /* kick-out process; sort of like in Cuckoo hashing */ \ |
275
|
|
|
|
|
|
|
khint_t k, i, step = 0; \ |
276
|
|
|
|
|
|
|
k = __hash_func(key); \ |
277
|
|
|
|
|
|
|
i = k & new_mask; \ |
278
|
|
|
|
|
|
|
while (!__ac_isempty(new_flags, i)) i = (i + (++step)) & new_mask; \ |
279
|
|
|
|
|
|
|
__ac_set_isempty_false(new_flags, i); \ |
280
|
|
|
|
|
|
|
if (i < h->n_buckets && __ac_iseither(h->flags, i) == 0) { /* kick out the existing element */ \ |
281
|
|
|
|
|
|
|
{ khkey_t tmp = h->keys[i]; h->keys[i] = key; key = tmp; } \ |
282
|
|
|
|
|
|
|
if (kh_is_map) { khval_t tmp = h->vals[i]; h->vals[i] = val; val = tmp; } \ |
283
|
|
|
|
|
|
|
__ac_set_isdel_true(h->flags, i); /* mark it as deleted in the old hash table */ \ |
284
|
|
|
|
|
|
|
} else { /* write the element and jump out of the loop */ \ |
285
|
|
|
|
|
|
|
h->keys[i] = key; \ |
286
|
|
|
|
|
|
|
if (kh_is_map) h->vals[i] = val; \ |
287
|
|
|
|
|
|
|
break; \ |
288
|
|
|
|
|
|
|
} \ |
289
|
|
|
|
|
|
|
} \ |
290
|
|
|
|
|
|
|
} \ |
291
|
|
|
|
|
|
|
} \ |
292
|
|
|
|
|
|
|
if (h->n_buckets > new_n_buckets) { /* shrink the hash table */ \ |
293
|
|
|
|
|
|
|
h->keys = (khkey_t*)kreallocarray((void *)h->keys, new_n_buckets, sizeof(khkey_t)); \ |
294
|
|
|
|
|
|
|
if (kh_is_map) h->vals = (khval_t*)kreallocarray((void *)h->vals, new_n_buckets, sizeof(khval_t)); \ |
295
|
|
|
|
|
|
|
} \ |
296
|
|
|
|
|
|
|
kfree(h->flags); /* free the working space */ \ |
297
|
|
|
|
|
|
|
h->flags = new_flags; \ |
298
|
|
|
|
|
|
|
h->n_buckets = new_n_buckets; \ |
299
|
|
|
|
|
|
|
h->n_occupied = h->size; \ |
300
|
|
|
|
|
|
|
h->upper_bound = (khint_t)(h->n_buckets * __ac_HASH_UPPER + 0.5); \ |
301
|
|
|
|
|
|
|
} \ |
302
|
|
|
|
|
|
|
return 0; \ |
303
|
|
|
|
|
|
|
} \ |
304
|
|
|
|
|
|
|
SCOPE khint_t kh_put_##name(kh_##name##_t *h, khkey_t key, int *ret) \ |
305
|
|
|
|
|
|
|
{ \ |
306
|
|
|
|
|
|
|
khint_t x; \ |
307
|
|
|
|
|
|
|
if (h->n_occupied >= h->upper_bound) { /* update the hash table */ \ |
308
|
|
|
|
|
|
|
if (h->n_buckets > (h->size<<1)) { \ |
309
|
|
|
|
|
|
|
if (kh_resize_##name(h, h->n_buckets - 1) < 0) { /* clear "deleted" elements */ \ |
310
|
|
|
|
|
|
|
*ret = -1; return h->n_buckets; \ |
311
|
|
|
|
|
|
|
} \ |
312
|
|
|
|
|
|
|
} else if (kh_resize_##name(h, h->n_buckets + 1) < 0) { /* expand the hash table */ \ |
313
|
|
|
|
|
|
|
*ret = -1; return h->n_buckets; \ |
314
|
|
|
|
|
|
|
} \ |
315
|
|
|
|
|
|
|
} /* TODO: to implement automatically shrinking; resize() already support shrinking */ \ |
316
|
|
|
|
|
|
|
{ \ |
317
|
|
|
|
|
|
|
khint_t k, i, site, last, mask = h->n_buckets - 1, step = 0; \ |
318
|
|
|
|
|
|
|
x = site = h->n_buckets; k = __hash_func(key); i = k & mask; \ |
319
|
|
|
|
|
|
|
if (__ac_isempty(h->flags, i)) x = i; /* for speed up */ \ |
320
|
|
|
|
|
|
|
else { \ |
321
|
|
|
|
|
|
|
last = i; \ |
322
|
|
|
|
|
|
|
while (!__ac_isempty(h->flags, i) && (__ac_isdel(h->flags, i) || !__hash_equal(h->keys[i], key))) { \ |
323
|
|
|
|
|
|
|
if (__ac_isdel(h->flags, i)) site = i; \ |
324
|
|
|
|
|
|
|
i = (i + (++step)) & mask; \ |
325
|
|
|
|
|
|
|
if (i == last) { x = site; break; } \ |
326
|
|
|
|
|
|
|
} \ |
327
|
|
|
|
|
|
|
if (x == h->n_buckets) { \ |
328
|
|
|
|
|
|
|
if (__ac_isempty(h->flags, i) && site != h->n_buckets) x = site; \ |
329
|
|
|
|
|
|
|
else x = i; \ |
330
|
|
|
|
|
|
|
} \ |
331
|
|
|
|
|
|
|
} \ |
332
|
|
|
|
|
|
|
} \ |
333
|
|
|
|
|
|
|
if (__ac_isempty(h->flags, x)) { /* not present at all */ \ |
334
|
|
|
|
|
|
|
h->keys[x] = key; \ |
335
|
|
|
|
|
|
|
__ac_set_isboth_false(h->flags, x); \ |
336
|
|
|
|
|
|
|
++h->size; ++h->n_occupied; \ |
337
|
|
|
|
|
|
|
*ret = 1; \ |
338
|
|
|
|
|
|
|
} else if (__ac_isdel(h->flags, x)) { /* deleted */ \ |
339
|
|
|
|
|
|
|
h->keys[x] = key; \ |
340
|
|
|
|
|
|
|
__ac_set_isboth_false(h->flags, x); \ |
341
|
|
|
|
|
|
|
++h->size; \ |
342
|
|
|
|
|
|
|
*ret = 2; \ |
343
|
|
|
|
|
|
|
} else *ret = 0; /* Don't touch h->keys[x] if present and not deleted */ \ |
344
|
|
|
|
|
|
|
return x; \ |
345
|
|
|
|
|
|
|
} \ |
346
|
|
|
|
|
|
|
SCOPE void kh_del_##name(kh_##name##_t *h, khint_t x) \ |
347
|
|
|
|
|
|
|
{ \ |
348
|
|
|
|
|
|
|
if (x != h->n_buckets && !__ac_iseither(h->flags, x)) { \ |
349
|
|
|
|
|
|
|
__ac_set_isdel_true(h->flags, x); \ |
350
|
|
|
|
|
|
|
--h->size; \ |
351
|
|
|
|
|
|
|
} \ |
352
|
|
|
|
|
|
|
} |
353
|
|
|
|
|
|
|
|
354
|
|
|
|
|
|
|
#define KHASH_DECLARE(name, khkey_t, khval_t) \ |
355
|
|
|
|
|
|
|
__KHASH_TYPE(name, khkey_t, khval_t) \ |
356
|
|
|
|
|
|
|
__KHASH_PROTOTYPES(name, khkey_t, khval_t) |
357
|
|
|
|
|
|
|
|
358
|
|
|
|
|
|
|
#define KHASH_INIT2(name, SCOPE, khkey_t, khval_t, kh_is_map, __hash_func, __hash_equal) \ |
359
|
|
|
|
|
|
|
__KHASH_TYPE(name, khkey_t, khval_t) \ |
360
|
|
|
|
|
|
|
__KHASH_IMPL(name, SCOPE, khkey_t, khval_t, kh_is_map, __hash_func, __hash_equal) |
361
|
|
|
|
|
|
|
|
362
|
|
|
|
|
|
|
#define KHASH_INIT(name, khkey_t, khval_t, kh_is_map, __hash_func, __hash_equal) \ |
363
|
|
|
|
|
|
|
KHASH_INIT2(name, static kh_inline, khkey_t, khval_t, kh_is_map, __hash_func, __hash_equal) |
364
|
|
|
|
|
|
|
|
365
|
|
|
|
|
|
|
/* --- BEGIN OF HASH FUNCTIONS --- */ |
366
|
|
|
|
|
|
|
|
367
|
|
|
|
|
|
|
/*! @function |
368
|
|
|
|
|
|
|
@abstract Integer hash function |
369
|
|
|
|
|
|
|
@param key The integer [khint32_t] |
370
|
|
|
|
|
|
|
@return The hash value [khint_t] |
371
|
|
|
|
|
|
|
*/ |
372
|
|
|
|
|
|
|
#define kh_int_hash_func(key) (khint32_t)(key) |
373
|
|
|
|
|
|
|
/*! @function |
374
|
|
|
|
|
|
|
@abstract Integer comparison function |
375
|
|
|
|
|
|
|
*/ |
376
|
|
|
|
|
|
|
#define kh_int_hash_equal(a, b) ((a) == (b)) |
377
|
|
|
|
|
|
|
/*! @function |
378
|
|
|
|
|
|
|
@abstract 64-bit integer hash function |
379
|
|
|
|
|
|
|
@param key The integer [khint64_t] |
380
|
|
|
|
|
|
|
@return The hash value [khint_t] |
381
|
|
|
|
|
|
|
*/ |
382
|
|
|
|
|
|
|
#define kh_int64_hash_func(key) (khint32_t)((key)>>33^(key)^(key)<<11) |
383
|
|
|
|
|
|
|
/*! @function |
384
|
|
|
|
|
|
|
@abstract 64-bit integer comparison function |
385
|
|
|
|
|
|
|
*/ |
386
|
|
|
|
|
|
|
#define kh_int64_hash_equal(a, b) ((a) == (b)) |
387
|
|
|
|
|
|
|
/*! @function |
388
|
|
|
|
|
|
|
@abstract const char* hash function |
389
|
|
|
|
|
|
|
@param s Pointer to a null terminated string |
390
|
|
|
|
|
|
|
@return The hash value |
391
|
|
|
|
|
|
|
*/ |
392
|
58854
|
|
|
|
|
|
static kh_inline khint_t __ac_X31_hash_string(const char *s) |
393
|
|
|
|
|
|
|
{ |
394
|
58854
|
|
|
|
|
|
khint_t h = (khint_t)*s; |
395
|
892044
|
50
|
|
|
|
|
if (h) for (++s ; *s; ++s) h = (h << 5) - h + (khint_t)*s; |
|
|
100
|
|
|
|
|
|
396
|
58854
|
|
|
|
|
|
return h; |
397
|
|
|
|
|
|
|
} |
398
|
|
|
|
|
|
|
/*! @function |
399
|
|
|
|
|
|
|
@abstract Another interface to const char* hash function |
400
|
|
|
|
|
|
|
@param key Pointer to a null terminated string [const char*] |
401
|
|
|
|
|
|
|
@return The hash value [khint_t] |
402
|
|
|
|
|
|
|
*/ |
403
|
|
|
|
|
|
|
#define kh_str_hash_func(key) __ac_X31_hash_string(key) |
404
|
|
|
|
|
|
|
/*! @function |
405
|
|
|
|
|
|
|
@abstract Const char* comparison function |
406
|
|
|
|
|
|
|
*/ |
407
|
|
|
|
|
|
|
#define kh_str_hash_equal(a, b) (strcmp(a, b) == 0) |
408
|
|
|
|
|
|
|
|
409
|
|
|
|
|
|
|
static kh_inline khint_t __ac_Wang_hash(khint_t key) |
410
|
|
|
|
|
|
|
{ |
411
|
|
|
|
|
|
|
key += ~(key << 15); |
412
|
|
|
|
|
|
|
key ^= (key >> 10); |
413
|
|
|
|
|
|
|
key += (key << 3); |
414
|
|
|
|
|
|
|
key ^= (key >> 6); |
415
|
|
|
|
|
|
|
key += ~(key << 11); |
416
|
|
|
|
|
|
|
key ^= (key >> 16); |
417
|
|
|
|
|
|
|
return key; |
418
|
|
|
|
|
|
|
} |
419
|
|
|
|
|
|
|
#define kh_int_hash_func2(k) __ac_Wang_hash((khint_t)key) |
420
|
|
|
|
|
|
|
|
421
|
|
|
|
|
|
|
/* --- END OF HASH FUNCTIONS --- */ |
422
|
|
|
|
|
|
|
|
423
|
|
|
|
|
|
|
/* Other convenient macros... */ |
424
|
|
|
|
|
|
|
|
425
|
|
|
|
|
|
|
/*! |
426
|
|
|
|
|
|
|
@abstract Type of the hash table. |
427
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
428
|
|
|
|
|
|
|
*/ |
429
|
|
|
|
|
|
|
#define khash_t(name) kh_##name##_t |
430
|
|
|
|
|
|
|
|
431
|
|
|
|
|
|
|
/*! @function |
432
|
|
|
|
|
|
|
@abstract Initiate a hash table. |
433
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
434
|
|
|
|
|
|
|
@return Pointer to the hash table [khash_t(name)*] |
435
|
|
|
|
|
|
|
*/ |
436
|
|
|
|
|
|
|
#define kh_init(name) kh_init_##name() |
437
|
|
|
|
|
|
|
|
438
|
|
|
|
|
|
|
/*! @function |
439
|
|
|
|
|
|
|
@abstract Destroy a hash table. |
440
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
441
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
442
|
|
|
|
|
|
|
*/ |
443
|
|
|
|
|
|
|
#define kh_destroy(name, h) kh_destroy_##name(h) |
444
|
|
|
|
|
|
|
|
445
|
|
|
|
|
|
|
/*! @function |
446
|
|
|
|
|
|
|
@abstract Reset a hash table without deallocating memory. |
447
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
448
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
449
|
|
|
|
|
|
|
*/ |
450
|
|
|
|
|
|
|
#define kh_clear(name, h) kh_clear_##name(h) |
451
|
|
|
|
|
|
|
|
452
|
|
|
|
|
|
|
/*! @function |
453
|
|
|
|
|
|
|
@abstract Resize a hash table. |
454
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
455
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
456
|
|
|
|
|
|
|
@param s New size [khint_t] |
457
|
|
|
|
|
|
|
*/ |
458
|
|
|
|
|
|
|
#define kh_resize(name, h, s) kh_resize_##name(h, s) |
459
|
|
|
|
|
|
|
|
460
|
|
|
|
|
|
|
/*! @function |
461
|
|
|
|
|
|
|
@abstract Insert a key to the hash table. |
462
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
463
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
464
|
|
|
|
|
|
|
@param k Key [type of keys] |
465
|
|
|
|
|
|
|
@param r Extra return code: -1 if the operation failed; |
466
|
|
|
|
|
|
|
0 if the key is present in the hash table; |
467
|
|
|
|
|
|
|
1 if the bucket is empty (never used); 2 if the element in |
468
|
|
|
|
|
|
|
the bucket has been deleted [int*] |
469
|
|
|
|
|
|
|
@return Iterator to the inserted element [khint_t] |
470
|
|
|
|
|
|
|
*/ |
471
|
|
|
|
|
|
|
#define kh_put(name, h, k, r) kh_put_##name(h, k, r) |
472
|
|
|
|
|
|
|
|
473
|
|
|
|
|
|
|
/*! @function |
474
|
|
|
|
|
|
|
@abstract Retrieve a key from the hash table. |
475
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
476
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
477
|
|
|
|
|
|
|
@param k Key [type of keys] |
478
|
|
|
|
|
|
|
@return Iterator to the found element, or kh_end(h) if the element is absent [khint_t] |
479
|
|
|
|
|
|
|
*/ |
480
|
|
|
|
|
|
|
#define kh_get(name, h, k) kh_get_##name(h, k) |
481
|
|
|
|
|
|
|
|
482
|
|
|
|
|
|
|
/*! @function |
483
|
|
|
|
|
|
|
@abstract Remove a key from the hash table. |
484
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
485
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
486
|
|
|
|
|
|
|
@param k Iterator to the element to be deleted [khint_t] |
487
|
|
|
|
|
|
|
*/ |
488
|
|
|
|
|
|
|
#define kh_del(name, h, k) kh_del_##name(h, k) |
489
|
|
|
|
|
|
|
|
490
|
|
|
|
|
|
|
/*! @function |
491
|
|
|
|
|
|
|
@abstract Test whether a bucket contains data. |
492
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
493
|
|
|
|
|
|
|
@param x Iterator to the bucket [khint_t] |
494
|
|
|
|
|
|
|
@return 1 if containing data; 0 otherwise [int] |
495
|
|
|
|
|
|
|
*/ |
496
|
|
|
|
|
|
|
#define kh_exist(h, x) (!__ac_iseither((h)->flags, (x))) |
497
|
|
|
|
|
|
|
|
498
|
|
|
|
|
|
|
/*! @function |
499
|
|
|
|
|
|
|
@abstract Get key given an iterator |
500
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
501
|
|
|
|
|
|
|
@param x Iterator to the bucket [khint_t] |
502
|
|
|
|
|
|
|
@return Key [type of keys] |
503
|
|
|
|
|
|
|
*/ |
504
|
|
|
|
|
|
|
#define kh_key(h, x) ((h)->keys[x]) |
505
|
|
|
|
|
|
|
|
506
|
|
|
|
|
|
|
/*! @function |
507
|
|
|
|
|
|
|
@abstract Get value given an iterator |
508
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
509
|
|
|
|
|
|
|
@param x Iterator to the bucket [khint_t] |
510
|
|
|
|
|
|
|
@return Value [type of values] |
511
|
|
|
|
|
|
|
@discussion For hash sets, calling this results in segfault. |
512
|
|
|
|
|
|
|
*/ |
513
|
|
|
|
|
|
|
#define kh_val(h, x) ((h)->vals[x]) |
514
|
|
|
|
|
|
|
|
515
|
|
|
|
|
|
|
/*! @function |
516
|
|
|
|
|
|
|
@abstract Alias of kh_val() |
517
|
|
|
|
|
|
|
*/ |
518
|
|
|
|
|
|
|
#define kh_value(h, x) ((h)->vals[x]) |
519
|
|
|
|
|
|
|
|
520
|
|
|
|
|
|
|
/*! @function |
521
|
|
|
|
|
|
|
@abstract Get the start iterator |
522
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
523
|
|
|
|
|
|
|
@return The start iterator [khint_t] |
524
|
|
|
|
|
|
|
*/ |
525
|
|
|
|
|
|
|
#define kh_begin(h) (khint_t)(0) |
526
|
|
|
|
|
|
|
|
527
|
|
|
|
|
|
|
/*! @function |
528
|
|
|
|
|
|
|
@abstract Get the end iterator |
529
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
530
|
|
|
|
|
|
|
@return The end iterator [khint_t] |
531
|
|
|
|
|
|
|
*/ |
532
|
|
|
|
|
|
|
#define kh_end(h) ((h)->n_buckets) |
533
|
|
|
|
|
|
|
|
534
|
|
|
|
|
|
|
/*! @function |
535
|
|
|
|
|
|
|
@abstract Get the number of elements in the hash table |
536
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
537
|
|
|
|
|
|
|
@return Number of elements in the hash table [khint_t] |
538
|
|
|
|
|
|
|
*/ |
539
|
|
|
|
|
|
|
#define kh_size(h) ((h)->size) |
540
|
|
|
|
|
|
|
|
541
|
|
|
|
|
|
|
/*! @function |
542
|
|
|
|
|
|
|
@abstract Get the number of buckets in the hash table |
543
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
544
|
|
|
|
|
|
|
@return Number of buckets in the hash table [khint_t] |
545
|
|
|
|
|
|
|
*/ |
546
|
|
|
|
|
|
|
#define kh_n_buckets(h) ((h)->n_buckets) |
547
|
|
|
|
|
|
|
|
548
|
|
|
|
|
|
|
/*! @function |
549
|
|
|
|
|
|
|
@abstract Iterate over the entries in the hash table |
550
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
551
|
|
|
|
|
|
|
@param kvar Variable to which key will be assigned |
552
|
|
|
|
|
|
|
@param vvar Variable to which value will be assigned |
553
|
|
|
|
|
|
|
@param code Block of code to execute |
554
|
|
|
|
|
|
|
*/ |
555
|
|
|
|
|
|
|
#define kh_foreach(h, kvar, vvar, code) { khint_t __i; \ |
556
|
|
|
|
|
|
|
for (__i = kh_begin(h); __i != kh_end(h); ++__i) { \ |
557
|
|
|
|
|
|
|
if (!kh_exist(h,__i)) continue; \ |
558
|
|
|
|
|
|
|
(kvar) = kh_key(h,__i); \ |
559
|
|
|
|
|
|
|
(vvar) = kh_val(h,__i); \ |
560
|
|
|
|
|
|
|
code; \ |
561
|
|
|
|
|
|
|
} } |
562
|
|
|
|
|
|
|
|
563
|
|
|
|
|
|
|
/*! @function |
564
|
|
|
|
|
|
|
@abstract Iterate over the values in the hash table |
565
|
|
|
|
|
|
|
@param h Pointer to the hash table [khash_t(name)*] |
566
|
|
|
|
|
|
|
@param vvar Variable to which value will be assigned |
567
|
|
|
|
|
|
|
@param code Block of code to execute |
568
|
|
|
|
|
|
|
*/ |
569
|
|
|
|
|
|
|
#define kh_foreach_value(h, vvar, code) { khint_t __i; \ |
570
|
|
|
|
|
|
|
for (__i = kh_begin(h); __i != kh_end(h); ++__i) { \ |
571
|
|
|
|
|
|
|
if (!kh_exist(h,__i)) continue; \ |
572
|
|
|
|
|
|
|
(vvar) = kh_val(h,__i); \ |
573
|
|
|
|
|
|
|
code; \ |
574
|
|
|
|
|
|
|
} } |
575
|
|
|
|
|
|
|
|
576
|
|
|
|
|
|
|
/* More conenient interfaces */ |
577
|
|
|
|
|
|
|
|
578
|
|
|
|
|
|
|
/*! @function |
579
|
|
|
|
|
|
|
@abstract Instantiate a hash set containing integer keys |
580
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
581
|
|
|
|
|
|
|
*/ |
582
|
|
|
|
|
|
|
#define KHASH_SET_INIT_INT(name) \ |
583
|
|
|
|
|
|
|
KHASH_INIT(name, khint32_t, char, 0, kh_int_hash_func, kh_int_hash_equal) |
584
|
|
|
|
|
|
|
|
585
|
|
|
|
|
|
|
/*! @function |
586
|
|
|
|
|
|
|
@abstract Instantiate a hash map containing integer keys |
587
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
588
|
|
|
|
|
|
|
@param khval_t Type of values [type] |
589
|
|
|
|
|
|
|
*/ |
590
|
|
|
|
|
|
|
#define KHASH_MAP_INIT_INT(name, khval_t) \ |
591
|
|
|
|
|
|
|
KHASH_INIT(name, khint32_t, khval_t, 1, kh_int_hash_func, kh_int_hash_equal) |
592
|
|
|
|
|
|
|
|
593
|
|
|
|
|
|
|
/*! @function |
594
|
|
|
|
|
|
|
@abstract Instantiate a hash map containing 64-bit integer keys |
595
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
596
|
|
|
|
|
|
|
*/ |
597
|
|
|
|
|
|
|
#define KHASH_SET_INIT_INT64(name) \ |
598
|
|
|
|
|
|
|
KHASH_INIT(name, khint64_t, char, 0, kh_int64_hash_func, kh_int64_hash_equal) |
599
|
|
|
|
|
|
|
|
600
|
|
|
|
|
|
|
/*! @function |
601
|
|
|
|
|
|
|
@abstract Instantiate a hash map containing 64-bit integer keys |
602
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
603
|
|
|
|
|
|
|
@param khval_t Type of values [type] |
604
|
|
|
|
|
|
|
*/ |
605
|
|
|
|
|
|
|
#define KHASH_MAP_INIT_INT64(name, khval_t) \ |
606
|
|
|
|
|
|
|
KHASH_INIT(name, khint64_t, khval_t, 1, kh_int64_hash_func, kh_int64_hash_equal) |
607
|
|
|
|
|
|
|
|
608
|
|
|
|
|
|
|
typedef const char *kh_cstr_t; |
609
|
|
|
|
|
|
|
/*! @function |
610
|
|
|
|
|
|
|
@abstract Instantiate a hash map containing const char* keys |
611
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
612
|
|
|
|
|
|
|
*/ |
613
|
|
|
|
|
|
|
#define KHASH_SET_INIT_STR(name) \ |
614
|
|
|
|
|
|
|
KHASH_INIT(name, kh_cstr_t, char, 0, kh_str_hash_func, kh_str_hash_equal) |
615
|
|
|
|
|
|
|
|
616
|
|
|
|
|
|
|
/*! @function |
617
|
|
|
|
|
|
|
@abstract Instantiate a hash map containing const char* keys |
618
|
|
|
|
|
|
|
@param name Name of the hash table [symbol] |
619
|
|
|
|
|
|
|
@param khval_t Type of values [type] |
620
|
|
|
|
|
|
|
*/ |
621
|
|
|
|
|
|
|
#define KHASH_MAP_INIT_STR(name, khval_t) \ |
622
|
|
|
|
|
|
|
KHASH_INIT(name, kh_cstr_t, khval_t, 1, kh_str_hash_func, kh_str_hash_equal) |
623
|
|
|
|
|
|
|
|
624
|
|
|
|
|
|
|
#endif /* __AC_KHASH_H */ |