/* Copyright � 1999 CERN - European Organization for Nuclear Research. Permission to use, copy, modify, distribute and sell this software and its documentation for any purpose is hereby granted without fee, provided that the above copyright notice appear in all copies and that both that copyright notice and this permission notice appear in supporting documentation. CERN makes no representations about the suitability of this software for any purpose. It is provided "as is" without expressed or implied warranty. */ package cern.colt.map; import cern.colt.function.DoubleIntProcedure; import cern.colt.function.DoubleProcedure; import cern.colt.list.DoubleArrayList; import cern.colt.list.IntArrayList; /** Abstract base class for hash maps holding (key,value) associations of type <tt>(double-->int)</tt>. First see the <a href="package-summary.html">package summary</a> and javadoc <a href="package-tree.html">tree view</a> to get the broad picture. <p> <b>Implementation</b>: <p> Almost all methods are expressed in terms of {@link #forEachKey(DoubleProcedure)}. As such they are fully functional, but inefficient. Override them in subclasses if necessary. @author wolfgang.hoschek@cern.ch @version 1.0, 09/24/99 @see java.util.HashMap */ public abstract class AbstractDoubleIntMap extends AbstractMap { //public static int hashCollisions = 0; // for debug only /** * Makes this class non instantiable, but still let's others inherit from it. */ protected AbstractDoubleIntMap() {} /** * Returns <tt>true</tt> if the receiver contains the specified key. * * @return <tt>true</tt> if the receiver contains the specified key. */ public boolean containsKey(final double key) { return ! forEachKey( new DoubleProcedure() { public boolean apply(double iterKey) { return (key != iterKey); } } ); } /** * Returns <tt>true</tt> if the receiver contains the specified value. * * @return <tt>true</tt> if the receiver contains the specified value. */ public boolean containsValue(final int value) { return ! forEachPair( new DoubleIntProcedure() { public boolean apply(double iterKey, int iterValue) { return (value != iterValue); } } ); } /** * Returns a deep copy of the receiver; uses <code>clone()</code> and casts the result. * * @return a deep copy of the receiver. */ public AbstractDoubleIntMap copy() { return (AbstractDoubleIntMap) clone(); } /** * Compares the specified object with this map for equality. Returns * <tt>true</tt> if the given object is also a map and the two maps * represent the same mappings. More formally, two maps <tt>m1</tt> and * <tt>m2</tt> represent the same mappings iff * <pre> * m1.forEachPair( * new DoubleIntProcedure() { * public boolean apply(double key, int value) { * return m2.containsKey(key) && m2.get(key) == value; * } * } * ) * && * m2.forEachPair( * new DoubleIntProcedure() { * public boolean apply(double key, int value) { * return m1.containsKey(key) && m1.get(key) == value; * } * } * ); * </pre> * * This implementation first checks if the specified object is this map; * if so it returns <tt>true</tt>. Then, it checks if the specified * object is a map whose size is identical to the size of this set; if * not, it it returns <tt>false</tt>. If so, it applies the iteration as described above. * * @param obj object to be compared for equality with this map. * @return <tt>true</tt> if the specified object is equal to this map. */ public boolean equals(Object obj) { if (obj == this) return true; if (!(obj instanceof AbstractDoubleIntMap)) return false; final AbstractDoubleIntMap other = (AbstractDoubleIntMap) obj; if (other.size() != size()) return false; return forEachPair( new DoubleIntProcedure() { public boolean apply(double key, int value) { return other.containsKey(key) && other.get(key) == value; } } ) && other.forEachPair( new DoubleIntProcedure() { public boolean apply(double key, int value) { return containsKey(key) && get(key) == value; } } ); } /** * Applies a procedure to each key of the receiver, if any. * Note: Iterates over the keys in no particular order. * Subclasses can define a particular order, for example, "sorted by key". * All methods which <i>can</i> be expressed in terms of this method (most methods can) <i>must guarantee</i> to use the <i>same</i> order defined by this method, even if it is no particular order. * This is necessary so that, for example, methods <tt>keys</tt> and <tt>values</tt> will yield association pairs, not two uncorrelated lists. * * @param procedure the procedure to be applied. Stops iteration if the procedure returns <tt>false</tt>, otherwise continues. * @return <tt>false</tt> if the procedure stopped before all keys where iterated over, <tt>true</tt> otherwise. */ public abstract boolean forEachKey(DoubleProcedure procedure); /** * Applies a procedure to each (key,value) pair of the receiver, if any. * Iteration order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * * @param procedure the procedure to be applied. Stops iteration if the procedure returns <tt>false</tt>, otherwise continues. * @return <tt>false</tt> if the procedure stopped before all keys where iterated over, <tt>true</tt> otherwise. */ public boolean forEachPair(final DoubleIntProcedure procedure) { return forEachKey( new DoubleProcedure() { public boolean apply(double key) { return procedure.apply(key,get(key)); } } ); } /** * Returns the value associated with the specified key. * It is often a good idea to first check with {@link #containsKey(double)} whether the given key has a value associated or not, i.e. whether there exists an association for the given key or not. * * @param key the key to be searched for. * @return the value associated with the specified key; <tt>0</tt> if no such key is present. */ public abstract int get(double key); /** * Returns the first key the given value is associated with. * It is often a good idea to first check with {@link #containsValue(int)} whether there exists an association from a key to this value. * Search order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * * @param value the value to search for. * @return the first key for which holds <tt>get(key) == value</tt>; * returns <tt>Double.NaN</tt> if no such key exists. */ public double keyOf(final int value) { final double[] foundKey = new double[1]; boolean notFound = forEachPair( new DoubleIntProcedure() { public boolean apply(double iterKey, int iterValue) { boolean found = value == iterValue; if (found) foundKey[0] = iterKey; return !found; } } ); if (notFound) return Double.NaN; return foundKey[0]; } /** * Returns a list filled with all keys contained in the receiver. * The returned list has a size that equals <tt>this.size()</tt>. * Note: Keys are filled into the list in no particular order. * However, the order is <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * <p> * This method can be used to iterate over the keys of the receiver. * * @return the keys. */ public DoubleArrayList keys() { DoubleArrayList list = new DoubleArrayList(size()); keys(list); return list; } /** * Fills all keys contained in the receiver into the specified list. * Fills the list, starting at index 0. * After this call returns the specified list has a new size that equals <tt>this.size()</tt>. * Iteration order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * <p> * This method can be used to iterate over the keys of the receiver. * * @param list the list to be filled, can have any size. */ public void keys(final DoubleArrayList list) { list.clear(); forEachKey( new DoubleProcedure() { public boolean apply(double key) { list.add(key); return true; } } ); } /** * Fills all keys <i>sorted ascending by their associated value</i> into the specified list. * Fills into the list, starting at index 0. * After this call returns the specified list has a new size that equals <tt>this.size()</tt>. * Primary sort criterium is "value", secondary sort criterium is "key". * This means that if any two values are equal, the smaller key comes first. * <p> * <b>Example:</b> * <br> * <tt>keys = (8,7,6), values = (1,2,2) --> keyList = (8,6,7)</tt> * * @param keyList the list to be filled, can have any size. */ public void keysSortedByValue(final DoubleArrayList keyList) { pairsSortedByValue(keyList, new IntArrayList(size())); } /** Fills all pairs satisfying a given condition into the specified lists. Fills into the lists, starting at index 0. After this call returns the specified lists both have a new size, the number of pairs satisfying the condition. Iteration order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. <p> <b>Example:</b> <br> <pre> DoubleIntProcedure condition = new DoubleIntProcedure() { // match even values only public boolean apply(double key, int value) { return value%2==0; } } keys = (8,7,6), values = (1,2,2) --> keyList = (6,8), valueList = (2,1)</tt> </pre> @param condition the condition to be matched. Takes the current key as first and the current value as second argument. @param keyList the list to be filled with keys, can have any size. @param valueList the list to be filled with values, can have any size. */ public void pairsMatching(final DoubleIntProcedure condition, final DoubleArrayList keyList, final IntArrayList valueList) { keyList.clear(); valueList.clear(); forEachPair( new DoubleIntProcedure() { public boolean apply(double key, int value) { if (condition.apply(key,value)) { keyList.add(key); valueList.add(value); } return true; } } ); } /** * Fills all keys and values <i>sorted ascending by key</i> into the specified lists. * Fills into the lists, starting at index 0. * After this call returns the specified lists both have a new size that equals <tt>this.size()</tt>. * <p> * <b>Example:</b> * <br> * <tt>keys = (8,7,6), values = (1,2,2) --> keyList = (6,7,8), valueList = (2,2,1)</tt> * * @param keyList the list to be filled with keys, can have any size. * @param valueList the list to be filled with values, can have any size. */ public void pairsSortedByKey(final DoubleArrayList keyList, final IntArrayList valueList) { /* keys(keyList); values(valueList); final double[] k = keyList.elements(); final int[] v = valueList.elements(); cern.colt.Swapper swapper = new cern.colt.Swapper() { public void swap(int a, int b) { int t1; double t2; t1 = v[a]; v[a] = v[b]; v[b] = t1; t2 = k[a]; k[a] = k[b]; k[b] = t2; } }; cern.colt.function.IntComparator comp = new cern.colt.function.IntComparator() { public int compare(int a, int b) { return k[a]<k[b] ? -1 : k[a]==k[b] ? 0 : 1; } }; cern.colt.MultiSorting.sort(0,keyList.size(),comp,swapper); */ // this variant may be quicker //cern.colt.map.OpenDoubleIntHashMap.hashCollisions = 0; //System.out.println("collisions="+cern.colt.map.OpenDoubleIntHashMap.hashCollisions); keys(keyList); keyList.sort(); valueList.setSize(keyList.size()); for (int i=keyList.size(); --i >= 0; ) { valueList.setQuick(i,get(keyList.getQuick(i))); } //System.out.println("collisions="+cern.colt.map.OpenDoubleIntHashMap.hashCollisions); } /** * Fills all keys and values <i>sorted ascending by value</i> into the specified lists. * Fills into the lists, starting at index 0. * After this call returns the specified lists both have a new size that equals <tt>this.size()</tt>. * Primary sort criterium is "value", secondary sort criterium is "key". * This means that if any two values are equal, the smaller key comes first. * <p> * <b>Example:</b> * <br> * <tt>keys = (8,7,6), values = (1,2,2) --> keyList = (8,6,7), valueList = (1,2,2)</tt> * * @param keyList the list to be filled with keys, can have any size. * @param valueList the list to be filled with values, can have any size. */ public void pairsSortedByValue(final DoubleArrayList keyList, final IntArrayList valueList) { keys(keyList); values(valueList); final double[] k = keyList.elements(); final int[] v = valueList.elements(); cern.colt.Swapper swapper = new cern.colt.Swapper() { public void swap(int a, int b) { int t1; double t2; t1 = v[a]; v[a] = v[b]; v[b] = t1; t2 = k[a]; k[a] = k[b]; k[b] = t2; } }; cern.colt.function.IntComparator comp = new cern.colt.function.IntComparator() { public int compare(int a, int b) { return v[a]<v[b] ? -1 : v[a]>v[b] ? 1 : (k[a]<k[b] ? -1 : (k[a]==k[b] ? 0 : 1)); } }; //cern.colt.map.OpenDoubleIntHashMap.hashCollisions = 0; cern.colt.GenericSorting.quickSort(0,keyList.size(),comp,swapper); //System.out.println("collisions="+cern.colt.map.OpenDoubleIntHashMap.hashCollisions); } /** * Associates the given key with the given value. * Replaces any old <tt>(key,someOtherValue)</tt> association, if existing. * * @param key the key the value shall be associated with. * @param value the value to be associated. * @return <tt>true</tt> if the receiver did not already contain such a key; * <tt>false</tt> if the receiver did already contain such a key - the new value has now replaced the formerly associated value. */ public abstract boolean put(double key, int value); /** * Removes the given key with its associated element from the receiver, if present. * * @param key the key to be removed from the receiver. * @return <tt>true</tt> if the receiver contained the specified key, <tt>false</tt> otherwise. */ public abstract boolean removeKey(double key); /** * Returns a string representation of the receiver, containing * the String representation of each key-value pair, sorted ascending by key. */ public String toString() { DoubleArrayList theKeys = keys(); theKeys.sort(); StringBuffer buf = new StringBuffer(); buf.append("["); int maxIndex = theKeys.size() - 1; for (int i = 0; i <= maxIndex; i++) { double key = theKeys.get(i); buf.append(String.valueOf(key)); buf.append("->"); buf.append(String.valueOf(get(key))); if (i < maxIndex) buf.append(", "); } buf.append("]"); return buf.toString(); } /** * Returns a string representation of the receiver, containing * the String representation of each key-value pair, sorted ascending by value. */ public String toStringByValue() { DoubleArrayList theKeys = new DoubleArrayList(); keysSortedByValue(theKeys); StringBuffer buf = new StringBuffer(); buf.append("["); int maxIndex = theKeys.size() - 1; for (int i = 0; i <= maxIndex; i++) { double key = theKeys.get(i); buf.append(String.valueOf(key)); buf.append("->"); buf.append(String.valueOf(get(key))); if (i < maxIndex) buf.append(", "); } buf.append("]"); return buf.toString(); } /** * Returns a list filled with all values contained in the receiver. * The returned list has a size that equals <tt>this.size()</tt>. * Iteration order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * <p> * This method can be used to iterate over the values of the receiver. * * @return the values. */ public IntArrayList values() { IntArrayList list = new IntArrayList(size()); values(list); return list; } /** * Fills all values contained in the receiver into the specified list. * Fills the list, starting at index 0. * After this call returns the specified list has a new size that equals <tt>this.size()</tt>. * Iteration order is guaranteed to be <i>identical</i> to the order used by method {@link #forEachKey(DoubleProcedure)}. * <p> * This method can be used to iterate over the values of the receiver. * * @param list the list to be filled, can have any size. */ public void values(final IntArrayList list) { list.clear(); forEachKey( new DoubleProcedure() { public boolean apply(double key) { list.add(get(key)); return true; } } ); } }