6#ifndef DUNE_GRID_IO_FILE_VTK_BASICWRITER_HH
7#define DUNE_GRID_IO_FILE_VTK_BASICWRITER_HH
20#include <dune/geometry/referenceelements.hh>
23#include <dune/grid/io/file/vtk/functionwriter.hh>
24#include <dune/grid/io/file/vtk/pvtuwriter.hh>
25#include <dune/grid/io/file/vtk/vtuwriter.hh>
34 template<
typename IteratorFactory>
36 typedef typename IteratorFactory::CellIterator CellIterator;
37 typedef typename IteratorFactory::CornerIterator CornerIterator;
38 typedef typename IteratorFactory::PointIterator PointIterator;
40 typedef typename IteratorFactory::Cell Cell;
43 typedef FunctionWriterBase<Cell> FunctionWriter;
46 typedef std::list<std::shared_ptr<FunctionWriter> > WriterList;
47 typedef typename WriterList::const_iterator WIterator;
49 typedef typename Cell::Geometry::ctype ctype;
50 static const unsigned celldim = Cell::mydimension;
51 typedef ReferenceElements<ctype, celldim> Refelems;
53 static const FileType fileType = celldim == 1
56 const IteratorFactory& factory;
61 CoordinatesWriter<Cell> coords;
62 typename IteratorFactory::ConnectivityWriter connectivity;
63 OffsetsWriter<Cell> offsets;
64 TypesWriter<Cell> types;
67 BasicWriter(
const IteratorFactory& factory_)
68 : factory(factory_), connectivity(factory.makeConnectivity())
76 void addCellData(
const std::shared_ptr<FunctionWriter>& writer) {
77 cellData.push_back(writer);
80 void addPointData(
const std::shared_ptr<FunctionWriter>& writer) {
81 pointData.push_back(writer);
95 void writeCellFunction(VTUWriter& vtuWriter,
96 FunctionWriter& functionWriter,
97 unsigned ncells)
const
99 if(functionWriter.beginWrite(vtuWriter, ncells)) {
100 const CellIterator& cellend = factory.endCells();
101 for(CellIterator cellit = factory.beginCells(); cellit != cellend;
106 functionWriter.endWrite();
109 void writePointFunction(VTUWriter& vtuWriter,
110 FunctionWriter& functionWriter,
111 unsigned npoints)
const
113 if(functionWriter.beginWrite(vtuWriter, npoints)) {
114 const PointIterator& pend = factory.endPoints();
115 for(PointIterator pit = factory.beginPoints(); pit != pend; ++pit)
116 functionWriter.write(pit->cell(), pit->duneIndex());
118 functionWriter.endWrite();
121 void writeCornerFunction(VTUWriter& vtuWriter,
122 FunctionWriter& functionWriter,
123 unsigned ncorners)
const
125 if(functionWriter.beginWrite(vtuWriter, ncorners)) {
126 const CornerIterator& cend = factory.endCorners();
127 for(CornerIterator cit = factory.beginCorners(); cit != cend; ++cit)
128 functionWriter.write(cit->cell(), cit->duneIndex());
130 functionWriter.endWrite();
138 static std::string getFirstScalar(
const WriterList& data) {
140 if (d->ncomps() == 1)
145 static std::string getFirstVector(
const WriterList& data) {
147 if (d->ncomps() == 3)
152 void writeCellData(VTUWriter& vtuWriter,
unsigned ncells)
const {
153 if(cellData.empty())
return;
155 vtuWriter.beginCellData(getFirstScalar(cellData),
156 getFirstVector(cellData));
158 for (
auto&& cD : cellData)
159 writeCellFunction(vtuWriter, *cD, ncells);
160 vtuWriter.endCellData();
163 void writePointData(VTUWriter& vtuWriter,
unsigned npoints)
const {
164 if(pointData.empty())
return;
166 vtuWriter.beginPointData(getFirstScalar(pointData),
167 getFirstVector(pointData));
169 for (
auto&& pD : pointData)
170 writePointFunction(vtuWriter, *pD, npoints);
171 vtuWriter.endPointData();
174 void writeGrid(VTUWriter& vtuWriter,
unsigned ncells,
unsigned npoints,
176 vtuWriter.beginPoints();
177 writePointFunction(vtuWriter, coords, npoints);
178 vtuWriter.endPoints();
180 vtuWriter.beginCells();
181 writeCornerFunction(vtuWriter, connectivity, ncorners);
182 writeCellFunction(vtuWriter, offsets, ncells);
183 if(fileType != polyData)
184 writeCellFunction(vtuWriter, types, ncells);
185 vtuWriter.endCells();
188 void writeAll(VTUWriter& vtuWriter,
unsigned ncells,
unsigned npoints,
190 writeCellData(vtuWriter, ncells);
191 writePointData(vtuWriter, npoints);
192 writeGrid(vtuWriter, ncells, npoints, ncorners);
196 void writePiece(
const std::string& filename, OutputType outputType) {
197 std::ofstream stream;
198 stream.exceptions(std::ios_base::badbit | std::ios_base::failbit |
199 std::ios_base::eofbit);
200 stream.open(filename.c_str(), std::ios::binary);
202 VTUWriter vtuWriter(stream, outputType, fileType);
204 unsigned ncells = std::distance(factory.beginCells(),
206 unsigned npoints = std::distance(factory.beginPoints(),
207 factory.endPoints());
208 unsigned ncorners = std::distance(factory.beginCorners(),
209 factory.endCorners());
211 vtuWriter.beginMain(ncells, npoints);
212 writeAll(vtuWriter, ncells, npoints, ncorners);
215 if(vtuWriter.beginAppended())
216 writeAll(vtuWriter, ncells, npoints, ncorners);
217 vtuWriter.endAppended();
238 void writeCollection(
const std::string name,
239 const std::string& piecename,
240 const std::string& piecepath)
242 std::ofstream stream;
243 stream.exceptions(std::ios_base::badbit | std::ios_base::failbit |
244 std::ios_base::eofbit);
245 stream.open(name.c_str(), std::ios::binary);
247 PVTUWriter writer(stream, fileType);
252 writer.beginPointData(getFirstScalar(pointData),
253 getFirstVector(pointData));
254 for(WIterator it=pointData.begin(); it!=pointData.end(); ++it)
255 (*it)->addArray(writer);
256 writer.endPointData();
259 writer.beginCellData(getFirstScalar(cellData),
260 getFirstVector(cellData));
261 for(WIterator it=cellData.begin(); it!=cellData.end(); ++it)
262 (*it)->addArray(writer);
263 writer.endCellData();
266 writer.beginPoints();
267 coords.addArray(writer);
271 for(
int i = 0; i < factory.comm().
size(); ++i )
272 writer.addPiece(getParallelPieceName(piecename, piecepath, i));
293 std::string getParallelPieceName(
const std::string& name,
294 const std::string& path,
int rank)
const
296 std::ostringstream s;
297 if(path.size() > 0) {
299 if(path[path.size()-1] !=
'/')
302 s <<
's' << std::setw(4) << std::setfill(
'0') << factory.comm().size()
304 s <<
'p' << std::setw(4) << std::setfill(
'0') << rank <<
':';
323 std::string getParallelHeaderName(
const std::string& name,
324 const std::string& path)
const
326 std::ostringstream s;
327 if(path.size() > 0) {
329 if(path[path.size()-1] !=
'/')
332 s <<
's' << std::setw(4) << std::setfill(
'0') << factory.comm().size()
336 case polyData : s <<
".pvtp";
break;
355 std::string getSerialPieceName(
const std::string& name,
356 const std::string& path)
const
393 std::string pwrite(
const std::string& name,
const std::string& path,
394 const std::string& extendpath, OutputType outputType)
396 MPIGuard guard(factory.comm());
401 file.exceptions(std::ios_base::badbit | std::ios_base::failbit |
402 std::ios_base::eofbit);
403 std::string piecepath =
concatPaths(path, extendpath);
404 std::string relpiecepath =
relativePath(path, piecepath);
407 std::string fullname = getParallelPieceName(name, piecepath,
408 factory.comm().rank());
409 writePiece(fullname, outputType);
412 fullname = getParallelHeaderName(name, path);
413 if(factory.comm().rank() == 0)
414 writeCollection(fullname, name, relpiecepath);
434 std::string write(
const std::string &name, OutputType outputType)
438 if(factory.comm().size() > 1)
439 return pwrite(name,
"",
"", outputType);
442 std::string pieceName = getSerialPieceName(name,
"");
444 writePiece(pieceName, outputType);
Common stuff for the VTKWriter.
FileType
which type of VTK file to write
Definition: common.hh:252
@ polyData
for .vtp files (PolyData)
Definition: common.hh:254
@ unstructuredGrid
for .vtu files (UnstructuredGrid)
Definition: common.hh:256
std::string relativePath(const std::string &newbase, const std::string &p)
compute a relative path between two paths
Definition: path.cc:149
std::string concatPaths(const std::string &base, const std::string &p)
concatenate two paths
Definition: path.cc:28
Implements a MPIGuard which detects an error on a remote process.
Dune namespace.
Definition: alignedallocator.hh:13
constexpr std::integral_constant< std::size_t, sizeof...(II)> size(std::integer_sequence< T, II... >)
Return the size of the sequence.
Definition: integersequence.hh:75
Utilities for handling filesystem paths.
static const ReferenceElement & general(const GeometryType &type)
get general reference elements
Definition: referenceelements.hh:156