misc.h

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
#include <cmath>
#include <functional>
#include <vector>
 
#include <petscdmda.h>
#include <petscsys.h>
 
#include <petibm/mesh.h>
#include <petibm/type.h>
 
namespace petibm
{
namespace misc
{
PetscErrorCode checkPeriodicBC(const type::IntVec2D &bcTypes,
                               type::BoolVec2D &periodic);
 
PetscErrorCode checkBoundaryProc(const DM &da, const type::IntVec1D &n,
                                 const type::BCLoc &loc, PetscBool &onThisProc);
 
PetscErrorCode getGhostPointList(const type::Mesh &mesh,
                                 const type::Field &field,
                                 const type::BCLoc &loc,
                                 type::GhostPointsList &points);
 
PetscErrorCode getPerpendAxes(const PetscInt &self, type::IntVec1D &pAxes);
 
PetscErrorCode getGhostTargetStencil(const type::IntVec1D &n,
                                     const type::BCLoc &loc,
                                     const type::IntVec1D &pIdx,
                                     MatStencil &ghost, MatStencil &target);
 
inline PetscErrorCode stretchGrid(const PetscReal &bg, const PetscReal &ed,
                                  const PetscInt &n, const PetscReal &r,
                                  type::RealVec1D &dL)
{
    PetscFunctionBeginUser;
 
    dL.resize(n);
 
    // calculate the size of the first cell
    dL[0] = (ed - bg) * (r - 1.0) / (std::pow(r, n) - 1.0);
 
    // dL[i] = dL[i-1] * r
    for (auto it = dL.begin() + 1; it < dL.end(); ++it) *it = *(it - 1) * r;
 
    PetscFunctionReturn(0);
}  // stretchGrid
 
struct LoopBound
{
    const PetscInt &bg, &ed;
};
 
inline PetscErrorCode doubleLoops(
    const LoopBound &bound1, const LoopBound &bound2,
    const std::function<PetscErrorCode(const PetscInt &, const PetscInt &)> &f)
{
    PetscErrorCode ierr;
 
    PetscFunctionBeginUser;
 
    for (PetscInt i = bound1.bg; i < bound1.ed; i++)
        for (PetscInt j = bound2.bg; j < bound2.ed; j++)
        {
            ierr = f(i, j); CHKERRQ(ierr);
        }
 
    PetscFunctionReturn(0);
}  // doubleLoops
 
inline PetscErrorCode tripleLoops(
    const LoopBound &bound1, const LoopBound &bound2, const LoopBound &bound3,
    const std::function<PetscErrorCode(const PetscInt &, const PetscInt &,
                                       const PetscInt &)> &f)
{
    using namespace std::placeholders;
 
    PetscErrorCode ierr;
 
    PetscFunctionBeginUser;
 
    for (PetscInt i = bound1.bg; i < bound1.ed; ++i)
    {
        ierr = doubleLoops(bound2, bound3, std::bind(f, i, _1, _2));
        CHKERRQ(ierr);
    }
 
    PetscFunctionReturn(0);
}  // tripleLoops
 
}  // end of namespace misc
 
}  // end of namespace petibm