stringstore.cpp

struct Stringstore {
    Hashmap<s64> string_map;
    Array_dyn<Offset<u8>> strings;
    Array_dyn<u8> string_data;
    bool c_str = false;
    s64 size = 0;
};

void stringstore_free(Stringstore* store) {
    hashmap_free(&store->string_map);
    array_free(&store->strings);
    array_free(&store->string_data);
}

s64 stringstore_get_id(Stringstore* store, Array<u8> str, bool nocreate=false) {
    u64 hash_base = hash_str(str);
    hash_base += hash_base == 0;
    
    for (u64 adj = 0;; ++adj) {
        auto* ptr = hashmap_getcreate(&store->string_map, hash_base + adj, (s64)-1);
        if (*ptr == -1) {
            if (nocreate) return -1;
            *ptr = store->strings.size;
            array_push(&store->strings, array_append(&store->string_data, str));
            if (store->c_str) array_push(&store->string_data, 0);
            ++store->size;
            return *ptr;
        } else {
            auto str_stored = array_suboffset(store->string_data, store->strings[*ptr]);
            if (array_equal(str_stored, str)) return *ptr;
        }
    }
}

// Deprecated! This used to return 0 on non-existence. Instead call stringstore_get_id(..., nocreate=true) and check for -1.
//u64 stringstore_get_id_if_exists(Stringstore* store, Array<u8> str);

Array<u8> stringstore_get_str(Stringstore* store, s64 id) {
    return array_suboffset(store->string_data, store->strings[id]);
}
char* stringstore_get_cstr(Stringstore* store, s64 id) {
    assert(store->c_str);
    return (char*)array_suboffset(store->string_data, store->strings[id]).data;
}