bodypack.h

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
#include <string>
#include <vector>
 
#include <petscdm.h>
#include <petscsys.h>
#include <yaml-cpp/yaml.h>
 
#include <petibm/singlebody.h>
 
namespace petibm
{
namespace body
{
class BodyPackBase
{
public:
    PetscInt dim;
 
    PetscInt nBodies;
 
    PetscInt nPts;
 
    PetscInt nLclPts;
 
    std::vector<type::SingleBody> bodies;
 
    DM dmPack;
 
    std::string info;
 
    BodyPackBase() = default;
 
    BodyPackBase(const MPI_Comm &comm, const PetscInt &dim,
                 const YAML::Node &node);
 
    virtual ~BodyPackBase();
 
    virtual PetscErrorCode destroy();
 
    PetscErrorCode printInfo() const;
 
    PetscErrorCode findProc(const PetscInt &bIdx, const PetscInt &ptIdx,
                            PetscMPIInt &proc) const;
 
    PetscErrorCode getGlobalIndex(const PetscInt &bIdx, const PetscInt &ptIdx,
                                  const PetscInt &dof, PetscInt &idx) const;
 
    PetscErrorCode getGlobalIndex(const PetscInt &bIdx, const MatStencil &s,
                                  PetscInt &idx) const;
 
    PetscErrorCode getPackedGlobalIndex(const PetscInt &bIdx,
                                        const PetscInt &ptIdx,
                                        const PetscInt &dof,
                                        PetscInt &idx) const;
 
    PetscErrorCode getPackedGlobalIndex(const PetscInt &bIdx,
                                        const MatStencil &s,
                                        PetscInt &idx) const;
 
    PetscErrorCode calculateAvgForces(const Vec &f, type::RealVec2D &fAvg);
 
    PetscErrorCode updateMeshIdx(const type::Mesh &mesh);
 
protected:
    PetscErrorCode init(const MPI_Comm &comm, const PetscInt &dim,
                        const YAML::Node &node);
 
    MPI_Comm comm;
 
    PetscMPIInt mpiSize;
 
    PetscMPIInt mpiRank;
 
    type::IntVec1D nLclAllProcs;
 
    type::IntVec1D offsetsAllProcs;
 
    PetscErrorCode createDmPack();
 
    PetscErrorCode createInfoString();
 
};  // BodyPackBase
 
}  // end of namespace body
 
namespace type
{
typedef std::shared_ptr<body::BodyPackBase> BodyPack;
 
}  // end of namespace type
 
namespace body
{
PetscErrorCode createBodyPack(const MPI_Comm &comm, const PetscInt &dim,
                              const YAML::Node &node, type::BodyPack &bodies);
 
}  // end of namespace body
 
}  // end of namespace petibm