wxmpimsg.h

Go to the documentation of this file.

#ifndef wxmpimsg_h
#define wxmpimsg_h
 
#ifndef _DONT_USE_MPI_
#include <mpi.h>
#endif
 
// WarpX includes
#include "wxmsgbase.h"
#include "wxmpimsgtmpl.h"
#include "wxdatatypes.h"
 
// std includes
 
class WxMpiMsg : public WxMsgBase
{
public:
    WxMpiMsg();
 
    int rank() const override
    {
        int r;
        MPI_Comm_rank(_comm, &r);
        return r;
    }
 
    unsigned numProcs() const override
    {
        int np;
        MPI_Comm_size(_comm, &np);
        return np;
    }
 
    WxMsgBase* createSubComm(const std::vector<int>& ranks) override;
 
    WxMsgBase* splitComm(int color, int order) override;
 
    void barrier() const override
    {
        MPI_Barrier(_comm);
    }
 
    void* finishRecv(WxMsgStatus ms) const override;
 
    void finishSend(WxMsgStatus ms) const override;
 
    bool checkRecv(WxMsgStatus ms) const override;
 
    MPI_Comm getMpiComm() const
    {
        return _comm;
    }
 
private:
    WxMpiMsg(WxMpiMsg* parent, MPI_Comm comm);
 
    MPI_Comm _comm;
};
 
#endif //  wxmpimsg_h