4 #ifndef DUNE_SUBSAMPLINGVTKWRITER_HH
5 #define DUNE_SUBSAMPLINGVTKWRITER_HH
10 #include <dune/common/indent.hh>
11 #include <dune/geometry/type.hh>
12 #include <dune/geometry/virtualrefinement.hh>
35 template<
class Gr
idView >
42 typedef typename GridView::Grid::ctype ctype;
43 typedef typename GridView::template Codim< 0 >::Entity Entity;
44 typedef VirtualRefinement<dim, ctype> Refinement;
45 typedef typename Refinement::IndexVector IndexVector;
46 typedef typename Refinement::ElementIterator SubElementIterator;
47 typedef typename Refinement::VertexIterator SubVertexIterator;
79 Dune::RefinementIntervals intervals_,
bool coerceToSimplex_ =
false,
82 , intervals(intervals_), coerceToSimplex(coerceToSimplex_)
84 if(intervals_.intervals() < 1) {
85 DUNE_THROW(Dune::IOError,
"SubsamplingVTKWriter: Refinement intervals must be larger than zero! (One interval means no subsampling)");
95 template<
typename SubIterator>
96 struct IteratorSelector
99 SubElementIterator refinementBegin(
const Refinement& refinement, Dune::RefinementIntervals intervals, IteratorSelector<SubElementIterator>)
101 return refinement.eBegin(intervals);
104 SubVertexIterator refinementBegin(
const Refinement& refinement, Dune::RefinementIntervals intervals, IteratorSelector<SubVertexIterator>)
106 return refinement.vBegin(intervals);
109 SubElementIterator refinementEnd(
const Refinement& refinement, Dune::RefinementIntervals intervals, IteratorSelector<SubElementIterator>)
111 return refinement.eEnd(intervals);
114 SubVertexIterator refinementEnd(
const Refinement& refinement, Dune::RefinementIntervals intervals, IteratorSelector<SubVertexIterator>)
116 return refinement.vEnd(intervals);
119 template<
typename Data,
typename Iterator,
typename SubIterator>
120 void writeData(VTK::VTUWriter& writer,
const Data& data,
const Iterator begin,
const Iterator end,
int nentries, IteratorSelector<SubIterator> sis)
122 for (
auto it = data.begin(),
128 VTK::FieldInfo fieldInfo = f.fieldInfo();
129 std::size_t writecomps = fieldInfo.size();
130 switch (fieldInfo.type())
138 DUNE_THROW(IOError,
"Cannot write VTK vectors with more than 3 components (components was " << writecomps <<
")");
142 DUNE_THROW(NotImplemented,
"VTK output for tensors not implemented yet");
144 std::shared_ptr<VTK::DataArrayWriter> p
145 (writer.makeArrayWriter(f.name(), writecomps, nentries, fieldInfo.precision()));
146 if(!p->writeIsNoop())
147 for (Iterator eit = begin; eit!=end; ++eit)
149 const Entity & e = *eit;
151 Refinement &refinement =
152 buildRefinement<dim, ctype>(eit->type(),
153 subsampledGeometryType(eit->type()));
154 for(SubIterator sit = refinementBegin(refinement,intervals,sis),
155 send = refinementEnd(refinement,intervals,sis);
159 f.write(sit.coords(),*p);
161 for(
unsigned j = f.fieldInfo().size(); j < writecomps; j++)
172 virtual void countEntities(
int &nvertices_,
int &ncells_,
int &ncorners_);
195 void addVertexData (
const V& v,
const std::string &name,
int ncomps=1);
197 void addCellData (
const V& v,
const std::string &name,
int ncomps=1);
199 Dune::RefinementIntervals intervals;
200 bool coerceToSimplex;
204 template <
class Gr
idView>
210 for (
CellIterator it=this->cellBegin(); it!=cellEnd(); ++it)
212 Refinement &refinement = buildRefinement<dim, ctype>(it->type(), subsampledGeometryType(it->type()));
214 ncells_ += refinement.nElements(intervals);
215 nvertices_ += refinement.nVertices(intervals);
216 ncorners_ += refinement.nElements(intervals) * refinement.eBegin(intervals).vertexIndices().size();
222 template <
class Gr
idView>
225 if(celldata.size() == 0)
231 std::string defaultScalarField, defaultVectorField;
232 std::tie(defaultScalarField, defaultVectorField) = this->getDataNames(celldata);
234 writer.
beginCellData(defaultScalarField, defaultVectorField);
235 writeData(writer,celldata,cellBegin(),cellEnd(),ncells,IteratorSelector<SubElementIterator>());
240 template <
class Gr
idView>
243 if(vertexdata.size() == 0)
249 std::string defaultScalarField, defaultVectorField;
250 std::tie(defaultScalarField, defaultVectorField) = this->getDataNames(vertexdata);
253 writeData(writer,vertexdata,cellBegin(),cellEnd(),nvertices,IteratorSelector<SubVertexIterator>());
258 template <
class Gr
idView>
263 std::shared_ptr<VTK::DataArrayWriter> p
264 (writer.
makeArrayWriter(
"Coordinates", 3, nvertices, this->coordPrecision()));
265 if(!p->writeIsNoop())
268 Refinement &refinement =
269 buildRefinement<dim, ctype>(i->type(),
270 subsampledGeometryType(i->type()));
271 for(SubVertexIterator sit = refinement.vBegin(intervals),
272 send = refinement.vEnd(intervals);
275 FieldVector<ctype, dimw> coords = i->geometry().global(sit.coords());
276 for (
int j=0; j<
std::min(
int(dimw),3); j++)
278 for (
int j=
std::min(
int(dimw),3); j<3; j++)
289 template <
class Gr
idView>
296 std::shared_ptr<VTK::DataArrayWriter> p1
299 if(!p1->writeIsNoop()) {
303 GeometryType coercedToType = subsampledGeometryType(i->type());
304 Refinement &refinement =
305 buildRefinement<dim, ctype>(i->type(), coercedToType);
306 for(SubElementIterator sit = refinement.eBegin(intervals),
307 send = refinement.eEnd(intervals);
310 IndexVector indices = sit.vertexIndices();
311 for(
unsigned int ii = 0; ii < indices.size(); ++ii)
314 offset += refinement.nVertices(intervals);
321 std::shared_ptr<VTK::DataArrayWriter> p2
323 if(!p2->writeIsNoop()) {
328 Refinement &refinement =
329 buildRefinement<dim, ctype>(i->type(),
330 subsampledGeometryType(i->type()));
331 unsigned int verticesPerCell =
332 refinement.eBegin(intervals).vertexIndices().size();
333 for(
int element = 0; element < refinement.nElements(intervals);
336 offset += verticesPerCell;
346 std::shared_ptr<VTK::DataArrayWriter> p3
348 if(!p3->writeIsNoop())
351 GeometryType coerceTo = subsampledGeometryType(it->type());
352 Refinement &refinement =
353 buildRefinement<dim, ctype>(it->type(), coerceTo);
355 for(
int i = 0; i < refinement.nElements(intervals); ++i)
Provides file i/o for the visualization toolkit.
@ dimensionworld
The dimension of the world the grid lives in.
Definition: common/gridview.hh:134
@ dimension
The dimension of the grid.
Definition: common/gridview.hh:130
Include standard header files.
Definition: agrid.hh:58
int min(const DofVectorPointer< int > &dofVector)
Definition: dofvector.hh:346
Precision
which precision to use when writing out data to vtk files
Definition: common.hh:269
int renumber(const Dune::GeometryType &t, int i)
renumber VTK <-> Dune
Definition: common.hh:184
@ nonconforming
Output non-conforming data.
Definition: common.hh:79
GeometryType geometryType(const Dune::GeometryType &t)
mapping from GeometryType to VTKGeometryType
Definition: common.hh:149
GeometryType
Type representing VTK's entity geometry types.
Definition: common.hh:130
Grid view abstract base class.
Definition: common/gridview.hh:63
@ tensor
tensor field (always 3x3)
@ vector
vector-valued field (always 3D, will be padded if necessary)
Writer for the output of subsampled grid functions in the vtk format.
Definition: subsamplingvtkwriter.hh:38
void addVertexData(const std::shared_ptr< const VTKFunction > &p)
Add a grid function that lives on the vertices of the grid to the visualization.
Definition: vtkwriter.hh:711
void addCellData(const std::shared_ptr< const VTKFunction > &p)
Add a grid function that lives on the cells of the grid to the visualization.
Definition: vtkwriter.hh:647
virtual void writeGridPoints(VTK::VTUWriter &writer)
write the positions of vertices
Definition: subsamplingvtkwriter.hh:259
virtual void writeVertexData(VTK::VTUWriter &writer)
write vertex data
Definition: subsamplingvtkwriter.hh:241
SubsamplingVTKWriter(const GridView &gridView, Dune::RefinementIntervals intervals_, bool coerceToSimplex_=false, VTK::Precision coordPrecision=VTK::Precision::float32)
Construct a SubsamplingVTKWriter working on a specific GridView.
Definition: subsamplingvtkwriter.hh:78
virtual void countEntities(int &nvertices_, int &ncells_, int &ncorners_)
count the vertices, cells and corners
Definition: subsamplingvtkwriter.hh:205
virtual void writeCellData(VTK::VTUWriter &writer)
write cell data
Definition: subsamplingvtkwriter.hh:223
virtual void writeGridCells(VTK::VTUWriter &writer)
write the connectivity array
Definition: subsamplingvtkwriter.hh:290
Writer for the ouput of grid functions in the vtk format.
Definition: vtkwriter.hh:93
VertexIterator vertexBegin() const
Definition: vtkwriter.hh:506
void addVertexData(const std::shared_ptr< const VTKFunction > &p)
Add a grid function that lives on the vertices of the grid to the visualization.
Definition: vtkwriter.hh:711
CellIterator cellEnd() const
Definition: vtkwriter.hh:400
std::list< VTKLocalFunction > vertexdata
Definition: vtkwriter.hh:1583
void addCellData(const std::shared_ptr< const VTKFunction > &p)
Add a grid function that lives on the cells of the grid to the visualization.
Definition: vtkwriter.hh:647
CellIterator cellBegin() const
Definition: vtkwriter.hh:395
VTK::OutputType outputtype
Definition: vtkwriter.hh:1607
std::list< VTKLocalFunction > celldata
Definition: vtkwriter.hh:1582
VTK::Precision coordPrecision() const
get the precision with which coordinates are written out
Definition: vtkwriter.hh:780
std::list< VTKLocalFunction >::const_iterator FunctionIterator
Definition: vtkwriter.hh:374
int nvertices
Definition: vtkwriter.hh:1590
int ncells
Definition: vtkwriter.hh:1589
VertexIterator vertexEnd() const
Definition: vtkwriter.hh:513
int ncorners
Definition: vtkwriter.hh:1591
Iterator over the grids elements.
Definition: vtkwriter.hh:383
Dump a .vtu/.vtp files contents to a stream.
Definition: vtuwriter.hh:96
void endCellData()
finish CellData section
Definition: vtuwriter.hh:218
void beginCells()
start section for the grid cells/PolyData lines
Definition: vtuwriter.hh:272
void endPointData()
finish PointData section
Definition: vtuwriter.hh:180
void beginCellData(const std::string &scalars="", const std::string &vectors="")
start CellData section
Definition: vtuwriter.hh:203
void beginPointData(const std::string &scalars="", const std::string &vectors="")
start PointData section
Definition: vtuwriter.hh:165
DataArrayWriter * makeArrayWriter(const std::string &name, unsigned ncomps, unsigned nitems, Precision prec)
acquire a DataArrayWriter
Definition: vtuwriter.hh:378
void endPoints()
finish section for the point coordinates
Definition: vtuwriter.hh:247
void endCells()
start section for the grid cells/PolyData lines
Definition: vtuwriter.hh:283
void beginPoints()
start section for the point coordinates
Definition: vtuwriter.hh:236