simpactpopulation.h

#ifndef SIMPACTPOPULATION_H

#define SIMPACTPOPULATION_H

#include "population.h"
#include <assert.h>

class PopulationDistribution;
class Person;
class Man;
class Woman;

class SimpactPopulationConfig : public errut::ErrorBase
{
public:
        SimpactPopulationConfig();
        ~SimpactPopulationConfig();

        void setInitialMen(int number)                                          { m_initialMen = number; }
        void setInitialWomen(int number)                                        { m_initialWomen = number; }
        void setEyeCapsFraction(double f)                                       { m_eyeCapsFraction = f; }

        int getInitialMen() const                                               { return m_initialMen; }
        int getInitialWomen() const                                             { return m_initialWomen; }
        double getEyeCapsFraction() const                                       { return m_eyeCapsFraction; }
private:
        int m_initialMen, m_initialWomen;
        double m_eyeCapsFraction;
};

class SimpactPopulation : public Population
{
public:
        SimpactPopulation(bool parallel, GslRandomNumberGenerator *pRng);
        ~SimpactPopulation();

        bool init(const SimpactPopulationConfig &popConfig, const PopulationDistribution &popDist);

        Person **getAllPeople()                                                 { return reinterpret_cast<Person**>(Population::getAllPeople()); }
        Man **getMen()                                                          { return reinterpret_cast<Man**>(Population::getMen()); }
        Woman **getWomen()                                                      { return reinterpret_cast<Woman**>(Population::getWomen()); }
        Person **getDeceasedPeople()                                            { return reinterpret_cast<Person**>(Population::getDeceasedPeople()); }

        int getInitialPopulationSize() const                                    { return m_initialPopulationSize; }
        double getEyeCapsFraction() const                                       { return m_eyeCapsFraction; }
protected:
        bool createInitialPopulation(const SimpactPopulationConfig &config, const PopulationDistribution &popDist);
        bool scheduleInitialEvents();
private:
        void onAboutToFire(EventBase *pEvt);

        int m_initialPopulationSize;
        double m_eyeCapsFraction;

        bool m_init;
};

inline SimpactPopulation &SIMPACTPOPULATION(State *pState)
{
        assert(pState != 0);
        return static_cast<SimpactPopulation &>(*pState);
}

inline const SimpactPopulation &SIMPACTPOPULATION(const State *pState)
{
        assert(pState != 0);
        return static_cast<const SimpactPopulation &>(*pState);
}

#endif // SIMPACTPOPULATION_H