28#ifndef EWOMS_VTK_MULTI_WRITER_HH
29#define EWOMS_VTK_MULTI_WRITER_HH
38#include <opm/material/common/Valgrind.hpp>
40#include <dune/common/fvector.hh>
41#include <dune/common/version.hh>
42#include <dune/istl/bvector.hh>
43#include <dune/grid/io/file/vtk/vtkwriter.hh>
64template <
class Gr
idView,
int vtkFormat>
78 if (multiWriter_.commSize_ > 1)
79 fileName = multiWriter_.curWriter_->pwrite(multiWriter_.curOutFileName_,
80 multiWriter_.outputDir_,
82 static_cast<Dune::VTK::OutputType
>(vtkFormat));
84 fileName = multiWriter_.curWriter_->write(multiWriter_.outputDir_ +
"/" + multiWriter_.curOutFileName_,
85 static_cast<Dune::VTK::OutputType
>(vtkFormat));
90 const std::filesystem::path
fullPath{fileName};
92 multiWriter_.multiFile_.precision(16);
93 multiWriter_.multiFile_ <<
" <DataSet timestep=\"" << multiWriter_.curTime_ <<
"\" file=\""
101 enum { dim = GridView::dimension };
103 using VertexMapper = Dune::MultipleCodimMultipleGeomTypeMapper<GridView>;
104 using ElementMapper = Dune::MultipleCodimMultipleGeomTypeMapper<GridView>;
107 using Scalar = BaseOutputWriter::Scalar;
108 using Vector = BaseOutputWriter::Vector;
109 using Tensor = BaseOutputWriter::Tensor;
110 using ScalarBuffer = BaseOutputWriter::ScalarBuffer;
111 using VectorBuffer = BaseOutputWriter::VectorBuffer;
112 using TensorBuffer = BaseOutputWriter::TensorBuffer;
114 using VtkWriter = Dune::VTKWriter<GridView>;
115 using FunctionPtr = std::shared_ptr< Dune::VTKFunction< GridView > >;
118 const GridView& gridView,
119 const std::string& outputDir,
120 const std::string&
simName =
"",
122 : gridView_(gridView)
123 , elementMapper_(gridView, Dune::mcmgElementLayout())
124 , vertexMapper_(gridView, Dune::mcmgVertexLayout())
125 , curWriter_(
nullptr)
129 outputDir_ = outputDir;
135 if (multiFileName_.empty())
136 multiFileName_ = outputDir_+
"/"+simName_+
".pvd";
138 commRank_ = gridView.comm().rank();
139 commSize_ = gridView.comm().size();
156 {
return curWriterNum_; }
167#if DUNE_VERSION_GTE(DUNE_GRID, 2, 8)
168 elementMapper_.update(gridView_);
169 vertexMapper_.update(gridView_);
171 elementMapper_.update();
172 vertexMapper_.update();
181 if (!multiFile_.is_open()) {
182 startMultiFile_(multiFileName_);
191 curOutFileName_ = fileName_();
193 curWriter_ =
new VtkWriter(gridView_, Dune::VTK::conforming);
206 managedScalarBuffers_.push_back(
buf);
219 for (
size_t i = 0; i <
numOuter; ++ i)
222 managedVectorBuffers_.push_back(
buf);
244 sanitizeScalarBuffer_(
buf);
252 curWriter_->addVertexData(
fnPtr);
272 sanitizeScalarBuffer_(
buf);
280 curWriter_->addCellData(
fnPtr);
301 sanitizeVectorBuffer_(
buf);
309 curWriter_->addVertexData(
fnPtr);
320 std::ostringstream
oss;
329 curWriter_->addVertexData(
fnPtr);
350 sanitizeVectorBuffer_(
buf);
358 curWriter_->addCellData(
fnPtr);
369 std::ostringstream
oss;
378 curWriter_->addCellData(
fnPtr);
392 auto tasklet = std::make_shared<WriteDataTasklet>(*
this);
407 template <
class Restarter>
410 res.serializeSectionBegin(
"VTKMultiWriter");
411 res.serializeStream() << curWriterNum_ <<
"\n";
413 if (commRank_ == 0) {
416 if (multiFile_.is_open()) {
419 multiFile_.seekp(0, std::ios::end);
435 res.serializeSectionEnd();
441 template <
class Restarter>
444 res.deserializeSectionBegin(
"VTKMultiWriter");
445 res.deserializeStream() >> curWriterNum_;
447 if (commRank_ == 0) {
449 std::getline(
res.deserializeStream(),
dummy);
455 std::getline(
res.deserializeStream(),
dummy);
456 if (multiFile_.is_open())
460 multiFile_.open(multiFileName_.c_str());
464 multiFile_.write(tmp,
fileLen);
472 std::getline(
res.deserializeStream(), tmp);
474 res.deserializeSectionEnd();
478 std::string fileName_()
481 std::ostringstream
oss;
482 oss << simName_ <<
"-"
483 << std::setw(5) << std::setfill(
'0') << curWriterNum_;
487 std::string fileSuffix_()
488 {
return (GridView::dimension == 1) ?
"vtp" :
"vtu"; }
493 if (commRank_ == 0) {
496 multiFile_ <<
"<?xml version=\"1.0\"?>\n"
497 "<VTKFile type=\"Collection\"\n"
499 " byte_order=\"LittleEndian\"\n"
500 " compressor=\"vtkZLibDataCompressor\">\n"
505 void finishMultiFile_()
508 if (commRank_ == 0) {
510 std::ofstream::pos_type pos = multiFile_.tellp();
511 multiFile_ <<
" </Collection>\n"
513 multiFile_.seekp(pos);
520 void sanitizeScalarBuffer_(ScalarBuffer&)
525 void sanitizeVectorBuffer_(VectorBuffer&)
531 void releaseBuffers_()
535 curWriter_ =
nullptr;
536 while (managedScalarBuffers_.begin() != managedScalarBuffers_.end()) {
537 delete managedScalarBuffers_.front();
538 managedScalarBuffers_.pop_front();
540 while (managedVectorBuffers_.begin() != managedVectorBuffers_.end()) {
541 delete managedVectorBuffers_.front();
542 managedVectorBuffers_.pop_front();
546 const GridView gridView_;
547 ElementMapper elementMapper_;
548 VertexMapper vertexMapper_;
550 std::string outputDir_;
551 std::string simName_;
552 std::ofstream multiFile_;
553 std::string multiFileName_;
558 VtkWriter *curWriter_;
560 std::string curOutFileName_;
563 std::list<ScalarBuffer *> managedScalarBuffers_;
564 std::list<VectorBuffer *> managedVectorBuffers_;
566 TaskletRunner taskletRunner_;
The base class for all output writers.
The base class for all output writers.
Definition baseoutputwriter.hh:44
The base class for tasklets.
Definition tasklets.hpp:44
void barrier()
Make sure that all tasklets have been completed after this method has been called.
Definition tasklets.cpp:134
void dispatch(std::shared_ptr< TaskletInterface > tasklet)
Add a new tasklet.
Definition tasklets.cpp:101
Simplifies writing multi-file VTK datasets.
Definition vtkmultiwriter.hh:66
void endWrite(bool onlyDiscard=false)
Finalizes the current writer.
Definition vtkmultiwriter.hh:389
void attachTensorVertexData(TensorBuffer &buf, std::string name)
Add a finished vertex-centered tensor field to the output.
Definition vtkmultiwriter.hh:315
void deserialize(Restarter &res)
Read the multi-writer's state from a restart file.
Definition vtkmultiwriter.hh:442
void attachTensorElementData(TensorBuffer &buf, std::string name)
Add a finished element-centered tensor field to the output.
Definition vtkmultiwriter.hh:364
int curWriterNum() const
Returns the number of the current VTK file.
Definition vtkmultiwriter.hh:155
VectorBuffer * allocateManagedVectorBuffer(size_t numOuter, size_t numInner)
Allocate a managed buffer for a vector field.
Definition vtkmultiwriter.hh:216
void gridChanged()
Updates the internal data structures after mesh refinement.
Definition vtkmultiwriter.hh:165
void beginWrite(double t)
Called whenever a new time step must be written.
Definition vtkmultiwriter.hh:179
ScalarBuffer * allocateManagedScalarBuffer(size_t numEntities)
Allocate a managed buffer for a scalar field.
Definition vtkmultiwriter.hh:203
void attachScalarVertexData(ScalarBuffer &buf, std::string name)
Add a finished vertex centered vector field to the output.
Definition vtkmultiwriter.hh:242
void attachVectorElementData(VectorBuffer &buf, std::string name)
Add a element centered quantity to the output.
Definition vtkmultiwriter.hh:348
void attachScalarElementData(ScalarBuffer &buf, std::string name)
Add a element centered quantity to the output.
Definition vtkmultiwriter.hh:270
void serialize(Restarter &res)
Write the multi-writer's state to a restart file.
Definition vtkmultiwriter.hh:408
void attachVectorVertexData(VectorBuffer &buf, std::string name)
Add a finished vertex centered vector field to the output.
Definition vtkmultiwriter.hh:299
Provides a vector-valued function using Dune::FieldVectors as elements.
Definition vtkscalarfunction.hh:50
Provides a tensor-valued function using Dune::FieldMatrix objects as elements.
Definition vtktensorfunction.hh:47
Provides a vector-valued function using Dune::FieldVectors as elements.
Definition vtkvectorfunction.hh:50
This file contains a set of helper functions used by VFPProd / VFPInj.
Definition blackoilboundaryratevector.hh:37
constexpr auto getPropValue()
get the value data member of a property
Definition propertysystem.hh:242
Provides a mechanism to dispatch work to separate threads.
Provides a vector-valued function using Dune::FieldVectors as elements.
Provides a tensor-valued function using Dune::FieldMatrix objects as elements.
Provides a vector-valued function using Dune::FieldVectors as elements.