vtkhdf5exportmodule.h

Go to the documentation of this file.

/*
 *
 *                 #####    #####   ######  ######  ###   ###
 *               ##   ##  ##   ##  ##      ##      ## ### ##
 *              ##   ##  ##   ##  ####    ####    ##  #  ##
 *             ##   ##  ##   ##  ##      ##      ##     ##
 *            ##   ##  ##   ##  ##      ##      ##     ##
 *            #####    #####   ##      ######  ##     ##
 *
 *
 *             OOFEM : Object Oriented Finite Element Code
 *
 *               Copyright (C) 1993 - 2025   Borek Patzak
 *
 *
 *
 *       Czech Technical University, Faculty of Civil Engineering,
 *   Department of Structural Mechanics, 166 29 Prague, Czech Republic
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
 
#ifndef vtkhdf5exportmodule_h
#define vtkhdf5exportmodule_h
 
#include "vtkbaseexportmodule.h"
#include "intarray.h"
#include "nodalrecoverymodel.h"
#include "interface.h"
#include "internalstatevaluetype.h"
#include "integrationrule.h"
#include "xfem/xfemmanager.h"
#include <iostream>
#include <fstream>
#include <iomanip>
 
 
#ifdef _WIN32
 #define NULL_DEVICE "NUL:"
#else
 #define NULL_DEVICE "/dev/null"
#endif
 
 
#include <string>
#include <list>
 
#ifdef __HDF_MODULE
#include "H5Cpp.h"
#endif
 
#ifndef H5_NO_NAMESPACE
//    using namespace H5;
#endif


#define _IFT_VTKHDF5ExportModule_Name "vtkhdf5"
#define _IFT_VTKHDF5ExportModule_cellvars "cellvars"
#define _IFT_VTKHDF5ExportModule_vars "vars"
#define _IFT_VTKHDF5ExportModule_primvars "primvars"
#define _IFT_VTKHDF5ExportModule_externalForces "externalforces"
#define _IFT_VTKHDF5ExportModule_ipvars "ipvars"
#define _IFT_VTKHDF5ExportModule_stype "stype"
 
using namespace std;
namespace oofem {
class Node;

class OOFEM_EXPORT VTKHDF5ExportModule : public VTKBaseExportModule
{
protected:
    IntArray internalVarsToExport;
    IntArray primaryVarsToExport;
    IntArray externalForcesToExport;
    IntArray cellVarsToExport;
    IntArray ipInternalVarsToExport;
 
    

    NodalRecoveryModel::NodalRecoveryModelType stype;
    std::unique_ptr< NodalRecoveryModel >smoother;
    std::unique_ptr< NodalRecoveryModel >primVarSmoother;
 
   
public:
    VTKHDF5ExportModule(int n, EngngModel *e);
    virtual ~VTKHDF5ExportModule();
 
    void initializeFrom(InputRecord &ir) override;
    void doOutput(TimeStep *tStep, bool forcedOutput = false) override;
    void initialize() override;
    void terminate() override;
    const char *giveClassName() const override { return "VTKHDF5ExportModule"; }
    //void exportPointDataHeader(std::ofstream fileStream, TimeStep *tStep);
    //virtual void giveDataHeaders(std::string &pointHeader, std::string &cellHeader);     // returns the headers
    NodalRecoveryModel *giveSmoother();
    NodalRecoveryModel *givePrimVarSmoother();
    
 
#ifdef __VTK_MODULE
    vtkSmartPointer< vtkUnstructuredGrid >fileStream;
    vtkSmartPointer< vtkPoints >nodes;
    vtkSmartPointer< vtkIdList >elemNodeArray;
 
    vtkSmartPointer< vtkDoubleArray >intVarArray;
    vtkSmartPointer< vtkDoubleArray >primVarArray;
#endif
#ifdef __HDF_MODULE
    std::string fileName;
    H5::H5File *file;
    H5::Group *topGroup, *pointDataGroup, *cellDataGroup, *stepsGroup;
    unsigned int pointCounter, cellCounter, connCounter, offsetCounter;
#endif
 
    ExportRegion defaultVTKPiece;
 
    std::vector< ExportRegion >defaultVTKPieces;
    
    ExportRegion& getExportRegions() {return this->defaultVTKPiece;}
protected:
 
#ifdef __HDF_MODULE
    void writeIntVars(ExportRegion &vtkPiece);
    void writeXFEMVars(ExportRegion &vtkPiece);
    void writePrimaryVars(ExportRegion &vtkPiece);
    void writeCellVars(ExportRegion &vtkPiece);
    void writeExternalForces(ExportRegion &vtkPiece);
    bool writeVTKPieceEpilog(ExportRegion &vtkPiece, TimeStep *tStep);
    bool writeVTKPieceVariables(ExportRegion &vtkPiece, TimeStep *tStep);
    bool writeVTKPieceProlog(ExportRegion &vtkPiece, TimeStep *tStep, unsigned int& pointCount, unsigned int &cellCounter, unsigned int &connCounter, unsigned int& offsetCounter,
                            unsigned int& stepPointCounter, unsigned int& stepCellCounter, unsigned int& stepConnCounter,
                            H5::DataSet& points, H5::DataSet& conn, H5::DataSet &typeDSet, H5::DataSet& types);
    void exportIntVarsInGpAs(IntArray valIDs, TimeStep *tStep);
 
 
    void updateDataSet (H5::DataSet&, int rank, hsize_t* dim, hsize_t* offset, H5::DataType, const void* data);
#endif
 
#ifdef __VTK_MODULE
    void writeVTKPointData(const char *name, vtkSmartPointer< vtkDoubleArray >varArray);
#else
    void writeVTKPointData(FloatArray &valueArray);
#endif
 
#ifdef __VTK_MODULE
    void writeVTKCellData(const char *name, vtkSmartPointer< vtkDoubleArray >varArray);
#else
    void writeVTKCellData(FloatArray &valueArray);
#endif
 
    // Export of composite elements (built up from several subcells)
    void exportCompositeElement(ExportRegion &vtkPiece, Element *el, TimeStep *tStep);
    void exportCompositeElement(std::vector< ExportRegion > &vtkPieces, Element *el, TimeStep *tStep);
};
 
} // end namespace oofem
#endif // vtkhdf5exportmodule_h