SeapodymCohortManager.h

#include <map>
#include <vector>
#include <set>
 
#ifndef SEAPODYM_COHORT_MANAGER
#define SEAPODYM_COHORT_MANAGER

 
class SeapodymCohortManager {
 
    private:
 
        // number of age groups for each time step, ideally matching the number of workers
        int numAgeGroups; 
 
        // number of workers 0...numWorkers-1
        int numWorkers;
 
        // total number of time steps across all cohorts
        int numTimeSteps;
 
        // number of cohorts
        int numCohorts = this->numAgeGroups + this->numTimeSteps - 1;
 
        // assign cohort Ids to workers
        std::map<int, std::vector<int> > worker2cohort;
 
        // map age index to worker ID
        std::map<int, int> ageIndex2Worker;
 
    public:

        SeapodymCohortManager(int numAgeGroups, int numWorkers, int numTimeSteps);

        std::vector<int> getInitCohortIds(int workerId) const;

        int getNumSteps(int cohortId) const;

        std::set<int> getDependencies(int cohortId) const;

        int getNextCohort(int cohortId) const;

        int getNewCohortWorker(int timeStep) const;
 
};
 
#endif // SEAPODYM_COHORT_MANAGER