strmap: enable allocations to come from a mem_pool

[thirdparty/git.git] / oidmap.c

diff --git a/oidmap.c b/oidmap.c
index 6db4fffcdb272d2227c8a7779a19597dc6102971..286a04a53c20119fe567eec4accbaba4d9f04a97 100644 (file)
--- a/oidmap.c
+++ b/oidmap.c
@@ -1,51 +1,61 @@
 #include "cache.h"
 #include "oidmap.h"
 
-static int cmpfn(const void *hashmap_cmp_fn_data,
-                const void *entry, const void *entry_or_key,
-                const void *keydata)
+static int oidmap_neq(const void *hashmap_cmp_fn_data,
+                     const struct hashmap_entry *e1,
+                     const struct hashmap_entry *e2,
+                     const void *keydata)
 {
-       const struct oidmap_entry *entry_ = entry;
-       if (keydata)
-               return oidcmp(&entry_->oid, (const struct object_id *) keydata);
-       return oidcmp(&entry_->oid,
-                     &((const struct oidmap_entry *) entry_or_key)->oid);
-}
+       const struct oidmap_entry *a, *b;
 
-static int hash(const struct object_id *oid)
-{
-       int hash;
-       memcpy(&hash, oid->hash, sizeof(hash));
-       return hash;
+       a = container_of(e1, const struct oidmap_entry, internal_entry);
+       b = container_of(e2, const struct oidmap_entry, internal_entry);
+
+       if (keydata)
+               return !oideq(&a->oid, (const struct object_id *) keydata);
+       return !oideq(&a->oid, &b->oid);
 }
 
 void oidmap_init(struct oidmap *map, size_t initial_size)
 {
-       hashmap_init(&map->map, cmpfn, NULL, initial_size);
+       hashmap_init(&map->map, oidmap_neq, NULL, initial_size);
 }
 
 void oidmap_free(struct oidmap *map, int free_entries)
 {
        if (!map)
                return;
-       hashmap_free(&map->map, free_entries);
+
+       /* TODO: make oidmap itself not depend on struct layouts */
+       hashmap_clear_(&map->map, free_entries ? 0 : -1);
 }
 
 void *oidmap_get(const struct oidmap *map, const struct object_id *key)
 {
-       return hashmap_get_from_hash(&map->map, hash(key), key);
+       if (!map->map.cmpfn)
+               return NULL;
+
+       return hashmap_get_from_hash(&map->map, oidhash(key), key);
 }
 
 void *oidmap_remove(struct oidmap *map, const struct object_id *key)
 {
        struct hashmap_entry entry;
-       hashmap_entry_init(&entry, hash(key));
+
+       if (!map->map.cmpfn)
+               oidmap_init(map, 0);
+
+       hashmap_entry_init(&entry, oidhash(key));
        return hashmap_remove(&map->map, &entry, key);
 }
 
 void *oidmap_put(struct oidmap *map, void *entry)
 {
        struct oidmap_entry *to_put = entry;
-       hashmap_entry_init(&to_put->internal_entry, hash(&to_put->oid));
-       return hashmap_put(&map->map, to_put);
+
+       if (!map->map.cmpfn)
+               oidmap_init(map, 0);
+
+       hashmap_entry_init(&to_put->internal_entry, oidhash(&to_put->oid));
+       return hashmap_put(&map->map, &to_put->internal_entry);
 }