singlebody.h

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
 
#include <petscdm.h>
#include <petscsys.h>
 
#include <petibm/mesh.h>
#include <petibm/type.h>
 
namespace petibm
{
namespace body
{
// TODO: the way we store coords and meshIdx is different from CartesianMesh.
//       Will this cause confusion?
class SingleBodyBase
{
    friend class BodyPackBase;
 
public:
    PetscInt dim;
 
    std::string name;
 
    std::string filePath;
 
    PetscInt nPts;
 
    type::RealVec2D coords;
 
    type::RealVec2D coords0;
 
    PetscInt nLclPts;
 
    type::IntVec2D meshIdx;
 
    PetscInt bgPt;
 
    PetscInt edPt;
 
    DM da;
 
    std::string info;
 
    SingleBodyBase(const MPI_Comm &comm, const PetscInt &dim,
                   const std::string &name, const std::string &filePath);
 
    SingleBodyBase() = default;
 
    virtual ~SingleBodyBase();
 
    virtual PetscErrorCode destroy();
 
    PetscErrorCode printInfo() const;
 
    virtual PetscErrorCode findProc(const PetscInt &i,
                                    PetscMPIInt &p) const = 0;
 
    virtual PetscErrorCode getGlobalIndex(const PetscInt &i,
                                          const PetscInt &dof,
                                          PetscInt &idx) const = 0;
 
    virtual PetscErrorCode getGlobalIndex(const MatStencil &s,
                                          PetscInt &idx) const = 0;
 
    virtual PetscErrorCode calculateAvgForces(const Vec &f,
                                              type::RealVec1D &fAvg) const = 0;
 
    virtual PetscErrorCode updateMeshIdx(const type::Mesh &mesh) = 0;
 
    virtual PetscErrorCode readBody(const std::string &filepath) = 0;
 
    virtual PetscErrorCode writeBody(const std::string &filepath) = 0;
 
 
protected:
    MPI_Comm comm;
 
    PetscMPIInt mpiSize;
 
    PetscMPIInt mpiRank;
 
    type::IntVec1D nLclAllProcs;
 
    type::IntVec1D offsetsAllProcs;
 
};  // SingleBodyBase
 
}  // end of namespace body
 
namespace type
{
typedef std::shared_ptr<body::SingleBodyBase> SingleBody;
}  // end of namespace type
 
namespace body
{
PetscErrorCode createSingleBody(const MPI_Comm &comm, const PetscInt &dim,
                                const std::string &type,
                                const std::string &name,
                                const std::string &filePath,
                                type::SingleBody &body);
 
}  // end of namespace body
 
}  // end of namespace petibm