hdf5_writer.hh

Go to the documentation of this file.

#pragma once
 
// DiFfRG
#include <DiFfRG/discretization/data/hdf5_frame.hh>
#include <DiFfRG/discretization/data/output_timings.hh>
 
// standard library
#include <condition_variable>
#include <deque>
#include <exception>
#include <mutex>
#include <thread>
 
namespace DiFfRG
{
  class HDF5FrameWriter
  {
  public:
    explicit HDF5FrameWriter(unsigned int queue_depth);
    ~HDF5FrameWriter() noexcept;
 
    HDF5FrameWriter(const HDF5FrameWriter &) = delete;
    HDF5FrameWriter &operator=(const HDF5FrameWriter &) = delete;
 
    bool asynchronous() const noexcept { return depth > 0; }
 
    void submit(HDF5Frame frame, FrameTimings *timings = nullptr);
 
    void drain();
 
    void finish();
 
    FrameTimings worker_totals() const;
 
  private:
    void loop();
    void rethrow_stored();
 
    const unsigned int depth;
 
    mutable std::mutex mutex;
    std::condition_variable producer_condition, consumer_condition;
    std::deque<HDF5Frame> queue;
    bool stopping = false;
    bool writing = false; 
    bool finished = false;
 
    std::exception_ptr stored_exception;
    FrameTimings worker_timings;
 
    std::thread worker;
  };
} // namespace DiFfRG