hdf5-reader.hpp

Go to the documentation of this file.

#ifndef OPENEV_READERS_HDF5_READER_HPP
#define OPENEV_READERS_HDF5_READER_HPP
 
#include "openev/readers/abstract-reader.hpp"
#include <H5Cpp.h>
#include <cstddef>
#include <string>
#include <vector>
 
namespace ev {
[[maybe_unused]] constexpr bool USING_HDF5_READER_HPP = true;

class HDF5Reader : public AbstractReader_ {
public:
  explicit HDF5Reader(const std::string &filename,
                      const std::string &t_path = "/events/t",
                      const std::string &x_path = "/events/x",
                      const std::string &y_path = "/events/y",
                      const std::string &p_path = "/events/p");
  ~HDF5Reader() override;

  HDF5Reader(const HDF5Reader &) = delete;
  HDF5Reader(HDF5Reader &&) noexcept = delete;
  HDF5Reader &operator=(const HDF5Reader &) = delete;
  HDF5Reader &operator=(HDF5Reader &&) noexcept = delete;
 
private:
  H5::H5File file_;
  H5::DataSet t_ds_, x_ds_, y_ds_, p_ds_;
  hsize_t total_{0};
  hsize_t file_pos_{0};
 
  static constexpr hsize_t kChunk = 4096;
  std::vector<double> t_buf_;
  std::vector<int> x_buf_;
  std::vector<int> y_buf_;
  std::vector<int> p_buf_;
  hsize_t buf_pos_{0};
  hsize_t buf_size_{0};
 
  std::size_t read_(Event *events, const std::size_t max) override;
  void reset_() override;
};
 
} // namespace ev
 
#endif // OPENEV_READERS_HDF5_READER_HPP