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;
76 int level_,
bool coerceToSimplex_ =
false)
78 , level(level_), coerceToSimplex(coerceToSimplex_)
81 DUNE_THROW(Dune::IOError,
"SubsamplingVTKWriter: Negative Subsampling " << level_ <<
" must not be used!");
93 template<
typename SubIterator>
94 struct IteratorSelector
97 SubElementIterator refinementBegin(
const Refinement& refinement,
int level, IteratorSelector<SubElementIterator>)
99 return refinement.eBegin(level);
102 SubVertexIterator refinementBegin(
const Refinement& refinement,
int level, IteratorSelector<SubVertexIterator>)
104 return refinement.vBegin(level);
107 SubElementIterator refinementEnd(
const Refinement& refinement,
int level, IteratorSelector<SubElementIterator>)
109 return refinement.eEnd(level);
112 SubVertexIterator refinementEnd(
const Refinement& refinement,
int level, IteratorSelector<SubVertexIterator>)
114 return refinement.vEnd(level);
117 template<
typename Data,
typename Iterator,
typename SubIterator>
118 void writeData(VTK::VTUWriter& writer,
const Data& data,
const Iterator begin,
const Iterator end,
int nentries, IteratorSelector<SubIterator> sis)
120 for (
auto it = data.begin(),
126 VTK::FieldInfo fieldInfo = f.fieldInfo();
127 std::size_t writecomps = fieldInfo.size();
128 switch (fieldInfo.type())
136 DUNE_THROW(IOError,
"Cannot write VTK vectors with more than 3 components (components was " << writecomps <<
")");
140 DUNE_THROW(NotImplemented,
"VTK output for tensors not implemented yet");
142 std::shared_ptr<VTK::DataArrayWriter<float> > p
143 (writer.makeArrayWriter<
float>(f.name(), writecomps, nentries));
144 if(!p->writeIsNoop())
145 for (Iterator eit = begin; eit!=end; ++eit)
147 const Entity & e = *eit;
149 Refinement &refinement =
150 buildRefinement<dim, ctype>(eit->type(),
151 subsampledGeometryType(eit->type()));
152 for(SubIterator sit = refinementBegin(refinement,level,sis),
153 send = refinementEnd(refinement,level,sis);
157 f.write(sit.coords(),*p);
159 for(
unsigned j = f.fieldInfo().size(); j < writecomps; j++)
193 void addVertexData (
const V& v,
const std::string &name,
int ncomps=1);
195 void addCellData (
const V& v,
const std::string &name,
int ncomps=1);
198 bool coerceToSimplex;
202 template <
class Gr
idView>
208 for (
CellIterator it=this->cellBegin(); it!=cellEnd(); ++it)
210 Refinement &refinement = buildRefinement<dim, ctype>(it->type(), subsampledGeometryType(it->type()));
212 ncells += refinement.nElements(level);
213 nvertices += refinement.nVertices(level);
214 ncorners += refinement.nElements(level) * refinement.eBegin(level).vertexIndices().size();
220 template <
class Gr
idView>
223 if(celldata.size() == 0)
229 std::string defaultScalarField, defaultVectorField;
230 std::tie(defaultScalarField, defaultVectorField) = this->getDataNames(celldata);
232 writer.
beginCellData(defaultScalarField, defaultVectorField);
233 writeData(writer,celldata,cellBegin(),cellEnd(),ncells,IteratorSelector<SubElementIterator>());
238 template <
class Gr
idView>
241 if(vertexdata.size() == 0)
247 std::string defaultScalarField, defaultVectorField;
248 std::tie(defaultScalarField, defaultVectorField) = this->getDataNames(vertexdata);
251 writeData(writer,vertexdata,cellBegin(),cellEnd(),nvertices,IteratorSelector<SubVertexIterator>());
256 template <
class Gr
idView>
261 std::shared_ptr<VTK::DataArrayWriter<float> > p
263 if(!p->writeIsNoop())
266 Refinement &refinement =
267 buildRefinement<dim, ctype>(i->type(),
268 subsampledGeometryType(i->type()));
269 for(SubVertexIterator sit = refinement.vBegin(level),
270 send = refinement.vEnd(level);
273 FieldVector<ctype, dimw> coords = i->geometry().global(sit.coords());
274 for (
int j=0; j<
std::min(
int(dimw),3); j++)
276 for (
int j=
std::min(
int(dimw),3); j<3; j++)
287 template <
class Gr
idView>
294 std::shared_ptr<VTK::DataArrayWriter<int> > p1
297 if(!p1->writeIsNoop()) {
301 GeometryType coercedToType = subsampledGeometryType(i->type());
302 Refinement &refinement =
303 buildRefinement<dim, ctype>(i->type(), coercedToType);
304 for(SubElementIterator sit = refinement.eBegin(level),
305 send = refinement.eEnd(level);
308 IndexVector indices = sit.vertexIndices();
309 for(
unsigned int ii = 0; ii < indices.size(); ++ii)
312 offset += refinement.nVertices(level);
319 std::shared_ptr<VTK::DataArrayWriter<int> > p2
321 if(!p2->writeIsNoop()) {
326 Refinement &refinement =
327 buildRefinement<dim, ctype>(i->type(),
328 subsampledGeometryType(i->type()));
329 unsigned int verticesPerCell =
330 refinement.eBegin(level).vertexIndices().size();
331 for(
int element = 0; element < refinement.nElements(level);
334 offset += verticesPerCell;
344 std::shared_ptr<VTK::DataArrayWriter<unsigned char> > p3
346 if(!p3->writeIsNoop())
349 GeometryType coerceTo = subsampledGeometryType(it->type());
350 Refinement &refinement =
351 buildRefinement<dim, ctype>(it->type(), coerceTo);
353 for(
int i = 0; i < refinement.nElements(level); ++i)
362 #endif // DUNE_SUBSAMPLINGVTKWRITER_HH