Homec4science
Diffusion akantu (features/joss-2023) Browse

rAKA/python/py_boundary_conditions.cc
bbecb2ab64c4features/joss-2023

Last Change
Raw File

py_boundary_conditions.cc
View Options

/* -------------------------------------------------------------------------- */
#include "py_boundary_conditions.hh"
#include "py_aka_array.hh"
#include "py_akantu_pybind11_compatibility.hh"
/* -------------------------------------------------------------------------- */
#include <boundary_condition_functor.hh>
/* -------------------------------------------------------------------------- */
#include <pybind11/pybind11.h>
/* -------------------------------------------------------------------------- */
namespace py = pybind11;
namespace akantu {
namespace {
/* ------------------------------------------------------------------------ */
template <typename daughter = BC::Dirichlet::DirichletFunctor>
class PyDirichletFunctor : public daughter {
public:
/* Inherit the constructors */
using daughter::daughter;
/* Trampoline (need one for each virtual function) */
void operator()(UInt node, Vector<bool> & flags, Vector<Real> & primal,
const Vector<Real> & coord) const override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_NAME(void, daughter, "__call__", operator(), node,
flags, primal, coord);
}
};
/* ------------------------------------------------------------------------ */
template <typename daughter = BC::Neumann::NeumannFunctor>
class PyNeumannFunctor : public daughter {
public:
/* Inherit the constructors */
using daughter::daughter;
/* Trampoline (need one for each virtual function) */
void operator()(const IntegrationPoint & quad_point, Vector<Real> & dual,
const Vector<Real> & coord,
const Vector<Real> & normals) const override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_PURE_NAME(void, daughter, "__call__", operator(),
quad_point, dual, coord, normals);
}
};
/* ------------------------------------------------------------------------ */
template <typename Functor, typename Constructor>
decltype(auto) register_dirichlet_functor(py::module mod, const char * name,
Constructor && cons) {
py::class_<Functor, PyDirichletFunctor<Functor>,
BC::Dirichlet::DirichletFunctor>(mod, name)
.def(cons);
}
/* ------------------------------------------------------------------------ */
template <typename Functor, typename Constructor>
decltype(auto) register_neumann_functor(py::module mod, const char * name,
Constructor && cons) {
py::class_<Functor, PyNeumannFunctor<Functor>, BC::Neumann::NeumannFunctor>(
mod, name)
.def(cons);
}
} // namespace
/* -------------------------------------------------------------------------- */
void register_boundary_conditions(py::module & mod) {
py::class_<BC::Functor>(mod, "BCFunctor");
py::class_<BC::Dirichlet::DirichletFunctor, PyDirichletFunctor<>,
BC::Functor>(mod, "DirichletFunctor")
.def(py::init())
.def(py::init<SpatialDirection>());
py::class_<BC::Neumann::NeumannFunctor, PyNeumannFunctor<>, BC::Functor>(
mod, "NeumannFunctor")
.def(py::init());
register_dirichlet_functor<BC::Dirichlet::FixedValue>(
mod, "FixedValue", py::init<Real, BC::Axis>());
register_dirichlet_functor<BC::Dirichlet::IncrementValue>(
mod, "IncrementValue", py::init<Real, BC::Axis>());
register_dirichlet_functor<BC::Dirichlet::Increment>(
mod, "Increment", py::init<Vector<Real> &>());
register_neumann_functor<BC::Neumann::FromHigherDim>(
mod, "FromHigherDim", py::init<Matrix<Real> &>());
register_neumann_functor<BC::Neumann::FromSameDim>(
mod, "FromSameDim", py::init<Vector<Real> &>());
register_neumann_functor<BC::Neumann::FreeBoundary>(mod, "FreeBoundary",
py::init());
}
} // namespace akantu
c4science · Help