be_mpi_distributor.h

Go to the documentation of this file.

 
#ifndef _BE_MPI_DISTRIBUTOR_H
#define _BE_MPI_DISTRIBUTOR_H
 
#include <memory>
#include <set>
#include <string>
 
#include <be_error_exception.h>
#include <be_io_logsheet.h>
#include <be_io_propertiesfile.h>
#include <be_mpi.h>
#include <be_mpi_resources.h>
#include <be_mpi_workpackage.h>
 
namespace BiometricEvaluation {
        namespace MPI {
                class Distributor {
                public:
                        static const std::string CHECKPOINTFILENAME;
 
                        static const std::string CHECKPOINTREASON;
 
                        static const std::string CHECKPOINTPID;
 
                        Distributor(const std::string &propertiesFileName);
 
                        virtual ~Distributor();
                        
                        void start();
 
                protected:
                        virtual void createWorkPackage(
                            MPI::WorkPackage &workPackage) = 0;
 
                        virtual void checkpointSave(
                            const std::string &reason) = 0;
 
                        virtual void checkpointRestore() = 0;
 
                        std::shared_ptr<IO::Logsheet> getLogsheet() const;
 
                        std::shared_ptr<IO::PropertiesFile>
                        getCheckpointData() const;
 
                private:
                        void distributeWork();
 
                        void sendWorkPackage(
                            MPI::WorkPackage &workPackage,
                            int MPITask);
 
                        void shutdown();
 
                        std::unique_ptr<MPI::Resources> _resources;
 
                        /* The list of tasks accepting work */
                        std::set<int> _activeMpiTasks;
 
                        std::shared_ptr<IO::Logsheet> _logsheet;
                        std::shared_ptr<IO::PropertiesFile> _checkpointData;
                };
        }
}
 
#endif /* _BE_MPI_DISTRIBUTOR_H */