MPIAllToAllCommunicator.h

Go to the documentation of this file.

#ifndef MPIALLTOALLCOMMUNICATOR_H_
#define MPIALLTOALLCOMMUNICATOR_H_

#include <mpi.h>
#include "MPIInputBuffer.h"
#include "MPIOutputBuffer.h"


class MPIAllToAllCommunicator
{
public:

    MPIAllToAllCommunicator(MPIInputBufferVector & mpiInputBuffers,
                            MPIOutputBufferVector & mpiOutputBuffers,
                            MPI::Intracomm & comm,                            
                            vector<bool> &incomingConnections,
                            vector<bool> &outgoingConnections);

    virtual ~MPIAllToAllCommunicator();


    bool doAllToAllExchange();


    // Gives the rank of current mpi process. [used for testing only]
    int getRank();

protected:
    // Actual implementation of the all-to-all buffer exchange algorithm on the mpi level.
    virtual void doExchangeAlgorithm() = 0;
    
    virtual void prepare() = 0;
    

    bool finished;

    MPIInputBufferVector &inputBuffers;

    MPIOutputBufferVector &outputBuffers;

    vector<bool> &incoming_connections;

    vector<bool> &outgoing_connections;

    int numNodes;

    MPI::Intracomm & mpi_comm;
    
    vector<bool> hasNextToSend;

    vector<bool> needsToSend;

    vector<bool> needsToReceive;
    
    int *send_displ;

    int *recv_displ;

    int *send_count;

    int *recv_count;

};

#endif /*MPIALLTOALLCOMMUNICATOR_H_*/

---