Homec4science
Diffusion akantu (features/68-automatic-differentiation) Browse

rAKA/python/pybind11/py_aka_boundary_conditions.cc
d351f96065bdfeatures/68-automatic…

Last Change
Raw File

py_aka_boundary_conditions.cc
View Options

/* -------------------------------------------------------------------------- */
#include "py_aka_boundary_conditions.hh"
#include "py_aka_array.hh"
/* -------------------------------------------------------------------------- */
#include <aka_common.hh>
#include <boundary_condition_functor.hh>
/* -------------------------------------------------------------------------- */
#include <pybind11/operators.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
/* -------------------------------------------------------------------------- */
namespace py = pybind11;
namespace akantu {
/* -------------------------------------------------------------------------- */
template <typename daughter = BC::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 {
PYBIND11_OVERLOAD_NAME(void, daughter, "__call__", operator(), node, flags,
primal, coord);
}
};
/* -------------------------------------------------------------------------- */
template <typename daughter = BC::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 {
PYBIND11_OVERLOAD_PURE_NAME(void, daughter, "__call__", operator(),
quad_point, dual, coord, normals);
}
};
/* -------------------------------------------------------------------------- */
template <typename Functor, typename Constructor>
decltype(auto) declareDirichletFunctor(py::module mod, const char * name,
Constructor && cons) {
py::class_<Functor, PyDirichletFunctor<Functor>, BC::DirichletFunctor>(mod,
name)
.def(cons);
}
template <typename Functor, typename Constructor>
decltype(auto) declareNeumannFunctor(py::module mod, const char * name,
Constructor && cons) {
py::class_<Functor, PyNeumannFunctor<Functor>, BC::NeumannFunctor>(mod, name)
.def(cons);
}
/* -------------------------------------------------------------------------- */
__attribute__((visibility("default"))) void
register_boundary_conditions(py::module & mod) {
py::class_<BC::Functor>(mod, "BCFunctor");
py::class_<BC::DirichletFunctor, PyDirichletFunctor<>, BC::Functor>(
mod, "DirichletFunctor")
.def(py::init())
.def(py::init<SpatialDirection>());
py::class_<BC::NeumannFunctor, PyNeumannFunctor<>, BC::Functor>(
mod, "NeumannFunctor")
.def(py::init());
declareDirichletFunctor<BC::FixedValue>(mod, "FixedValue",
py::init<Real, BC::Axis>());
declareDirichletFunctor<BC::IncrementValue>(mod, "IncrementValue",
py::init<Real, BC::Axis>());
declareDirichletFunctor<BC::Increment>(mod, "Increment",
py::init<Vector<Real> &>());
declareNeumannFunctor<BC::FromHigherDim>(mod, "FromHigherDim",
py::init<Matrix<Real> &>());
declareNeumannFunctor<BC::FromSameDim>(mod, "FromSameDim",
py::init<Vector<Real> &>());
declareNeumannFunctor<BC::FreeBoundary>(mod, "FreeBoundary", py::init());
}
} // namespace akantu
c4science · Help