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() {}
    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
    unsigned int **buf_int; //is [2][size]: replicate [0] and generation [1] done by each worker
    unsigned int buf_stride, size; //size = 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); }
    unsigned int workerCount() const { return size-1; }
    unsigned int workerRank() const  { return rank; }
    std::string  hostName() const   { return host; }
 
 private:
    unsigned int size;
    unsigned int rank;
    std::string  host;
 
};
 
#endif