313 lines
8.8 KiB
C
313 lines
8.8 KiB
C
|
/**
|
||
|
|
* @file CHash_impl.h
|
||
|
|
* @author Ambroz Bizjak <ambrop7@gmail.com>
|
||
|
|
*
|
||
|
|
* @section LICENSE
|
||
|
|
*
|
||
|
|
* Redistribution and use in source and binary forms, with or without
|
||
|
|
* modification, are permitted provided that the following conditions are met:
|
||
|
|
* 1. Redistributions of source code must retain the above copyright
|
||
|
|
* notice, this list of conditions and the following disclaimer.
|
||
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
||
|
|
* notice, this list of conditions and the following disclaimer in the
|
||
|
|
* documentation and/or other materials provided with the distribution.
|
||
|
|
* 3. Neither the name of the author nor the
|
||
|
|
* names of its contributors may be used to endorse or promote products
|
||
|
|
* derived from this software without specific prior written permission.
|
||
|
|
*
|
||
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||
|
|
* DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
||
|
|
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||
|
|
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||
|
|
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||
|
|
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "CHash_header.h"
|
||
|
|
|
||
|
|
static void CHash_assert_valid_entry (CHashArg arg, CHashRef entry)
|
||
|
|
{
|
||
|
|
ASSERT(entry.link != CHashNullLink())
|
||
|
|
ASSERT(entry.ptr == CHASH_PARAM_DEREF(arg, entry.link))
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashLink CHashNullLink (void)
|
||
|
|
{
|
||
|
|
return CHASH_PARAM_NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashRef CHashNullRef (void)
|
||
|
|
{
|
||
|
|
CHashRef entry = {NULL, CHashNullLink()};
|
||
|
|
return entry;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int CHashIsNullLink (CHashLink link)
|
||
|
|
{
|
||
|
|
return (link == CHashNullLink());
|
||
|
|
}
|
||
|
|
|
||
|
|
static int CHashIsNullRef (CHashRef ref)
|
||
|
|
{
|
||
|
|
return CHashIsNullLink(ref.link);
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashRef CHashDerefMayNull (CHashArg arg, CHashLink link)
|
||
|
|
{
|
||
|
|
if (link == CHashNullLink()) {
|
||
|
|
return CHashNullRef();
|
||
|
|
}
|
||
|
|
|
||
|
|
CHashRef entry = {CHASH_PARAM_DEREF(arg, link), link};
|
||
|
|
ASSERT(entry.ptr)
|
||
|
|
|
||
|
|
return entry;
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashRef CHashDerefNonNull (CHashArg arg, CHashLink link)
|
||
|
|
{
|
||
|
|
ASSERT(link != CHashNullLink())
|
||
|
|
|
||
|
|
CHashRef entry = {CHASH_PARAM_DEREF(arg, link), link};
|
||
|
|
ASSERT(entry.ptr)
|
||
|
|
|
||
|
|
return entry;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int CHash_Init (CHash *o, size_t num_buckets)
|
||
|
|
{
|
||
|
|
if (num_buckets == 0) {
|
||
|
|
num_buckets = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
o->num_buckets = num_buckets;
|
||
|
|
|
||
|
|
o->buckets = (CHashLink *)BAllocArray(o->num_buckets, sizeof(o->buckets[0]));
|
||
|
|
if (!o->buckets) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (size_t i = 0; i < o->num_buckets; i++) {
|
||
|
|
o->buckets[i] = CHashNullLink();
|
||
|
|
}
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void CHash_Free (CHash *o)
|
||
|
|
{
|
||
|
|
BFree(o->buckets);
|
||
|
|
}
|
||
|
|
|
||
|
|
static int CHash_Insert (CHash *o, CHashArg arg, CHashRef entry, CHashRef *out_existing)
|
||
|
|
{
|
||
|
|
CHash_assert_valid_entry(arg, entry);
|
||
|
|
|
||
|
|
size_t index = CHASH_PARAM_ENTRYHASH(arg, entry) % o->num_buckets;
|
||
|
|
|
||
|
|
CHashLink link = o->buckets[index];
|
||
|
|
while (link != CHashNullLink()) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
if (CHASH_PARAM_COMPARE_ENTRIES(arg, cur, entry)) {
|
||
|
|
if (out_existing) {
|
||
|
|
*out_existing = cur;
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
link = CHash_next(cur);
|
||
|
|
}
|
||
|
|
|
||
|
|
CHash_next(entry) = o->buckets[index];
|
||
|
|
o->buckets[index] = entry.link;
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void CHash_InsertMulti (CHash *o, CHashArg arg, CHashRef entry)
|
||
|
|
{
|
||
|
|
CHash_assert_valid_entry(arg, entry);
|
||
|
|
|
||
|
|
size_t index = CHASH_PARAM_ENTRYHASH(arg, entry) % o->num_buckets;
|
||
|
|
|
||
|
|
CHashRef prev = CHashNullRef();
|
||
|
|
CHashLink link = o->buckets[index];
|
||
|
|
while (link != CHashNullLink()) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
if (CHASH_PARAM_COMPARE_ENTRIES(arg, cur, entry)) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
prev = cur;
|
||
|
|
link = CHash_next(cur);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (link == CHashNullLink() || prev.link == CHashNullLink()) {
|
||
|
|
CHash_next(entry) = o->buckets[index];
|
||
|
|
o->buckets[index] = entry.link;
|
||
|
|
} else {
|
||
|
|
CHash_next(entry) = link;
|
||
|
|
CHash_next(prev) = entry.link;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static void CHash_Remove (CHash *o, CHashArg arg, CHashRef entry)
|
||
|
|
{
|
||
|
|
CHash_assert_valid_entry(arg, entry);
|
||
|
|
|
||
|
|
size_t index = CHASH_PARAM_ENTRYHASH(arg, entry) % o->num_buckets;
|
||
|
|
|
||
|
|
CHashRef prev = CHashNullRef();
|
||
|
|
CHashLink link = o->buckets[index];
|
||
|
|
while (link != entry.link) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
prev = cur;
|
||
|
|
link = CHash_next(cur);
|
||
|
|
ASSERT(link != CHashNullLink())
|
||
|
|
}
|
||
|
|
|
||
|
|
if (prev.link == CHashNullLink()) {
|
||
|
|
o->buckets[index] = CHash_next(entry);
|
||
|
|
} else {
|
||
|
|
CHash_next(prev) = CHash_next(entry);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashRef CHash_Lookup (const CHash *o, CHashArg arg, CHashKey key)
|
||
|
|
{
|
||
|
|
size_t hash = CHASH_PARAM_KEYHASH(arg, key);
|
||
|
|
size_t index = hash % o->num_buckets;
|
||
|
|
|
||
|
|
CHashLink link = o->buckets[index];
|
||
|
|
while (link != CHashNullLink()) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
#if CHASH_PARAM_ENTRYHASH_IS_CHEAP
|
||
|
|
if (CHASH_PARAM_ENTRYHASH(arg, cur) == hash && CHASH_PARAM_COMPARE_KEY_ENTRY(arg, key, cur)) {
|
||
|
|
#else
|
||
|
|
if (CHASH_PARAM_COMPARE_KEY_ENTRY(arg, key, cur)) {
|
||
|
|
#endif
|
||
|
|
return cur;
|
||
|
|
}
|
||
|
|
link = CHash_next(cur);
|
||
|
|
}
|
||
|
|
|
||
|
|
return CHashNullRef();
|
||
|
|
}
|
||
|
|
|
||
|
|
static CHashRef CHash_GetNextEqual (const CHash *o, CHashArg arg, CHashRef entry)
|
||
|
|
{
|
||
|
|
CHash_assert_valid_entry(arg, entry);
|
||
|
|
|
||
|
|
CHashLink next = CHash_next(entry);
|
||
|
|
|
||
|
|
if (next == CHashNullLink()) {
|
||
|
|
return CHashNullRef();
|
||
|
|
}
|
||
|
|
|
||
|
|
CHashRef next_ref = CHashDerefNonNull(arg, next);
|
||
|
|
if (!CHASH_PARAM_COMPARE_ENTRIES(arg, next_ref, entry)) {
|
||
|
|
return CHashNullRef();
|
||
|
|
}
|
||
|
|
|
||
|
|
return next_ref;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int CHash_MultiplyBuckets (CHash *o, CHashArg arg, int exp)
|
||
|
|
{
|
||
|
|
ASSERT(exp > 0)
|
||
|
|
|
||
|
|
size_t new_num_buckets = o->num_buckets;
|
||
|
|
while (exp-- > 0) {
|
||
|
|
if (new_num_buckets > SIZE_MAX / 2) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
new_num_buckets *= 2;
|
||
|
|
}
|
||
|
|
|
||
|
|
CHashLink *new_buckets = (CHashLink *)BReallocArray(o->buckets, new_num_buckets, sizeof(new_buckets[0]));
|
||
|
|
if (!new_buckets) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
o->buckets = new_buckets;
|
||
|
|
|
||
|
|
for (size_t i = o->num_buckets; i < new_num_buckets; i++) {
|
||
|
|
o->buckets[i] = CHashNullLink();
|
||
|
|
}
|
||
|
|
|
||
|
|
for (size_t i = 0; i < o->num_buckets; i++) {
|
||
|
|
CHashRef prev = CHashNullRef();
|
||
|
|
CHashLink link = o->buckets[i];
|
||
|
|
|
||
|
|
while (link != CHashNullLink()) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
link = CHash_next(cur);
|
||
|
|
|
||
|
|
size_t new_index = CHASH_PARAM_ENTRYHASH(arg, cur) % new_num_buckets;
|
||
|
|
if (new_index == i) {
|
||
|
|
prev = cur;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (CHashIsNullRef(prev)) {
|
||
|
|
o->buckets[i] = CHash_next(cur);
|
||
|
|
} else {
|
||
|
|
CHash_next(prev) = CHash_next(cur);
|
||
|
|
}
|
||
|
|
|
||
|
|
CHash_next(cur) = o->buckets[new_index];
|
||
|
|
o->buckets[new_index] = cur.link;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for (size_t i = o->num_buckets; i < new_num_buckets; i++) {
|
||
|
|
CHashLink new_bucket_link = CHashNullLink();
|
||
|
|
|
||
|
|
CHashLink link = o->buckets[i];
|
||
|
|
while (link != CHashNullLink()) {
|
||
|
|
CHashRef cur = CHashDerefNonNull(arg, link);
|
||
|
|
link = CHash_next(cur);
|
||
|
|
|
||
|
|
CHash_next(cur) = new_bucket_link;
|
||
|
|
new_bucket_link = cur.link;
|
||
|
|
}
|
||
|
|
|
||
|
|
o->buckets[i] = new_bucket_link;
|
||
|
|
}
|
||
|
|
|
||
|
|
o->num_buckets = new_num_buckets;
|
||
|
|
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void CHash_Verify (const CHash *o, CHashArg arg)
|
||
|
|
{
|
||
|
|
ASSERT_FORCE(o->num_buckets > 0)
|
||
|
|
ASSERT_FORCE(o->buckets)
|
||
|
|
|
||
|
|
for (size_t i = 0; i < o->num_buckets; i++) {
|
||
|
|
CHashRef cur = CHashDerefMayNull(arg, o->buckets[i]);
|
||
|
|
CHashRef same_first = cur;
|
||
|
|
|
||
|
|
while (!CHashIsNullRef(cur)) {
|
||
|
|
size_t index = CHASH_PARAM_ENTRYHASH(arg, cur) % o->num_buckets;
|
||
|
|
ASSERT_FORCE(index == i)
|
||
|
|
|
||
|
|
if (!CHASH_PARAM_COMPARE_ENTRIES(arg, cur, same_first)) {
|
||
|
|
same_first = cur;
|
||
|
|
}
|
||
|
|
|
||
|
|
CHashRef ccur = CHashDerefNonNull(arg, o->buckets[i]);
|
||
|
|
while (ccur.link != same_first.link) {
|
||
|
|
ASSERT_FORCE(!CHASH_PARAM_COMPARE_ENTRIES(arg, ccur, cur))
|
||
|
|
ccur = CHashDerefMayNull(arg, CHash_next(ccur));
|
||
|
|
}
|
||
|
|
|
||
|
|
cur = CHashDerefMayNull(arg, CHash_next(cur));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#include "CHash_footer.h"
|