//-----------------------------------------------------------------------------
// t-eoMGE.cpp
//-----------------------------------------------------------------------------

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#ifndef __GNUG__
// to avoid long name warnings
#pragma warning(disable:4786)
#endif // __GNUG__

#include "eo"
#include "ga/eoBitOp.h"
#include "RoyalRoad.h"

// Viri
#include "VirusOp.h"
#include "eoVirus.h"
#include "eoInitVirus.h"

//-----------------------------------------------------------------------------

typedef eoVirus<double> Chrom;

//-----------------------------------------------------------------------------

int main()
{
  const unsigned POP_SIZE = 10, CHROM_SIZE = 12;
  unsigned i;
  eoBooleanGenerator gen;

  // the populations:
  eoPop<Chrom> pop;

  // Evaluation
  RoyalRoad<Chrom> rr( 8 );
  eoEvalFuncCounter<Chrom> eval( rr );

  eoInitVirus1bit<double> random(CHROM_SIZE, gen);
  for (i = 0; i < POP_SIZE; ++i) {
      Chrom chrom;
      random(chrom);
      eval(chrom);
      pop.push_back(chrom);
  }

  std::cout << "population:" << std::endl;
  for (i = 0; i < pop.size(); ++i)
    std::cout << "\t" << pop[i] << " " << pop[i].fitness() << std::endl;

  // selection
  eoStochTournamentSelect<Chrom> lottery(0.9 );

  // breeder
  VirusShiftMutation<double> vm;
  VirusTransmission<double> vt;
  VirusBitFlip<double> vf;
  eoUBitXover<Chrom> xover;
  eoProportionalOp<Chrom> propSel;
  eoGeneralBreeder<Chrom> breeder( lottery, propSel );
  propSel.add(vm, 0.8);
  propSel.add(vf, 0.05);
  propSel.add(vt, 0.05);
  propSel.add(xover, 0.1);

  // Replace a single one
  eoCommaReplacement<Chrom> replace;

  // Terminators
  eoGenContinue<Chrom> continuator1(10);
  eoFitContinue<Chrom> continuator2(CHROM_SIZE);
  eoCombinedContinue<Chrom> continuator(continuator1);
  continuator.add(continuator2);
  eoCheckPoint<Chrom> checkpoint(continuator);
  eoStdoutMonitor monitor;
  checkpoint.add(monitor);
  eoSecondMomentStats<Chrom> stats;
  eoPopStat<Chrom> dumper( 10 );
  monitor.add(stats);
  checkpoint.add(dumper);
  checkpoint.add(stats);

  // GA generation
  eoEasyEA<Chrom> ea(checkpoint, eval, breeder, replace);

  // evolution
  try {
      ea(pop);
  } catch (std::exception& e) {
      std::cerr << "exception: " << e.what() << std::endl;;
      exit(EXIT_FAILURE);
  }

  std::cout << "pop" << std::endl;
  for (i = 0; i < pop.size(); ++i)
      std::cout << "\t" <<  pop[i] << " " << pop[i].fitness() << std::endl;

  std::cout << "\n --> Number of Evaluations = " << eval.getValue() << std::endl;
  return EXIT_SUCCESS;
}

//-----------------------------------------------------------------------------



// Local Variables:
// mode: C++
// c-file-style: "Stroustrup"
// End:
