eoTruncatedSelectMany.h

//-----------------------------------------------------------------------------

#ifndef _eoTruncatedSelectMany_h
#define _eoTruncatedSelectMany_h


//-----------------------------------------------------------------------------
#include <eoSelect.h>
#include <eoSelectOne.h>
#include <utils/eoHowMany.h>
#include <math.h>
//-----------------------------------------------------------------------------

template<class EOT>
class eoTruncatedSelectMany : public eoSelect<EOT>
{
 public:
     eoTruncatedSelectMany(eoSelectOne<EOT>& _select,
                  double  _rateGenitors, double  _rateFertile,
                  bool _interpret_as_rateG = true,
                  bool _interpret_as_rateF = true)
         : select(_select),
           howManyGenitors(_rateGenitors, _interpret_as_rateG),
           howManyFertile(_rateFertile, _interpret_as_rateF) {}

     // Ctor with eoHowManys
     eoTruncatedSelectMany(eoSelectOne<EOT>& _select,
                  eoHowMany _howManyGenitors, eoHowMany _howManyFertile)
         : select(_select), howManyGenitors(_howManyGenitors),
           howManyFertile(_howManyFertile) {}

  virtual void operator()(const eoPop<EOT>& _source, eoPop<EOT>& _dest)
  {
    unsigned target = howManyGenitors(_source.size());

    _dest.resize(target);

    unsigned nbFertile = howManyFertile(_source.size());

    //revert to standard selection (see eoSelectMany) if no truncation
    if (nbFertile == _source.size())
      {
        select.setup(_source);

        for (size_t i = 0; i < _dest.size(); ++i)
          _dest[i] = select(_source);
      }
    else
      {
    // at the moment, brute force (rush rush, no good)
    // what we would need otherwise is a std::vector<EOT &> class
    // and selectors that act on such a thing
        eoPop<EOT> tmpPop = _source; // hum hum, could be a pain in the ass

        tmpPop.sort();             // maybe we could only do partial sort?
        tmpPop.resize(nbFertile);  // only the best guys here now
        tmpPop.shuffle();          // as some selectors are order-sensitive

        select.setup(tmpPop);

        for (size_t i = 0; i < _dest.size(); ++i)
          _dest[i] = select(tmpPop);
      }
  }

private :
  eoSelectOne<EOT>& select;        // selector for one guy
  eoHowMany howManyGenitors;       // number of guys to select
  eoHowMany howManyFertile;        // number of fertile guys
};

#endif