Writer.h

#pragma once
 
#include <boost/core/demangle.hpp>
#include <utility>
 
// using HighFive
#include <highfive/H5File.hpp>
 
#include "hdtree/Atomic.h"
#include "hdtree/Constants.h"
#include "hdtree/Exception.h"
 
namespace hdtree {
 
class Writer {
 public:
  Writer(const std::pair<std::string, std::string>& file_tree_path,
         bool inplace = false, int rows_per_chunk = 10000, bool shuffle = true,
         int compression_level = 6);
 
  ~Writer();
 
  int getRowsPerChunk() const { return rows_per_chunk_; }
 
  void flush();
 
  void increment();
 
  const std::string& name() const;
 
  void structure(const std::string& branch_name,
                 const std::pair<std::string, int>& type);
 
  HighFive::DataSet createDataSet(const std::string& branch_name,
                                  HighFive::DataType data_type);
 
  inline std::size_t entries() const { return entries_; }
 
  friend std::ostream& operator<<(std::ostream& s, const Writer& w) {
    return s << "Writer(" << w.name() << ")";
  }
 
  Writer(const Writer&) = delete;
  void operator=(const Writer&) = delete;
 
 private:
  HighFive::File file_;
  HighFive::Group tree_;
  HighFive::DataSetCreateProps create_props_;
  HighFive::DataSpace space_;
  std::size_t entries_;
  std::size_t rows_per_chunk_;
};
 
}  // namespace hdtree