package org.codehaus.plexus.util;
   
   /*
    * J.A.D.E. Java(TM) Addition to Default Environment.
    * Latest release available at http://jade.dautelle.com/
    * This class is public domain (not copyrighted).
    */
   
   import java.util.Collection;
  import java.util.Collections;
  import java.util.Map;
  import java.util.Set;
  
  /**
   * <p>
   * This class provides cache access to <code>Map</code> collections.
   * </p>
   * <p>
   * Instance of this class can be used as "proxy" for any collection implementing the <code>java.util.Map</code>
   * interface.
   * </p>
   * <p>
   * Typically, {@link CachedMap} are used to accelerate access to large collections when the access to the collection is
   * not evenly distributed (associative cache). The performance gain is about 50% for the fastest hash map collections
   * (e.g. {@link FastMap}). For slower collections such as <code>java.util.TreeMap</code>, non-resizable {@link FastMap}
   * (real-time) or database access, performance can be of several orders of magnitude.
   * </p>
   * <p>
   * <b>Note:</b> The keys used to access elements of a {@link CachedMap} do not need to be immutable as they are not
   * stored in the cache (only keys specified by the {@link #put} method are). In other words, access can be performed
   * using mutable keys as long as these keys can be compared for equality with the real map's keys (e.g. same
   * <code>hashCode</code> values).
   * </p>
   * <p>
   * This implementation is not synchronized. Multiple threads accessing or modifying the collection must be synchronized
   * externally.
   * </p>
   * <p>
   * <i> This class is <b>public domain</b> (not copyrighted).</i>
   * </p>
   *
   * @author <a href="mailto:jean-marie@dautelle.com">Jean-Marie Dautelle</a>
   * @version 5.3, October 30, 2003
   */
  public final class CachedMap implements Map {
  
      /**
       * Holds the FastMap backing this collection (<code>null</code> if generic backing map).
       */
      private final FastMap _backingFastMap;
  
      /**
       * Holds the generic map backing this collection.
       */
      private final Map _backingMap;
  
      /**
       * Holds the keys of the backing map (key-to-key mapping). (<code>null</code> if FastMap backing map).
       */
      private final FastMap _keysMap;
  
      /**
       * Holds the cache's mask (capacity - 1).
       */
      private final int _mask;
  
      /**
       * Holds the keys being cached.
       */
      private final Object[] _keys;
  
      /**
       * Holds the values being cached.
       */
      private final Object[] _values;
  
      /**
       * Creates a cached map backed by a {@link FastMap}. The default cache size and map capacity is set to
       * <code>256</code> entries.
       */
      public CachedMap() {
          this(256, new FastMap());
      }
  
      /**
       * Creates a cached map backed by a {@link FastMap} and having the specified cache size.
       *
       * @param cacheSize the cache size, the actual cache size is the first power of 2 greater or equal to
       *            <code>cacheSize</code>. This is also the initial capacity of the backing map.
       */
      public CachedMap(int cacheSize) {
          this(cacheSize, new FastMap(cacheSize));
      }
  
      /**
       * Creates a cached map backed by the specified map and having the specified cache size. In order to maintain cache
       * veracity, it is critical that <b>all</b> update to the backing map is accomplished through the {@link CachedMap}
       * instance; otherwise {@link #flush} has to be called.
       *
      * @param cacheSize the cache size, the actual cache size is the first power of 2 greater or equal to
      *            <code>cacheSize</code>.
      * @param backingMap the backing map to be "wrapped" in a cached map.
      */
     public CachedMap(int cacheSize, Map backingMap) {
 
         // Find a power of 2 >= minimalCache
         int actualCacheSize = 1;
         while (actualCacheSize < cacheSize) {
             actualCacheSize <<= 1;
         }
 
         // Sets up cache.
         _keys = new Object[actualCacheSize];
         _values = new Object[actualCacheSize];
         _mask = actualCacheSize - 1;
 
         // Sets backing map references.
         if (backingMap instanceof FastMap) {
             _backingFastMap = (FastMap) backingMap;
             _backingMap = _backingFastMap;
             _keysMap = null;
         } else {
             _backingFastMap = null;
             _backingMap = backingMap;
             _keysMap = new FastMap(backingMap.size());
             for (Object key : backingMap.keySet()) {
                 _keysMap.put(key, key);
             }
         }
     }
 
     /**
      * Returns the actual cache size.
      *
      * @return the cache size (power of 2).
      */
     public int getCacheSize() {
         return _keys.length;
     }
 
     /**
      * Returns the backing map. If the backing map is modified directly, this {@link CachedMap} has to be flushed.
      *
      * @return the backing map.
      * @see #flush
      */
     public Map getBackingMap() {
         return (_backingFastMap != null) ? _backingFastMap : _backingMap;
     }
 
     /**
      * Flushes the key/value pairs being cached. This method should be called if the backing map is externally modified.
      */
     public void flush() {
         for (int i = 0; i < _keys.length; i++) {
             _keys[i] = null;
             _values[i] = null;
         }
 
         if (_keysMap != null) {
             // Re-populates keys from backing map.
             for (Object key : _backingMap.keySet()) {
                 _keysMap.put(key, key);
             }
         }
     }
 
     /**
      * Returns the value to which this map maps the specified key. First, the cache is being checked, then if the cache
      * does not contains the specified key, the backing map is accessed and the key/value pair is stored in the cache.
      *
      * @param key the key whose associated value is to be returned.
      * @return the value to which this map maps the specified key, or <code>null</code> if the map contains no mapping
      *         for this key.
      * @throws ClassCastException if the key is of an inappropriate type for the backing map (optional).
      * @throws NullPointerException if the key is <code>null</code>.
      */
     @Override
     public Object get(Object key) {
         int index = key.hashCode() & _mask;
         return key.equals(_keys[index]) ? _values[index] : getCacheMissed(key, index);
     }
 
     private Object getCacheMissed(Object key, int index) {
         if (_backingFastMap != null) {
             Map.Entry entry = _backingFastMap.getEntry(key);
             if (entry != null) {
                 _keys[index] = entry.getKey();
                 Object value = entry.getValue();
                 _values[index] = value;
                 return value;
             } else {
                 return null;
             }
         } else { // Generic backing map.
             Object mapKey = _keysMap.get(key);
             if (mapKey != null) {
                 _keys[index] = mapKey;
                 Object value = _backingMap.get(key);
                 _values[index] = value;
                 return value;
             } else {
                 return null;
             }
         }
     }
 
     /**
      * Associates the specified value with the specified key in this map.
      *
      * @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 specified key, or <code>null</code> if there was no mapping for the
      *         key.
      * @throws UnsupportedOperationException if the <code>put</code> operation is not supported by the backing map.
      * @throws ClassCastException if the class of the specified key or value prevents it from being stored in this map.
      * @throws IllegalArgumentException if some aspect of this key or value prevents it from being stored in this map.
      * @throws NullPointerException if the key is <code>null</code>.
      */
     @Override
     public Object put(Object key, Object value) {
         // Updates the cache.
         int index = key.hashCode() & _mask;
         if (key.equals(_keys[index])) {
             _values[index] = value;
         } else if (_keysMap != null) { // Possibly a new key.
             _keysMap.put(key, key);
         }
 
         // Updates the backing map.
         return _backingMap.put(key, value);
     }
 
     /**
      * Removes the mapping for this key from this map if it is present.
      *
      * @param key key whose mapping is to be removed from the map.
      * @return previous value associated with specified key, or <code>null</code> if there was no mapping for key.
      * @throws ClassCastException if the key is of an inappropriate type for the backing map (optional).
      * @throws NullPointerException if the key is <code>null</code>.
      * @throws UnsupportedOperationException if the <code>remove</code> method is not supported by the backing map.
      */
     @Override
     public Object remove(Object key) {
         // Removes from cache.
         int index = key.hashCode() & _mask;
         if (key.equals(_keys[index])) {
             _keys[index] = null;
         }
         // Removes from key map.
         if (_keysMap != null) {
             _keysMap.remove(key);
         }
         // Removes from backing map.
         return _backingMap.remove(key);
     }
 
     /**
      * Indicates if this map contains a mapping for the specified key.
      *
      * @param key the key whose presence in this map is to be tested.
      * @return <code>true</code> if this map contains a mapping for the specified key; <code>false</code> otherwise.
      */
     @Override
     public boolean containsKey(Object key) {
         // Checks the cache.
         int index = key.hashCode() & _mask;
         if (key.equals(_keys[index])) {
             return true;
         } else { // Checks the backing map.
             return _backingMap.containsKey(key);
         }
     }
 
     /**
      * Returns the number of key-value mappings in this map. If the map contains more than
      * <code>Integer.MAX_VALUE</code> elements, returns <code>Integer.MAX_VALUE</code>.
      *
      * @return the number of key-value mappings in this map.
      */
     @Override
     public int size() {
         return _backingMap.size();
     }
 
     /**
      * Returns <code>true</code> if this map contains no key-value mappings.
      *
      * @return <code>true</code> if this map contains no key-value mappings.
      */
     @Override
     public boolean isEmpty() {
         return _backingMap.isEmpty();
     }
 
     /**
      * Returns <code>true</code> if this map maps one or more keys to the specified value.
      *
      * @param value value whose presence in this map is to be tested.
      * @return <code>true</code> if this map maps one or more keys to the specified value.
      * @throws ClassCastException if the value is of an inappropriate type for the backing map (optional).
      * @throws NullPointerException if the value is <code>null</code> and the backing map does not not permit
      *             <code>null</code> values.
      */
     @Override
     public boolean containsValue(Object value) {
         return _backingMap.containsValue(value);
     }
 
     /**
      * Copies all of the mappings from the specified map to this map (optional operation). This method automatically
      * flushes the cache.
      *
      * @param map the mappings to be stored in this map.
      * @throws UnsupportedOperationException if the <code>putAll</code> method is not supported by the backing map.
      * @throws ClassCastException if the class of a key or value in the specified map prevents it from being stored in
      *             this map.
      * @throws IllegalArgumentException some aspect of a key or value in the specified map prevents it from being stored
      *             in this map.
      * @throws NullPointerException the specified map is <code>null</code>, or if the backing map does not permit
      *             <code>null</code> keys or values, and the specified map contains <code>null</code> keys or values.
      */
     @Override
     public void putAll(Map map) {
         _backingMap.putAll(map);
         flush();
     }
 
     /**
      * Removes all mappings from this map (optional operation). This method automatically flushes the cache.
      *
      * @throws UnsupportedOperationException if clear is not supported by the backing map.
      */
     @Override
     public void clear() {
         _backingMap.clear();
         flush();
     }
 
     /**
      * Returns an <b>unmodifiable</b> view of the keys contained in this map.
      *
      * @return an unmodifiable view of the keys contained in this map.
      */
     @Override
     public Set keySet() {
         return Collections.unmodifiableSet(_backingMap.keySet());
     }
 
     /**
      * Returns an <b>unmodifiable</b> view of the values contained in this map.
      *
      * @return an unmodifiable view of the values contained in this map.
      */
     @Override
     public Collection values() {
         return Collections.unmodifiableCollection(_backingMap.values());
     }
 
     /**
      * Returns an <b>unmodifiable</b> view of the mappings contained in this map. Each element in the returned set is a
      * <code>Map.Entry</code>.
      *
      * @return an unmodifiable view of the mappings contained in this map.
      */
     @Override
     public Set entrySet() {
         return Collections.unmodifiableSet(_backingMap.entrySet());
     }
 
     /**
      * Compares the specified object with this map for equality. Returns <code>true</code> if the given object is also a map
      * and the two Maps represent the same mappings.
      *
      * @param o object to be compared for equality with this map.
      * @return <code>true</code> if the specified object is equal to this map.
      */
     @Override
     public boolean equals(Object o) {
         return _backingMap.equals(o);
     }
 
     /**
      * Returns the hash code value for this map.
      *
      * @return the hash code value for this map.
      */
     @Override
     public int hashCode() {
         return _backingMap.hashCode();
     }
 }