moeoCrowdingDistanceDiversityAssignment.h

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

//-----------------------------------------------------------------------------
// moeoCrowdingDistanceDiversityAssignment.h
// (c) OPAC Team (LIFL), Dolphin Project (INRIA), 2007
/*
    This library...

    Contact: paradiseo-help@lists.gforge.inria.fr, http://paradiseo.gforge.inria.fr
 */
//-----------------------------------------------------------------------------

#ifndef MOEOCROWDINGDISTANCEDIVERSITYASSIGNMENT_H_
#define MOEOCROWDINGDISTANCEDIVERSITYASSIGNMENT_H_

#include <eoPop.h>
#include <comparator/moeoOneObjectiveComparator.h>
#include <diversity/moeoDiversityAssignment.h>

template < class MOEOT >
class moeoCrowdingDistanceDiversityAssignment : public moeoDiversityAssignment < MOEOT >
{
public:

    typedef typename MOEOT::ObjectiveVector ObjectiveVector;


    double inf() const
    {
        return std::numeric_limits<double>::max();
    }


    double tiny() const
    {
        return 1e-6;
    }


    void operator()(eoPop < MOEOT > & _pop)
    {
        if (_pop.size() <= 2)
        {
            for (unsigned int i=0; i<_pop.size(); i++)
            {
                _pop[i].diversity(inf());
            }
        }
        else
        {
            setDistances(_pop);
        }
    }


    void updateByDeleting(eoPop < MOEOT > & _pop, ObjectiveVector & _objVec)
    {
        std::cout << "WARNING : updateByDeleting not implemented in moeoCrowdingDiversityAssignment" << std::endl;
    }


protected:

    virtual void setDistances (eoPop < MOEOT > & _pop)
    {
        double min, max, distance;
        unsigned int nObjectives = MOEOT::ObjectiveVector::nObjectives();
        // set diversity to 0
        for (unsigned int i=0; i<_pop.size(); i++)
        {
            _pop[i].diversity(0);
        }
        // for each objective
        for (unsigned int obj=0; obj<nObjectives; obj++)
        {
            // comparator
            moeoOneObjectiveComparator < MOEOT > objComp(obj);
            // sort
            std::sort(_pop.begin(), _pop.end(), objComp);
            // min & max
            min = _pop[0].objectiveVector()[obj];
            max = _pop[_pop.size()-1].objectiveVector()[obj];
            // set the diversity value to infiny for min and max
            _pop[0].diversity(inf());
            _pop[_pop.size()-1].diversity(inf());
            for (unsigned int i=1; i<_pop.size()-1; i++)
            {
                distance = (_pop[i+1].objectiveVector()[obj] - _pop[i-1].objectiveVector()[obj]) / (max-min);
                _pop[i].diversity(_pop[i].diversity() + distance);
            }
        }
    }

};

#endif /*MOEOCROWDINGDISTANCEDIVERSITYASSIGNMENT_H_*/

---