mesh.h

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <vector>
 
#include <petscao.h>
#include <petscdmcomposite.h>
#include <petscdmda.h>
#include <petscsys.h>
#include <yaml-cpp/yaml.h>
 
#include <petibm/type.h>
 
namespace petibm
{
namespace mesh
{
class MeshBase
{
public:
    PetscInt dim = -1;
 
    type::RealVec1D min;
 
    type::RealVec1D max;
 
    type::IntVec2D n;
 
    type::BoolVec2D periodic;
 
    type::GhostedVec3D coord;
 
    type::GhostedVec3D dL;
 
    PetscInt UN;
 
    PetscInt pN;
 
    std::string info;
 
    // PETSc stuffs
    std::vector<DM> da;
 
    type::IntVec1D nProc;
 
    type::IntVec2D bg;
 
    type::IntVec2D ed;
 
    type::IntVec2D m;
 
    PetscInt UNLocal;
 
    PetscInt pNLocal;
 
    DM UPack;
 
    // MPI stuffs
    MPI_Comm comm;
 
    PetscMPIInt mpiSize;
 
    PetscMPIInt mpiRank;
 
    MeshBase() = default;
 
    MeshBase(const MPI_Comm &world, const YAML::Node &node){};
 
    virtual ~MeshBase();
 
    virtual PetscErrorCode destroy();
 
    PetscErrorCode printInfo() const;
 
    virtual PetscErrorCode write(const std::string &filePath) const = 0;
 
    virtual PetscBool isPointOnLocalProc(const type::RealVec1D &point,
                                         const type::Field &field) = 0;
 
    virtual PetscErrorCode getLocalIndex(const PetscInt &f, const MatStencil &s,
                                         PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getLocalIndex(const PetscInt &f, const PetscInt &i,
                                         const PetscInt &j, const PetscInt &k,
                                         PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getNaturalIndex(const PetscInt &f,
                                           const MatStencil &s,
                                           PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getNaturalIndex(const PetscInt &f, const PetscInt &i,
                                           const PetscInt &j, const PetscInt &k,
                                           PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getGlobalIndex(const PetscInt &f,
                                          const MatStencil &s,
                                          PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getGlobalIndex(const PetscInt &f, const PetscInt &i,
                                          const PetscInt &j, const PetscInt &k,
                                          PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getPackedGlobalIndex(const PetscInt &f,
                                                const MatStencil &s,
                                                PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getPackedGlobalIndex(const PetscInt &f,
                                                const PetscInt &i,
                                                const PetscInt &j,
                                                const PetscInt &k,
                                                PetscInt &idx) const = 0;
 
protected:
    virtual PetscErrorCode init(const MPI_Comm &world,
                                const YAML::Node &node) = 0;
 
};  // MeshBase
 
}  // end of namespace mesh
 
namespace type
{
typedef std::shared_ptr<mesh::MeshBase> Mesh;
 
}  // end of namespace type
 
namespace mesh
{
PetscErrorCode createMesh(const MPI_Comm &comm, const YAML::Node &node,
                          type::Mesh &mesh);
 
}  // end of namespace mesh
 
}  // end of namespace petibm