|
1 | 1 | #ifndef OIDSET_H
|
2 | 2 | #define OIDSET_H
|
3 | 3 |
|
4 |
| -#include "oidmap.h" |
| 4 | +#include "hashmap.h" |
| 5 | +#include "khash.h" |
5 | 6 |
|
6 | 7 | /**
|
7 | 8 | * This API is similar to sha1-array, in that it maintains a set of object ids
|
|
15 | 16 | * table overhead.
|
16 | 17 | */
|
17 | 18 |
|
| 19 | +static inline unsigned int oid_hash(struct object_id oid) |
| 20 | +{ |
| 21 | + return sha1hash(oid.hash); |
| 22 | +} |
| 23 | + |
| 24 | +static inline int oid_equal(struct object_id a, struct object_id b) |
| 25 | +{ |
| 26 | + return oideq(&a, &b); |
| 27 | +} |
| 28 | + |
| 29 | +KHASH_INIT(oid, struct object_id, int, 0, oid_hash, oid_equal) |
| 30 | + |
18 | 31 | /**
|
19 | 32 | * A single oidset; should be zero-initialized (or use OIDSET_INIT).
|
20 | 33 | */
|
21 | 34 | struct oidset {
|
22 |
| - struct oidmap map; |
| 35 | + kh_oid_t set; |
23 | 36 | };
|
24 | 37 |
|
25 |
| -#define OIDSET_INIT { OIDMAP_INIT } |
| 38 | +#define OIDSET_INIT { { 0 } } |
26 | 39 |
|
27 | 40 |
|
28 | 41 | static inline void oidset_init(struct oidset *set, size_t initial_size)
|
29 | 42 | {
|
30 |
| - oidmap_init(&set->map, initial_size); |
| 43 | + memset(&set->set, 0, sizeof(set->set)); |
| 44 | + if (initial_size) |
| 45 | + kh_resize_oid(&set->set, initial_size); |
31 | 46 | }
|
32 | 47 |
|
33 | 48 | /**
|
@@ -58,19 +73,24 @@ int oidset_remove(struct oidset *set, const struct object_id *oid);
|
58 | 73 | void oidset_clear(struct oidset *set);
|
59 | 74 |
|
60 | 75 | struct oidset_iter {
|
61 |
| - struct oidmap_iter m_iter; |
| 76 | + kh_oid_t *set; |
| 77 | + khiter_t iter; |
62 | 78 | };
|
63 | 79 |
|
64 | 80 | static inline void oidset_iter_init(struct oidset *set,
|
65 | 81 | struct oidset_iter *iter)
|
66 | 82 | {
|
67 |
| - oidmap_iter_init(&set->map, &iter->m_iter); |
| 83 | + iter->set = &set->set; |
| 84 | + iter->iter = kh_begin(iter->set); |
68 | 85 | }
|
69 | 86 |
|
70 | 87 | static inline struct object_id *oidset_iter_next(struct oidset_iter *iter)
|
71 | 88 | {
|
72 |
| - struct oidmap_entry *e = oidmap_iter_next(&iter->m_iter); |
73 |
| - return e ? &e->oid : NULL; |
| 89 | + for (; iter->iter != kh_end(iter->set); iter->iter++) { |
| 90 | + if (kh_exist(iter->set, iter->iter)) |
| 91 | + return &kh_key(iter->set, iter->iter++); |
| 92 | + } |
| 93 | + return NULL; |
74 | 94 | }
|
75 | 95 |
|
76 | 96 | static inline struct object_id *oidset_iter_first(struct oidset *set,
|
|
0 commit comments