/* Copyright 2009-2016 David Hadka
 *
 * This file is part of the MOEA Framework.
 *
 * The MOEA Framework 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 3 of the License, or (at your
 * option) any later version.
 *
 * The MOEA Framework 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 Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with the MOEA Framework.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.moeaframework.core;

import java.util.Iterator;

import org.moeaframework.core.comparator.EpsilonBoxDominanceComparator;

/**
 * A non-dominated population using an ε-box dominance comparator.  
 * ε-Dominance archives provide several important theoretical 
 * properties, such as guaranteed convergence and diversity if certain other 
 * conditions hold [1].  In addition, this archive also tracks 
 * ε-progress [2] by counting the number of ε-box improvements.
 * <p>
 * References:
 * <ol>
 *   <li>Laumanns, M. et al.  "Combining Convergence and Diversity in
 *       Evolutionary Multi-Objective Optimization."  Evolutionary Computation,
 *       10(3):263-282, 2002.
 *   <li>Hadka, D. and Reed, P.  "Borg: An Auto-Adaptive Many-Objective
 *       Evolutionary Computing Framework."  Evolutionary Computation,
 *       21(2):231-259, 2013.
 * </ol>
 */
public class EpsilonBoxDominanceArchive extends NondominatedPopulation {

	/**
	 * The number of ε-box improvements that have occurred.
	 */
	private int numberOfImprovements;

	/**
	 * The number of ε-box improvements dominating an existing solution
	 * that have occurred.
	 */
	private int numberOfDominatingImprovements;

	/**
	 * Constructs an empty ε-box dominance archive using an additive
	 * ε-box dominance comparator with the specified ε.
	 * 
	 * @param epsilon the ε value used by the additive ε-box
	 *        dominance comparator
	 */
	public EpsilonBoxDominanceArchive(double epsilon) {
		super(new EpsilonBoxDominanceComparator(epsilon));
	}

	/**
	 * Constructs an ε-box dominance archive using an additive
	 * ε-box dominance comparator with the specified ε and
	 * initialized with the specified solutions.
	 * 
	 * @param epsilon the ε value used by the additive ε-box
	 *        dominance comparator
	 * @param iterable the solutions used to initialize this archive
	 */
	public EpsilonBoxDominanceArchive(double epsilon,
			Iterable<? extends Solution> iterable) {
		super(new EpsilonBoxDominanceComparator(epsilon), iterable);
	}
	
	/**
	 * Constructs an empty ε-box dominance archive using an additive
	 * ε-box dominance comparator with the specified ε values.
	 * 
	 * @param epsilon the ε values used by the additive ε-box
	 *        dominance comparator
	 */
	public EpsilonBoxDominanceArchive(double[] epsilon) {
		super(new EpsilonBoxDominanceComparator(epsilon));
	}
	
	/**
	 * Constructs an ε-box dominance archive using an additive
	 * ε-box dominance comparator with the specified ε values
	 * and initialized with the specified solutions.
	 * 
	 * @param epsilon the ε values used by the additive ε-box
	 *        dominance comparator
	 * @param iterable the solutions used to initialize this archive
	 */
	public EpsilonBoxDominanceArchive(double[] epsilon,
			Iterable<? extends Solution> iterable) {
		super(new EpsilonBoxDominanceComparator(epsilon), iterable);
	}

	/**
	 * Constructs an empty ε-box dominance archive using the specified
	 * ε-box dominance comparator.
	 * 
	 * @param comparator the ε-box dominance comparator used by this
	 *        archive
	 */
	public EpsilonBoxDominanceArchive(EpsilonBoxDominanceComparator comparator) {
		super(comparator);
	}

	/**
	 * Constructs an ε-box dominance archive using the specified
	 * ε-box dominance comparator and initialized with the specified
	 * solutions.
	 * 
	 * @param comparator the ε-box dominance comparator used by this
	 *        archive
	 * @param iterable the solutions used to initialize this archive
	 */
	public EpsilonBoxDominanceArchive(EpsilonBoxDominanceComparator comparator,
			Iterable<? extends Solution> iterable) {
		super(comparator, iterable);
	}

	@Override
	public boolean add(Solution newSolution) {
		Iterator<Solution> iterator = iterator();

		boolean same = false;
		boolean dominates = false;

		while (iterator.hasNext()) {
			Solution oldSolution = iterator.next();
			int flag = getComparator().compare(newSolution, oldSolution);

			if (flag < 0) {
				if (getComparator().isSameBox()) {
					same = true;
				} else {
					dominates = true;
				}

				iterator.remove();
			} else if (flag > 0) {
				return false;
			}
		}

		if (!same) {
			numberOfImprovements++;

			if (dominates) {
				numberOfDominatingImprovements++;
			}
		}

		return forceAddWithoutCheck(newSolution);
	}

	/**
	 * Returns the ε-box dominance comparator used by this archive.
	 * 
	 * @return the ε-box dominance comparator used by this archive
	 */
	@Override
	public EpsilonBoxDominanceComparator getComparator() {
		return (EpsilonBoxDominanceComparator)super.getComparator();
	}

	/**
	 * Returns the number of ε-box improvements that have occurred.
	 * 
	 * @return the number of ε-box improvements that have occurred
	 */
	public int getNumberOfImprovements() {
		return numberOfImprovements;
	}

	/**
	 * Returns the number of ε-box improvements dominating existing
	 * solutions that have occurred.
	 * 
	 * @return the number of ε-box improvements dominating existing
	 *         solutions that have occurred
	 */
	public int getNumberOfDominatingImprovements() {
		return numberOfDominatingImprovements;
	}

}