msgMpiExec.h

Go to the documentation of this file.

 
// Copyright (c) 2013-2015 OpenM++
// This code is licensed under the MIT license (see LICENSE.txt for details)
 
#ifndef MSG_MPI_EXEC_H
#define MSG_MPI_EXEC_H
 
using namespace std;
 
#include "msgCommon.h"
#include "msgExecBase.h"
#include "msgMpiPacked.h"
#include "msgMpiSend.h"
#include "msgMpiRecv.h"
 
namespace openm
{
    class MpiExec : public IMsgExec, public MsgExecBase
    {
    public:
        MpiExec(int & argc, char ** & argv, IFinalState * i_final);
 
        ~MpiExec(void) noexcept;
 
        int worldSize(void) const noexcept override { return MsgExecBase::worldSize(); }
 
        int rank(void) const noexcept override { return MsgExecBase::rank(); }
 
        int groupRank(void) const noexcept override { return MsgExecBase::groupRank(); }
 
        void setCleanExit(bool i_isClean = false) override;
 
        void createGroups(int i_groupSize, int i_groupCount) override;
 
        void bcastValue(int i_groupOne, const type_info & i_type, void * io_value) override;
 
        void bcastSend(int i_groupOne, const type_info & i_type, size_t i_size, void * io_valueArr) override;
 
        void bcastReceive(int i_groupOne, const type_info & i_type, size_t i_size, void * io_valueArr) override;
 
        void bcastSendPacked(int i_groupOne, IRowBaseVec & io_rowVec, const IPackedAdapter & i_adapter) override;
 
        void bcastReceivePacked(int i_groupOne, IRowBaseVec & io_rowVec, const IPackedAdapter & i_adapter) override;
 
        void startSend(int i_sendTo, MsgTag i_msgTag, const type_info & i_type, size_t i_size, void * i_valueArr) override
        { MsgExecBase::startSend(i_sendTo, i_msgTag, i_type, i_size, i_valueArr); }
 
        void startSendPacked(int i_sendTo, const IRowBaseVec & i_rowVec, const IPackedAdapter & i_adapter) override
        { MsgExecBase::startSendPacked(i_sendTo, i_rowVec, i_adapter); }
 
        bool tryReceive(int i_recvFrom, IRowBaseVec & io_resultRowVec, const IPackedAdapter & i_adapter) const override
        { return MsgExecBase::tryReceive(i_recvFrom, io_resultRowVec, i_adapter); }
 
        bool tryReceive(int i_recvFrom, MsgTag i_msgTag, const type_info & i_type, size_t i_size, void * io_valueArr) const override
        { return MsgExecBase::tryReceive(i_recvFrom, i_msgTag, i_type, i_size, io_valueArr); }
 
        void waitSendAll(void) override { MsgExecBase::waitSendAll(); }
 
    private:
        bool isCleanExit;                   // if false then process exit by error or exception
        MPI_Group worldGroup;               // MPI world global group
        MPI_Comm groupComm;                 // modeling group communicator for current process
        vector<MPI_Group> mpiGroupVec;      // handles of modeling groups
        vector<MPI_Comm> mpiCommVec;        // handles of communicators
 
        // cleanup MPI resources
        void cleanup(void) noexcept;
 
        MPI_Comm commByGroupOne(int i_groupOne);
 
        void waitRequest(int i_pollTime, MPI_Request & io_request) const;
 
    private:
        MpiExec(const MpiExec & i_exec) = delete;
        MpiExec & operator=(const MpiExec & i_exec) = delete;
    };
}
 
#endif  // MSG_MPI_EXEC_H