MPICommHandle.h

Go to the documentation of this file.

#ifndef MPICOMMHANDLE_H_
#define MPICOMMHANDLE_H_

#include "mpi.h"

#include <vector>

using std::vector;

 class MPICommHandle {
public:

        MPICommHandle()
        {
                comm =  MPI::COMM_WORLD;
        }

    MPICommHandle(MPICommHandle & base_comm, const vector<unsigned> &ranks)
        {
                MPI::Group grp = base_comm.comm.Get_group();
                int * rank_arr = new int[ranks.size()];
                for (int i = 0; i < (int)ranks.size(); ++i)
                        rank_arr[i] = (int)ranks[i];             
                MPI::Group subgrp = grp.Incl(ranks.size(), rank_arr);
                comm = base_comm.comm.Create(subgrp);           
                delete [] rank_arr;
        }       
        
        int mpi_rank()
        {
                return comm.Get_rank(); 
        }
        
        int mpi_size() 
        {
                return comm.Get_size();
        }
        
        static MPICommHandle WORLD;
        
        
protected:
        MPI::Intracomm & getComm()
        {
                return comm;    
        }                                                               

        MPI::Intracomm comm;
};



#endif /*MPICOMMHANDLE_H_*/

---