/**
    * Copyright (c) 2011, The University of Southampton and the individual contributors.
    * All rights reserved.
    *
    * Redistribution and use in source and binary forms, with or without modification,
    * are permitted provided that the following conditions are met:
    *
    *   * 	Redistributions of source code must retain the above copyright notice,
    * 	this list of conditions and the following disclaimer.
   *
   *   *	Redistributions in binary form must reproduce the above copyright notice,
   * 	this list of conditions and the following disclaimer in the documentation
   * 	and/or other materials provided with the distribution.
   *
   *   *	Neither the name of the University of Southampton nor the names of its
   * 	contributors may be used to endorse or promote products derived from this
   * 	software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
   * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
   * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
   * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
   * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
   * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
   * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
   * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
   * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   */
  package org.openimaj.workinprogress.optimisation.params;
  
  import gnu.trove.iterator.TObjectDoubleIterator;
  import gnu.trove.map.hash.TObjectDoubleHashMap;
  import gnu.trove.procedure.TObjectDoubleProcedure;
  
  import java.util.Iterator;
  
  import org.openimaj.workinprogress.optimisation.params.KeyedParameters.ObjectDoubleEntry;
  
  public final class KeyedParameters<KEY> implements Parameters<KeyedParameters<KEY>>, Iterable<ObjectDoubleEntry<KEY>> {
  	public static class ObjectDoubleEntry<KEY> {
  		public KEY key;
  		public double value;
  	}
  
  	private TObjectDoubleHashMap<KEY> paramsMap = new TObjectDoubleHashMap<KEY>();
  
  	/*
  	 * (non-Javadoc)
  	 *
  	 * @see
  	 * org.openimaj.workinprogress.params.Parameters#multiplyInplace(org.openimaj
  	 * .workinprogress.params.KeyedParameters)
  	 */
  	@Override
  	public void multiplyInplace(final KeyedParameters<KEY> other) {
  		paramsMap.forEachEntry(new TObjectDoubleProcedure<KEY>() {
  			@Override
  			public boolean execute(KEY key, double value) {
  				if (other.contains(key)) {
  					paramsMap.put(key, other.get(key) * value);
  				}
  				return true;
  			}
  		});
  	}
  
  	public void multiplyInplace(KEY key, double value) {
  		if (paramsMap.contains(key)) {
  			paramsMap.put(key, paramsMap.get(key) * value);
  		}
  	}
  
  	@Override
  	public void addInplace(final KeyedParameters<KEY> other) {
  		other.paramsMap.forEachEntry(new TObjectDoubleProcedure<KEY>() {
  			@Override
  			public boolean execute(KEY key, double value) {
  				paramsMap.adjustOrPutValue(key, value, value);
  				return true;
  			}
  		});
  	}
  
  	public void addInplace(KEY key, double value) {
  		paramsMap.adjustOrPutValue(key, value, value);
  	}
  
  	public void set(KEY key, double value) {
  		paramsMap.put(key, value);
  	}
  
  	public double get(KEY key) {
  		return paramsMap.get(key);
  	}
  
  	public boolean contains(KEY key) {
  		return paramsMap.containsKey(key);
  	}
 
 	@Override
 	public void multiplyInplace(final double value) {
 		paramsMap.forEachEntry(new TObjectDoubleProcedure<KEY>() {
 			@Override
 			public boolean execute(KEY key, double v) {
 				paramsMap.put(key, value * v);
 				return true;
 			}
 		});
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see org.openimaj.workinprogress.params.Parameters#addInplace(KEY)
 	 */
 	@Override
 	public void addInplace(final double value) {
 		paramsMap.forEachEntry(new TObjectDoubleProcedure<KEY>() {
 			@Override
 			public boolean execute(KEY key, double v) {
 				paramsMap.put(key, value * v);
 				return true;
 			}
 		});
 	}
 
 	@Override
 	public Iterator<ObjectDoubleEntry<KEY>> iterator() {
 		return new Iterator<ObjectDoubleEntry<KEY>>() {
 			TObjectDoubleIterator<KEY> iter = paramsMap.iterator();
 			ObjectDoubleEntry<KEY> entry = new ObjectDoubleEntry<KEY>();
 
 			@Override
 			public boolean hasNext() {
 				return iter.hasNext();
 			}
 
 			@Override
 			public ObjectDoubleEntry<KEY> next() {
 				iter.advance();
 				entry.key = iter.key();
 				entry.value = iter.value();
 				return entry;
 			}
 
 			@Override
 			public void remove() {
 				throw new UnsupportedOperationException("Not supported");
 			}
 		};
 	}
 }