eoFlOrMonOp.h

// -*- mode: c++; c-indent-level: 4; c++-member-init-indent: 8; comment-column: 35; -*-

//-----------------------------------------------------------------------------
// eoFlOrMonOp.h
// (c) Marc Schoenauer - Maarten Keijzer 2000-2003
/*
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    Contact: Marc.Schoenauer@polytechnique.fr
             mkeijzer@cs.vu.nl
 */
//-----------------------------------------------------------------------------

#ifndef _eoFlOrMonOp_h
#define _eoFlOrMonOp_h

#include <eoFunctor.h>
#include <eoOp.h>
#include <eoInit.h>

template <class EOT>
class eoFlOrAllMutation : public eoMonOp<EOT>
{
public :

  typedef typename EOT::AtomType AtomType;

  eoFlOrAllMutation(eoMonOp<AtomType> & _atomMutation, double _rate=1.0) :
    atomMutation(_atomMutation), rate(_rate) {}

  bool operator()(EOT & _eo)
  {
    bool modified=false;
    for (unsigned i=0; i<_eo.size(); i++)
      if (eo::rng.flip(rate))
        if (atomMutation(_eo[i]))
          modified = true;

    return modified;
  }

  virtual std::string className() const
  {
    return "eoFlOrAllMutation(" + atomMutation.className() + ")";
  }

private:
  eoMonOp<AtomType> & atomMutation; // the atom mutation
  double rate;                     // the mutation rate PER ATOM
};

template <class EOT>
class eoFlOrKMutation : public eoMonOp<EOT>
{
public :

  typedef typename EOT::AtomType AtomType;

  eoFlOrKMutation(eoMonOp<AtomType> & _atomMutation, unsigned _nb=1) :
    nb(_nb), atomMutation(_atomMutation) {}


  bool operator()(EOT & _eo)
  {
    bool modified=false;
    for (unsigned k=0; k<nb; k++)
      {
        unsigned i = rng.random(_eo.size()); // we don't test for duplicates...
        if (atomMutation(_eo[i]))
          modified = true;
      }
    return modified;
  }

  virtual std::string className() const
  {
    return "eoFlOrKMutation(" + atomMutation.className() + ")";
  }

private:
  unsigned nb;                     // the number of atoms to mutate
  eoMonOp<AtomType> & atomMutation; // the atom mutation
};

#endif