349 lines
12 KiB
C++
349 lines
12 KiB
C++
#pragma mark
|
|
|
|
#include <cstring>
|
|
#include <functional>
|
|
|
|
#include "utils/crtp.hpp"
|
|
#include "utils/likely.hpp"
|
|
#include "utils/option_ptr.hpp"
|
|
#include "data_structures/map/rh_common.hpp"
|
|
|
|
// HashMultiMap with RobinHood collision resolution policy.
|
|
// Single threaded.
|
|
// Entrys are POINTERS alligned to 8B.
|
|
// Entrys must know thers key.
|
|
// D must have method K& get_key()
|
|
// K must be comparable with ==.
|
|
// HashMap behaves as if it isn't owner of entrys.
|
|
template <class K, class D, size_t init_size_pow2 = 2>
|
|
class RhHashMultiMap : public RhBase<K, D, init_size_pow2>
|
|
{
|
|
typedef RhBase<K, D, init_size_pow2> base;
|
|
using base::array;
|
|
using base::index;
|
|
using base::capacity;
|
|
using base::count;
|
|
using typename base::Combined;
|
|
using base::before_index;
|
|
using base::create_it;
|
|
|
|
void increase_size()
|
|
{
|
|
size_t old_size = capacity;
|
|
auto a = array;
|
|
if (base::increase_size()) {
|
|
for (int i = 0; i < old_size; i++) {
|
|
if (a[i].valid()) {
|
|
add(a[i].ptr());
|
|
}
|
|
}
|
|
}
|
|
|
|
free(a);
|
|
}
|
|
|
|
public:
|
|
using base::RhBase;
|
|
using base::end;
|
|
using typename base::ConstIterator;
|
|
using typename base::Iterator;
|
|
|
|
bool contains(const K &key) { return find(key) != end(); }
|
|
|
|
Iterator find(const K &key_in)
|
|
{
|
|
if (count > 0) {
|
|
auto key = std::ref(key_in);
|
|
size_t mask = this->mask();
|
|
size_t now = index(key, mask);
|
|
size_t off = 0;
|
|
size_t border = 8 <= capacity ? 8 : capacity;
|
|
Combined other = array[now];
|
|
while (other.valid() && off < border) {
|
|
auto other_off = other.off();
|
|
if (other_off == off && key == other.ptr()->get_key()) {
|
|
return create_it(now);
|
|
|
|
} else if (other_off < off) { // Other is rich
|
|
break;
|
|
|
|
} else { // Else other has equal or greater off, so he is poor.
|
|
if (UNLIKELY(skip(now, other, other_off, mask))) {
|
|
break;
|
|
}
|
|
off++;
|
|
}
|
|
}
|
|
}
|
|
|
|
return end();
|
|
}
|
|
|
|
// Inserts element.
|
|
void add(D *data) { add(data->get_key(), data); }
|
|
|
|
// Inserts element with the given key.
|
|
void add(const K &key_in, D *data)
|
|
{
|
|
assert(key_in == data->get_key());
|
|
|
|
if (count < capacity) {
|
|
auto key = std::ref(key_in);
|
|
size_t mask = this->mask();
|
|
size_t now = index(key, mask);
|
|
size_t start = now;
|
|
size_t off = 0;
|
|
size_t border = 8 <= capacity ? 8 : capacity;
|
|
|
|
Combined other = array[now];
|
|
while (off < border) {
|
|
if (other.valid()) {
|
|
const size_t other_off = other.off();
|
|
bool multi = false;
|
|
if (other_off == off && other.ptr()->get_key() == key) {
|
|
// Found the same
|
|
do {
|
|
now = (now + 1) & mask;
|
|
other = array[now];
|
|
if (!other.valid()) {
|
|
set(now, data, off);
|
|
return;
|
|
}
|
|
} while (other.equal(key, off));
|
|
multi = true;
|
|
} else if (other_off > off ||
|
|
other_poor(other, mask, start,
|
|
now)) { // Else other has equal or
|
|
// greater off, so he is poor.
|
|
skip(now, other, other_off, mask); // TRUE IS IMPOSSIBLE
|
|
off++;
|
|
continue;
|
|
}
|
|
|
|
array[now] = Combined(data, off);
|
|
auto start_insert = now;
|
|
while (is_off_adjusted(other, mask, start_insert, now,
|
|
multi) ||
|
|
other.increment_off()) {
|
|
now = (now + 1) & mask;
|
|
auto tmp = array[now];
|
|
array[now] = other;
|
|
other = tmp;
|
|
if (!other.valid()) {
|
|
count++;
|
|
return;
|
|
}
|
|
}
|
|
data = other.ptr();
|
|
break; // Cant insert removed element
|
|
} else {
|
|
set(now, data, off);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
increase_size();
|
|
add(data);
|
|
}
|
|
|
|
// Removes element equal by key and value. Returns true if it existed.
|
|
bool remove(D *data)
|
|
{
|
|
if (count > 0) {
|
|
auto key = std::ref(data->get_key());
|
|
size_t mask = this->mask();
|
|
size_t now = index(key, mask);
|
|
size_t off = 0;
|
|
size_t border = 8 <= capacity ? 8 : capacity;
|
|
Combined other = array[now];
|
|
while (other.valid() && off < border) {
|
|
const size_t other_off = other.off();
|
|
if (other_off == off && key == other.ptr()->get_key()) {
|
|
auto founded = capacity;
|
|
size_t started = now;
|
|
bool multi = false;
|
|
do {
|
|
if (other.ptr() == data) {
|
|
founded = now;
|
|
}
|
|
now = (now + 1) & mask;
|
|
other = array[now];
|
|
if (!other.valid() || UNLIKELY(started == now)) {
|
|
// Reason is possibility of map full of same values.
|
|
break;
|
|
}
|
|
} while (other.equal(key, off) && (multi = true));
|
|
// multi = true is correct
|
|
if (founded == capacity) {
|
|
return false;
|
|
}
|
|
|
|
auto bef = before_index(now, mask);
|
|
array[founded] = array[bef];
|
|
|
|
auto start_rem = bef;
|
|
while (other.valid() &&
|
|
(is_off_adjusted_rem(other, mask, start_rem, bef,
|
|
now, multi) ||
|
|
other.decrement_off())) {
|
|
array[bef] = other;
|
|
bef = now;
|
|
now = (now + 1) & mask;
|
|
other = array[now];
|
|
}
|
|
|
|
array[bef] = Combined();
|
|
count--;
|
|
return true;
|
|
|
|
} else if (other_off < off) { // Other is rich
|
|
break;
|
|
|
|
} else { // Else other has equal or greater off, so he is poor.
|
|
if (UNLIKELY(skip(now, other, other_off, mask))) {
|
|
break;
|
|
}
|
|
off++;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private:
|
|
// Skips same key valus as other. true if whole map is full of same key
|
|
// values.
|
|
bool skip(size_t &now, Combined &other, size_t other_off, size_t mask)
|
|
{
|
|
auto other_key = other.ptr()->get_key();
|
|
size_t start = now;
|
|
do {
|
|
now = (now + 1) & mask;
|
|
other = array[now];
|
|
if (UNLIKELY(start == now)) { // Reason is possibility of map
|
|
// full of same values.
|
|
return true;
|
|
}
|
|
} while (other.valid() && other.equal(other_key, other_off));
|
|
return false;
|
|
}
|
|
|
|
void set(size_t now, D *data, size_t off)
|
|
{
|
|
array[now] = Combined(data, off);
|
|
count++;
|
|
}
|
|
|
|
// True if no adjusment is needed, false otherwise.
|
|
bool is_off_adjusted(Combined &com, size_t mask, size_t start, size_t now,
|
|
bool multi)
|
|
{
|
|
if (com.off() == 0) { // Must be adjusted
|
|
return false;
|
|
}
|
|
size_t cin = index(com.ptr()->get_key(), mask);
|
|
if (outside(start, now, cin)) { // Outside [start,now] interval
|
|
return multi;
|
|
}
|
|
auto a = array[cin];
|
|
auto b = array[(cin + 1) & mask];
|
|
return a == b;
|
|
// Check if different key has eneterd in to
|
|
// range of other.
|
|
}
|
|
|
|
bool other_poor(Combined other, size_t mask, size_t start, size_t now)
|
|
{
|
|
// If other index is smaller then he is poorer.
|
|
return outside_left_weak(start, now,
|
|
index(other.ptr()->get_key(), mask));
|
|
}
|
|
|
|
// True if no adjusment is needed, false otherwise.
|
|
bool is_off_adjusted_rem(Combined &com, size_t mask, size_t start,
|
|
size_t bef, size_t now, bool multi)
|
|
{
|
|
if (com.off() == 0) { // Must be adjusted
|
|
return false;
|
|
}
|
|
size_t cin = index(com.ptr()->get_key(), mask);
|
|
if (cin == bef) {
|
|
return false;
|
|
}
|
|
if (outside(start, now, cin)) {
|
|
return multi;
|
|
}
|
|
auto a = array[cin];
|
|
auto b = array[before_index(cin, mask)];
|
|
return b.valid() && a == b;
|
|
// Check if different key has eneterd in to
|
|
// range of other.
|
|
}
|
|
|
|
// True if p is uutside [start,end] interval
|
|
bool outside(size_t start, size_t end, size_t p)
|
|
{
|
|
return (start <= end && (p < start || p > end)) ||
|
|
(end < start && p < start && p > end);
|
|
}
|
|
|
|
// True if p is outside <start,end] interval
|
|
bool outside_left_weak(size_t start, size_t end, size_t p)
|
|
{
|
|
return (start <= end && (p <= start || p > end)) ||
|
|
(end < start && p <= start && p > end);
|
|
}
|
|
};
|
|
|
|
// Unnecessary
|
|
// // Removes element. Returns removed element if it existed. It doesn't
|
|
// // specify which element from same key group will be removed.
|
|
// OptionPtr<D> remove(const K &key_in)
|
|
// {
|
|
//
|
|
// if (count > 0) {
|
|
// auto key = std::ref(key_in);
|
|
// size_t mask = this->mask();
|
|
// size_t now = index(key, mask);
|
|
// size_t off = 0;
|
|
// size_t checked = 0;
|
|
// size_t border = 8 <= capacity ? 8 : capacity;
|
|
// Combined other = array[now];
|
|
// while (other.valid() && off < border) {
|
|
// auto other_off = other.off();
|
|
// bool multi = false;
|
|
// if (other_off == off && key == other.ptr()->get_key()) {
|
|
// do {
|
|
// now = (now + 1) & mask;
|
|
// other = array[now];
|
|
// if (!other.valid()) {
|
|
// break;
|
|
// }
|
|
// other_off = other.off();
|
|
// } while (other_off == off &&
|
|
// other.ptr()->get_key() == key &&
|
|
// (multi = true)); // multi = true is correct
|
|
//
|
|
// auto bef = before_index(now, mask);
|
|
// auto ret = OptionPtr<D>(array[bef].ptr());
|
|
//
|
|
// move_before(now, bef, other, mask, multi);
|
|
// return ret;
|
|
//
|
|
// } else if (other_off < off) { // Other is rich
|
|
// break;
|
|
//
|
|
// } else { // Else other has equal or greater off, so he is
|
|
// poor.
|
|
// if (UNLIKELY(skip(now, other, other_off, mask))) {
|
|
// break;
|
|
// }
|
|
// off++;
|
|
// }
|
|
// }
|
|
// }
|
|
//
|
|
// return OptionPtr<D>();
|
|
// }
|