package jcgp.tests;

import static org.junit.Assert.*;

import java.util.Random;

import jcgp.Parameters;
import jcgp.Utilities;
import jcgp.fitness.ParameterMismatchException;
import jcgp.function.Addition;
import jcgp.function.FunctionSet;
import jcgp.function.Subtraction;
import jcgp.population.Chromosome;
import jcgp.population.Connection;
import jcgp.population.Input;
import jcgp.population.MutableElement;
import jcgp.population.Node;
import jcgp.population.Output;

import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * 
 * Tests which cover the behaviour specified for a chromosome.
 * 
 *  - The chromosome should be able to return a specified node, input or output.
 *  - It should be able to return a random MutableElement.
 *  - It should be able to return a random allowed connection given a column.
 *  - It should be able to return a random connection.
 *  - It should contain a freely modifiable fitness value.
 *  - For truth table evaluations, it should be able to have its inputs set.
 *  - For truth table evaluations, the output should return a value according to what
 *    it is set to.
 *  - It should feature a clone constructor, which creates a deep copy of a 
 *    specified Chromosome object.
 *  - It should be able to return a list of active nodes.
 *  
 *  TODO: bashing (strange value ranges, etc)
 *  
 *  WARNING: changing parameters may cause the tests to incorrectly fail!
 * 
 * @author Eduardo Pedroni
 *
 */
public class ChromosomeTests {

	private Chromosome chromosome;

	@BeforeClass
	public static void setUpBeforeClass() {
		// initialise function set
		FunctionSet functionSet = new FunctionSet(new Addition(), new Subtraction());

		// initialise utilities
		Utilities.setResources(new Random(1234), functionSet);

		// initialise parameters
		Parameters.setColumns(1);
		Parameters.setRows(20);
		Parameters.setInputs(2);
		Parameters.setOutputs(4);
		Parameters.setLevelsBack(1);
		Parameters.setMutationRate(10);
		Parameters.setTotalGenerations(100);
		Parameters.setTotalRuns(5);
		Parameters.setMaxArity(functionSet.getMaxArity());
	}

	@Before
	public void setUp() throws Exception {
		chromosome = new Chromosome();
	}

	/**
	 * 
	 */
	@Test
	public void cloneTest() {
		int[] testInputs;
		// create a clone, check to see if it really is a clone
		Chromosome clone = new Chromosome(chromosome);

		// set input values
		testInputs = new int[Parameters.getInputs()];
		for (int i = 0; i < Parameters.getInputs(); i++) {
			testInputs[i] = i * 2 - 3;
		}
		chromosome.setInputs(testInputs);
		clone.setInputs(testInputs);

		// connect outputs to inputs, check that the outputs match
		for (int i = 0; i < Parameters.getOutputs(); i++) {
			chromosome.getOutput(i).setConnection(chromosome.getInput(i % Parameters.getInputs()));
			clone.getOutput(i).setConnection(clone.getInput(i % Parameters.getInputs()));
			assertTrue("Incorrect output returned.", chromosome.getOutput(i).calculate() == 
					clone.getOutput(i).calculate());
		}

		// change clone inputs, outputs should no longer match
		testInputs = new int[Parameters.getInputs()];
		for (int i = 0; i < Parameters.getInputs(); i++) {
			testInputs[i] = i * 2;
		}
		clone.setInputs(testInputs);
		for (int i = 0; i < Parameters.getOutputs(); i++) {
			assertTrue("Incorrect output returned.", chromosome.getOutput(i).calculate() != 
					clone.getOutput(i).calculate());
		}
	}

	/**
	 * 
	 */
	@Test
	public void fitnessTest() {
		// set a fitness value, check if returned value is the same
		chromosome.setFitness(10);
		assertTrue("Incorrect fitness returned.", chromosome.getFitness() == 10);
	}

	/**
	 * 
	 */
	@Test
	public void randomConnectionTest() {
		// get random connections with column 0, check that they are all inputs
		for (int i = 0; i < 10000; i++) {
			boolean connectionReturn = chromosome.getRandomConnection(0) instanceof Input;
			assertTrue("Connection is not an input.", connectionReturn);
		}

		// get random connections with column 1
		// they should all be nodes, and their columns should be within range
		int connectionNodes = 0, connectionOutOfRange = 0, connectionInputs = 0, connectionPicks = 100000;
		int chosenColumn = 1;
		for (int i = 0; i < connectionPicks; i++) {
			Connection c = chromosome.getRandomConnection(chosenColumn);
			if (c instanceof Node) {
				connectionNodes++;
				if (((Node) c).getColumn() >= chosenColumn) {
					connectionOutOfRange++;
				}
				assertTrue("Connection is not allowed : " + ((Node) c).getColumn(), ((Node) c).getColumn() < chosenColumn && ((Node) c).getColumn() < chosenColumn);
			} else if (c instanceof Input) {
				connectionInputs++;
			} else {
				fail("Return is neither Node nor Input.");

			}	
		}
		System.out.println("Out of " + connectionPicks + " connections picked from " + ((chosenColumn >= Parameters.getLevelsBack()) ? Parameters.getLevelsBack() : chosenColumn) * Parameters.getRows() +
				" allowed nodes and " + Parameters.getInputs() + " inputs, " + connectionNodes + " were nodes and " + connectionInputs + " were inputs.");

		System.out.println("Node/input ratio: " + ((double) ((chosenColumn >= Parameters.getLevelsBack()) ? Parameters.getLevelsBack() : chosenColumn) * Parameters.getRows()) / (double) Parameters.getInputs() +
				", picked ratio: " + (double) connectionNodes / (double) connectionInputs);

		System.out.println(connectionOutOfRange + " nodes that disrespected levels back were picked.");
	}

	/**
	 * 
	 */
	@Test
	public void randomMutableTest() {
		// get mutable elements, check Node to Output ratio
		int mutablePicks = 100000;
		int mutableNodes = 0, mutableOutputs = 0;
		for (int i = 0; i < mutablePicks; i++) {
			MutableElement m = chromosome.getRandomMutableElement();

			if (m instanceof Node) {
				mutableNodes++;
			} else if (m instanceof Output) {
				mutableOutputs++;
			} else {
				fail("Return is neither Node nor Output.");
			}
		}
		System.out.println("Out of " + mutablePicks + " mutable elements picked from " + Parameters.getNodeCount() +
				" nodes and " + Parameters.getOutputs() + " outputs, " + mutableNodes + " were nodes and " +
				mutableOutputs + " were outputs.");
		System.out.println("Node/output ratio: " + (double) Parameters.getNodeCount() / (double) Parameters.getOutputs() +
				", picked ratio: " + (double) mutableNodes / (double) mutableOutputs + "\n");
	}

	/**
	 * 
	 */
	@Test
	public void getOutputsTest() {
		// connect outputs to inputs, check that calculated outputs return input values
		for (int i = 0; i < Parameters.getOutputs(); i++) {
			chromosome.getOutput(i).setConnection(chromosome.getInput(i % Parameters.getInputs()));
			assertTrue("Incorrect output returned.", chromosome.getOutput(i).calculate() == 
					chromosome.getInput(i % Parameters.getInputs()).getValue());
		}
	}

	/**
	 * @throws ParameterMismatchException
	 */
	@Test
	public void setInputTest() throws ParameterMismatchException {
		// set input values, check that acquired values are correct
		int[] testInputs = new int[Parameters.getInputs()];
		for (int i = 0; i < Parameters.getInputs(); i++) {
			testInputs[i] = i * 2 - 3;
		}
		chromosome.setInputs(testInputs);
		for (int i = 0; i < Parameters.getInputs(); i++) {
			assertTrue("Incorrect input returned.", chromosome.getInput(i).getValue() == i * 2 - 3);
		}
	}

	/**
	 * 
	 */
	@Test
	public void getNodeTest() {
		// get all nodes one by one, check that they are all correct
		for (int r = 0; r < Parameters.getRows(); r++) {
			for (int c = 0; c < Parameters.getColumns(); c++) {
				assertTrue("Incorrect node returned.", chromosome.getNode(r, c).getColumn() == c &&
						chromosome.getNode(r, c).getRow() == r);
			}
		}
	}

	/**
	 * 
	 */
	@Test
	public void activeNodeTest() {
		// active node detection happens recursively, the user only calls a single method
		// set connections to a known configuration
		for (int i = 0; i < Parameters.getOutputs(); i++) {
			chromosome.getOutput(i).setConnection(chromosome.getNode(0, 0));
		}

		chromosome.getNode(0, 0).setConnection(chromosome.getInput(0));

		assertTrue("Active connection not in list.", chromosome.getActiveNodes().contains(chromosome.getInput(0)));
		assertTrue("Active connection not in list.", chromosome.getActiveNodes().contains(chromosome.getNode(0, 0)));

		// change outputs, print list
		chromosome.getOutput(0).setConnection(chromosome.getNode(0, Parameters.getColumns() - 1));

		System.out.println("Active connections: " + chromosome.getActiveNodes().toString() + "\n");
	}
}