unstructuredgridwriter.hh

Go to the documentation of this file.

#pragma once
 
#include <array>
#include <iosfwd>
#include <map>
 
#include <dune/vtk/filewriter.hh>
#include <dune/vtk/types.hh>
#include <dune/vtk/vtkwriterbase.hh>
#include <dune/vtk/datacollectors/continuousdatacollector.hh>
#include <dune/vtk/utility/concepts.hh>
 
namespace Dune::Vtk
{
 
  template <class GridView, class DataCollector = ContinuousDataCollector<GridView>>
  class UnstructuredGridWriter
      : public VtkWriterBase<GridView, DataCollector>
  {
    template <class> friend class TimeseriesWriter;
 
    using Super = VtkWriterBase<GridView, DataCollector>;
    using pos_type = typename Super::pos_type;
 
  public:
    using Super::Super;
 
  private:
    virtual void writeSerialFile (std::ofstream& out) const override;
 
    virtual void writeParallelFile (std::ofstream& out, std::string const& pfilename, int size) const override;
 
 
    void writeTimeseriesSerialFile (std::ofstream& out,
                                    std::string const& filenameMesh,
                                    std::vector<std::pair<double, std::string>> const& timesteps,
                                    std::vector<std::uint64_t> const& blocks) const;
 
    void writeTimeseriesParallelFile (std::ofstream& out,
                                      std::string const& pfilename, int size,
                                      std::vector<std::pair<double, std::string>> const& timesteps) const;
 
    virtual std::string fileExtension () const override
    {
      return "vtu";
    }
 
    virtual void writeGridAppended (std::ofstream& out, std::vector<std::uint64_t>& blocks) const override;
 
    // Write the element connectivity to the output stream `out`. In case
    // of binary format, stores the streampos of XML attributes "offset" in the
    // vector `offsets`.
    void writeCells (std::ofstream& out,
                     std::vector<pos_type>& offsets,
                     std::optional<std::size_t> timestep = {}) const;
 
    void writePointIds (std::ofstream& out,
                        std::vector<pos_type>& offsets,
                        std::optional<std::size_t> timestep = {}) const;
 
  private:
    using Super::dataCollector_;
    using Super::format_;
    using Super::datatype_;
    using Super::headertype_;
 
    // attached data
    using Super::pointData_;
    using Super::cellData_;
  };
 
  // deduction guides
  template <class GridView, class... Args,
    IsGridView<GridView> = true>
  UnstructuredGridWriter(GridView, Args...)
    -> UnstructuredGridWriter<GridView, ContinuousDataCollector<GridView>>;
 
  template <class DataCollector, class... Args,
    IsDataCollector<DataCollector> = true>
  UnstructuredGridWriter(DataCollector&, Args...)
    -> UnstructuredGridWriter<typename DataCollector::GridView, DataCollector>;
 
  template <class DataCollector, class... Args,
    Vtk::IsDataCollector<DataCollector> = true>
  UnstructuredGridWriter(DataCollector&&, Args...)
    -> UnstructuredGridWriter<typename DataCollector::GridView, DataCollector>;
 
  template <class DataCollector, class... Args,
    IsDataCollector<DataCollector> = true>
  UnstructuredGridWriter(std::shared_ptr<DataCollector>, Args...)
    -> UnstructuredGridWriter<typename DataCollector::GridView, DataCollector>;
 
} // end namespace Dune::Vtk
 
#include "unstructuredgridwriter.impl.hh"