PeanoHDF5PatchFileWriter.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 "PatchWriter.h"
#include "config.h"
 
#include "tarch/logging/Log.h"
 
#include <vector>
 
#ifdef UseHDF5
#include "hdf5.h"
#endif
 
namespace tarch {
  namespace plotter {
    namespace griddata {
      namespace blockstructured {
        class PeanoHDF5PatchFileWriter;
      }
    }
  }
}
 
class tarch::plotter::griddata::blockstructured::PeanoHDF5PatchFileWriter: public tarch::plotter::griddata::blockstructured::PatchWriter {
  protected:
    static tarch::logging::Log _log;
    static const std::string HEADER;
 
    const int  _dimensions;
    const bool _compress;
 
    int _vertexCounter;
    int _cellCounter;
 
    #ifdef UseHDF5
    hid_t       _file;
 
    hid_t  createDataTableProperties(int lineWidth, int rowCount) const;
    #endif
 
    bool        _isOpen;
 
    std::vector<double>  _geometryData;
 
    int         _numberOfActiveDataset;
 
    std::string  getNameOfCurrentDataset() const;
 
  public:
    class CellDataWriter: public tarch::plotter::griddata::blockstructured::PatchWriter::CellDataWriter {
      protected:
        tarch::plotter::griddata::blockstructured::PeanoHDF5PatchFileWriter& _writer;
 
        const std::string    _identifier;
        const int            _numberOfUnknowns;
        std::vector<double>  _data;
 
      public:
        CellDataWriter(
          const std::string& identifier,
          int                unknownsPerAxis,
          int                numberOfUnknowns, const std::string& description,
          const std::string& metaData,
          double*            mapping,
          tarch::plotter::griddata::blockstructured::PeanoHDF5PatchFileWriter& writer
        );
 
        virtual ~CellDataWriter();
 
        virtual void plotCell( int index, double value ) override;
        virtual void plotCell( int index, double* values ) override;
 
        virtual void close() override;
 
        virtual void assignRemainingCellsDefaultValues() override;
 
        virtual int getFirstCellWithinPatch(int index) const override;
     };
 
     class VertexDataWriter: public tarch::plotter::griddata::blockstructured::PatchWriter::VertexDataWriter {
       protected:
         tarch::plotter::griddata::blockstructured::PeanoHDF5PatchFileWriter& _writer;
 
         const std::string    _identifier;
         const int            _numberOfUnknowns;
         std::vector<double>  _data;
 
       public:
         VertexDataWriter(
           const std::string& identifier,
           int                unknownsPerAxis,
           int                numberOfUnknowns, const std::string& description,
           const std::string& metaData,
           double*            mapping,
           tarch::plotter::griddata::blockstructured::PeanoHDF5PatchFileWriter& writer
         );
 
         ~VertexDataWriter();
 
         virtual void plotVertex( int index, double value ) override;
         virtual void plotVertex( int index, double* values ) override;
 
         virtual void close() override;
 
         virtual void assignRemainingVerticesDefaultValues() override;
 
         virtual int getFirstVertexWithinPatch(int index) const override;
     };
 
    PeanoHDF5PatchFileWriter(
      int                  dimension,
      const std::string&   filename,
      bool                 append,
      bool                 compress
    );
 
    virtual ~PeanoHDF5PatchFileWriter();
 
    virtual CellDataWriter*    createCellDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerCell, const std::string& description ) override;
    virtual CellDataWriter*    createCellDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerCell, const std::string& description, const std::string& metaData ) override;
 
    virtual CellDataWriter*    createCellDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerCell, const std::string& description, const std::string& metaData, double* mapping ) override;
 
    virtual VertexDataWriter*  createVertexDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerVertex, const std::string& description ) override;
    virtual VertexDataWriter*  createVertexDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerVertex, const std::string& description, const std::string& metaData  ) override;
    virtual VertexDataWriter*  createVertexDataWriter( const std::string& identifier, int unknownsPerAxis, int recordsPerVertex, const std::string& description, const std::string& metaData, double* mapping ) override;
 
    virtual int plotPatch(
      const tarch::la::Vector<2,double>& offset,
      const tarch::la::Vector<2,double>& size
    ) override;
 
    virtual int plotPatch(
      const tarch::la::Vector<3,double>& offset,
      const tarch::la::Vector<3,double>& size
    ) override;
 
    virtual bool writeToFile() override;
 
    virtual bool isOpen() override;
 
    virtual void clear() override;
 
    void addMetaData(const std::string& metaData);
};