MPImanager.h

Go to the documentation of this file.

#ifndef __MPIMANAGER_H
#define __MPIMANAGER_H

#include <queue>
#include <string>
#include <map>
#include "MPStatHandler.h"

#ifdef USE_MPI
#include <mpi.h>
#endif

class MPImanager {

 public:
    MPImanager() {}
    virtual ~MPImanager() {}
    virtual unsigned int init( MPStatHandler *SH ) = 0;
    virtual void finish( MPStatHandler *SH, unsigned int *_gen, unsigned int _repl );
    virtual void iterate( Metapop *_pop, MPStatHandler *SH, unsigned int *_gen, unsigned int *_repl ) = 0;

 protected:
    double **buf_dbl;
    unsigned int **buf_int;
    unsigned int buf_stride, size;

};

#ifdef USE_MPI

class MPImaster : public MPImanager {

 public:
    MPImaster( const unsigned int count );
    ~MPImaster()                   { delete[] _request; }
    unsigned int init( MPStatHandler *SH );
    void finish( MPStatHandler *SH, unsigned int *_gen, unsigned int _repl );
    void iterate( Metapop *_pop, MPStatHandler *SH, unsigned int *_gen, unsigned int *_repl );

 private:
    MPI::Request *_request;
    std::vector< unsigned int > slave_job;
    unsigned int waitSlave();
    unsigned int assign( const unsigned int job, const unsigned int slave );
    
};

class MPIslave : public MPImanager {

 public:
    MPIslave() {}
    unsigned int init( MPStatHandler *SH );
    void iterate( Metapop *_pop, MPStatHandler *SH, 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 slaveCount() const { return size-1; }
    unsigned int slaveRank() const  { return rank; }
    std::string  hostName() const   { return host; }

 private:
    unsigned int size;
    unsigned int rank;
    std::string  host;

};

#endif