TPZAcademicGeoMesh.h

Go to the documentation of this file.

//
//  TPZAcademicGeoMesh.hpp
//  PZ
//
//  Created by Philippe Devloo on 5/29/16.
//
//

#ifndef TPZAcademicGeoMesh_hpp
#define TPZAcademicGeoMesh_hpp

#include <stdio.h>
#include "pzmanvector.h"
#include "pzreal.h"
#include "pzgmesh.h"
#include <iostream>


class TPZAcademicGeoMesh
{
public:
    static TPZManVector<REAL,3> fX0, fEps;
    
    TPZGeoMesh fDeformed;
    
    enum MMeshType {ENone, EHexa, EPrism, ETetrahedra, EPyramid };

protected:
    
    MMeshType fMeshType;
    
    TPZManVector<int,6> fBCNumbers;
    
    bool fShouldDeform;
    
    int64_t fNumberElements;
    
    int fMaterialId;
    
    void GenerateNodes(TPZGeoMesh *gmesh);
    
    void DeformGMesh(TPZGeoMesh &gmesh);
    
    TPZGeoMesh *HexahedralMesh();
    TPZGeoMesh *TetrahedralMeshUsingRefinement();
    
    int AddBoundaryElements(TPZGeoMesh *gmesh);
  
    int AddBoundaryElementsByCoord(TPZGeoMesh *gmesh);
      
    void CheckConsistency(TPZGeoMesh *gmesh);

    
public:
    
    TPZAcademicGeoMesh();
    
    TPZAcademicGeoMesh(int numelements, MMeshType meshtype, bool deform = false) : fMeshType(meshtype), fBCNumbers(6,-1), fShouldDeform(deform),
    fNumberElements(numelements), fMaterialId(1)
    {
        
    }
    
    void SetMeshType(MMeshType meshtype)
    {
        if (meshtype == ENone) {
            DebugStop();
        }
        fMeshType = meshtype;
    }
    
    void SetBoundaryIndices(TPZVec<int> &BCNumbers);
    
    void ShouldDeform(bool deform);
    
    void SetNumberElements(int64_t numelements)
    {
#ifdef PZDEBUG
        if(numelements < 0)
        {
            DebugStop();
        }
#endif
        fNumberElements = numelements;
    }
    void SetMaterialId(int id)
    {
#ifdef PZDEBUG
        if(id == 0)
        {
            DebugStop();
        }
#endif
        fMaterialId = id;
    }
    
    TPZGeoMesh *CreateGeoMesh()
    {
        switch (fMeshType) {
            case EHexa:
                return HexahedralMesh();
                break;
            case ETetrahedra:
                return TetrahedralMesh();
                break;
            case EPyramid:
                return PyramidalAndTetrahedralMesh();
                break;
            case EPrism:
                std::cout << "Not implemented\n";

            default:
                DebugStop();
                break;
        }
        return 0;
    }
  
    TPZGeoMesh *PyramidalAndTetrahedralMesh();
    
    TPZGeoMesh *RedBlackPyramidalAndHexagonalMesh();

    TPZGeoMesh *TetrahedralMesh();
  
  void SetBCIDVector(TPZVec<int> &BCNumbers){
    if(fBCNumbers.size() != BCNumbers.size()){
      DebugStop();
    }
    for (int i = 0; i < BCNumbers.size(); i++) {
      fBCNumbers[i] = BCNumbers[i];
    }
  }
  
    
};

#endif /* TPZAcademicGeoMesh_hpp */