SeapodymCohortDependencyAnalyzer.h

#include <map>
#include <set>
#include <array>
 
#ifndef SEAPODYM_COHORT_DEPENDENCY_ANALYZER
#define SEAPODYM_COHORT_DEPENDENCY_ANALYZER

class SeapodymCohortDependencyAnalyzer {
 
private:
 
    // number of (normal) age groups, excluding the A+ "plus group"
    int numAgeGroups;
 
    // number of time steps
    int numTimeSteps;
 
    // total number of cohorts (including A+ if aPlusCohort=true)
    int numIds;
 
    // cohort Id: starting step index
    std::map<int, int> stepBegMap;
 
    // cohort Id: ending step index + 1
    std::map<int, int> stepEndMap;
 
    // cohort Id: {(taskId, step), ...}
    std::map<int, std::set<std::array<int, 2>>> dependencyMap;
 
public:

    SeapodymCohortDependencyAnalyzer(int numAgeGroups, int numTimeSteps, int ageMature = 0, bool aPlusCohort = false);

    int getNumberOfCohorts() const;

    int getNumberOfCohortSteps() const;

    std::map<int, int> getStepBegMap() const;

    std::map<int, int> getStepEndMap() const;

    std::map<int, std::set<std::array<int, 2>>>  getDependencyMap() const;

    int getFirstAPlusCohortId() const;
 
};
 
#endif // SEAPODYM_COHORT_DEPENDENCY_ANALYZER