/*

 * Copyright (c) 2000, 2018, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package java.util;

import java.lang.reflect.Array;

import java.util.function.BiConsumer;

import java.util.function.BiFunction;

import java.util.function.Consumer;

/**

 * This class implements the {@code Map} interface with a hash table, using

 * reference-equality in place of object-equality when comparing keys (and

 * values).  In other words, in an {@code IdentityHashMap}, two keys

 * {@code k1} and {@code k2} are considered equal if and only if

 * {@code (k1==k2)}.  (In normal {@code Map} implementations (like

 * {@code HashMap}) two keys {@code k1} and {@code k2} are considered equal

 * if and only if {@code (k1==null ? k2==null : k1.equals(k2))}.)

 *

 * <p><b>This class is <i>not</i> a general-purpose {@code Map}

 * implementation!  While this class implements the {@code Map} interface, it

 * intentionally violates {@code Map's} general contract, which mandates the

 * use of the {@code equals} method when comparing objects.  This class is

 * designed for use only in the rare cases wherein reference-equality

 * semantics are required.</b>

 *

 * <p>A typical use of this class is <i>topology-preserving object graph

 * transformations</i>, such as serialization or deep-copying.  To perform such

 * a transformation, a program must maintain a "node table" that keeps track

 * of all the object references that have already been processed.  The node

 * table must not equate distinct objects even if they happen to be equal.

 * Another typical use of this class is to maintain <i>proxy objects</i>.  For

 * example, a debugging facility might wish to maintain a proxy object for

 * each object in the program being debugged.

 *

 * <p>This class provides all of the optional map operations, and permits

 * {@code null} values and the {@code null} key.  This class makes no

 * guarantees as to the order of the map; in particular, it does not guarantee

 * that the order will remain constant over time.

 *

 * <p>This class provides constant-time performance for the basic

 * operations ({@code get} and {@code put}), assuming the system

 * identity hash function ({@link System#identityHashCode(Object)})

 * disperses elements properly among the buckets.

 *

 * <p>This class has one tuning parameter (which affects performance but not

 * semantics): <i>expected maximum size</i>.  This parameter is the maximum

 * number of key-value mappings that the map is expected to hold.  Internally,

 * this parameter is used to determine the number of buckets initially

 * comprising the hash table.  The precise relationship between the expected

 * maximum size and the number of buckets is unspecified.

 *

 * <p>If the size of the map (the number of key-value mappings) sufficiently

 * exceeds the expected maximum size, the number of buckets is increased.

 * Increasing the number of buckets ("rehashing") may be fairly expensive, so

 * it pays to create identity hash maps with a sufficiently large expected

 * maximum size.  On the other hand, iteration over collection views requires

 * time proportional to the number of buckets in the hash table, so it

 * pays not to set the expected maximum size too high if you are especially

 * concerned with iteration performance or memory usage.

 *

 * <p><strong>Note that this implementation is not synchronized.</strong>

 * If multiple threads access an identity hash map concurrently, and at

 * least one of the threads modifies the map structurally, it <i>must</i>

 * be synchronized externally.  (A structural modification is any operation

 * that adds or deletes one or more mappings; merely changing the value

 * associated with a key that an instance already contains is not a

 * structural modification.)  This is typically accomplished by

 * synchronizing on some object that naturally encapsulates the map.

 *

 * If no such object exists, the map should be "wrapped" using the

 * {@link Collections#synchronizedMap Collections.synchronizedMap}

 * method.  This is best done at creation time, to prevent accidental

 * unsynchronized access to the map:<pre>

 *   Map m = Collections.synchronizedMap(new IdentityHashMap(...));</pre>

 *

 * <p>The iterators returned by the {@code iterator} method of the

 * collections returned by all of this class's "collection view

 * methods" are <i>fail-fast</i>: if the map is structurally modified

 * at any time after the iterator is created, in any way except

 * through the iterator's own {@code remove} method, the iterator

 * will throw a {@link ConcurrentModificationException}.  Thus, in the

 * face of concurrent modification, the iterator fails quickly and

 * cleanly, rather than risking arbitrary, non-deterministic behavior

 * at an undetermined time in the future.

 *

 * <p>Note that the fail-fast behavior of an iterator cannot be guaranteed

 * as it is, generally speaking, impossible to make any hard guarantees in the

 * presence of unsynchronized concurrent modification.  Fail-fast iterators

 * throw {@code ConcurrentModificationException} on a best-effort basis.

 * Therefore, it would be wrong to write a program that depended on this

 * exception for its correctness: <i>fail-fast iterators should be used only

 * to detect bugs.</i>

 *

 * <p>Implementation note: This is a simple <i>linear-probe</i> hash table,

 * as described for example in texts by Sedgewick and Knuth.  The array

 * alternates holding keys and values.  (This has better locality for large

 * tables than does using separate arrays.)  For many JRE implementations

 * and operation mixes, this class will yield better performance than

 * {@link HashMap} (which uses <i>chaining</i> rather than linear-probing).

 *

 * <p>This class is a member of the

 * <a href="{@docRoot}/java.base/java/util/package-summary.html#CollectionsFramework">

 * Java Collections Framework</a>.

 *

 * @see     System#identityHashCode(Object)

 * @see     Object#hashCode()

 * @see     Collection

 * @see     Map

 * @see     HashMap

 * @see     TreeMap

 * @author  Doug Lea and Josh Bloch

 * @since   1.4

 */

public class IdentityHashMap<K,V>

    extends AbstractMap<K,V>

    implements Map<K,V>, java.io.Serializable, Cloneable

{

    /**

     * The initial capacity used by the no-args constructor.

     * MUST be a power of two.  The value 32 corresponds to the

     * (specified) expected maximum size of 21, given a load factor

     * of 2/3.

     */

    private static final int DEFAULT_CAPACITY = 32;

    /**

     * The minimum capacity, used if a lower value is implicitly specified

     * by either of the constructors with arguments.  The value 4 corresponds

     * to an expected maximum size of 2, given a load factor of 2/3.

     * MUST be a power of two.

     */

    private static final int MINIMUM_CAPACITY = 4;

    /**

     * The maximum capacity, used if a higher value is implicitly specified

     * by either of the constructors with arguments.

     * MUST be a power of two <= 1<<29.

     *

     * In fact, the map can hold no more than MAXIMUM_CAPACITY-1 items

     * because it has to have at least one slot with the key == null

     * in order to avoid infinite loops in get(), put(), remove()

     */

    private static final int MAXIMUM_CAPACITY = 1 << 29;

    /**

     * The table, resized as necessary. Length MUST always be a power of two.

     */

    transient Object[] table; // non-private to simplify nested class access

    /**

     * The number of key-value mappings contained in this identity hash map.

     *

     * @serial

     */

    int size;

    /**

     * The number of modifications, to support fast-fail iterators

     */

    transient int modCount;

    /**

     * Value representing null keys inside tables.

     */

    static final Object NULL_KEY = new Object();

    /**

     * Use NULL_KEY for key if it is null.

     */

    private static Object maskNull(Object key) {

        return (key == null ? NULL_KEY : key);

    }

    /**

     * Returns internal representation of null key back to caller as null.

     */

    static final Object unmaskNull(Object key) {

        return (key == NULL_KEY ? null : key);

    }

    /**

     * Constructs a new, empty identity hash map with a default expected

     * maximum size (21).

     */

    public IdentityHashMap() {

        init(DEFAULT_CAPACITY);

    }

    /**

     * Constructs a new, empty map with the specified expected maximum size.

     * Putting more than the expected number of key-value mappings into

     * the map may cause the internal data structure to grow, which may be

     * somewhat time-consuming.

     *

     * @param expectedMaxSize the expected maximum size of the map

     * @throws IllegalArgumentException if {@code expectedMaxSize} is negative

     */

    public IdentityHashMap(int expectedMaxSize) {

        if (expectedMaxSize < 0)

            throw new IllegalArgumentException("expectedMaxSize is negative: "

                                               + expectedMaxSize);

        init(capacity(expectedMaxSize));

    }

    /**

     * Returns the appropriate capacity for the given expected maximum size.

     * Returns the smallest power of two between MINIMUM_CAPACITY and

     * MAXIMUM_CAPACITY, inclusive, that is greater than (3 *

     * expectedMaxSize)/2, if such a number exists.  Otherwise returns

     * MAXIMUM_CAPACITY.

     */

    private static int capacity(int expectedMaxSize) {

        // assert expectedMaxSize >= 0;

        return

            (expectedMaxSize > MAXIMUM_CAPACITY / 3) ? MAXIMUM_CAPACITY :

            (expectedMaxSize <= 2 * MINIMUM_CAPACITY / 3) ? MINIMUM_CAPACITY :

            Integer.highestOneBit(expectedMaxSize + (expectedMaxSize << 1));

    }

    /**

     * Initializes object to be an empty map with the specified initial

     * capacity, which is assumed to be a power of two between

     * MINIMUM_CAPACITY and MAXIMUM_CAPACITY inclusive.

     */

    private void init(int initCapacity) {

        // assert (initCapacity & -initCapacity) == initCapacity; // power of 2

        // assert initCapacity >= MINIMUM_CAPACITY;

        // assert initCapacity <= MAXIMUM_CAPACITY;

        table = new Object[2 * initCapacity];

    }

    /**

     * Constructs a new identity hash map containing the keys-value mappings

     * in the specified map.

     *

     * @param m the map whose mappings are to be placed into this map

     * @throws NullPointerException if the specified map is null

     */

    public IdentityHashMap(Map<? extends K, ? extends V> m) {

        // Allow for a bit of growth

        this((int) ((1 + m.size()) * 1.1));

        putAll(m);

    }

    /**

     * Returns the number of key-value mappings in this identity hash map.

     *

     * @return the number of key-value mappings in this map

     */

    public int size() {

        return size;

    }

    /**

     * Returns {@code true} if this identity hash map contains no key-value

     * mappings.

     *

     * @return {@code true} if this identity hash map contains no key-value

     *         mappings

     */

    public boolean isEmpty() {

        return size == 0;

    }

    /**

     * Returns index for Object x.

     */

    private static int hash(Object x, int length) {

        int h = System.identityHashCode(x);

        // Multiply by -127, and left-shift to use least bit as part of hash

        return ((h << 1) - (h << 8)) & (length - 1);

    }

    /**

     * Circularly traverses table of size len.

     */

    private static int nextKeyIndex(int i, int len) {

        return (i + 2 < len ? i + 2 : 0);

    }

    /**

     * Returns the value to which the specified key is mapped,

     * or {@code null} if this map contains no mapping for the key.

     *

     * <p>More formally, if this map contains a mapping from a key

     * {@code k} to a value {@code v} such that {@code (key == k)},

     * then this method returns {@code v}; otherwise it returns

     * {@code null}.  (There can be at most one such mapping.)

     *

     * <p>A return value of {@code null} does not <i>necessarily</i>

     * indicate that the map contains no mapping for the key; it's also

     * possible that the map explicitly maps the key to {@code null}.

     * The {@link #containsKey containsKey} operation may be used to

     * distinguish these two cases.

     *

     * @see #put(Object, Object)

     */

    @SuppressWarnings("unchecked")

    public V get(Object key) {

        Object k = maskNull(key);

        Object[] tab = table;

        int len = tab.length;

        int i = hash(k, len);

        while (true) {

            Object item = tab[i];

            if (item == k)

                return (V) tab[i + 1];

            if (item == null)

                return null;

            i = nextKeyIndex(i, len);

        }

    }

    /**

     * Tests whether the specified object reference is a key in this identity

     * hash map.

     *

     * @param   key   possible key

     * @return  {@code true} if the specified object reference is a key

     *          in this map

     * @see     #containsValue(Object)

     */

    public boolean containsKey(Object key) {

        Object k = maskNull(key);

        Object[] tab = table;

        int len = tab.length;

        int i = hash(k, len);

        while (true) {

            Object item = tab[i];

            if (item == k)

                return true;

            if (item == null)

                return false;

            i = nextKeyIndex(i, len);

        }

    }

    /**

     * Tests whether the specified object reference is a value in this identity

     * hash map.

     *

     * @param value value whose presence in this map is to be tested

     * @return {@code true} if this map maps one or more keys to the

     *         specified object reference

     * @see     #containsKey(Object)

     */

    public boolean containsValue(Object value) {

        Object[] tab = table;

        for (int i = 1; i < tab.length; i += 2)

            if (tab[i] == value && tab[i - 1] != null)

                return true;

        return false;

    }

    /**

     * Tests if the specified key-value mapping is in the map.

     *

     * @param   key   possible key

     * @param   value possible value

     * @return  {@code true} if and only if the specified key-value

     *          mapping is in the map

     */

    private boolean containsMapping(Object key, Object value) {

        Object k = maskNull(key);

        Object[] tab = table;

        int len = tab.length;

        int i = hash(k, len);

        while (true) {

            Object item = tab[i];

            if (item == k)

                return tab[i + 1] == value;

            if (item == null)

                return false;

            i = nextKeyIndex(i, len);

        }

    }

    /**

     * Associates the specified value with the specified key in this identity

     * hash map.  If the map previously contained a mapping for the key, the

     * old value is replaced.

     *

     * @param key the key with which the specified value is to be associated

     * @param value the value to be associated with the specified key

     * @return the previous value associated with {@code key}, or

     *         {@code null} if there was no mapping for {@code key}.

     *         (A {@code null} return can also indicate that the map

     *         previously associated {@code null} with {@code key}.)

     * @see     Object#equals(Object)

     * @see     #get(Object)

     * @see     #containsKey(Object)

     */

    public V put(K key, V value) {

        final Object k = maskNull(key);

        retryAfterResize: for (;;) {

            final Object[] tab = table;

            final int len = tab.length;

            int i = hash(k, len);

            for (Object item; (item = tab[i]) != null;

                 i = nextKeyIndex(i, len)) {

                if (item == k) {

                    @SuppressWarnings("unchecked")

                        V oldValue = (V) tab[i + 1];

                    tab[i + 1] = value;

                    return oldValue;

                }

            }

            final int s = size + 1;

            // Use optimized form of 3 * s.

            // Next capacity is len, 2 * current capacity.

            if (s + (s << 1) > len && resize(len))

                continue retryAfterResize;

            modCount++;

            tab[i] = k;

            tab[i + 1] = value;

            size = s;

            return null;

        }

    }

    /**

     * Resizes the table if necessary to hold given capacity.

     *

     * @param newCapacity the new capacity, must be a power of two.

     * @return whether a resize did in fact take place

     */

    private boolean resize(int newCapacity) {

        // assert (newCapacity & -newCapacity) == newCapacity; // power of 2

        int newLength = newCapacity * 2;

        Object[] oldTable = table;

        int oldLength = oldTable.length;

        if (oldLength == 2 * MAXIMUM_CAPACITY) { // can't expand any further

            if (size == MAXIMUM_CAPACITY - 1)

                throw new IllegalStateException("Capacity exhausted.");

            return false;

        }

        if (oldLength >= newLength)

            return false;

        Object[] newTable = new Object[newLength];

        for (int j = 0; j < oldLength; j += 2) {

            Object key = oldTable[j];

            if (key != null) {

                Object value = oldTable[j+1];

                oldTable[j] = null;

                oldTable[j+1] = null;

                int i = hash(key, newLength);

                while (newTable[i] != null)

                    i = nextKeyIndex(i, newLength);

                newTable[i] = key;

                newTable[i + 1] = value;

            }

        }

        table = newTable;

        return true;

    }

    /**

     * Copies all of the mappings from the specified map to this map.

     * These mappings will replace any mappings that this map had for

     * any of the keys currently in the specified map.

     *

     * @param m mappings to be stored in this map

     * @throws NullPointerException if the specified map is null

     */

    public void putAll(Map<? extends K, ? extends V> m) {

        int n = m.size();

        if (n == 0)

            return;

        if (n > size)

            resize(capacity(n)); // conservatively pre-expand

        for (Entry<? extends K, ? extends V> e : m.entrySet())

            put(e.getKey(), e.getValue());

    }

    /**

     * Removes the mapping for this key from this map if present.

     *

     * @param key key whose mapping is to be removed from the map

     * @return the previous value associated with {@code key}, or

     *         {@code null} if there was no mapping for {@code key}.

     *         (A {@code null} return can also indicate that the map