Homec4science
Diffusion GooseFEM (rstcheck) Browse

rGOOSEFEM/python/VectorPartitioned.hpp
d1faf5e55424rstcheck

Last Change
Raw File

VectorPartitioned.hpp
View Options

/* =================================================================================================
(c - GPLv3) T.W.J. de Geus (Tom) | tom@geus.me | www.geus.me | github.com/tdegeus/GooseFEM
================================================================================================= */
#include <pybind11/pybind11.h>
#include <pyxtensor/pyxtensor.hpp>
#include "../include/GooseFEM/GooseFEM.h"
// =================================================================================================
namespace py = pybind11;
// =================================================================================================
void init_VectorPartitioned(py::module& m)
{
py::class_<GooseFEM::VectorPartitioned>(m, "VectorPartitioned")
.def(py::init<
const xt::xtensor<size_t,2>&,
const xt::xtensor<size_t,2>&,
const xt::xtensor<size_t,1>&>(),
"Switch between dofval/nodevec/elemvec",
py::arg("conn"),
py::arg("dofs"),
py::arg("iip"))
.def("nelem",
&GooseFEM::VectorPartitioned::nelem,
"Number of element")
.def("nne",
&GooseFEM::VectorPartitioned::nne,
"Number of nodes per element")
.def("nnode",
&GooseFEM::VectorPartitioned::nnode,
"Number of nodes")
.def("ndim",
&GooseFEM::VectorPartitioned::ndim,
"Number of dimensions")
.def("ndof",
&GooseFEM::VectorPartitioned::ndof,
"Number of degrees-of-freedom")
.def("nnu",
&GooseFEM::VectorPartitioned::nnu,
"Number of unknown degrees-of-freedom")
.def("nnp",
&GooseFEM::VectorPartitioned::nnp,
"Number of prescribed degrees-of-freedom")
.def("dofs",
&GooseFEM::VectorPartitioned::dofs,
"Return degrees-of-freedom")
.def("iiu",
&GooseFEM::VectorPartitioned::iiu,
"Return unknown degrees-of-freedom")
.def("iip",
&GooseFEM::VectorPartitioned::iip,
"Return prescribed degrees-of-freedom")
.def("AsDofs",
py::overload_cast<const xt::xtensor<double,1>&,const xt::xtensor<double,1>&>(
&GooseFEM::VectorPartitioned::AsDofs, py::const_),
"Set 'dofval",
py::arg("dofval_u"),
py::arg("dofval_p"))
.def("AsDofs",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AsDofs, py::const_),
"Set 'dofval",
py::arg("nodevec"))
.def("AsDofs",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AsDofs, py::const_),
"Set 'dofval",
py::arg("elemvec"))
.def("AsDofs_u",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AsDofs_u, py::const_),
"Set 'dofval",
py::arg("nodevec"))
.def("AsDofs_u",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AsDofs_u, py::const_),
"Set 'dofval",
py::arg("elemvec"))
.def("AsDofs_p",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AsDofs_p, py::const_),
"Set 'dofval",
py::arg("nodevec"))
.def("AsDofs_p",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AsDofs_p, py::const_),
"Set 'dofval",
py::arg("elemvec"))
.def("AsNode",
py::overload_cast<const xt::xtensor<double,1>&,const xt::xtensor<double,1>&>(
&GooseFEM::VectorPartitioned::AsNode, py::const_),
"Set 'nodevec",
py::arg("dofval_u"),
py::arg("dofval_p"))
.def("AsNode",
py::overload_cast<const xt::xtensor<double,1>&>(
&GooseFEM::VectorPartitioned::AsNode, py::const_),
"Set 'nodevec",
py::arg("dofval"))
.def("AsNode",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AsNode, py::const_),
"Set 'nodevec",
py::arg("elemvec"))
.def("AsElement",
py::overload_cast<const xt::xtensor<double,1>&,const xt::xtensor<double,1>&>(
&GooseFEM::VectorPartitioned::AsElement, py::const_),
"Set 'elemvec",
py::arg("dofval_u"),
py::arg("dofval_p"))
.def("AsElement",
py::overload_cast<const xt::xtensor<double,1>&>(
&GooseFEM::VectorPartitioned::AsElement, py::const_),
"Set 'elemvec",
py::arg("dofval"))
.def("AsElement",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AsElement, py::const_),
"Set 'elemvec",
py::arg("nodevec"))
.def("AssembleDofs",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AssembleDofs, py::const_),
"Assemble 'dofval'",
py::arg("nodevec"))
.def("AssembleDofs",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AssembleDofs, py::const_),
"Assemble 'dofval'",
py::arg("elemvec"))
.def("AssembleDofs_u",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AssembleDofs_u, py::const_),
"Assemble 'dofval'",
py::arg("nodevec"))
.def("AssembleDofs_u",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AssembleDofs_u, py::const_),
"Assemble 'dofval'",
py::arg("elemvec"))
.def("AssembleDofs_p",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::VectorPartitioned::AssembleDofs_p, py::const_),
"Assemble 'dofval'",
py::arg("nodevec"))
.def("AssembleDofs_p",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AssembleDofs_p, py::const_),
"Assemble 'dofval'",
py::arg("elemvec"))
.def("AssembleNode",
py::overload_cast<const xt::xtensor<double,3>&>(
&GooseFEM::VectorPartitioned::AssembleNode, py::const_),
"Assemble 'nodevec'",
py::arg("elemvec"))
.def("Copy",
&GooseFEM::VectorPartitioned::Copy,
"Copy")
.def("Copy_u",
&GooseFEM::VectorPartitioned::Copy_u,
"Copy iiu")
.def("Copy_p",
&GooseFEM::VectorPartitioned::Copy_p,
"Copy iip")
.def("__repr__", [](
const GooseFEM::VectorPartitioned&){ return "<GooseFEM.VectorPartitioned>"; });
}
// =================================================================================================
c4science · Help