timeserieswriter.hh

Go to the documentation of this file.

#pragma once
 
#include <cassert>
#include <optional>
#include <string>
#include <tuple>
#include <vector>
 
#include <dune/vtk/filewriter.hh>
#include <dune/vtk/types.hh>
#include <dune/vtk/utility/filesystem.hh>
#include <dune/vtk/utility/uid.hh>
 
namespace Dune::Vtk
{
 
  template <class VtkWriter>
  class TimeseriesWriter
      : public Vtk::FileWriter
  {
  protected:
    using Self = TimeseriesWriter;
    using pos_type = typename std::ostream::pos_type;
 
    template <class W>
    using HasWriteTimeseriesSerialFile = decltype(&W::writeTimeseriesSerialFile);
    // static_assert(Std::is_detected<HasWriteTimeseriesSerialFile, VtkWriter>::value, "");
 
    template <class W>
    using HasWriteTimeseriesParallelFile = decltype(&W::writeTimeseriesParallelFile);
    // static_assert(Std::is_detected<HasWriteTimeseriesParallelFile, VtkWriter>::value, "");
 
  public:
    template <class... Args,
      disableCopyMove<Self, Args...> = 0>
    TimeseriesWriter (Args&&... args)
      : vtkWriter_{std::forward<Args>(args)...}
    {
      assert(vtkWriter_.format_ != Vtk::FormatTypes::ASCII && "Timeseries writer requires APPENDED mode");
      std::srand(std::time(nullptr));
 
      // put temporary file to /tmp directory
      tmpDir_ = Vtk::Path("/tmp/TimeseriesWriter_" + Vtk::uid(10) + "/");
    }
 
    ~TimeseriesWriter ();
 
 
    void writeTimestep (double time, std::string const& fn,
                        std::optional<std::string> tmpDir = {},
                        bool writeCollection = true) const;
 
    // NOTE: requires an aforegoing call to \ref writeTimestep
    virtual std::string write (std::string const& fn, std::optional<std::string> dir = {}) const override;
 
    template <class Function, class... Args>
    TimeseriesWriter& addPointData (Function const& fct, Args&&... args)
    {
      vtkWriter_.addPointData(fct, std::forward<Args>(args)...);
      return *this;
    }
 
    template <class Function, class... Args>
    TimeseriesWriter& addCellData (Function const& fct, Args&&... args)
    {
      vtkWriter_.addCellData(fct, std::forward<Args>(args)...);
      return *this;
    }
 
  protected:
    VtkWriter vtkWriter_;
    Vtk::Path tmpDir_;
 
    mutable bool initialized_ = false;
 
    // block size of attached data
    mutable std::vector<std::uint64_t> blocks_;
 
    mutable std::string filenameMesh_;
    mutable std::vector<std::pair<double, std::string>> timesteps_;
  };
 
} // end namespace Dune::Vtk
 
#include "timeserieswriter.impl.hh"