DistrArrayGA.h

#ifndef LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYGA_H
#define LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYGA_H
#include <map>
 
#include "molpro/linalg/array/DistrArray.h"
 
namespace molpro::linalg::array {
 
class DistrArrayGA : public DistrArray {
protected:
  int m_comm_rank = 0;         
  int m_comm_size = 0;         
  int m_ga_handle = 0;         
  int m_ga_pgroup = 0;         
  int m_ga_chunk = 1;          
  bool m_ga_allocated = false; 
  static std::map<MPI_Comm, int> _ga_pgroups;
  std::unique_ptr<Distribution> m_distribution;
 
public:
  DistrArrayGA() = delete;
  DistrArrayGA(size_t dimension, MPI_Comm commun);
  DistrArrayGA(const DistrArrayGA &source);
  DistrArrayGA &operator=(const DistrArrayGA &source);
  DistrArrayGA(DistrArrayGA &&source) noexcept;
  DistrArrayGA &operator=(DistrArrayGA &&source) noexcept;
  ~DistrArrayGA() override;
 
  friend void swap(DistrArrayGA &a1, DistrArrayGA &a2) noexcept;
  void sync() const override;
 
protected:
  struct LocalBufferGA : public DistrArray::LocalBuffer {
    explicit LocalBufferGA(DistrArrayGA &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;
  void error(const std::string &message) const override;
 
protected:
  void check_ga_ind_overlow(index_type ind) const;
 
  Distribution make_distribution() const;
  void allocate_buffer();
};
 
} // namespace molpro::linalg::array
 
#endif // LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYGA_H