VTKWriter.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#pragma once
 
 
#include "tarch/la/Vector.h"
#include "tarch/plotter/griddata/unstructured/vtk/VTKTextFileWriter.h"
#include "tarch/mpi/BooleanSemaphore.h"
#include "../PointWriter.h"
 
 
namespace tarch {
  namespace plotter {
    namespace pointdata {
      namespace vtk {
        class VTKWriter;
      }
    }
  }
}
 
 
class tarch::plotter::pointdata::vtk::VTKWriter: public tarch::plotter::pointdata::PointWriter {
  private:
    static tarch::mpi::BooleanSemaphore _sempahore;
 
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter*                 _vtkWriter;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::VertexWriter*   _vertexWriter;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::CellWriter*     _cellWriter;
 
    VTKWriter(const VTKWriter& writer) = delete;
 
    VTKWriter& operator=(const VTKWriter& writer) = delete;
 
  public:
    VTKWriter(bool binaryFile, const std::string&  fileName, const std::string&  indexFileName, tarch::plotter::PVDTimeSeriesWriter::IndexFileMode modeFile, double timeStamp);
 
    virtual ~VTKWriter();
 
    virtual bool writeToFile() override;
 
    virtual bool isOpen() override;
 
    virtual void clear() override;
 
    virtual int plotPoint(const tarch::la::Vector<2,double>& position) override;
    virtual int plotPoint(const tarch::la::Vector<3,double>& position) override;
 
    class PointDataWriter: public PointWriter::PointDataWriter {
      private:
        VTKWriter& _myWriter;
 
        tarch::plotter::griddata::unstructured::UnstructuredGridWriter::VertexDataWriter*  _particleDataWriter;
      public:
        PointDataWriter(VTKWriter& myWriter, const std::string& identifier, int recordsPerPoint);
        virtual ~PointDataWriter();
 
        virtual void plot( int index, double value ) override;
        virtual void plot( int index, const tarch::la::Vector<1,double>& value ) override;
        virtual void plot( int index, const tarch::la::Vector<2,double>& value ) override;
        virtual void plot( int index, const tarch::la::Vector<3,double>& value ) override;
        virtual void plot( int index, double* values, int numberOfValues ) override;
 
        virtual void close() override;
 
        virtual void assignRemainingPointsDefaultValues() override;
    };
 
    virtual PointWriter::PointDataWriter*    createPointDataWriter( const std::string& identifier, int recordsPerPoint ) override;
};