iteration_payload.hpp

#ifndef __ITERATION_PAYLOAD_HPP__
#define __ITERATION_PAYLOAD_HPP__
 
#include <cma_read/flow_iterator.hpp>
#include <cma_read/neighbors.hpp>
#include <cma_read/reactorstate.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__