wxmsgtmpl.h

Go to the documentation of this file.

#ifndef wxmsgtmpl_h
#define wxmsgtmpl_h
 
#include <mpi.h>
 
// std includes
#include <vector>
 
struct WxMsgStatus_v
{
};
typedef WxMsgStatus_v* WxMsgStatus;
 
template<typename T> class WxMsgTmpl
{
public:
    virtual ~WxMsgTmpl()
    {
        delete[] _sendBuff;
        delete[] _recvBuff;
    }
 
    virtual void send(unsigned num, const T* array, unsigned recvRank, int tag) const = 0;
 
    virtual void finishSend(WxMsgStatus ms) const = 0;
 
    virtual WxMsgStatus
    startSend(unsigned num, const T* arr, unsigned recvRank, int tag) const = 0;
 
    virtual void
    recv(unsigned num, std::vector<T>& array, unsigned sendRank, int tag) const = 0;
 
    virtual void recv(unsigned num, T* array, unsigned sendRank, int tag) const = 0;
 
    virtual WxMsgStatus
    startRecv(unsigned num, unsigned sendRank, int tag = -1) const = 0;
 
    virtual WxMsgStatus
    startRecv(unsigned num, T* arr, unsigned sendRank, int tag = -1) const = 0;
 
    virtual void allReduce(unsigned num, T* sendBuff, T* recvBuff, MPI_Op op) const = 0;
 
    virtual void broadcast(unsigned num, T* buffer, int root) const = 0;
 
protected:
    WxMsgTmpl() : _sendSize(0), _sendBuff(0), _recvSize(0), _recvBuff(0)
    {
    }
 
    void resizeArray(T*& array, unsigned& numElem, unsigned reqNumElem) const
    {
        if (numElem < reqNumElem)
        {
            // delete old array and allocate new memory
            delete[] array;
            array = new T[reqNumElem];
        }
        // set number of elements
        numElem = reqNumElem;
        return;
    }
 
    void resizeSendBuff(unsigned reqSize)
    {
        if (_sendSize < reqSize)
        {
            // delete old array and allocate new memory
            delete[] _sendBuff;
            _sendSize = reqSize;
            _sendBuff = new T[_sendSize];
        }
        return;
    }
 
    void resizeRecvBuff(unsigned reqSize)
    {
        if (_recvSize < reqSize)
        {
            delete[] _recvBuff;
            _recvSize = reqSize;
            _recvBuff = new T[_recvSize];
        }
        return;
    }
 
    unsigned _sendSize;
    T* _sendBuff; // buffer for sending
 
    unsigned _recvSize;
    T* _recvBuff; // buffer for receiving
};
 
#endif // wxmsgtmpl_h