TaskStepWorker.h

#include <mpi.h>
#include <functional>
#include <map>
 
#ifndef TASK_STEP_WORKER
#define TASK_STEP_WORKER

 
class TaskStepWorker {
 
    private:
 
        // communicator
        MPI_Comm comm; 
 
        // task function, takes task_id, stepBeg and stepEnd as input 
        // and returns a code/result
        std::function<void(int, int, int, MPI_Comm)> taskFunc;
 
        // task Id to first step index map
        std::map<int, int> stepBegMap;
 
        // task Id to last step index + 1 map
        std::map<int, int> stepEndMap;
 
        // local rank;
        int rank;
 
    public:

        TaskStepWorker(MPI_Comm comm, 
            std::function<void(int, int, int, MPI_Comm)> taskFunc, 
            const std::map<int, int>& stepBegMap,
            const std::map<int, int>& stepEndMap);

        void run() const;
 
};
 
#endif // TASK_STEP_WORKER