pzgengrid.h

Go to the documentation of this file.

#ifndef _TPZGENGRIDHH_
#define _TPZGENGRIDHH_

class TPZCompMesh;
class TPZGeoMesh;

#include "pzvec.h"
#include "pzmanvector.h"

#include <stdio.h>
#include <iostream>
#include "pzreal.h"
#include "tpzautopointer.h"

#include "pzeltype.h"

#include <fstream>

class TPZGenGrid {
                
public:
                
    TPZGenGrid(TPZVec<int> &nx, TPZVec<REAL> &x0, TPZVec<REAL> &x1, int numl = 1, REAL rot = 0.5);
                
    virtual ~TPZGenGrid();

    void SetData(TPZVec<int> &nx, TPZVec<REAL> &x0, TPZVec<REAL> &x1, MElementType eltype = EQuadrilateral, int numl = 1, REAL rot = 0.5);
    
    void SetZigZagPattern()
    {
        fZigZag = true;
    }
    
    void SetDistortion(REAL distortion)
    {
        if (distortion) {
            fTrapeze = true;
            fDistortion = distortion;
        }
        else
        {
            fTrapeze = false;
            fDistortion = 0.;
        }
    }
    virtual short Read(TPZGeoMesh *mesh,int matid = 1);
    short Read(TPZAutoPointer<TPZGeoMesh> &mesh, int matid = 1);
    
    static REAL GeometricProgression(REAL minsize, REAL size, int numdiv);
                
    void ComputeGeometricProgression(TPZVec<REAL> &minsizes, TPZVec<REAL> &progression);
    
    void SetGeometricProgression(TPZVec<REAL> &progression);
    
    virtual void SetBC(TPZGeoMesh *gr, int side, int bc);
    
    virtual void SetBC(TPZGeoMesh *gr, TPZVec<REAL> &start, TPZVec<REAL> &end, int bc);
    
    void SetBC(TPZAutoPointer<TPZGeoMesh> gr, int side, int bc)
    {
        SetBC(gr.operator->(), side, bc);
    }
    
    void SetBC(TPZAutoPointer<TPZGeoMesh> gr, TPZVec<REAL> &start, TPZVec<REAL> &end, int bc)
    {
        SetBC(gr.operator->(), start, end, bc);
    }
    
    void SetPointBC(TPZGeoMesh *gr, TPZVec<REAL> &x, int bc);
                
    void SetPointBC(TPZAutoPointer<TPZGeoMesh> gr, TPZVec<REAL> &x, int bc)
    {
        SetPointBC(gr.operator->(), x, bc);
    }
                
    virtual void Print( char *name = NULL, std::ostream &out = std::cout );
    
    virtual void SetElementType(MElementType type);
    
    int64_t ElemId(int64_t iel,int64_t jel, int layer);
    
                static REAL Distance(TPZVec<REAL> &x1,TPZVec<REAL> &x2);
                
                bool ReadAndMergeGeoMesh(TPZGeoMesh * grid,TPZGeoMesh * grid2);
                bool ReadAndMergeGeoMesh(TPZGeoMesh* grid,TPZGeoMesh* grid2,int matid);
                bool MergeGeoMesh(TPZGeoMesh* grid,TPZGeoMesh* grid2,int matid = 1);
    
    void SetRefpatternElements(bool refpat)
    {
        fRefPattern = refpat;
    }
                
protected:
    virtual void Coord(int i, TPZVec<REAL> &coord);
    
    virtual int GlobalI(int ix, int iy, int layer);
    
                void ElementConnectivity(int64_t iel, TPZVec<int64_t> &nodes);
    
    void ElementConnectivityZigZag(int64_t iel, TPZVec<int64_t> &nodes);
    
    virtual bool GenerateNodes(TPZGeoMesh *grid);
    
    virtual bool GenerateElements(TPZGeoMesh *grid,int matid);

    virtual bool GenerateElementsZigZag(TPZGeoMesh *grid,int matid);

                TPZManVector<int> fNx;
                TPZManVector<REAL> fX0;
                TPZManVector<REAL> fX1;
                TPZManVector<REAL> fDelx;
                TPZManVector<REAL> fGeometricProgression;
                int64_t fNumNodes;
                MElementType fElementType;
    
    int fNumLayers;
    REAL fRotAngle;
    
    bool fZigZag;
    
    bool fTrapeze;
    
    REAL fDistortion;
    
    bool fRefPattern;
                
};

#endif // _TGENGRIDHH_