pzgraphmesh.h

Go to the documentation of this file.

#ifndef GRAFGRIDH
#define GRAFGRIDH

#include "pzcompel.h"
#include "pzadmchunk.h"
#include "pzvec.h"
#include "tpzautopointer.h"
#include "TPZMaterial.h"
#include "pzfmatrix.h"
#include "pzgraphnode.h"
#include "TPZDrawStyle.h"
#include "pzgraphel.h"

#include <iostream>
class TPZCompMesh;
template<class TVar>
class TPZBlock;

class TPZGraphMesh : public TPZSavable {

public:
    
    TPZGraphMesh(TPZCompMesh *cm, int dimension, const std::set<int> & matids, const TPZVec<std::string> &scalarnames,const TPZVec<std::string> &vecnames);
    
    TPZGraphMesh(TPZCompMesh *cm, int dimension, const std::set<int> & matids, const TPZVec<std::string> &scalarnames, const TPZVec<std::string> &vecnames, const TPZVec<std::string> &tensornames);
    
                virtual ~TPZGraphMesh(void);
        
    int ClassId() const override;

    void Read(TPZStream &buf, void *context) override;

    void Write(TPZStream &buf, int withclassid) const override;

                TPZGraphNode &FindNode(int64_t side);
                TPZGraphEl *FindElement(int64_t sid);
                TPZAdmChunkVector<TPZGraphEl *> &ElementList();
                TPZAdmChunkVector<TPZGraphNode> &NodeMap();
                int64_t NPoints();
                int64_t NElements(MElementType type);
                int Res() {return fResolution;}
    virtual void SetCompMesh(TPZCompMesh *mesh, const std::set<int> & matids);
                virtual void DrawNodes();
                virtual void DrawMesh(int numcases);
                virtual void DrawConnectivity(MElementType type);
                virtual void DrawSolution(int step, REAL time);
                TPZDrawStyle Style();
                virtual void SetFileName(const std::string &filename);
                
                std::ostream &Out()
                {
                                return fOutFile;
                }
                
                void SetResolution(int res){ fResolution = res; SequenceNodes();}
                
                void Print(std::ostream &out);
    
                void SetNames(const TPZVec<std::string>&scalarnames, const TPZVec<std::string>&vecnames);

    void SetNames(const TPZVec<std::string>&scalarnames, const TPZVec<std::string>&vecnames, const TPZVec<std::string>& tensornames);
                
    void SetMaterialIds(const std::set<int> & matids);
    
    std::set<int> MaterialIds();
    
    bool Material_Is_PostProcessed(int matid);
    
protected:
    
                TPZCompMesh *fCompMesh;
                TPZGeoMesh *fGeoMesh;
    std::set<int> fMaterialIds;
                int fDimension;
                TPZAdmChunkVector<TPZGraphEl *> fElementList;
                TPZAdmChunkVector<TPZGraphNode> fNodeMap;
                int fResolution;
                TPZDrawStyle fStyle;
                std::ofstream fOutFile;
                std::string fFileName;
                TPZVec<std::string> fScalarNames, fVecNames, fTensorNames;
                virtual void SequenceNodes();
                
                TPZCompEl *FindFirstInterpolatedElement(TPZCompMesh *mesh,int dimension);
                
public:
    
//    /** @brief Return of the material for graphical mesh */
//    virtual TPZMaterial * Material();
                virtual TPZCompMesh *Mesh() { return fCompMesh;}
    
    TPZVec<std::string> ScalarNames()
    {
        return fScalarNames;
    }
    
    TPZVec<std::string> VecNames()
    {
        return fVecNames;
    }
    
    TPZVec<std::string> TensorNames()
    {
        return fTensorNames;
    }
};

inline TPZDrawStyle TPZGraphMesh::Style(){
                return fStyle;
}

#endif