package it.unimi.dsi.fastutil.longs;
import it.unimi.dsi.fastutil.Hash;
import it.unimi.dsi.fastutil.HashCommon;
import static it.unimi.dsi.fastutil.HashCommon.arraySize;
import static it.unimi.dsi.fastutil.HashCommon.maxFill;
import java.util.Map;
import java.util.Arrays;
import java.util.NoSuchElementException;
import java.util.function.Consumer;
import it.unimi.dsi.fastutil.objects.ReferenceCollection;
import it.unimi.dsi.fastutil.objects.AbstractReferenceCollection;
import it.unimi.dsi.fastutil.objects.AbstractObjectSet;
import it.unimi.dsi.fastutil.objects.ObjectIterator;
public class Long2ReferenceOpenHashMap<V> extends AbstractLong2ReferenceMap<V>
implements
java.io.Serializable,
Cloneable,
Hash {
private static final long serialVersionUID = 0L;
private static final boolean ASSERTS = false;
protected transient long[] key;
protected transient V[] value;
protected transient int mask;
protected transient boolean containsNullKey;
protected transient int n;
protected transient int maxFill;
protected final transient int minN;
protected int size;
protected final float f;
protected transient FastEntrySet<V> entries;
protected transient LongSet keys;
protected transient ReferenceCollection<V> values;
@SuppressWarnings("unchecked")
public Long2ReferenceOpenHashMap(final int expected, final float f) {
if (f <= 0 || f > 1)
throw new IllegalArgumentException("Load factor must be greater than 0 and smaller than or equal to 1");
if (expected < 0)
throw new IllegalArgumentException("The expected number of elements must be nonnegative");
this.f = f;
minN = n = arraySize(expected, f);
mask = n - 1;
maxFill = maxFill(n, f);
key = new long[n + 1];
value = (V[]) new Object[n + 1];
}
public Long2ReferenceOpenHashMap(final int expected) {
this(expected, DEFAULT_LOAD_FACTOR);
}
public Long2ReferenceOpenHashMap() {
this(DEFAULT_INITIAL_SIZE, DEFAULT_LOAD_FACTOR);
}
public Long2ReferenceOpenHashMap(final Map<? extends Long, ? extends V> m, final float f) {
this(m.size(), f);
putAll(m);
}
public Long2ReferenceOpenHashMap(final Map<? extends Long, ? extends V> m) {
this(m, DEFAULT_LOAD_FACTOR);
}
public Long2ReferenceOpenHashMap(final Long2ReferenceMap<V> m, final float f) {
this(m.size(), f);
putAll(m);
}
public Long2ReferenceOpenHashMap(final Long2ReferenceMap<V> m) {
this(m, DEFAULT_LOAD_FACTOR);
}
public Long2ReferenceOpenHashMap(final long[] k, final V[] v, final float f) {
this(k.length, f);
if (k.length != v.length)
throw new IllegalArgumentException(
"The key array and the value array have different lengths (" + k.length + " and " + v.length + ")");
for (int i = 0; i < k.length; i++)
this.put(k[i], v[i]);
}
public Long2ReferenceOpenHashMap(final long[] k, final V[] v) {
this(k, v, DEFAULT_LOAD_FACTOR);
}
private int realSize() {
return containsNullKey ? size - 1 : size;
}
private void ensureCapacity(final int capacity) {
final int needed = arraySize(capacity, f);
if (needed > n)
rehash(needed);
}
private void tryCapacity(final long capacity) {
final int needed = (int) Math.min(1 << 30,
Math.max(2, HashCommon.nextPowerOfTwo((long) Math.ceil(capacity / f))));
if (needed > n)
rehash(needed);
}
private V removeEntry(final int pos) {
final V oldValue = value[pos];
value[pos] = null;
size--;
shiftKeys(pos);
if (n > minN && size < maxFill / 4 && n > DEFAULT_INITIAL_SIZE)
rehash(n / 2);
return oldValue;
}
private V removeNullEntry() {
containsNullKey = false;
final V oldValue = value[n];
value[n] = null;
size--;
if (n > minN && size < maxFill / 4 && n > DEFAULT_INITIAL_SIZE)
rehash(n / 2);
return oldValue;
}
@Override
public void putAll(Map<? extends Long, ? extends V> m) {
if (f <= .5)
ensureCapacity(m.size());
else
tryCapacity(size() + m.size());
super.putAll(m);
}
private int find(final long k) {
if (((k) == (0)))
return containsNullKey ? n : -(n + 1);
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return -(pos + 1);
if (((k) == (curr)))
return pos;
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return -(pos + 1);
if (((k) == (curr)))
return pos;
}
}
private void insert(final int pos, final long k, final V v) {
if (pos == n)
containsNullKey = true;
key[pos] = k;
value[pos] = v;
if (size++ >= maxFill)
rehash(arraySize(size + 1, f));
if (ASSERTS)
checkTable();
}
@Override
public V put(final long k, final V v) {
final int pos = find(k);
if (pos < 0) {
insert(-pos - 1, k, v);
return defRetValue;
}
final V oldValue = value[pos];
value[pos] = v;
return oldValue;
}
protected final void shiftKeys(int pos) {
int last, slot;
long curr;
final long[] key = this.key;
for (;;) {
pos = ((last = pos) + 1) & mask;
for (;;) {
if (((curr = key[pos]) == (0))) {
key[last] = (0);
value[last] = null;
return;
}
slot = (int) it.unimi.dsi.fastutil.HashCommon.mix((curr)) & mask;
if (last <= pos ? last >= slot || slot > pos : last >= slot && slot > pos)
break;
pos = (pos + 1) & mask;
}
key[last] = curr;
value[last] = value[pos];
}
}
@Override
public V remove(final long k) {
if (((k) == (0))) {
if (containsNullKey)
return removeNullEntry();
return defRetValue;
}
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return defRetValue;
if (((k) == (curr)))
return removeEntry(pos);
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return defRetValue;
if (((k) == (curr)))
return removeEntry(pos);
}
}
@Override
public V get(final long k) {
if (((k) == (0)))
return containsNullKey ? value[n] : defRetValue;
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return defRetValue;
if (((k) == (curr)))
return value[pos];
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return defRetValue;
if (((k) == (curr)))
return value[pos];
}
}
@Override
public boolean containsKey(final long k) {
if (((k) == (0)))
return containsNullKey;
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return false;
if (((k) == (curr)))
return true;
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return false;
if (((k) == (curr)))
return true;
}
}
@Override
public boolean containsValue(final Object v) {
final V value[] = this.value;
final long key[] = this.key;
if (containsNullKey && ((value[n]) == (v)))
return true;
for (int i = n; i-- != 0;)
if (!((key[i]) == (0)) && ((value[i]) == (v)))
return true;
return false;
}
@Override
public V getOrDefault(final long k, final V defaultValue) {
if (((k) == (0)))
return containsNullKey ? value[n] : defaultValue;
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return defaultValue;
if (((k) == (curr)))
return value[pos];
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return defaultValue;
if (((k) == (curr)))
return value[pos];
}
}
@Override
public V putIfAbsent(final long k, final V v) {
final int pos = find(k);
if (pos >= 0)
return value[pos];
insert(-pos - 1, k, v);
return defRetValue;
}
@Override
public boolean remove(final long k, final Object v) {
if (((k) == (0))) {
if (containsNullKey && ((v) == (value[n]))) {
removeNullEntry();
return true;
}
return false;
}
long curr;
final long[] key = this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return false;
if (((k) == (curr)) && ((v) == (value[pos]))) {
removeEntry(pos);
return true;
}
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return false;
if (((k) == (curr)) && ((v) == (value[pos]))) {
removeEntry(pos);
return true;
}
}
}
@Override
public boolean replace(final long k, final V oldValue, final V v) {
final int pos = find(k);
if (pos < 0 || !((oldValue) == (value[pos])))
return false;
value[pos] = v;
return true;
}
@Override
public V replace(final long k, final V v) {
final int pos = find(k);
if (pos < 0)
return defRetValue;
final V oldValue = value[pos];
value[pos] = v;
return oldValue;
}
@Override
public V computeIfAbsent(final long k, final java.util.function.LongFunction<? extends V> mappingFunction) {
java.util.Objects.requireNonNull(mappingFunction);
final int pos = find(k);
if (pos >= 0)
return value[pos];
final V newValue = mappingFunction.apply(k);
insert(-pos - 1, k, newValue);
return newValue;
}
@Override
public V computeIfPresent(final long k,
final java.util.function.BiFunction<? super Long, ? super V, ? extends V> remappingFunction) {
java.util.Objects.requireNonNull(remappingFunction);
final int pos = find(k);
if (pos < 0)
return defRetValue;
final V newValue = remappingFunction.apply(Long.valueOf(k), (value[pos]));
if (newValue == null) {
if (((k) == (0)))
removeNullEntry();
else
removeEntry(pos);
return defRetValue;
}
return value[pos] = (newValue);
}
@Override
public V compute(final long k,
final java.util.function.BiFunction<? super Long, ? super V, ? extends V> remappingFunction) {
java.util.Objects.requireNonNull(remappingFunction);
final int pos = find(k);
final V newValue = remappingFunction.apply(Long.valueOf(k), pos >= 0 ? (value[pos]) : null);
if (newValue == null) {
if (pos >= 0) {
if (((k) == (0)))
removeNullEntry();
else
removeEntry(pos);
}
return defRetValue;
}
V newVal = (newValue);
if (pos < 0) {
insert(-pos - 1, k, newVal);
return newVal;
}
return value[pos] = newVal;
}
@Override
public V merge(final long k, final V v,
final java.util.function.BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
java.util.Objects.requireNonNull(remappingFunction);
final int pos = find(k);
if (pos < 0 || value[pos] == null) {
if (v == null)
return defRetValue;
insert(-pos - 1, k, v);
return v;
}
final V newValue = remappingFunction.apply((value[pos]), (v));
if (newValue == null) {
if (((k) == (0)))
removeNullEntry();
else
removeEntry(pos);
return defRetValue;
}
return value[pos] = (newValue);
}
@Override
public void clear() {
if (size == 0)
return;
size = 0;
containsNullKey = false;
Arrays.fill(key, (0));
Arrays.fill(value, null);
}
@Override
public int size() {
return size;
}
@Override
public boolean isEmpty() {
return size == 0;
}
final class MapEntry implements Long2ReferenceMap.Entry<V>, Map.Entry<Long, V> {
int index;
MapEntry(final int index) {
this.index = index;
}
MapEntry() {
}
@Override
public long getLongKey() {
return key[index];
}
@Override
public V getValue() {
return value[index];
}
@Override
public V setValue(final V v) {
final V oldValue = value[index];
value[index] = v;
return oldValue;
}
@Deprecated
@Override
public Long getKey() {
return Long.valueOf(key[index]);
}
@SuppressWarnings("unchecked")
@Override
public boolean equals(final Object o) {
if (!(o instanceof Map.Entry))
return false;
Map.Entry<Long, V> e = (Map.Entry<Long, V>) o;
return ((key[index]) == ((e.getKey()).longValue())) && ((value[index]) == ((e.getValue())));
}
@Override
public int hashCode() {
return it.unimi.dsi.fastutil.HashCommon.long2int(key[index])
^ ((value[index]) == null ? 0 : System.identityHashCode(value[index]));
}
@Override
public String toString() {
return key[index] + "=>" + value[index];
}
}
private class MapIterator {
int pos = n;
int last = -1;
int c = size;
boolean mustReturnNullKey = Long2ReferenceOpenHashMap.this.containsNullKey;
LongArrayList wrapped;
public boolean hasNext() {
return c != 0;
}
public int nextEntry() {
if (!hasNext())
throw new NoSuchElementException();
c--;
if (mustReturnNullKey) {
mustReturnNullKey = false;
return last = n;
}
final long key[] = Long2ReferenceOpenHashMap.this.key;
for (;;) {
if (--pos < 0) {
last = Integer.MIN_VALUE;
final long k = wrapped.getLong(-pos - 1);
int p = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask;
while (!((k) == (key[p])))
p = (p + 1) & mask;
return p;
}
if (!((key[pos]) == (0)))
return last = pos;
}
}
private void shiftKeys(int pos) {
int last, slot;
long curr;
final long[] key = Long2ReferenceOpenHashMap.this.key;
for (;;) {
pos = ((last = pos) + 1) & mask;
for (;;) {
if (((curr = key[pos]) == (0))) {
key[last] = (0);
value[last] = null;
return;
}
slot = (int) it.unimi.dsi.fastutil.HashCommon.mix((curr)) & mask;
if (last <= pos ? last >= slot || slot > pos : last >= slot && slot > pos)
break;
pos = (pos + 1) & mask;
}
if (pos < last) {
if (wrapped == null)
wrapped = new LongArrayList(2);
wrapped.add(key[pos]);
}
key[last] = curr;
value[last] = value[pos];
}
}
public void remove() {
if (last == -1)
throw new IllegalStateException();
if (last == n) {
containsNullKey = false;
value[n] = null;
} else if (pos >= 0)
shiftKeys(last);
else {
Long2ReferenceOpenHashMap.this.remove(wrapped.getLong(-pos - 1));
last = -1;
return;
}
size--;
last = -1;
if (ASSERTS)
checkTable();
}
public int skip(final int n) {
int i = n;
while (i-- != 0 && hasNext())
nextEntry();
return n - i - 1;
}
}
private class EntryIterator extends MapIterator implements ObjectIterator<Long2ReferenceMap.Entry<V>> {
private MapEntry entry;
@Override
public MapEntry next() {
return entry = new MapEntry(nextEntry());
}
@Override
public void remove() {
super.remove();
entry.index = -1;
}
}
private class FastEntryIterator extends MapIterator implements ObjectIterator<Long2ReferenceMap.Entry<V>> {
private final MapEntry entry = new MapEntry();
@Override
public MapEntry next() {
entry.index = nextEntry();
return entry;
}
}
private final class MapEntrySet extends AbstractObjectSet<Long2ReferenceMap.Entry<V>> implements FastEntrySet<V> {
@Override
public ObjectIterator<Long2ReferenceMap.Entry<V>> iterator() {
return new EntryIterator();
}
@Override
public ObjectIterator<Long2ReferenceMap.Entry<V>> fastIterator() {
return new FastEntryIterator();
}
@Override
@SuppressWarnings("unchecked")
public boolean contains(final Object o) {
if (!(o instanceof Map.Entry))
return false;
final Map.Entry<?, ?> e = (Map.Entry<?, ?>) o;
if (e.getKey() == null || !(e.getKey() instanceof Long))
return false;
final long k = ((Long) (e.getKey())).longValue();
final V v = ((V) e.getValue());
if (((k) == (0)))
return Long2ReferenceOpenHashMap.this.containsNullKey && ((value[n]) == (v));
long curr;
final long[] key = Long2ReferenceOpenHashMap.this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return false;
if (((k) == (curr)))
return ((value[pos]) == (v));
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return false;
if (((k) == (curr)))
return ((value[pos]) == (v));
}
}
@Override
@SuppressWarnings("unchecked")
public boolean remove(final Object o) {
if (!(o instanceof Map.Entry))
return false;
final Map.Entry<?, ?> e = (Map.Entry<?, ?>) o;
if (e.getKey() == null || !(e.getKey() instanceof Long))
return false;
final long k = ((Long) (e.getKey())).longValue();
final V v = ((V) e.getValue());
if (((k) == (0))) {
if (containsNullKey && ((value[n]) == (v))) {
removeNullEntry();
return true;
}
return false;
}
long curr;
final long[] key = Long2ReferenceOpenHashMap.this.key;
int pos;
if (((curr = key[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask]) == (0)))
return false;
if (((curr) == (k))) {
if (((value[pos]) == (v))) {
removeEntry(pos);
return true;
}
return false;
}
while (true) {
if (((curr = key[pos = (pos + 1) & mask]) == (0)))
return false;
if (((curr) == (k))) {
if (((value[pos]) == (v))) {
removeEntry(pos);
return true;
}
}
}
}
@Override
public int size() {
return size;
}
@Override
public void clear() {
Long2ReferenceOpenHashMap.this.clear();
}
@Override
public void forEach(final Consumer<? super Long2ReferenceMap.Entry<V>> consumer) {
if (containsNullKey)
consumer.accept(new AbstractLong2ReferenceMap.BasicEntry<V>(key[n], value[n]));
for (int pos = n; pos-- != 0;)
if (!((key[pos]) == (0)))
consumer.accept(new AbstractLong2ReferenceMap.BasicEntry<V>(key[pos], value[pos]));
}
@Override
public void fastForEach(final Consumer<? super Long2ReferenceMap.Entry<V>> consumer) {
final AbstractLong2ReferenceMap.BasicEntry<V> entry = new AbstractLong2ReferenceMap.BasicEntry<>();
if (containsNullKey) {
entry.key = key[n];
entry.value = value[n];
consumer.accept(entry);
}
for (int pos = n; pos-- != 0;)
if (!((key[pos]) == (0))) {
entry.key = key[pos];
entry.value = value[pos];
consumer.accept(entry);
}
}
}
@Override
public FastEntrySet<V> long2ReferenceEntrySet() {
if (entries == null)
entries = new MapEntrySet();
return entries;
}
private final class KeyIterator extends MapIterator implements LongIterator {
public KeyIterator() {
super();
}
@Override
public long nextLong() {
return key[nextEntry()];
}
}
private final class KeySet extends AbstractLongSet {
@Override
public LongIterator iterator() {
return new KeyIterator();
}
@Override
public void forEach(final java.util.function.LongConsumer consumer) {
if (containsNullKey)
consumer.accept(key[n]);
for (int pos = n; pos-- != 0;) {
final long k = key[pos];
if (!((k) == (0)))
consumer.accept(k);
}
}
@Override
public int size() {
return size;
}
@Override
public boolean contains(long k) {
return containsKey(k);
}
@Override
public boolean remove(long k) {
final int oldSize = size;
Long2ReferenceOpenHashMap.this.remove(k);
return size != oldSize;
}
@Override
public void clear() {
Long2ReferenceOpenHashMap.this.clear();
}
}
@Override
public LongSet keySet() {
if (keys == null)
keys = new KeySet();
return keys;
}
private final class ValueIterator extends MapIterator implements ObjectIterator<V> {
public ValueIterator() {
super();
}
@Override
public V next() {
return value[nextEntry()];
}
}
@Override
public ReferenceCollection<V> values() {
if (values == null)
values = new AbstractReferenceCollection<V>() {
@Override
public ObjectIterator<V> iterator() {
return new ValueIterator();
}
@Override
public int size() {
return size;
}
@Override
public boolean contains(Object v) {
return containsValue(v);
}
@Override
public void clear() {
Long2ReferenceOpenHashMap.this.clear();
}
@Override
public void forEach(final Consumer<? super V> consumer) {
if (containsNullKey)
consumer.accept(value[n]);
for (int pos = n; pos-- != 0;)
if (!((key[pos]) == (0)))
consumer.accept(value[pos]);
}
};
return values;
}
public boolean trim() {
return trim(size);
}
public boolean trim(final int n) {
final int l = HashCommon.nextPowerOfTwo((int) Math.ceil(n / f));
if (l >= this.n || size > maxFill(l, f))
return true;
try {
rehash(l);
} catch (OutOfMemoryError cantDoIt) {
return false;
}
return true;
}
@SuppressWarnings("unchecked")
protected void rehash(final int newN) {
final long key[] = this.key;
final V value[] = this.value;
final int mask = newN - 1;
final long newKey[] = new long[newN + 1];
final V newValue[] = (V[]) new Object[newN + 1];
int i = n, pos;
for (int j = realSize(); j-- != 0;) {
while (((key[--i]) == (0)));
if (!((newKey[pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((key[i])) & mask]) == (0)))
while (!((newKey[pos = (pos + 1) & mask]) == (0)));
newKey[pos] = key[i];
newValue[pos] = value[i];
}
newValue[newN] = value[n];
n = newN;
this.mask = mask;
maxFill = maxFill(n, f);
this.key = newKey;
this.value = newValue;
}
@Override
@SuppressWarnings("unchecked")
public Long2ReferenceOpenHashMap<V> clone() {
Long2ReferenceOpenHashMap<V> c;
try {
c = (Long2ReferenceOpenHashMap<V>) super.clone();
} catch (CloneNotSupportedException cantHappen) {
throw new InternalError();
}
c.keys = null;
c.values = null;
c.entries = null;
c.containsNullKey = containsNullKey;
c.key = key.clone();
c.value = value.clone();
return c;
}
@Override
public int hashCode() {
int h = 0;
for (int j = realSize(), i = 0, t = 0; j-- != 0;) {
while (((key[i]) == (0)))
i++;
t = it.unimi.dsi.fastutil.HashCommon.long2int(key[i]);
if (this != value[i])
t ^= ((value[i]) == null ? 0 : System.identityHashCode(value[i]));
h += t;
i++;
}
if (containsNullKey)
h += ((value[n]) == null ? 0 : System.identityHashCode(value[n]));
return h;
}
private void writeObject(java.io.ObjectOutputStream s) throws java.io.IOException {
final long key[] = this.key;
final V value[] = this.value;
final MapIterator i = new MapIterator();
s.defaultWriteObject();
for (int j = size, e; j-- != 0;) {
e = i.nextEntry();
s.writeLong(key[e]);
s.writeObject(value[e]);
}
}
@SuppressWarnings("unchecked")
private void readObject(java.io.ObjectInputStream s) throws java.io.IOException, ClassNotFoundException {
s.defaultReadObject();
n = arraySize(size, f);
maxFill = maxFill(n, f);
mask = n - 1;
final long key[] = this.key = new long[n + 1];
final V value[] = this.value = (V[]) new Object[n + 1];
long k;
V v;
for (int i = size, pos; i-- != 0;) {
k = s.readLong();
v = (V) s.readObject();
if (((k) == (0))) {
pos = n;
containsNullKey = true;
} else {
pos = (int) it.unimi.dsi.fastutil.HashCommon.mix((k)) & mask;
while (!((key[pos]) == (0)))
pos = (pos + 1) & mask;
}
key[pos] = k;
value[pos] = v;
}
if (ASSERTS)
checkTable();
}
private void checkTable() {
}
}