cartesianmesh.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/mesh.h>
#include <petibm/type.h>
 
namespace petibm
{
namespace mesh
{
class CartesianMesh : public MeshBase
{
public:
    CartesianMesh(const MPI_Comm &world, const YAML::Node &node);
 
    virtual ~CartesianMesh();
 
    // doc is the same as MeshBase::destroy
    virtual PetscErrorCode destroy();
 
    virtual PetscBool isPointOnLocalProc(const type::RealVec1D &point,
                                         const type::Field &field);
 
    // doc is the same as MeshBase::write
    virtual PetscErrorCode write(const std::string &filePath) const;
 
    // doc is the same as MeshBase::getLocalIndex
    virtual PetscErrorCode getLocalIndex(const PetscInt &f, const MatStencil &s,
                                         PetscInt &idx) const;
 
    // doc is the same as MeshBase::getLocalIndex
    virtual PetscErrorCode getLocalIndex(const PetscInt &f, const PetscInt &i,
                                         const PetscInt &j, const PetscInt &k,
                                         PetscInt &idx) const;
 
    // doc is the same as MeshBase::getNaturalIndex
    virtual PetscErrorCode getNaturalIndex(const PetscInt &f,
                                           const MatStencil &s,
                                           PetscInt &idx) const;
 
    // doc is the same as MeshBase::getNaturalIndex
    virtual PetscErrorCode getNaturalIndex(const PetscInt &f, const PetscInt &i,
                                           const PetscInt &j, const PetscInt &k,
                                           PetscInt &idx) const;
 
    // doc is the same as MeshBase::getGlobalIndex
    virtual PetscErrorCode getGlobalIndex(const PetscInt &f,
                                          const MatStencil &s,
                                          PetscInt &idx) const;
 
    // doc is the same as MeshBase::getGlobalIndex
    virtual PetscErrorCode getGlobalIndex(const PetscInt &f, const PetscInt &i,
                                          const PetscInt &j, const PetscInt &k,
                                          PetscInt &idx) const;
 
    // doc is the same as MeshBase::getPackedGlobalIndex
    virtual PetscErrorCode getPackedGlobalIndex(const PetscInt &f,
                                                const MatStencil &s,
                                                PetscInt &idx) const;
 
    // doc is the same as MeshBase::getPackedGlobalIndex
    virtual PetscErrorCode getPackedGlobalIndex(const PetscInt &f,
                                                const PetscInt &i,
                                                const PetscInt &j,
                                                const PetscInt &k,
                                                PetscInt &idx) const;
 
protected:
    // doc is the same as MeshBase::init
    PetscErrorCode init(const MPI_Comm &world, const YAML::Node &node);
 
    type::RealVec3D dLTrue;
 
    type::RealVec3D coordTrue;
 
    std::vector<AO> ao;
 
    type::IntVec2D UNLocalAllProcs;
 
    type::IntVec1D UPackNLocalAllProcs;
 
    type::IntVec2D offsetsAllProcs;
 
    type::IntVec1D offsetsPackAllProcs;
 
    PetscErrorCode createVertexMesh();
 
    PetscErrorCode createPressureMesh();
 
    PetscErrorCode createVelocityMesh();
 
    PetscErrorCode createInfoString();
 
    PetscErrorCode addLocalInfoString(std::stringstream &ss);
 
    PetscErrorCode initDMDA();
 
    PetscErrorCode createSingleDMDA(const PetscInt &i);
 
    PetscErrorCode createPressureDMDA();
 
    PetscErrorCode createVelocityPack();
 
};  // CartesianMesh
 
}  // end of namespace mesh
 
}  // end of namespace petibm