Homec4science
Diffusion GooseFEM (rstcheck) Browse

rGOOSEFEM/python/Element.hpp
76076a666f83rstcheck

Last Change
Raw File

Element.hpp
View Options

/**
\file
\copyright Copyright 2017. Tom de Geus. All rights reserved.
\license This project is released under the GNU Public License (GPLv3).
*/
#ifndef PYGOOSEFEM_ELEMENT_H
#define PYGOOSEFEM_ELEMENT_H
#include <GooseFEM/Element.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <xtensor-python/pyarray.hpp>
#include <xtensor-python/pytensor.hpp>
namespace py = pybind11;
template <class C, class P>
void register_Element_QuadratureBase(P& cls)
{
cls.def("nelem", &C::nelem, "Number of elements");
cls.def("nne", &C::nne, "Number of nodes per element");
cls.def("ndim", &C::ndim, "Number of dimensions");
cls.def("tdim", &C::tdim, "Number of dimensions of tensors");
cls.def("nip", &C::nip, "Number of integration points");
// todo: https://github.com/xtensor-stack/xtensor-python/issues/265
// cls.def("asTensor",
// static_cast<void (C::*)(const xt::pyarray<double>&, xt::pyarray<double>&) const>(&C::asTensor),
// "Convert 'qscalar' to 'qtensor' of certain rank",
// py::arg("qscalar"),
// py::arg("qtensor"));
cls.def("AsTensor",
static_cast<xt::xarray<double> (C::*)(size_t, const xt::pyarray<double>&) const>(&C::AsTensor),
"Convert 'qscalar' to 'qtensor' of certain rank",
py::arg("rank"),
py::arg("qscalar"));
cls.def("shape_elemvec",
static_cast<std::array<size_t, 3> (C::*)() const>(&C::shape_elemvec),
"Shape of 'elemvec'");
cls.def("shape_elemvec",
static_cast<std::array<size_t, 3> (C::*)(size_t) const>(&C::shape_elemvec),
"Shape of 'elemvec'",
py::arg("tdim"));
cls.def("shape_elemmat",
&C::shape_elemmat,
"Shape of 'elemmat'");
cls.def("shape_qtensor",
static_cast<std::vector<size_t> (C::*)(size_t) const>(&C::shape_qtensor),
"Shape of 'qtensor'",
py::arg("rank"));
cls.def("shape_qtensor",
static_cast<std::vector<size_t> (C::*)(size_t, size_t) const>(&C::shape_qtensor),
"Shape of 'qtensor'",
py::arg("rank"),
py::arg("tdim"));
cls.def("shape_qscalar",
&C::shape_qscalar,
"Shape of 'qscalar'");
cls.def("shape_qvector",
static_cast<std::array<size_t, 3> (C::*)() const>(&C::shape_qvector),
"Shape of 'qvector'");
cls.def("shape_qvector",
static_cast<std::array<size_t, 3> (C::*)(size_t) const>(&C::shape_qvector),
"Shape of 'qvector'",
py::arg("tdim"));
}
template <class C, class P>
void register_Element_QuadratureBaseCartesian(P& cls)
{
cls.def("update_x",
&C::template update_x<xt::pytensor<double, 3>>,
"Update the nodal positions",
py::arg("x"));
cls.def("dV", &C::dV, "Integration point volume (qscalar)");
cls.def("InterpQuad_vector",
&C::template InterpQuad_vector<xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::InterpQuad_vector`.",
py::arg("elemvec"));
cls.def("interpQuad_vector",
&C::template interpQuad_vector<xt::pytensor<double, 3>, xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::interpQuad_vector`.",
py::arg("elemvec"),
py::arg("qvector"));
cls.def("GradN_vector",
&C::template GradN_vector<xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::GradN_vector`.",
py::arg("elemvec"));
cls.def("gradN_vector",
&C::template gradN_vector<xt::pytensor<double, 3>, xt::pytensor<double, 4>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::gradN_vector`.",
py::arg("elemvec"),
py::arg("qtensor"));
cls.def("GradN_vector_T",
&C::template GradN_vector_T<xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::GradN_vector_T`.",
py::arg("elemvec"));
cls.def("gradN_vector_T",
&C::template gradN_vector_T<xt::pytensor<double, 3>, xt::pytensor<double, 4>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::gradN_vector_T`.",
py::arg("elemvec"),
py::arg("qtensor"));
cls.def("SymGradN_vector",
&C::template SymGradN_vector<xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::SymGradN_vector`.",
py::arg("elemvec"));
cls.def("symGradN_vector",
&C::template symGradN_vector<xt::pytensor<double, 3>, xt::pytensor<double, 4>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::symGradN_vector`.",
py::arg("elemvec"),
py::arg("qtensor"));
cls.def("Int_N_vector_dV",
&C::template Int_N_vector_dV<xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::Int_N_vector_dV`.",
py::arg("qvector"));
cls.def("int_N_vector_dV",
&C::template int_N_vector_dV<xt::pytensor<double, 3>, xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::int_N_vector_dV`.",
py::arg("qvector"),
py::arg("elemvec"));
cls.def("Int_N_scalar_NT_dV",
&C::template Int_N_scalar_NT_dV<xt::pytensor<double, 2>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::Int_N_scalar_NT_dV`.",
py::arg("qscalar"));
cls.def("int_N_scalar_NT_dV",
&C::template int_N_scalar_NT_dV<xt::pytensor<double, 2>, xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::int_N_scalar_NT_dV`.",
py::arg("qscalar"),
py::arg("elemmat"));
cls.def("Int_gradN_dot_tensor2_dV",
&C::template Int_gradN_dot_tensor2_dV<xt::pytensor<double, 4>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::Int_gradN_dot_tensor2_dV`.",
py::arg("qtensor"));
cls.def("int_gradN_dot_tensor2_dV",
&C::template int_gradN_dot_tensor2_dV<xt::pytensor<double, 4>, xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::int_gradN_dot_tensor2_dV`.",
py::arg("qtensor"),
py::arg("elemvec"));
cls.def("Int_gradN_dot_tensor4_dot_gradNT_dV",
&C::template Int_gradN_dot_tensor4_dot_gradNT_dV<xt::pytensor<double, 6>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::Int_gradN_dot_tensor4_dot_gradNT_dV`.",
py::arg("qtensor"));
cls.def("int_gradN_dot_tensor4_dot_gradNT_dV",
&C::template int_gradN_dot_tensor4_dot_gradNT_dV<xt::pytensor<double, 6>, xt::pytensor<double, 3>>,
"See :cpp:class:`GooseFEM::Quad4::Quadrature::int_gradN_dot_tensor4_dot_gradNT_dV`.",
py::arg("qtensor"),
py::arg("elemmat"));
}
void init_Element(py::module& m)
{
m.def("asElementVector",
&GooseFEM::Element::asElementVector,
"Convert nodal vector [nnode, ndim] to nodal vector stored per element [nelem, nne, ndim]",
py::arg("conn"),
py::arg("nodevec"));
m.def("assembleElementVector",
&GooseFEM::Element::assembleNodeVector,
"Assemble nodal vector stored per element [nelem, nne, ndim] to nodal vector [nnode, ndim]",
py::arg("conn"),
py::arg("elemvec"));
}
#endif
c4science · Help