TaskStepManager.h

#include <mpi.h>
#include <map>
#include <set>
#include <array>
 
#ifndef TASK_DEPENDENCY_MANAGER
#define TASK_DEPENDENCY_MANAGER

 
// TaskId 
using dep_type = std::array<int, 2>;

class TaskStepManager {
 
    private:
 
        // Communicator
        MPI_Comm comm; 
 
        // number of tasks
        int numTasks;
 
        // taskId to first step index map
        std::map<int, int> stepBegMap;
 
        // taskId to last step index + 1 map
        std::map<int, int> stepEndMap;
 
        // dependencies
        std::map<int, std::set<dep_type> > deps;
 
    public:

        TaskStepManager(MPI_Comm comm, int numTasks, 
            const std::map<int, int>& stepBegMap,
            const std::map<int, int>& stepEndMap,
            const std::map<int, std::set<dep_type> >& dependencyMap);

        std::set< std::array<int, 3> > run() const;
 
};
 
#endif // TASK_DEPENDENCY_MANAGER