vtkxmlexportmodule.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 vtkxmlexportmodule_h
#define vtkxmlexportmodule_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 __VTK_MODULE
 #include <vtkUnstructuredGrid.h>
 #include <vtkSmartPointer.h>
#endif
 
 
#ifdef _WIN32
 #define NULL_DEVICE "NUL:"
#else
 #define NULL_DEVICE "/dev/null"
#endif
 
 
#include <string>
#include <list>


#define _IFT_VTKXMLExportModule_Name "vtkxml"
#define _IFT_VTKXMLExportModule_cellvars "cellvars"
#define _IFT_VTKXMLExportModule_vars "vars"
#define _IFT_VTKXMLExportModule_primvars "primvars"
#define _IFT_VTKXMLExportModule_externalForces "externalforces"
#define _IFT_VTKXMLExportModule_ipvars "ipvars"
#define _IFT_VTKXMLExportModule_stype "stype"
#define _IFT_VTKXMLExportModule_exportSetMembershipFlag "setmembership"

 
using namespace std;
namespace oofem {
class Node;

class OOFEM_EXPORT VTKXMLExportModule : public VTKBaseExportModule
{
protected:
    IntArray internalVarsToExport;
    IntArray primaryVarsToExport;
    IntArray externalForcesToExport;
    IntArray cellVarsToExport;
    IntArray ipInternalVarsToExport;
    bool exportSetMembershipFlag=0;
 
    

    NodalRecoveryModel::NodalRecoveryModelType stype;
    std::unique_ptr< NodalRecoveryModel >smoother;
    std::unique_ptr< NodalRecoveryModel >primVarSmoother;

    std::list< std::string >pvdBuffer;

    std::list< std::string >gpPvdBuffer;
 
 
public:
    VTKXMLExportModule(int n, EngngModel *e);
    virtual ~VTKXMLExportModule();
 
    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 "VTKXMLExportModule"; }
    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;
#else
    std::ofstream fileStream;
#endif
 
    ExportRegion defaultVTKPiece;
 
    std::vector< ExportRegion >defaultVTKPieces;
    
    ExportRegion& getExportRegions() {return this->defaultVTKPiece;}
protected:

    std::string giveOutputFileName(TimeStep *tStep);

    std::ofstream giveOutputStream(TimeStep *tStep);
 
    void writeIntVars(ExportRegion &vtkPiece);
    void writeXFEMVars(ExportRegion &vtkPiece);
    void writePrimaryVars(ExportRegion &vtkPiece);
    void writeCellVars(ExportRegion &vtkPiece);
    void writeExternalForces(ExportRegion &vtkPiece);
    void writeVertexSetMembership(ExportRegion &vtkPiece);
    void writeCellSetMembership(ExportRegion &vtkPiece);
    bool writeVTKPieceEpilog(ExportRegion &vtkPiece, TimeStep *tStep);
    bool writeVTKPieceVariables(ExportRegion &vtkPiece, TimeStep *tStep);
    bool writeVTKPieceProlog(ExportRegion &vtkPiece, TimeStep *tStep);
    void exportIntVarsInGpAs(IntArray valIDs, TimeStep *tStep);
    void writeVTKCollection();

    void writeGPVTKCollection();
 
#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);
};

class OOFEM_EXPORT VTKXMLExportModuleElementInterface : public Interface
{
 
public:
    VTKXMLExportModuleElementInterface() : Interface() { }
    virtual void giveCompositeExportData(ExportRegion &vtkPiece, IntArray &primaryVarsToExport, IntArray &internalVarsToExport, IntArray cellVarsToExport, TimeStep *tStep) { }
    virtual void giveCompositeExportData(std::vector< ExportRegion > &vtkPieces, IntArray &primaryVarsToExport, IntArray &internalVarsToExport, IntArray cellVarsToExport, TimeStep *tStep) { }
};
} // end namespace oofem
#endif // vtkxmlexportmodule_h