initial version of mastermind

eo/app/mastermind/mastermind.cc

@@ -0,0 +1,146 @@
+//-----------------------------------------------------------------------------
+// mastermind
+//-----------------------------------------------------------------------------
+
+#include <stdlib.h>                // EXIT_SUCCESS EXIT_FAILURE
+#include <stdexcept>               // exception 
+#include <iostream>                // cerr cout
+#include <fstream>                 // ifstream
+#include <string>                  // string
+#include <utils/eoParser.h>        // eoParser
+#include <eoPop.h>                 // eoPop
+#include <eoEvalFuncPtr.h>         // eoEvalFunc
+#include <eoProportional.h>        // eoProportional
+#include <eoGenContinue.h>         // eoGenContinue
+#include <eoFitContinue.h>         // eoFitContinue
+#include <eoCombinedContinue.h>    // eoCombinedContinue
+#include <utils/eoCheckPoint.h>    // eoCheckPoint
+#include <utils/eoStat.h>          // eoBestFitnessStat
+#include <utils/eoStdoutMonitor.h> // eoStdoutMonitor
+#include <eoSGA.h>                 // eoSGA
+#include "mastermind.h"            // Chrom eoChromInit eoChromMutation eoChromXover eoChromEvaluator
+
+//-----------------------------------------------------------------------------
+// global variables
+//-----------------------------------------------------------------------------
+
+unsigned in, out, hidden;
+
+//-----------------------------------------------------------------------------
+// parameters
+//-----------------------------------------------------------------------------
+
+eoValueParam<unsigned> pop_size(16, "pop_size", "population size", 'p');
+eoValueParam<unsigned> generations(100, "generations", "number of generation", 'g');
+eoValueParam<double> mut_rate(0.1, "mut_rate", "mutation rate", 'm');
+eoValueParam<double> xover_rate(0.5, "xover_rate", "default crossover rate", 'x');
+eoValueParam<unsigned> col_p(8, "number_of_colors", "number of colors", 'c');
+eoValueParam<unsigned> len_p(8, "solution_legth", "solution legth", 'l');
+eoValueParam<string> sol_p(default_solution, "solution", "problem solution", 's');
+
+//-----------------------------------------------------------------------------
+// auxiliar functions
+//-----------------------------------------------------------------------------
+
+void arg(int argc, char** argv);
+void ga();
+
+//-----------------------------------------------------------------------------
+// main
+//-----------------------------------------------------------------------------
+
+int main(int argc, char** argv)
+{
+  try
+    {
+      arg(argc, argv);
+      ga();
+    }
+  catch (exception& e)
+    {
+      cerr << argv[0] << ": " << e.what() << endl;
+      exit(EXIT_FAILURE);
+    }
+
+  return 0;
+}
+
+//-----------------------------------------------------------------------------
+// implementation
+//-----------------------------------------------------------------------------
+
+void arg(int argc, char** argv)
+{
+  eoParser parser(argc, argv);
+  
+  parser.processParam(pop_size,    "genetic operators");
+  parser.processParam(generations, "genetic operators");
+  parser.processParam(mut_rate,    "genetic operators");
+  parser.processParam(xover_rate,  "genetic operators");
+  parser.processParam(col_p,       "problem");
+  parser.processParam(len_p,       "problem");
+  parser.processParam(sol_p,       "problem");
+
+  if (parser.userNeedsHelp())
+    {
+      parser.printHelp(cout);
+      exit(EXIT_SUCCESS);
+    }
+
+  init_eoChromEvaluator(col_p.value(), len_p.value(), sol_p.value());
+  solution.fitness(eoChromEvaluator(solution));
+}
+
+//-----------------------------------------------------------------------------
+
+void ga()
+{
+  // create population
+  eoInitChrom init;
+  eoPop<Chrom> pop(pop_size.value(), init);
+  
+  // evaluate population
+  eoEvalFuncPtr<Chrom> evaluator(eoChromEvaluator);
+  apply<Chrom>(evaluator, pop);
+  
+  // selector
+  eoProportional<Chrom> select(pop);
+
+  // genetic operators
+  eoChromMutation mutation;
+  eoChromXover xover;
+  
+  // stop condition
+  eoGenContinue<Chrom> continuator1(generations.value());
+  eoFitContinue<Chrom> continuator2(solution.fitness());  
+  eoCombinedContinue<Chrom> continuator(continuator1, continuator2);
+
+  // checkpoint
+  eoCheckPoint<Chrom> checkpoint(continuator);
+    
+  // monitor
+  eoStdoutMonitor monitor;
+  checkpoint.add(monitor);
+  
+  // statistics
+  eoBestFitnessStat<Chrom> stats;
+  checkpoint.add(stats);
+  monitor.add(stats);
+    
+  // genetic algorithm
+  eoSGA<Chrom> sga(select,
+		   xover, xover_rate.value(), 
+		   mutation, mut_rate.value(), 
+		   evaluator, 
+		   checkpoint);
+  sga(pop);
+  
+  cout << "solution = " << solution << endl
+       << "best     = " << *max_element(pop.begin(), pop.end()) << endl;
+}
+
+//-----------------------------------------------------------------------------
+
+// Local Variables: 
+// mode:C++
+// End: