DataProvider.h

#include <mpi.h>
#include <cstddef>
#include <cstring>
#include <stdexcept>
#include <vector>
#include <string>
#include <unordered_map>
#include <tuple>
 
#ifndef DATA_PROVIDER_H
#define DATA_PROVIDER_H

class DataProvider {
 
public:
 
    DataProvider(const DataProvider&) = delete;
    DataProvider& operator=(const DataProvider&) = delete;
  
    DataProvider(MPI_Comm comm, const std::vector<std::pair<std::string, std::size_t> >& nameSizePairs);
 
    ~DataProvider();

    int getShmRank() const { return shmRank_; }

    bool isShmRoot() const { return shmRank_ == 0; }

    double* getDataPtr(const std::string& name) const {
        auto it = this->data_.find(name);
        return it != this->data_.end() ? std::get<0>(it->second) : nullptr;
    }

    size_t getNumElements(const std::string& name) const {
        auto it = this->data_.find(name);
        return it != this->data_.end() ? std::get<1>(it->second) : 0;
    }

    MPI_Comm getShmComm() const {
        return this->shmcomm_;
    }
 
private:
    MPI_Comm comm_;
    MPI_Comm shmcomm_;
    std::unordered_map< std::string, std::tuple<double*, std::size_t, MPI_Win> > data_;
    int shmRank_;
};
 
#endif // DATA_PROVIDER_H