/***************************************************************************
                          main.cpp  -  description
                             -------------------
    begin                : Thu Oct  3 11:44:13 CEST 2002
    copyright            : (C) 2002 by Rolland Balzon Philippe
    email                : prolland@free.fr
 ***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

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

#include <stdlib.h>
#include <map>
#include <iostream.h>
#include <fstream.h>
#include <GASimpleGA.h>	// the header for the GA we'll use
#include <GA2DBinStrGenome.h> // and the header for the genome we need
#include <graphBin.h>

#define pow2i(x)   ((x)<0 ? 0 : (1<<(x)))

float objective(GAGenome &);
int genome_bin2dec (GAGenome & ,int );

int
main(int argc, char *argv[])
{
  cout << "Example \n\n";
  cout << "This program reads in a data file a ascii \n";
  cout << "graph file then runs a simple GA whose \n";
  cout << "objective function tries to color the graph\n";
  cout << "By option option: f path_of_your_file_graph.txt you can load your graph.\n\n";

// 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.

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

//   Set the default values of the parameters and declare the params variable.
// We use the genetic algorithm's configure member to set up the parameter list
// so that it will parse for the appropriate arguments.  Notice that the
// params argument 'removes' from the argv list any arguments that it
// recognized (actually it just re-orders them and changes the value of argc).
//   Once the GA's parameters are registered, we set some values that we want
// that are different than the GAlib defaults.  Then we parse the command line.

  //int popsize  = 100;
  int popsize  = 400;
  //int ngen     = 100;
  int ngen     = 5000;
  //float pmut   = 0.001;
  float pmut   = 0.01;
  //float pcross = 0.9;
  float pcross = 0.6;

  GAParameterList params;
  GASimpleGA::registerDefaultParameters(params);
  params.set(gaNscoreFilename, "bog.dat");
  params.set(gaNflushFrequency, 5);
  params.set(gaNpMutation, 0.001);
  params.set(gaNpCrossover, 0.8);
  params.parse(argc, argv, gaFalse);



/*
  params.set(gaNflushFrequency, 50);
  params.set(gaNpMutation, 0.001);
  params.set(gaNpCrossover, 0.8);
  params.parse(argc, argv, gaFalse);
*/


  //char filename[128] = "graph20.txt";
  char filename[128] = "graph18.txt";
  //char filename[128] = "graph5.txt";

// Parse the command line for arguments.

  for(int i=1; i<argc; i++){
    if(strcmp("file", argv[i]) == 0 || strcmp("f", argv[i]) == 0){
      if(++i >= argc){
        cerr << argv[0] << ": the file option needs a filename.\n";
        exit(1);
      }
      else{
	sprintf(filename, argv[i]);
        continue;
      }
    }
    else if(strcmp("seed", argv[i]) == 0){
      if(++i < argc) continue;
      continue;
    }
    else {
      cerr << argv[0] << ":  unrecognized arguement: " << argv[i] << "\n\n";
      cerr << "valid arguments include standard GAlib arguments plus:\n";
      cerr << "  f\tfilename from which to read (" << filename << ")\n";
      cerr << "\n";
      exit(1);
    }
  }

  ManyGraphBin aGraphSet(filename);
  //aGraphSet.display_all_graphBin();
  printf("\n _nb_graph=%d\n", aGraphSet._nb_graph);

  //printf("\n _nb_graph=%d\n", aGraphSet._nb_graph);
 // for(int g=0;g < aGraphSet._nb_graph;g++){
   for(int g=0;g < 1;g++){
    OneGraphBin * aGraphBin = new OneGraphBin();
    aGraphBin = aGraphSet.selectOneGraphBin(g);
    short **target=aGraphBin->_mat;
    int width=aGraphBin->_n;
    int height=width;

    GA2DBinaryStringGenome genome(width, height, objective, (void *)target);
    GASimpleGA ga(genome);
    GASigmaTruncationScaling scaling;
    ga.parameters(params);
    ga.populationSize(popsize);
    ga.nGenerations(ngen);
    ga.pMutation(pmut);
    ga.pCrossover(pcross);
    ga.scaling(scaling);
    ga.evolve();
    //cout << "best of generation data are in '" << ga.scoreFilename() << "'\n";

    genome = ga.statistics().bestIndividual();
    aGraphBin->coloration (genome) ;
    aGraphBin->nb_color(genome.nb_color);
    //printf("\n    genome.nb_color %d",genome.nb_color);
    //cout << "the ga generated:\n";
    //aGraphSet.display_target (aGraphSet.selectOneGraphBin(g));
    //exit(0);
    }
  aGraphSet.display_all_graphBin();
  return 0;


}


int
genome_bin2dec (GAGenome & c,int k)
{
	GA2DBinaryStringGenome & genome = (GA2DBinaryStringGenome &)c;
	int temp =0;
 	for (int i=0; i<genome.width(); i++)
        	temp+=pow2i(genome.width()-i-1)*int(genome.gene(i,k));
	return temp;
}


// For the objective function we compare the contents of the genome with the
// target.  If a bit is set in the genome and it is also set in the target,
// then we add 1 to the score.  If the bits do not match, we don't do anything.

float
objective(GAGenome & c)
{
  GA2DBinaryStringGenome & genome = (GA2DBinaryStringGenome &)c;
  short **pattern = (short **)c.userData();
  int tmp,* tab = new int[genome.width()+1];

  float value=0.0;
    for(int j=0; j<genome.height(); j++){
	genome.color[j]=genome_bin2dec(genome,j);
    tab[j]=genome.color[j];
        for(int i=0; i<genome.width(); i++){
        // on selectionne les colorisation sans conflit
        if (pattern[i][j]==1){
                if ( genome.color[j] != genome.color[i])
			          value++;
		          else  {
			          value--;
                   //value=0;
                    }
		   }
	  else {
	    if ( genome.color[j] == genome.color[i])
			value+=2;
      else
          value--;
          //value=0;
		  }
	}//for(int i=0; i<genome.width(); i++)
	}//for(int j=0; j<genome.height(); j++)

    for(int i=0;i<genome.width();i++)   {
      for(int j=0;j<genome.width()-i;j++) {
        if (tab[j]>tab[j+1]){tmp= tab[j+1];tab[j+1]=tab[j];tab[j]=tmp;}
      }
    }
    genome.nb_color=0;
    //printf("\n\n");
     for(int i=0;i<genome.width();i++)   {
      //printf ("\n %d %d",i,tab[i]);
      if  (tab[i]!=tab[i+1])  genome.nb_color++;
      }

  //printf("\n genome.nb_color=%d",genome.nb_color);
  value+= (float) 1 / genome.nb_color;
  //printf("\n value=%f",value);
  // exit(0);
  return(value);
}