gnu.trove.TByteIntHashMap Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of trove4j Show documentation
Show all versions of trove4j Show documentation
Fork of trove4j library used in IntelliJ Platform.
The newest version!
///////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2001, Eric D. Friedman All Rights Reserved.
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
///////////////////////////////////////////////////////////////////////////////
// THIS FILE IS AUTOGENERATED, PLEASE DO NOT EDIT OR ELSE
package gnu.trove;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
/**
* An open addressed Map implementation for byte keys and int values.
*
* Created: Sun Nov 4 08:52:45 2001
*
* @author Eric D. Friedman
*/
public class TByteIntHashMap extends TByteHash {
/** the values of the map */
protected transient int[] _values;
/**
* Creates a new TByteIntHashMap instance with the default
* capacity and load factor.
*/
public TByteIntHashMap() {
}
/**
* Creates a new TByteIntHashMap instance with a prime
* capacity equal to or greater than initialCapacity and
* with the default load factor.
*
* @param initialCapacity an int value
*/
public TByteIntHashMap(int initialCapacity) {
super(initialCapacity);
}
/**
* Creates a new TByteIntHashMap instance with a prime
* capacity equal to or greater than initialCapacity and
* with the specified load factor.
*
* @param initialCapacity an int value
* @param loadFactor a float value
*/
public TByteIntHashMap(int initialCapacity, float loadFactor) {
super(initialCapacity, loadFactor);
}
/**
* Creates a new TByteIntHashMap instance with the default
* capacity and load factor.
* @param strategy used to compute hash codes and to compare keys.
*/
public TByteIntHashMap(TByteHashingStrategy strategy) {
super(strategy);
}
/**
* Creates a new TByteIntHashMap instance whose capacity
* is the next highest prime above initialCapacity + 1
* unless that value is already prime.
*
* @param initialCapacity an int value
* @param strategy used to compute hash codes and to compare keys.
*/
public TByteIntHashMap(int initialCapacity, TByteHashingStrategy strategy) {
super(initialCapacity, strategy);
}
/**
* Creates a new TByteIntHashMap instance with a prime
* value at or near the specified capacity and load factor.
*
* @param initialCapacity used to find a prime capacity for the table.
* @param loadFactor used to calculate the threshold over which
* rehashing takes place.
* @param strategy used to compute hash codes and to compare keys.
*/
public TByteIntHashMap(int initialCapacity, float loadFactor, TByteHashingStrategy strategy) {
super(initialCapacity, loadFactor, strategy);
}
/**
* @return a deep clone of this collection
*/
@Override
public Object clone() {
TByteIntHashMap m = (TByteIntHashMap)super.clone();
m._values = _values == null ? null : _values.clone();
return m;
}
/**
* @return a TByteIntIterator with access to this map's keys and values
*/
public TByteIntIterator iterator() {
return new TByteIntIterator(this);
}
/**
* initializes the hashtable to a prime capacity which is at least
* initialCapacity + 1.
*
* @param initialCapacity an int value
* @return the actual capacity chosen
*/
@Override
protected int setUp(int initialCapacity) {
int capacity = super.setUp(initialCapacity);
_values = initialCapacity == JUST_CREATED_CAPACITY ? null : new int[capacity];
return capacity;
}
/**
* Inserts a key/value pair into the map.
*
* @param key an byte value
* @param value an int value
* @return the previous value associated with key,
* or null if none was found.
*/
public int put(byte key, int value) {
int previous = 0;
int index = insertionIndex(key);
boolean isNewMapping = true;
if (index < 0) {
index = -index -1;
previous = _values[index];
isNewMapping = false;
}
byte previousState = _states[index];
_set[index] = key;
_states[index] = FULL;
_values[index] = value;
if (isNewMapping) {
postInsertHook(previousState == FREE);
}
return previous;
}
/**
* rehashes the map to the new capacity.
*
* @param newCapacity an int value
*/
@Override
protected void rehash(int newCapacity) {
int oldCapacity = capacity();
byte[] oldKeys = _set;
int[] oldVals = _values;
byte[] oldStates = _states;
_set = new byte[newCapacity];
_values = new int[newCapacity];
_states = new byte[newCapacity];
for (int i = oldCapacity; i-- > 0;) {
if(oldStates[i] == FULL) {
byte o = oldKeys[i];
int index = insertionIndex(o);
_set[index] = o;
_values[index] = oldVals[i];
_states[index] = FULL;
}
}
}
/**
* retrieves the value for key
*
* @param key an byte value
* @return the value of key or null if no such mapping exists.
*/
public int get(byte key) {
int index = index(key);
return index < 0 ? 0 : _values[index];
}
/**
* Empties the map.
*
*/
@Override
public void clear() {
super.clear();
byte[] keys = _set;
int[] vals = _values;
if (vals == null) return;
byte[] states = _states;
for (int i = keys.length; i-- > 0;) {
keys[i] = 0;
vals[i] = 0;
states[i] = FREE;
}
}
/**
* Deletes a key/value pair from the map.
*
* @param key an byte value
* @return an int value
*/
public int remove(byte key) {
int prev = 0;
int index = index(key);
if (index >= 0) {
prev = _values[index];
removeAt(index); // clear key,state; adjust size
}
return prev;
}
/**
* Compares this map with another map for equality of their stored
* entries.
*
* @param other an Object value
* @return a boolean value
*/
@Override
public boolean equals(Object other) {
if (! (other instanceof TByteIntHashMap)) {
return false;
}
TByteIntHashMap that = (TByteIntHashMap)other;
if (that.size() != size()) {
return false;
}
return forEachEntry(new EqProcedure(that));
}
@Override
public int hashCode() {
HashProcedure p = new HashProcedure();
forEachEntry(p);
return p.getHashCode();
}
private final class HashProcedure implements TByteIntProcedure {
private int h;
HashProcedure() {
}
public int getHashCode() {
return h;
}
public final boolean execute(byte key, int value) {
h += _hashingStrategy.computeHashCode(key) ^ HashFunctions.hash(value);
return true;
}
}
private static final class EqProcedure implements TByteIntProcedure {
private final TByteIntHashMap _otherMap;
EqProcedure(TByteIntHashMap otherMap) {
_otherMap = otherMap;
}
public final boolean execute(byte key, int value) {
int index = _otherMap.index(key);
return index >= 0 && eq(value, _otherMap.get(key));
}
/**
* Compare two ints for equality.
*/
private static boolean eq(int v1, int v2) {
return v1 == v2;
}
}
/**
* removes the mapping at index from the map.
*
* @param index an int value
*/
@Override
protected void removeAt(int index) {
_values[index] = 0;
super.removeAt(index); // clear key, state; adjust size
}
/**
* Returns the values of the map.
*
* @return a Collection value
*/
public int[] getValues() {
int[] vals = new int[size()];
int[] v = _values;
byte[] states = _states;
if (states != null) {
for (int i = states.length, j = 0; i-- > 0;) {
if (states[i] == FULL) {
vals[j++] = v[i];
}
}
}
return vals;
}
/**
* returns the keys of the map.
*
* @return a Set value
*/
public byte[] keys() {
byte[] keys = new byte[size()];
byte[] k = _set;
byte[] states = _states;
if (states != null) {
for (int i = states.length, j = 0; i-- > 0;) {
if (states[i] == FULL) {
keys[j++] = k[i];
}
}
}
return keys;
}
/**
* checks for the presence of val in the values of the map.
*
* @param val an int value
* @return a boolean value
*/
public boolean containsValue(int val) {
byte[] states = _states;
int[] vals = _values;
if (states != null) {
for (int i = states.length; i-- > 0;) {
if (states[i] == FULL && val == vals[i]) {
return true;
}
}
}
return false;
}
/**
* checks for the present of key in the keys of the map.
*
* @param key an byte value
* @return a boolean value
*/
public boolean containsKey(byte key) {
return contains(key);
}
/**
* Executes procedure for each key in the map.
*
* @param procedure a TByteProcedure value
* @return false if the loop over the keys terminated because
* the procedure returned false for some key.
*/
public boolean forEachKey(TByteProcedure procedure) {
return forEach(procedure);
}
/**
* Executes procedure for each value in the map.
*
* @param procedure a TIntProcedure value
* @return false if the loop over the values terminated because
* the procedure returned false for some value.
*/
public boolean forEachValue(TIntProcedure procedure) {
byte[] states = _states;
int[] values = _values;
if (states != null) {
for (int i = states.length; i-- > 0;) {
if (states[i] == FULL && ! procedure.execute(values[i])) {
return false;
}
}
}
return true;
}
/**
* Executes procedure for each key/value entry in the
* map.
*
* @param procedure a TByteIntProcedure value
* @return false if the loop over the entries terminated because
* the procedure returned false for some entry.
*/
public boolean forEachEntry(TByteIntProcedure procedure) {
byte[] states = _states;
byte[] keys = _set;
int[] values = _values;
if (states != null) {
for (int i = states.length; i-- > 0;) {
if (states[i] == FULL && ! procedure.execute(keys[i],values[i])) {
return false;
}
}
}
return true;
}
/**
* Retains only those entries in the map for which the procedure
* returns a true value.
*
* @param procedure determines which entries to keep
* @return true if the map was modified.
*/
public boolean retainEntries(TByteIntProcedure procedure) {
boolean modified = false;
byte[] states = _states;
byte[] keys = _set;
int[] values = _values;
if (states != null) {
for (int i = states.length; i-- > 0;) {
if (states[i] == FULL && ! procedure.execute(keys[i],values[i])) {
removeAt(i);
modified = true;
}
}
}
return modified;
}
/**
* Transform the values in this map using function.
*
* @param function a TIntFunction value
*/
public void transformValues(TIntFunction function) {
byte[] states = _states;
int[] values = _values;
if (states != null) {
for (int i = states.length; i-- > 0;) {
if (states[i] == FULL) {
values[i] = function.execute(values[i]);
}
}
}
}
/**
* Increments the primitive value mapped to key by 1
*
* @param key the key of the value to increment
* @return true if a mapping was found and modified.
*/
public boolean increment(byte key) {
return adjustValue(key, (int) 1);
}
/**
* Adjusts the primitive value mapped to key.
*
* @param key the key of the value to increment
* @param amount the amount to adjust the value by.
* @return true if a mapping was found and modified.
*/
public boolean adjustValue(byte key, int amount) {
int index = index(key);
if (index < 0) {
return false;
} else {
_values[index] += amount;
return true;
}
}
private void writeObject(ObjectOutputStream stream)
throws IOException {
stream.defaultWriteObject();
// number of entries
stream.writeInt(_size);
SerializationProcedure writeProcedure = new SerializationProcedure(stream);
if (! forEachEntry(writeProcedure)) {
throw writeProcedure.exception;
}
}
private void readObject(ObjectInputStream stream)
throws IOException, ClassNotFoundException {
stream.defaultReadObject();
int size = stream.readInt();
setUp(size);
while (size-- > 0) {
byte key = stream.readByte();
int val = stream.readInt();
put(key, val);
}
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder();
forEachEntry(new TByteIntProcedure() {
public boolean execute(byte key, int value) {
if (sb.length() != 0) {
sb.append(',').append(' ');
}
sb.append(key);
sb.append('=');
sb.append(value);
return true;
}
});
sb.append('}');
sb.insert(0, '{');
return sb.toString();
}
} // TByteIntHashMap