Homec4science
Diffusion akantu (bugfixes/python_interface_energy_computations) Browse

rAKA/python/py_contact_mechanics_model.cc
bacf19476747bugfixes/python_interfa…

Last Change
Raw File

py_contact_mechanics_model.cc
View Options

/* -------------------------------------------------------------------------- */
#include "py_aka_array.hh"
/* -------------------------------------------------------------------------- */
#include <contact_mechanics_model.hh>
#include <contact_detector.hh>
#include <surface_selector.hh>
#include <geometry_utils.hh>
#include <mesh_events.hh>
#include <parsable.hh>
/* -------------------------------------------------------------------------- */
#include <pybind11/pybind11.h>
/* -------------------------------------------------------------------------- */
namespace py = pybind11;
/* -------------------------------------------------------------------------- */
namespace akantu {
/* -------------------------------------------------------------------------- */
#define def_deprecated(func_name, mesg) \
def(func_name, [](py::args, py::kwargs) { AKANTU_ERROR(mesg); })
#define def_function_nocopy(func_name) \
def( \
#func_name, \
[](ContactMechanicsModel & self) -> decltype(auto) { \
return self.func_name(); \
}, \
py::return_value_policy::reference)
#define def_function(func_name) \
def(#func_name, [](ContactMechanicsModel & self) -> decltype(auto) { \
return self.func_name(); \
})
/* -------------------------------------------------------------------------- */
void register_contact_mechanics_model(py::module & mod) {
py::class_<ContactDetector>(mod, "ContactDetector", py::multiple_inheritance())
.def(py::init<Mesh &, const ID &>(),
py::arg("mesh"), py::arg("id") ="contact_detector")
.def(py::init<Mesh &, Array<Real>, const ID &>(),
py::arg("mesh"), py::arg("positions"),
py::arg("id") = "contact_detector")
.def("setSurfaceSelector", &ContactDetector::setSurfaceSelector);
py::class_<SurfaceSelector, std::shared_ptr<SurfaceSelector>>(mod, "SurfaceSelector", py::multiple_inheritance())
.def(py::init<Mesh &>(),
py::arg("mesh"));
py::class_<PhysicalSurfaceSelector, SurfaceSelector, std::shared_ptr<PhysicalSurfaceSelector>>(mod, "PhysicalSurfaceSelector")
.def(py::init<Mesh &>(),
py::arg("mesh"));
py::class_<CohesiveSurfaceSelector, SurfaceSelector, std::shared_ptr<CohesiveSurfaceSelector>>(mod, "CohesiveSurfaceSelector")
.def(py::init<Mesh &>(),
py::arg("mesh"));
py::class_<AllSurfaceSelector, SurfaceSelector, std::shared_ptr<AllSurfaceSelector>>(mod, "AllSurfaceSelector")
.def(py::init<Mesh &>(),
py::arg("mesh"));
py::class_<ContactMechanicsModelOptions>(mod, "ContactMechanicsModelOptions")
.def(py::init<AnalysisMethod>(),
py::arg("analysis_method") = _explicit_contact);
py::class_<ContactMechanicsModel, Model>(mod, "ContactMechanicsModel",
py::multiple_inheritance())
.def(py::init<Mesh &, UInt, const ID &, std::shared_ptr<DOFManager>,
const ModelType>(),
py::arg("mesh"), py::arg("spatial_dimension") = _all_dimensions,
py::arg("id") = "contact_mechanics_model", py::arg("dof_manager") = nullptr,
py::arg("model_type") = ModelType::_contact_mechanics_model)
.def(
"initFull",
[](ContactMechanicsModel & self,
const ContactMechanicsModelOptions & options) {
self.initFull(options);
},
py::arg("_analysis_method") = ContactMechanicsModelOptions())
.def(
"initFull",
[](ContactMechanicsModel & self,
const AnalysisMethod & analysis_method) {
self.initFull(_analysis_method = analysis_method);
},
py::arg("_analysis_method"))
.def_function(search)
.def_function(assembleStiffnessMatrix)
.def_function(assembleInternalForces)
.def_function_nocopy(getExternalForce)
.def_function_nocopy(getNormalForce)
.def_function_nocopy(getTangentialForce)
.def_function_nocopy(getInternalForce)
.def_function_nocopy(getGaps)
.def_function_nocopy(getNormals)
.def_function_nocopy(getNodalArea)
.def_function_nocopy(getContactDetector)
.def("dump", py::overload_cast<>(&ContactMechanicsModel::dump))
.def("dump",
py::overload_cast<const std::string &>(&ContactMechanicsModel::dump))
.def("dump", py::overload_cast<const std::string &, UInt>(
&ContactMechanicsModel::dump))
.def("dump", py::overload_cast<const std::string &, Real, UInt>(
&ContactMechanicsModel::dump));
py::class_<ContactElement>(mod, "ContactElement").def(py::init<>());
py::class_<GeometryUtils>(mod, "GeometryUtils")
.def_static(
"normal",
py::overload_cast<const Mesh &, const Array<Real> &, const Element &,
Vector<Real> &, bool>(&GeometryUtils::normal),
py::arg("mesh"), py::arg("positions"), py::arg("element"),
py::arg("normal"), py::arg("outward") = true)
.def_static(
"covariantBasis",
py::overload_cast<const Mesh &, const Array<Real> &, const Element &,
const Vector<Real> &, Vector<Real> &,
Matrix<Real> &>(&GeometryUtils::covariantBasis),
py::arg("mesh"), py::arg("positions"), py::arg("element"),
py::arg("normal"), py::arg("natural_projection"), py::arg("basis"))
.def_static("curvature", &GeometryUtils::curvature)
.def_static("contravariantBasis", &GeometryUtils::contravariantBasis,
py::arg("covariant_basis"), py::arg("basis"))
.def_static("realProjection",
py::overload_cast<const Mesh &, const Array<Real> &,
const Vector<Real> &, const Element &,
const Vector<Real> &, Vector<Real> &>(
&GeometryUtils::realProjection),
py::arg("mesh"), py::arg("positions"), py::arg("slave"),
py::arg("element"), py::arg("normal"), py::arg("projection"))
.def_static("isBoundaryElement", &GeometryUtils::isBoundaryElement);
}
} // namespace akantu
c4science · Help