iteration_payload.hpp

#ifndef __ITERATION_PAYLOAD_HPP__
#define __ITERATION_PAYLOAD_HPP__
 
#include "cma_read/reactorstate.hpp"
#include <cma_read/flow_iterator.hpp>
#include <cma_read/neighbors.hpp>
#include <mpi.h>
#include <span>
#include <vector>
 
namespace WrapMPI
{
  class IterationPayload
  {
  public:
    std::vector<double> liquid_flows;
 
    std::vector<double> liquid_volumes;
 
    std::vector<double> gas_volumes;
 
    std::vector<size_t> raw_neighbors;
 
    CmaRead::Neighbors::Neighbors_const_view_t neighbors;
 
    explicit IterationPayload(size_t size_flows, size_t volumes);
 
    bool recv(size_t source, MPI_Status* status) noexcept;
  };
 
  class HostIterationPayload
  {
  public:
    std::span<const double> liquid_flows;
    std::span<const double> liquid_volumes;
    std::span<const double> gas_volumes;
    CmaRead::Neighbors::Neighbors_const_view_t neighbors;
 
    void fill(const CmaRead::ReactorState& current_reactor_state);
 
    [[nodiscard]] bool sendAll(std::size_t n_rank)  noexcept;
    private:
    [[nodiscard]] bool send(size_t rank)  noexcept;
    
    static constexpr std::size_t n_vector_send = 4; 
    std::array<MPI_Request,n_vector_send> requests;
  };
 
} // namespace WrapMPI
 
#endif //__ITERATION_PAYLOAD_HPP__