/* ----------------------------------------------------------------------------
  btest.C   ( a slight modification of ex18.C)
---------------------------------------------------------------------------- */
#include <stdio.h>
#include <stdlib.h>
//#include <iostream.h>
//#include <fstream.h>
#include <ga/ga.h>
#include <ga/GARealGenome.h>
#include <ga/GARealGenome.C>

float objective(GAGenome &);

int
main(int argc, char *argv[])
{
  cout << "My test program for realGA\n\n";
  cout.flush();

// See if we've been given a seed to use (for testing purposes).  When you
// specify a random seed, the evolution will be exactly the same each time
// you use that seed number.

  unsigned int seed = 0;
  for(int ii=1; ii<argc; ii++) {
    if(strcmp(argv[ii++],"seed") == 0) {
      seed = atoi(argv[ii]);
    }
  }

// Set the default values of the parameters and declare the params variable.

  GAParameterList params;
  //GASimpleGA::registerDefaultParameters(params);
  //GASteadyStateGA::registerDefaultParameters(params);
  GAIncrementalGA::registerDefaultParameters(params);
  params.set(gaNpopulationSize, 30);	// population size
  params.set(gaNpCrossover, 0.9);	// probability of crossover
  params.set(gaNpMutation, 0.01);	// probability of mutation
  params.set(gaNnGenerations, 400);	// number of generations
  params.set(gaNpReplacement, 0.25);	// how much of pop to replace each gen
  params.set(gaNscoreFrequency, 10);	// how often to record scores
  params.set(gaNflushFrequency, 50);	// how often to dump scores to file
  params.set(gaNscoreFilename, "bog.dat");
  params.parse(argc, argv, gaFalse);    // parse command line for GAlib args


// This genome is created using an array of allele sets.  This means that each
// element of the genome will assume a value in its corresponding allele set.
// For example, since the first allele set is [0,10], the first element of the
// genome will be in [0,10].  Notice that you can add allele sets in many other
// ways than those shown.

  GARealAlleleSetArray myallele;
  /*
  myallele.add(-5,10);
  myallele.add(-10,5);  
  myallele.add(-20,5);  
  myallele.add(-1,1);
  myallele.add(-15,33);  
  */
  GARealGenome genome(myallele, objective);

  //GASimpleGA ga(genome);
  //GASteadyStateGA ga(genome);
  GAIncrementalGA ga(genome);
  ga.minimize();
  ga.parameters(params);
  ga.evolve(seed);      
  genome = ga.statistics().bestIndividual();

  cout << "\nthe best indiv: " << genome << "\n";
  cout.flush();
  return 0;
}
 


float objective(GAGenome& g)
{
  GARealGenome& genome = (GARealGenome&)g;
  float value=0.0;
  for(int i=0; i<genome.length(); i++)
    value += ((genome.gene(i)) * (genome.gene(i)));
  return value;
}