DistrArraySpan.h

#ifndef LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYSPAN_H
#define LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYSPAN_H
#include <molpro/linalg/array/DistrArray.h>
#include <molpro/mpi.h>
 
namespace molpro::linalg::array {
class DistrArraySpan : public DistrArray {
protected:
  std::unique_ptr<Distribution> m_distribution; 
  bool m_allocated = false;                     
  Span<value_type> m_span;                      
 
public:
  DistrArraySpan() = delete;
  DistrArraySpan(size_t dimension, Span<value_type> buffer, MPI_Comm commun = molpro::mpi::comm_global());
  DistrArraySpan(std::unique_ptr<Distribution> distribution, Span<value_type> buffer,
                 MPI_Comm commun = molpro::mpi::comm_global());
  DistrArraySpan(const DistrArraySpan &source);
  explicit DistrArraySpan(const DistrArray &source);
  DistrArraySpan(DistrArraySpan &&source) noexcept;
  DistrArraySpan &operator=(const DistrArraySpan &source);
  DistrArraySpan &operator=(DistrArraySpan &&source) noexcept;
 
  friend void swap(DistrArraySpan &a1, DistrArraySpan &a2) noexcept;
  // void sync() const override;
  void allocate_buffer(Span<value_type> buffer);
 
protected:
  struct LocalBufferSpan : public DistrArray::LocalBuffer {
    explicit LocalBufferSpan(DistrArraySpan &source);
    explicit LocalBufferSpan(const DistrArraySpan &source);
  };
 
public:
  [[nodiscard]] const Distribution &distribution() const override;
  [[nodiscard]] std::unique_ptr<LocalBuffer> local_buffer() override;
  [[nodiscard]] std::unique_ptr<const LocalBuffer> local_buffer() const override;
  [[nodiscard]] value_type at(index_type ind) const override;
  void set(index_type ind, value_type val) override;
  void get(index_type lo, index_type hi, value_type *buf) const override;
  [[nodiscard]] std::vector<value_type> get(index_type lo, index_type hi) const override;
  void put(index_type lo, index_type hi, const value_type *data) override;
  void acc(index_type lo, index_type hi, const value_type *data) override;
  [[nodiscard]] std::vector<value_type> gather(const std::vector<index_type> &indices) const override;
  void scatter(const std::vector<index_type> &indices, const std::vector<value_type> &data) override;
  void scatter_acc(std::vector<index_type> &indices, const std::vector<value_type> &data) override;
  [[nodiscard]] std::vector<value_type> vec() const override;
};
 
} // namespace molpro::linalg::array
#endif // LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYSPAN_H