initial version of mastermind

eo/app/mastermind/mastermind.h

@@ -0,0 +1,158 @@
+//-----------------------------------------------------------------------------
+// mastermind.h
+//-----------------------------------------------------------------------------
+
+#ifndef mastermind_h
+#define mastermind_h
+
+//-----------------------------------------------------------------------------
+
+#include <stdlib.h>               // exit EXIT_FAILURE
+#include <eoFixedLength.h>        // eoFixedLength
+#include <eoOp.h>                 // eoMonOp eoQuadraticOp
+#include <eoInit.h>               // eoInit
+#include <utils/rnd_generators.h> // uniform_generator
+
+//-----------------------------------------------------------------------------
+// phenotype
+//-----------------------------------------------------------------------------
+
+typedef float phenotype;
+
+//-----------------------------------------------------------------------------
+// genotype
+//-----------------------------------------------------------------------------
+
+typedef vector<int> genotype;
+
+//-----------------------------------------------------------------------------
+// Chrom
+//-----------------------------------------------------------------------------
+
+typedef eoFixedLength<phenotype, int> Chrom;
+
+//-----------------------------------------------------------------------------
+// eoChromEvaluator
+//-----------------------------------------------------------------------------
+
+const unsigned default_size = 8;
+const string default_solution = "01234567";
+
+Chrom solution;
+unsigned num_colors;
+
+void init_eoChromEvaluator(const unsigned& c, const unsigned& l, string s)
+{
+  num_colors = c;
+
+  // generate a random solution
+  if (s == default_solution || s.size() != l)
+    {
+      uniform_generator<char> color('0', static_cast<char>('0' + num_colors));
+      s.resize(l);
+      generate(s.begin(), s.end(), color);
+    }
+
+  // check solution
+  for (unsigned i = 0; i < solution.size(); ++i)
+    if (solution[i] >= num_colors)
+      {
+	cerr << "too high color number found!" << endl;
+	exit(EXIT_FAILURE);
+      }
+  
+  solution.resize(s.size());
+  for (unsigned i = 0; i < solution.size(); ++i)
+    solution[i] = s[i] - '0';
+}
+
+const unsigned points_per_black = 3, points_per_white = 1;
+
+phenotype eoChromEvaluator(const Chrom& chrom)
+{
+  Chrom tmp = solution;
+  unsigned black = 0, white = 0;
+
+  // look for blacks
+  for (unsigned i = 0; i < chrom.size(); ++i)
+    if (chrom[i] == tmp[i])
+      {
+	++black;
+	tmp[i] = -1;
+      }
+  
+  // look for whites
+  for (unsigned i = 0; i < chrom.size(); ++i)
+    for (unsigned j = 0; j < tmp.size(); ++j)
+      if (chrom[i] == tmp[j])
+	{
+	  ++white;
+	  tmp[j] = -1;
+	  break;
+	}
+
+  //  return black * points_per_black + white * points_per_white;
+  return black * chrom.size() + white;
+};
+
+//-----------------------------------------------------------------------------
+// eoChromInit
+//-----------------------------------------------------------------------------
+
+class eoInitChrom: public eoInit<Chrom>
+{
+public:
+  void operator()(Chrom& chrom)
+  {
+    uniform_generator<int> color(0, num_colors);
+    chrom.resize(solution.size());
+    generate(chrom.begin(), chrom.end(), color);
+    chrom.invalidate();
+  }
+};
+
+//-----------------------------------------------------------------------------
+// eoChromMutation
+//-----------------------------------------------------------------------------
+
+class eoChromMutation: public eoMonOp<Chrom>
+{
+  // two changes in one mutation :(
+  void operator()(Chrom& chrom)
+  {
+    uniform_generator<unsigned> position(0, chrom.size());
+
+    // random gene change
+    uniform_generator<int> color(0, num_colors);
+    chrom[position()] = color();
+    
+    // random gene swap
+    swap(chrom[position()], chrom[position()]);
+
+    chrom.invalidate();
+  }
+};
+
+//-----------------------------------------------------------------------------
+// eoChromXover
+//-----------------------------------------------------------------------------
+
+class eoChromXover: public eoQuadraticOp<Chrom>
+{
+public:
+  void operator()(Chrom& chrom1, Chrom& chrom2)
+  {
+    uniform_generator<unsigned> position(0, chrom1.size());
+    swap_ranges(chrom1.begin(), chrom1.begin() + position(), chrom2.begin());
+    chrom1.invalidate();
+    chrom2.invalidate();
+  }
+};
+
+//-----------------------------------------------------------------------------
+
+#endif // mastermind_h
+
+// Local Variables: 
+// mode:C++ 
+// End: