moeoFastNonDominatedSortingFitnessAssignment.h

/*
* <moeoFastNonDominatedSortingFitnessAssignment.h>
* Copyright (C) DOLPHIN Project-Team, INRIA Futurs, 2006-2007
* (C) OPAC Team, LIFL, 2002-2007
*
* Arnaud Liefooghe
*
* This software is governed by the CeCILL license under French law and
* abiding by the rules of distribution of free software.  You can  use,
* modify and/ or redistribute the software under the terms of the CeCILL
* license as circulated by CEA, CNRS and INRIA at the following URL
* "http://www.cecill.info".
*
* As a counterpart to the access to the source code and  rights to copy,
* modify and redistribute granted by the license, users are provided only
* with a limited warranty  and the software's author,  the holder of the
* economic rights,  and the successive licensors  have only  limited liability.
*
* In this respect, the user's attention is drawn to the risks associated
* with loading,  using,  modifying and/or developing or reproducing the
* software by the user in light of its specific status of free software,
* that may mean  that it is complicated to manipulate,  and  that  also
* therefore means  that it is reserved for developers  and  experienced
* professionals having in-depth computer knowledge. Users are therefore
* encouraged to load and test the software's suitability as regards their
* requirements in conditions enabling the security of their systems and/or
* data to be ensured and,  more generally, to use and operate it in the
* same conditions as regards security.
* The fact that you are presently reading this means that you have had
* knowledge of the CeCILL license and that you accept its terms.
*
* ParadisEO WebSite : http://paradiseo.gforge.inria.fr
* Contact: paradiseo-help@lists.gforge.inria.fr
*
*/
//-----------------------------------------------------------------------------

#ifndef MOEOFASTNONDOMINATEDSORTINGFITNESSASSIGNMENT_H_
#define MOEOFASTNONDOMINATEDSORTINGFITNESSASSIGNMENT_H_

#include <vector>
#include <eoPop.h>
#include <comparator/moeoObjectiveObjectiveVectorComparator.h>
#include <comparator/moeoObjectiveVectorComparator.h>
#include <comparator/moeoParetoObjectiveVectorComparator.h>
#include <fitness/moeoParetoBasedFitnessAssignment.h>


template < class MOEOT >
class moeoFastNonDominatedSortingFitnessAssignment : public moeoParetoBasedFitnessAssignment < MOEOT >
  {
  public:

    typedef typename MOEOT::ObjectiveVector ObjectiveVector;


    moeoFastNonDominatedSortingFitnessAssignment() : comparator(paretoComparator)
    {}


    moeoFastNonDominatedSortingFitnessAssignment(moeoObjectiveVectorComparator < ObjectiveVector > & _comparator) : comparator(_comparator)
    {}


    void operator()(eoPop < MOEOT > & _pop)
    {
      // number of objectives for the problem under consideration
      unsigned int nObjectives = MOEOT::ObjectiveVector::nObjectives();
      if (nObjectives == 1)
        {
          // one objective
          oneObjective(_pop);
        }
      else if (nObjectives == 2)
        {
          // two objectives (the two objectives function is still to implement)
          mObjectives(_pop);
        }
      else if (nObjectives > 2)
        {
          // more than two objectives
          mObjectives(_pop);
        }
      else
        {
          // problem with the number of objectives
          throw std::runtime_error("Problem with the number of objectives in moeoNonDominatedSortingFitnessAssignment");
        }
      // a higher fitness is better, so the values need to be inverted
      double max = _pop[0].fitness();
      for (unsigned int i=1 ; i<_pop.size() ; i++)
        {
          max = std::max(max, _pop[i].fitness());
        }
      for (unsigned int i=0 ; i<_pop.size() ; i++)
        {
          _pop[i].fitness(max - _pop[i].fitness());
        }
    }


    void updateByDeleting(eoPop < MOEOT > & _pop, ObjectiveVector & _objVec)
    {
      for (unsigned int i=0; i<_pop.size(); i++)
        {
          // if _pop[i] is dominated by _objVec
          if ( comparator(_pop[i].objectiveVector(), _objVec) )
            {
              _pop[i].fitness(_pop[i].fitness()+1);
            }
        }
    }


  private:

    moeoObjectiveVectorComparator < ObjectiveVector > & comparator;
    moeoParetoObjectiveVectorComparator < ObjectiveVector > paretoComparator;
  class ObjectiveComparator : public moeoComparator < MOEOT >
      {
      public:
        const bool operator()(const MOEOT & _moeo1, const MOEOT & _moeo2)
        {
          return cmp(_moeo1.objectiveVector(), _moeo2.objectiveVector());
        }
      private:
        moeoObjectiveObjectiveVectorComparator < ObjectiveVector > cmp;
      }
    objComparator;


    void oneObjective (eoPop < MOEOT > & _pop)
    {
      // sorts the population in the ascending order
      std::sort(_pop.begin(), _pop.end(), objComparator);
      // assign fitness values
      unsigned int rank = 1;
      _pop[_pop.size()-1].fitness(rank);
      for (unsigned int i=_pop.size()-2; i>=0; i--)
        {
          if (_pop[i].objectiveVector() != _pop[i+1].objectiveVector())
            {
              rank++;
            }
          _pop[i].fitness(rank);
        }
    }


    void twoObjectives (eoPop < MOEOT > & _pop)
    {
      //... TO DO !
    }


    void mObjectives (eoPop < MOEOT > & _pop)
    {
      // S[i] = indexes of the individuals dominated by _pop[i]
      std::vector < std::vector<unsigned int> > S(_pop.size());
      // n[i] = number of individuals that dominate the individual _pop[i]
      std::vector < unsigned int > n(_pop.size(), 0);
      // fronts: F[i] = indexes of the individuals contained in the ith front
      std::vector < std::vector<unsigned int> > F(_pop.size()+2);
      // used to store the number of the first front
      F[1].reserve(_pop.size());
      for (unsigned int p=0; p<_pop.size(); p++)
        {
          for (unsigned int q=0; q<_pop.size(); q++)
            {
              // if q is dominated by p
              if ( comparator(_pop[q].objectiveVector(), _pop[p].objectiveVector()) )
                {
                  // add q to the set of solutions dominated by p
                  S[p].push_back(q);
                }
              // if p is dominated by q
              else if  ( comparator(_pop[p].objectiveVector(), _pop[q].objectiveVector()) )
                {
                  // increment the domination counter of p
                  n[p]++;
                }
            }
          // if no individual dominates p
          if (n[p] == 0)
            {
              // p belongs to the first front
              _pop[p].fitness(1);
              F[1].push_back(p);
            }
        }
      // front counter
      unsigned int counter=1;
      unsigned int p,q;
      while (! F[counter].empty())
        {
          // used to store the number of the next front
          F[counter+1].reserve(_pop.size());
          for (unsigned int i=0; i<F[counter].size(); i++)
            {
              p = F[counter][i];
              for (unsigned int j=0; j<S[p].size(); j++)
                {
                  q = S[p][j];
                  n[q]--;
                  // if no individual dominates q anymore
                  if (n[q] == 0)
                    {
                      // q belongs to the next front
                      _pop[q].fitness(counter+1);
                      F[counter+1].push_back(q);
                    }
                }
            }
          counter++;
        }
    }

  };

#endif /*MOEOFASTNONDOMINATEDSORTINGFITNESSASSIGNMENT_H_*/

---