vtkexportmodule.h

Go to the documentation of this file.

/*
 *
 *                 #####    #####   ######  ######  ###   ###
 *               ##   ##  ##   ##  ##      ##      ## ### ##
 *              ##   ##  ##   ##  ####    ####    ##  #  ##
 *             ##   ##  ##   ##  ##      ##      ##     ##
 *            ##   ##  ##   ##  ##      ##      ##     ##
 *            #####    #####   ##      ######  ##     ##
 *
 *
 *             OOFEM : Object Oriented Finite Element Code
 *
 *               Copyright (C) 1993 - 2013   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 vtkexportmodule_h
#define vtkexportmodule_h

#include "exportmodule.h"
#include "intarray.h"
#include "nodalrecoverymodel.h"
#include "internalstatevaluetype.h"
#include "unknowntype.h"
#include "valuemodetype.h"


#define _IFT_VTKExportModule_Name "vtk"
#define _IFT_VTKExportModule_cellvars "cellvars"
#define _IFT_VTKExportModule_vars "vars"
#define _IFT_VTKExportModule_primvars "primvars"
#define _IFT_VTKExportModule_stype "stype"
#define _IFT_VTKExportModule_regionstoskip "regionstoskip"


namespace oofem {
class DofManager;

class OOFEM_EXPORT VTKExportModule : public ExportModule
{
protected:
    IntArray internalVarsToExport;
    IntArray primaryVarsToExport;
    IntArray cellVarsToExport;

    enum modeType {
        wdmode, 
        rbrmode 
    };

    modeType outMode;
    modeType mode;

    NodalRecoveryModel :: NodalRecoveryModelType stype;
    NodalRecoveryModel *smoother;
    IntArray regionsToSkip;

public:
    VTKExportModule(int n, EngngModel * e);
    virtual ~VTKExportModule();

    virtual IRResultType initializeFrom(InputRecord *ir);

    virtual void doOutput(TimeStep *tStep, bool forcedOutput = false);
    virtual void initialize();
    virtual void terminate();
    virtual const char *giveClassName() const { return "VTKExportModule"; }
    virtual const char *giveInputRecordName() const { return _IFT_VTKExportModule_Name; }

protected:
    NodalRecoveryModel *giveSmoother();

    FILE *giveOutputStream(TimeStep *tStep);
    int giveCellType(Element *tStep);
    int giveNumberOfElementCells(Element *);
    int giveNumberOfNodesPerCell(int cellType);
    void giveElementCell(IntArray &answer, Element *elem, int cell);
    void exportIntVars(FILE *stream, TimeStep *tStep);
    void exportPrimaryVars(FILE *stream, TimeStep *tStep);
    void exportIntVarAs(InternalStateType valID, InternalStateValueType type, FILE *stream, TimeStep *tStep);
    void exportPrimVarAs(UnknownType valID, FILE *stream, TimeStep *tStep);
    void exportCellVars(FILE *stream, int elemToProcess, TimeStep *tStep);

    int initRegionNodeNumbering(IntArray &regionNodalNumbers, int &regionDofMans,
                                int offset, Domain *domain, int reg, int mode);
    int giveTotalRBRNumberOfNodes(Domain *d);
    void getDofManPrimaryVariable(FloatArray &answer, DofManager *dman, IntArray &dofIDMask,
                                  ValueModeType mode, TimeStep *tStep, InternalStateType iType);
};
} // end namespace oofem
#endif // vtkexportmodule_h