TraversalVTKPlotter.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 "TraversalObserver.h"
 
#include "tarch/logging/Log.h"
#include "tarch/multicore/BooleanSemaphore.h"
#include "tarch/mpi/BooleanSemaphore.h"
 
#include "tarch/plotter/griddata/unstructured/vtk/VTUTextFileWriter.h"
 
#include "config.h"
 
namespace peano4 {
  namespace grid {
    class TraversalVTKPlotter;
  }
}
 
class peano4::grid::TraversalVTKPlotter: public peano4::grid::TraversalObserver {
  protected:
    static tarch::logging::Log                 _log;
 
    const std::string                                                                _filename;
    const int                                                                        _spacetreeId;
 
    void plotCell(
        const GridTraversalEvent&  event
    );
 
  private:
    tarch::plotter::griddata::unstructured::vtk::VTUTextFileWriter*                  _writer;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::VertexWriter*    _vertexWriter;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::CellWriter*      _cellWriter;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::CellDataWriter*  _spacetreeIdWriter;
    tarch::plotter::griddata::unstructured::UnstructuredGridWriter::CellDataWriter*  _coreWriter;
 
    static tarch::mpi::BooleanSemaphore                                              _sempahore;
 
  public:
    TraversalVTKPlotter( const std::string& filename, int treeId=-1 );
    virtual ~TraversalVTKPlotter();
 
    virtual void beginTraversal(
      const tarch::la::Vector<Dimensions,double>&  x,
      const tarch::la::Vector<Dimensions,double>&  h
    ) override;
 
    virtual void endTraversal(
      const tarch::la::Vector<Dimensions,double>&  x,
      const tarch::la::Vector<Dimensions,double>&  h
    ) override;
 
    virtual void loadCell(
      const GridTraversalEvent&  event
    ) override;
 
    virtual void storeCell(
      const GridTraversalEvent&  event
    ) override;
 
    virtual void enterCell(
      const GridTraversalEvent&  event
    ) override;
 
    virtual void leaveCell(
      const GridTraversalEvent&  event
    ) override;
 
    virtual TraversalObserver* clone(int spacetreeId) override;
 
    virtual std::vector< GridControlEvent > getGridControlEvents() const override;
};