SeapodymCourier.h

#include <mpi.h>
#include <set>
#include <vector>
#ifndef SEAPODYM_COURIER
#define SEAPODYM_COURIER
 
 
using Tuple3 = std::tuple<MPI_Win, double*, std::size_t>;

class SeapodymCourier {
    
    private:
    
        // MPI communicator to use for communication
        MPI_Comm comm;
 
        // Pointer to the data exposed by this worker. This class does not own this pointer, 
        // it is provided by the user. The data is expected to be allocated by the user and 
        // should remain valid for the lifetime of this SeapodymCourier instance.
        double *data;
        int data_size; 
 
        // MPI window for the exposed data
        MPI_Win win;
        MPI_Win winRecv;
 
        // data array to receive the accumulate operation
        std::vector<double> dataRecv;
 
        // Local MPI rank
        int local_rank;
    public:

    SeapodymCourier(MPI_Comm comm=MPI_COMM_WORLD);

    ~SeapodymCourier();

    double* getDataPtr() const { return this->data;}

    void expose(double* data, int data_size);

    std::vector<double> fetch(int source_worker);

    std::vector<double> accumulate(int targetWorker); 

    void free() {
        if (this->win != MPI_WIN_NULL) {
            MPI_Win_free(&this->win);
        }
    if (this->winRecv != MPI_WIN_NULL) {
            MPI_Win_free(&this->winRecv);
    }
        this->data = nullptr;
        this->data_size = 0;
    this->dataRecv.clear();
    }
 
    // Disable copy and assignment operations
    // to prevent accidental copying of the SeapodymCourier instance
    // This is important because the class manages an MPI window and data pointer
    // which should not be copied or assigned.
    SeapodymCourier(const SeapodymCourier&) = delete; // Disable copy constructor
    SeapodymCourier& operator=(const SeapodymCourier&) = delete; // Disable assignment operator
    SeapodymCourier(SeapodymCourier&& other) noexcept; // Move constructor
    SeapodymCourier& operator=(SeapodymCourier&& other) noexcept; // Move assignment operator
};
 
#endif // SEAPODYM_COURIER