template.h

Go to the documentation of this file.

 
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
 
#include <derive-c/core/helpers.h>
#include <derive-c/core/panic.h>
#include <derive-c/structures/hashmap/utils.h>
 
#include <derive-c/core/alloc/def.h>
#include <derive-c/core/self/def.h>
 
#if !defined KEY
    #if !defined __clang_analyzer__
        #error "KEY must be defined to for a hashmap template"
    #endif
typedef struct {
    int x;
} key_t;
    #define KEY key_t
static void key_delete(key_t* UNUSED(self)) {}
    #define KEY_DELETE key_delete
static key_t key_clone(key_t const* self) { return *self; }
    #define KEY_CLONE key_clone
static bool key_eq(key_t const* key_1, key_t const* key_2) { return key_1->x == key_2->x; }
    #define KEY_EQ key_eq
#endif
 
#if !defined KEY_HASH
    #if !defined __clang_analyzer__
        #error "The hash function for K must be defined"
    #endif
static size_t key_hash(key_t const* key);
    #define KEY_HASH key_hash
#endif
 
#if !defined KEY_EQ
    #define KEY_EQ(key_1, key_2) (*(key_1) == *(key_2))
#endif
 
#if !defined KEY_DELETE
    #define KEY_DELETE(key)
#endif
 
#if !defined KEY_CLONE
    #define KEY_CLONE(value) (*(value))
#endif
 
#if !defined VALUE
    #if !defined __clang_analyzer__
        #error "VALUE must be defined to for a hashmap template"
    #endif
typedef struct {
    int x;
} value_t;
    #define VALUE value_t
static void value_delete(value_t* UNUSED(self)) {}
    #define VALUE_DELETE value_delete
static value_t value_clone(value_t const* self) { return *self; }
    #define VALUE_CLONE value_clone
#endif
 
#if !defined VALUE_DELETE
    #define VALUE_DELETE(value)
#endif
 
#if !defined VALUE_CLONE
    #define VALUE_CLONE(value) (*(value))
#endif
 
#define KEY_ENTRY NS(SELF, key_entry)
typedef struct {
    bool present;
    uint16_t distance_from_desired;
    KEY key;
} KEY_ENTRY;
 
typedef struct {
    size_t capacity; // INVARIANT: A power of 2
    size_t items;
    // Split keys & values in an old hashmap I used, cannot remember why (many collisions better
    // decomposed), should probably use 1 buffer.
    KEY_ENTRY* keys;
    VALUE* values;
    ALLOC* alloc;
    gdb_marker derive_c_hashmap;
} SELF;
 
static SELF NS(SELF, new_with_capacity_for)(size_t capacity, ALLOC* alloc) {
    ASSERT(capacity > 0);
    size_t real_capacity = apply_capacity_policy(capacity);
    ASSERT(real_capacity > 0);
    // JUSTIFY: calloc of keys
    //  - A cheap way to get all precense flags as zeroed (os & allocater supported get zeroed page)
    //  - for the values, we do not need this (no precense checks are done on values)
    KEY_ENTRY* keys = (KEY_ENTRY*)NS(ALLOC, calloc)(alloc, sizeof(KEY_ENTRY), real_capacity);
    VALUE* values = (VALUE*)NS(ALLOC, malloc)(alloc, sizeof(VALUE) * real_capacity);
    ASSERT(keys && values);
    return (SELF){
        .capacity = real_capacity,
        .items = 0,
        .keys = keys,
        .values = values,
        .alloc = alloc,
    };
}
 
static SELF NS(SELF, new)(ALLOC* alloc) {
    return NS(SELF, new_with_capacity_for)(INITIAL_CAPACITY, alloc);
}
 
static SELF NS(SELF, clone)(SELF const* self) {
    DEBUG_ASSERT(self);
 
    // JUSTIFY: Naive copy
    //           - We could resize (potentially a smaller map) and rehash
    //           - Not confident it would be any better than just a copy.
    // JUSTIFY: Individually copy keys
    //           - Many entries are zeroed, no need to copy uninit data
 
    KEY_ENTRY* keys = (KEY_ENTRY*)NS(ALLOC, calloc)(self->alloc, sizeof(KEY_ENTRY), self->capacity);
    VALUE* values = (VALUE*)NS(ALLOC, malloc)(self->alloc, sizeof(VALUE) * self->capacity);
    ASSERT(keys && values);
 
    for (size_t i = 0; i < self->capacity; i++) {
        if (self->keys[i].present) {
            KEY_ENTRY const* old_entry = &self->keys[i];
            keys[i] = (KEY_ENTRY){
                .present = true,
                .distance_from_desired = old_entry->distance_from_desired,
                .key = KEY_CLONE(&old_entry->key),
            };
            values[i] = VALUE_CLONE(&self->values[i]);
        }
    }
 
    return (SELF){
        .capacity = self->capacity,
        .items = self->items,
        .keys = keys,
        .values = values,
        .alloc = self->alloc,
    };
}
 
static void NS(SELF, delete)(SELF* self);
 
static VALUE* NS(SELF, insert)(SELF* self, KEY key, VALUE value);
 
static void NS(SELF, extend_capacity_for)(SELF* self, size_t expected_items) {
    DEBUG_ASSERT(self);
    size_t target_capacity = apply_capacity_policy(expected_items);
    if (target_capacity > self->capacity) {
        SELF new_map = NS(SELF, new_with_capacity_for)(expected_items, self->alloc);
        for (size_t index = 0; index < self->capacity; index++) {
            KEY_ENTRY* entry = &self->keys[index];
            if (entry->present) {
                NS(SELF, insert)(&new_map, entry->key, self->values[index]);
            }
        }
        NS(ALLOC, free)(self->alloc, (void*)self->keys);
        NS(ALLOC, free)(self->alloc, (void*)self->values);
        *self = new_map;
    }
}
 
static VALUE* NS(SELF, try_insert)(SELF* self, KEY key, VALUE value) {
    DEBUG_ASSERT(self);
    if (apply_capacity_policy(self->items) > self->capacity / 2) {
        NS(SELF, extend_capacity_for)(self, self->items * 2);
    }
 
    uint16_t distance_from_desired = 0;
    size_t hash = KEY_HASH(&key);
    size_t index = modulus_power_of_2_capacity(hash, self->capacity);
    VALUE* placed_entry = NULL;
    for (;;) {
        KEY_ENTRY* entry = &self->keys[index];
        DEBUG_ASSERT(distance_from_desired < self->capacity);
 
        if (entry->present) {
            if (KEY_EQ(&entry->key, &key)) {
                return NULL;
            }
 
            if (entry->distance_from_desired < distance_from_desired) {
                KEY switch_key = entry->key;
                uint16_t switch_distance_from_desired = entry->distance_from_desired;
                VALUE switch_value = self->values[index];
 
                entry->key = key;
                entry->distance_from_desired = distance_from_desired;
                self->values[index] = value;
 
                key = switch_key;
                distance_from_desired = switch_distance_from_desired;
                value = switch_value;
 
                if (!placed_entry) {
                    placed_entry = &self->values[index];
                }
            }
 
            distance_from_desired++;
            index = modulus_power_of_2_capacity(index + 1, self->capacity);
        } else {
            entry->present = true;
            entry->distance_from_desired = distance_from_desired;
            entry->key = key;
            self->values[index] = value;
            if (!placed_entry) {
                placed_entry = &self->values[index];
            }
            self->items++;
            return placed_entry;
        }
    }
}
 
static VALUE* NS(SELF, insert)(SELF* self, KEY key, VALUE value) {
    VALUE* value_ptr = NS(SELF, try_insert)(self, key, value);
    ASSERT(value_ptr);
    return value_ptr;
}
 
static VALUE* NS(SELF, try_write)(SELF* self, KEY key) {
    DEBUG_ASSERT(self);
    size_t hash = KEY_HASH(&key);
    size_t index = modulus_power_of_2_capacity(hash, self->capacity);
 
    for (;;) {
        KEY_ENTRY* entry = &self->keys[index];
        if (entry->present) {
            if (KEY_EQ(&entry->key, &key)) {
                return &self->values[index];
            }
            index = modulus_power_of_2_capacity(index + 1, self->capacity);
        } else {
            return NULL;
        }
    }
}
 
static VALUE* NS(SELF, write)(SELF* self, KEY key) {
    VALUE* value = NS(SELF, try_write)(self, key);
    ASSERT(value);
    return value;
}
 
static VALUE const* NS(SELF, try_read)(SELF const* self, KEY key) {
    DEBUG_ASSERT(self);
    size_t hash = KEY_HASH(&key);
    size_t index = modulus_power_of_2_capacity(hash, self->capacity);
 
    for (;;) {
        KEY_ENTRY* entry = &self->keys[index];
        if (entry->present) {
            if (KEY_EQ(&entry->key, &key)) {
                return &self->values[index];
            }
            index = modulus_power_of_2_capacity(index + 1, self->capacity);
        } else {
            return NULL;
        }
    }
}
 
static VALUE const* NS(SELF, read)(SELF const* self, KEY key) {
    VALUE const* value = NS(SELF, try_read)(self, key);
    ASSERT(value);
    return value;
}
 
static bool NS(SELF, try_remove)(SELF* self, KEY key, VALUE* destination) {
    DEBUG_ASSERT(self);
    size_t hash = KEY_HASH(&key);
    size_t index = modulus_power_of_2_capacity(hash, self->capacity);
 
    for (;;) {
        KEY_ENTRY* entry = &self->keys[index];
        if (entry->present) {
            if (KEY_EQ(&entry->key, &key)) {
                self->items--;
 
                *destination = self->values[index];
                KEY_DELETE(&entry->key);
 
                // NOTE: For robin hood hashing, we need probe chains to be unbroken
                //        - Need to find the entries that might use this chain (
                //          distance_to_desired > 0 until the next not-present or
                //          distance_to_desired=0 slot)
                //        hence we shift entries the left (being careful with modulus index)
 
                size_t free_index = index;
                KEY_ENTRY* free_entry = entry;
 
                size_t check_index = modulus_power_of_2_capacity(free_index + 1, self->capacity);
                KEY_ENTRY* check_entry = &self->keys[check_index];
 
                while (check_entry->present && (check_entry->distance_from_desired > 0)) {
                    free_entry->key = check_entry->key;
                    free_entry->distance_from_desired = check_entry->distance_from_desired - 1;
                    self->values[free_index] = self->values[check_index];
 
                    free_index = check_index;
                    free_entry = check_entry;
 
                    check_index = modulus_power_of_2_capacity(free_index + 1, self->capacity);
                    check_entry = &self->keys[check_index];
                }
 
                // JUSTIFY: Only setting free entry to false
                //           - We remove, then shift down an index
                //           - The removed entry already has the flag set
                //           - the free entry was the last one removed/moved down an index, so it
                //             should be false.
 
                free_entry->present = false;
 
                return true;
            }
            index = modulus_power_of_2_capacity(index + 1, self->capacity);
        } else {
            return false;
        }
    }
}
 
static VALUE NS(SELF, remove)(SELF* self, KEY key) {
    VALUE value;
    ASSERT(NS(SELF, try_remove)(self, key, &value));
    return value;
}
 
static void NS(SELF, delete_entry)(SELF* self, KEY key) {
    VALUE value = NS(SELF, remove)(self, key);
    VALUE_DELETE(&value);
}
 
static size_t NS(SELF, size)(SELF const* self) {
    DEBUG_ASSERT(self);
    return self->items;
}
 
#define KV_PAIR NS(SELF, kv)
 
typedef struct {
    KEY const* key;
    VALUE* value;
} KV_PAIR;
 
#define ITER NS(SELF, iter)
 
typedef struct {
    SELF* map;
    size_t index;
    KV_PAIR curr;
} ITER;
 
static KV_PAIR const* NS(ITER, next)(ITER* iter) {
    DEBUG_ASSERT(iter);
    if (iter->index < iter->map->capacity) {
        iter->curr = (KV_PAIR){.key = &iter->map->keys[iter->index].key,
                               .value = &iter->map->values[iter->index]};
        iter->index++;
        while (iter->index < iter->map->capacity && !iter->map->keys[iter->index].present) {
            iter->index++;
        }
        return &iter->curr;
    }
    return NULL;
}
 
static bool NS(ITER, empty)(ITER const* iter) {
    DEBUG_ASSERT(iter);
    return iter->index >= iter->map->capacity;
}
 
static ITER NS(SELF, get_iter)(SELF* self) {
    DEBUG_ASSERT(self);
    size_t first_index = 0;
    while (first_index < self->capacity && !self->keys[first_index].present) {
        first_index++;
    }
    return (ITER){
        .map = self,
        .index = first_index,
        .curr = (KV_PAIR){.key = NULL, .value = NULL},
    };
}
 
static void NS(SELF, delete)(SELF* self) {
    DEBUG_ASSERT(self);
    ITER iter = NS(SELF, get_iter)(self);
 
    for (size_t i = 0; i < self->capacity; i++) {
        if (self->keys[i].present) {
            KEY_DELETE(&self->keys[i].key);
            VALUE_DELETE(&self->values[i]);
        }
    }
 
    NS(ALLOC, free)(self->alloc, (void*)self->keys);
    NS(ALLOC, free)(self->alloc, (void*)self->values);
}
 
#undef ITER
#undef KV_PAIR
 
#define KV_PAIR_CONST NS(SELF, kv_const)
 
typedef struct {
    KEY const* key;
    VALUE const* value;
} KV_PAIR_CONST;
 
#define ITER_CONST NS(SELF, iter_const)
 
typedef struct {
    SELF const* map;
    size_t index;
    KV_PAIR_CONST curr;
} ITER_CONST;
 
static KV_PAIR_CONST const* NS(ITER_CONST, next)(ITER_CONST* iter) {
    DEBUG_ASSERT(iter);
    if (iter->index < iter->map->capacity) {
        iter->curr = (KV_PAIR_CONST){.key = &iter->map->keys[iter->index].key,
                                     .value = &iter->map->values[iter->index]};
        iter->index++;
        while (iter->index < iter->map->capacity && !iter->map->keys[iter->index].present) {
            iter->index++;
        }
        return &iter->curr;
    }
    return NULL;
}
 
static bool NS(ITER_CONST, empty)(ITER_CONST const* iter) {
    DEBUG_ASSERT(iter);
    return iter->index >= iter->map->capacity;
}
 
static ITER_CONST NS(SELF, get_iter_const)(SELF const* self) {
    DEBUG_ASSERT(self);
    size_t first_index = 0;
    while (first_index < self->capacity && !self->keys[first_index].present) {
        first_index++;
    }
    return (ITER_CONST){
        .map = self,
        .index = first_index,
        .curr = (KV_PAIR_CONST){.key = NULL, .value = NULL},
    };
}
 
#undef ITER_CONST
#undef KV_PAIR_CONST
 
#undef KEY_ENTRY
 
#undef KEY
#undef KEY_HASH
#undef KEY_EQ
#undef KEY_DELETE
#undef KEY_CLONE
#undef VALUE
#undef VALUE_DELETE
#undef VALUE_CLONE
 
#include <derive-c/core/alloc/undef.h>
#include <derive-c/core/self/undef.h>