TaskScore.h

#include <mpi.h>
#include <vector>
#include <set>
 
#ifndef TASK_SCORE
#define TASK_SCORE

 
class TaskScore {
 
    private:
 
        // Communicator
        MPI_Comm comm; 
 
        // number of tasks
        int numTasks;
 
        // local MPI rank
        int me;
 
        // scoreboard 
        std::vector<int> status;
 
        // the MPI window to receive the score updates from remote processses
        MPI_Win win;
 
    public:
 
        enum Status {
            PENDING = -1,
            RUNNING = -2,
            SUCCEEDED = -3,
            FAILED = -4,
        };
 

        TaskScore(MPI_Comm comm, int numTasks);

        void store(int taskId, int status);

         void print() const;

        void free() {
            if (this->win != MPI_WIN_NULL) {
                MPI_Win_free(&this->win);
            }
        }

        ~TaskScore() {
            this->free();
        }

        std::set<int> get(int status) const;
 
};
 
#endif // TASK_SCORE