Object* LookupTable::store(STATE, Object* key, Object* val) {
unsigned int num_entries, num_bins, bin;
LookupTableBucket* entry;
LookupTableBucket* last = NULL;
num_entries = entries_->to_native();
num_bins = bins_->to_native();
if(max_density_p(num_entries, num_bins)) {
redistribute(state, num_bins <<= 1);
}
key_to_sym(key);
bin = find_bin(key_hash(key), num_bins);
entry = try_as<LookupTableBucket>(values_->at(state, bin));
while(entry) {
if(entry->key() == key) {
entry->value(state, val);
return val;
}
last = entry;
entry = try_as<LookupTableBucket>(entry->next());
}
if(last) {
last->next(state, LookupTableBucket::create(state, key, val));
} else {
values_->put(state, bin, LookupTableBucket::create(state, key, val));
}
entries(state, Fixnum::from(num_entries + 1));
return val;
}
Object* LookupTable::remove(STATE, Object* key, bool* removed) {
hashval bin;
LookupTableBucket* entry;
LookupTableBucket* last = NULL;
size_t num_entries = entries_->to_native();
size_t num_bins = bins_->to_native();
if(min_density_p(num_entries, num_bins) && (num_bins >> 1) >= LOOKUPTABLE_MIN_SIZE) {
redistribute(state, num_bins >>= 1);
}
key_to_sym(key);
bin = find_bin(key_hash(key), num_bins);
entry = try_as<LookupTableBucket>(values_->at(state, bin));
while(entry) {
if(entry->key() == key) {
Object *val = entry->value();
if(last) {
last->next(state, entry->next());
} else {
values_->put(state, bin, entry->next());
}
entries(state, Fixnum::from(entries_->to_native() - 1));
if(removed) *removed = true;
return val;
}
last = entry;
entry = try_as<LookupTableBucket>(entry->next());
}
return Qnil;
}
void LookupTable::redistribute(STATE, size_t size) {
size_t num = bins_->to_native();
Tuple* new_values = Tuple::create(state, size);
for(size_t i = 0; i < num; i++) {
LookupTableBucket* entry = try_as<LookupTableBucket>(values_->at(state, i));
while(entry) {
LookupTableBucket* link = try_as<LookupTableBucket>(entry->next());
entry->next(state, reinterpret_cast<LookupTableBucket *>(Qnil));
size_t bin = find_bin(key_hash(entry->key()), size);
LookupTableBucket* slot = try_as<LookupTableBucket>(new_values->at(state, bin));
if(slot) {
slot->append(state, entry);
} else {
new_values->put(state, bin, entry);
}
entry = link;
}
}
values(state, new_values);
bins(state, Fixnum::from(size));
}
LookupTableBucket* LookupTableBucket::create(STATE, Object *key, Object *value) {
LookupTableBucket *entry =
state->new_object<LookupTableBucket>(G(lookuptablebucket));
entry->key(state, key);
entry->value(state, value);
return entry;
}
void test_find_entry() {
Object* k = Fixnum::from(47);
tbl->store(state, k, cTrue);
LookupTableBucket* entry = tbl->find_entry(state, k);
TS_ASSERT(entry != cNil);
TS_ASSERT_EQUALS(entry->key(),k);
TS_ASSERT_EQUALS(entry->value(),cTrue);
entry = tbl->find_entry(state, Fixnum::from(40));
TS_ASSERT(entry == cNil);
}
LookupTableBucket* LookupTable::find_entry(STATE, Object* key) {
unsigned int bin;
key_to_sym(key);
bin = find_bin(key_hash(key), bins_->to_native());
LookupTableBucket *entry = try_as<LookupTableBucket>(values_->at(state, bin));
while(entry) {
if(entry->key() == key) {
return entry;
}
entry = try_as<LookupTableBucket>(entry->next());
}
return reinterpret_cast<LookupTableBucket *>(Qnil);
}
LookupTable* LookupTable::duplicate(STATE) {
size_t size, i;
LookupTable *dup;
size = bins_->to_native();
dup = LookupTable::create(state, size);
// Allow for subclassing.
dup->klass(state, class_object(state));
size_t num = entries_->to_native();
Array* entries = all_entries(state);
for(i = 0; i < num; i++) {
LookupTableBucket* entry = as<LookupTableBucket>(entries->get(state, i));
dup->store(state, entry->key(), entry->value());
}
return dup;
}
void test_store_handles_entries_in_same_bin() {
Object* k1 = Fixnum::from((4 << 4) | 15);
Object* k2 = Fixnum::from((10 << 4) | 15);
Object* k3 = Fixnum::from((11 << 4) | 15);
Object* v1 = cTrue;
Object* v2 = cFalse;
Object* v3 = cTrue;
tbl->store(state, k1, v1);
tbl->store(state, k2, v2);
tbl->store(state, k3, v3);
TS_ASSERT_EQUALS(as<Integer>(tbl->entries())->to_native(), 3);
LookupTableBucket* entry = tbl->find_entry(state, k1);
TS_ASSERT(entry != cNil);
TS_ASSERT(entry->next() != cNil);
TS_ASSERT_EQUALS(entry->next()->key(), k2);
entry = tbl->find_entry(state, k3);
TS_ASSERT(entry != cNil);
TS_ASSERT_EQUALS(entry->key(), k3);
}