MPImanager.h

Go to the documentation of this file.

 
#ifndef __MPIMANAGER_H
#define __MPIMANAGER_H
 
#include <queue>
#include <string>
#include <map>
 
#ifdef USE_MPI
#include <mpi.h>
#endif
 
class SimRunner;
class StatServices;
 
class MPImanager {
 
 public:
    MPImanager():buf_dbl(0),buf_int(0),buf_stride(0),num_worker(0) {}
    virtual ~MPImanager() {}
    virtual unsigned int init( StatServices* StatManager ) = 0;
    virtual void finish( StatServices* StatManager, unsigned int *_gen, unsigned int _repl );
    virtual void iterate( SimRunner *_sim, StatServices* StatManager, unsigned int *_gen, unsigned int *_repl ) = 0;
 
 protected:
    double       **buf_dbl; //receives stat records from each worker [num_worker][buf_stride]
    unsigned int **buf_int; //is [2][num_worker]: how many records [0] and which replicate [1] done by each worker
    unsigned int buf_stride;//number of stat recorders x number of generations recorded
    int num_worker;         //worker count
 
};
 
#ifdef USE_MPI
class MPImaster : public MPImanager {
 
 public:
    MPImaster( const unsigned int count );
    ~MPImaster()                   { delete[] _request; }
    unsigned int init( StatServices *StatManager );
    void finish( StatServices *StatManager, unsigned int *_gen, unsigned int _repl );
    void iterate( SimRunner *_sim, StatServices* StatManager, unsigned int *_gen, unsigned int *_repl );
 
 private:
    MPI_Request *_request;
    std::vector< unsigned int > worker_job; //records the current replicate done by worker i
    unsigned int waitWorker();
    unsigned int assign( const unsigned int job, const unsigned int worker );
    
};
 
class MPIworker : public MPImanager {
 
 public:
    MPIworker() {}
    unsigned int init( StatServices *StatManager );
    void iterate( SimRunner *_sim, StatServices *StatManager, unsigned int *_gen, unsigned int *_repl );
 
};
#endif
 
class MPIenv {
 
 public:
    MPIenv( int *argc, char ***argv, MPImanager *&_p );
    static void  abort( int i );
    static void  finish( MPImanager *p ); 
    bool         isMaster() const    { return (rank==0); }
    int          workerCount() const { return size-1; }
    int          workerRank() const  { return rank; }
    std::string  hostName() const    { return host; }
 
 private:
    int size;
    int rank;
    std::string  host;
 
};
 
#endif