/*
 * Encog(tm) Core v3.4 - Java Version
 * http://www.heatonresearch.com/encog/
 * https://github.com/encog/encog-java-core
 
 * Copyright 2008-2016 Heaton Research, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *   
 * For more information on Heaton Research copyrights, licenses 
 * and trademarks visit:
 * http://www.heatonresearch.com/copyright
 */
package org.encog.ml.ea.opp.selection;

import java.io.Serializable;
import java.util.Random;

import org.encog.ml.ea.genome.Genome;
import org.encog.ml.ea.species.Species;
import org.encog.ml.ea.train.EvolutionaryAlgorithm;
import org.encog.ml.ea.train.basic.BasicEA;

/**
 * Tournament select can be used to select a fit (or unfit) genome from a
 * species. The selection is run a set number of rounds. Each round two random
 * participants are chosen. The more fit participant continues to the next
 * round.
 *
 * http://en.wikipedia.org/wiki/Tournament_selection
 *
 */
public class TournamentSelection implements SelectionOperator, Serializable {

	/**
	 * The serial id.
	 */
	private static final long serialVersionUID = 1L;

	/**
	 * The trainer being used.
	 */
	private EvolutionaryAlgorithm trainer;

	/**
	 * The number of rounds.
	 */
	private int rounds;

	/**
	 * Construct a tournament selection.
	 * @param theTrainer The trainer.
	 * @param theRounds The number of rounds to use.
	 */
	public TournamentSelection(final EvolutionaryAlgorithm theTrainer,
			final int theRounds) {
		this.trainer = theTrainer;
		this.rounds = theRounds;
	}

	/**
	 * @return The number of rounds.
	 */
	public int getRounds() {
		return this.rounds;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public EvolutionaryAlgorithm getTrainer() {
		return this.trainer;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public int performAntiSelection(final Random rnd, final Species species) {
		int worstIndex = rnd.nextInt(species.getMembers().size());
		Genome worst = species.getMembers().get(worstIndex);
		BasicEA.calculateScoreAdjustment(worst,
				this.trainer.getScoreAdjusters());

		for (int i = 0; i < this.rounds; i++) {
			final int competitorIndex = rnd.nextInt(species.getMembers().size());
			final Genome competitor = species.getMembers().get(competitorIndex);

			// force an invalid genome to lose
			if (Double.isInfinite(competitor.getAdjustedScore())
					|| Double.isNaN(competitor.getAdjustedScore())) {
				return competitorIndex;
			}

			BasicEA.calculateScoreAdjustment(competitor,
					this.trainer.getScoreAdjusters());
			if (!this.trainer.getSelectionComparator().isBetterThan(competitor,
					worst)) {
				worst = competitor;
				worstIndex = competitorIndex;
			}
		}
		return worstIndex;
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public int performSelection(final Random rnd, final Species species) {
		int bestIndex = rnd.nextInt(species.getMembers().size());
		Genome best = species.getMembers().get(bestIndex);
		BasicEA.calculateScoreAdjustment(best, this.trainer.getScoreAdjusters());

		for (int i = 0; i < this.rounds; i++) {
			final int competitorIndex = rnd.nextInt(species.getMembers().size());
			final Genome competitor = species.getMembers().get(competitorIndex);

			// only evaluate valid genomes
			if (!Double.isInfinite(competitor.getAdjustedScore())
					&& !Double.isNaN(competitor.getAdjustedScore())) {
				BasicEA.calculateScoreAdjustment(competitor,
						this.trainer.getScoreAdjusters());
				if (this.trainer.getSelectionComparator().isBetterThan(
						competitor, best)) {
					best = competitor;
					bestIndex = competitorIndex;
				}
			}
		}
		return bestIndex;
	}

	/**
	 * Set the number of rounds.
	 * @param rounds The number of rounds.
	 */
	public void setRounds(final int rounds) {
		this.rounds = rounds;
	}

	/**
	 * Set the trainer.
	 * @param trainer The trainer.
	 */
	public void setTrainer(final EvolutionaryAlgorithm trainer) {
		this.trainer = trainer;
	}
}