DistrArrayDisk.h

#ifndef LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYDISK_H
#define LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYDISK_H
 
#include <future>
#include <molpro/Profiler.h>
#include <molpro/linalg/array/DistrArray.h>
#include <molpro/linalg/array/Span.h>
 
namespace molpro::linalg::array {
class DistrArrayDisk : public DistrArray {
public:
  using disk_array = void; 
protected:
  bool m_allocated = false;                     
  std::unique_ptr<Distribution> m_distribution; 
  size_t m_buffer_size = 8192;                  
  using DistrArray::DistrArray;
 
  DistrArrayDisk(std::unique_ptr<Distribution> distr, MPI_Comm commun);
  DistrArrayDisk();
  DistrArrayDisk(const DistrArray& source);
  DistrArrayDisk(DistrArrayDisk&& source) noexcept;
  ~DistrArrayDisk() override;
 
public:
  virtual void erase() = 0;
  [[nodiscard]] const Distribution& distribution() const override;
  [[nodiscard]] value_type dot(const DistrArrayDisk& y) const;
  [[nodiscard]] value_type dot(const DistrArray& y) const override;
  [[nodiscard]] value_type dot(const SparseArray& y) const override;
  void set_buffer_size(size_t buffer_size) { m_buffer_size = buffer_size; }
  void copy(const DistrArray& y) override;
 
protected:
  class LocalBufferDisk : public DistrArray::LocalBuffer {
  public:
    explicit LocalBufferDisk(DistrArrayDisk& source);
    explicit LocalBufferDisk(DistrArrayDisk& source, const span::Span<value_type>& buffer);
    ~LocalBufferDisk() override;
    bool do_dump = true;
 
  protected:
    std::vector<value_type> m_snapshot_buffer; 
    DistrArrayDisk& m_source;                  
  };
 
public:
  [[nodiscard]] std::unique_ptr<LocalBuffer> local_buffer() override;
  [[nodiscard]] std::unique_ptr<const LocalBuffer> local_buffer() const override;
  [[nodiscard]] std::unique_ptr<LocalBuffer> local_buffer(const span::Span<value_type>& buffer);
  [[nodiscard]] std::unique_ptr<const LocalBuffer> local_buffer(const span::Span<value_type>& buffer) const;
 
public:
};
 
double dot(const DistrArrayDisk& x, const DistrArrayDisk& y);
double dot(const DistrArrayDisk& x, const DistrArray& y);
double dot(const DistrArray& x, const DistrArrayDisk& y);
 
} // namespace molpro::linalg::array
 
#endif // LINEARALGEBRA_SRC_MOLPRO_LINALG_ARRAY_DISTRARRAYDISK_H