<?php define('__ROOT__', dirname(dirname(__FILE__)) );

	require_once __ROOT__.'/autoloader.php';

	use \neuralnetwork\core\Genome;
	use \neuralnetwork\core\NeuralNetwork;
	use \filemanager\core\FileManager;

	function behaviourtest1($in){ return [$in[0] + $in[1] - $in[2]]; }
	function behaviourtest2($in){ return [ 2*pow($in[0], 2) - 5*$in[1] + 8*$in[2]]; }


	$train = $argc > 1 && $argv[1] == 'train';
	$guess = !$train;

	if( $train && 'learning_process' ){

		$part = 1;


		echo "Welcome to neural-network.php\n";
		echo "-----------------------------\n\n";

		/* [1] Trying to load neural network
		=========================================================*/
		try{

			$nn = NeuralNetwork::load('test2/test2');
			echo "$part. NeuralNetwork loaded from 'test2/test2'\n"; $part++;

		/* [2] Else, creates it
		=========================================================*/
		}catch(\Exception $e){

			$nn = NeuralNetwork::create(50, 100);

			$nn->setHiddenLayersCount(5);       // 3 Hidden layers
			$nn->setHiddenLayerNeuronsCount(3); // Composed with 3 neurons each
			$nn->setInputLayerCount(3);         // 3 inputs
			$nn->setOutputLayerCount(1);        // 1 output
			$nn->setMutationThreshold(0.5);     // mutation 30% each generation
			$nn->setFitnessEnd(-1.5);           // Algorithm is done when fitness reaches 0
			$nn->setAntiRegression(true);       // That repeats a generation while its fitness is lower than the previous one

			echo "$part. NeuralNetwork configured\n"; $part++;

			$d = [0, 0, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 0, 1]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 1, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 1, 1]; $nn->addSample($d, behaviourtest2($d));
			$d = [1, 0, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [1, 0, 1]; $nn->addSample($d, behaviourtest2($d));
			$d = [1, 1, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [1, 1, 1]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 0, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 0, 2]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 2, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [0, 2, 2]; $nn->addSample($d, behaviourtest2($d));
			$d = [2, 0, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [2, 0, 2]; $nn->addSample($d, behaviourtest2($d));
			$d = [2, 2, 0]; $nn->addSample($d, behaviourtest2($d));
			$d = [2, 2, 2]; $nn->addSample($d, behaviourtest2($d));
			echo "$part. Samples added to NeuralNetwork\n"; $part++;

			$nn->store('test2/test2', true);
			echo "$part. NeuralNetwork stored to 'test2/test2'\n"; $part++;

		}

		/* [2] Initializing learning routine
		=========================================================*/
		$fitness = 0;
		$max_fit = 0;
		$nn->loadLearningRoutine(function($input, $output){
			global $fitness;
			$fitness -= abs(round($output[0]) - behaviourtest2($input)[0]);
		});
		echo "$part. Learning routine initialized.\n"; $part++;


		/* [3] Learning through generations and genomes
		=========================================================*/
		/* (1) For each generation */
		$last_gnr = -1;
		$gen_repeat = 0;
		while( true ){

			if( $nn->gnr > $last_gnr)
				$start = microtime(true);

			$last_gnr = $nn->gnr;

			$max_fit = -1e9;

			/* (2) For each genome */
			while( true ){
				$fitness = 0;

				/* (2.1) Get current genome */
				$g = $nn->getGenome();
				echo "\r[x] gnm ".($nn->gnm+1)."/100 on gnr ".($nn->gnr+1)."/50 - x".($gen_repeat+1)." - fit[$max_fit]                 ";

				/* (2.2) Train genome with random samples */
				for( $r = 0 ; $r < 500 ; $r++ )
					$g->train([rand(0,100), rand(0,100), rand(0,100)]);

				/* (2.3) Set fitness & go to next genome */
				if( $fitness > $max_fit ) $max_fit = $fitness;

				$g->setFitness($fitness);

				if( $nn->gnm >= 100-1 )
					break;

				$nn->nextGenome();
			}

			$nn->nextGenome();


			// If generation evolution, notify
			if( $nn->gnr > $last_gnr){
				echo "\n\t".((microtime(true)-$start))."s\n";
				$gen_repeat = 0;
			}else $gen_repeat++;


			if( is_null($nn->gnr) || $nn->gnr == 50-1 )
				break;

		}


	}


	if( $guess && 'guessing_process' ){
		$part = 1;


		echo "Welcome to neural-network.php\n";
		echo "-----------------------------\n\n";

		/* [1] Trying to load neural network
		=========================================================*/
		try{

			$nn = NeuralNetwork::load('test2/test2');
			echo "$part. NeuralNetwork loaded from 'test2/test2'\n"; $part++;

		/* [2] Else, creates it
		=========================================================*/
		}catch(\Exception $e){

			echo "You must create/train your neural network before using it.\n";
			exit();
		}


		/* [2] Fetch trained genome
		=========================================================*/
		$genome = $nn->getTrainedGenome();
		$genome->setCallback(function($in, $out){
			echo "callback input:  ".implode(',', $in)."\n";
			echo "callback output: ".$out[0]."\n";
			echo "callback result: ".implode(',', behaviourtest2($in))."\n";
		});

		$genome->train([rand(0,10), rand(0,10), rand(0,10)]);


	}


	// REWRITE TEST
	// for( $a = 0, $al = 50 ; $a < $al ; $a++ )
	// 	for( $b = 0, $bl = 20 ; $b < $bl ; $b++ ){
	// 		print "genome $b/$bl on generation $a/$al     \r";
	// 		usleep(1000*10);
	// 	}

?>