Homec4science
Diffusion GooseFEM (rstcheck) Browse

rGOOSEFEM/python/MeshQuad4.hpp
f64ccfd6ac5drstcheck

Last Change
Raw File

MeshQuad4.hpp
View Options

/* =================================================================================================
(c - GPLv3) T.W.J. de Geus (Tom) | tom@geus.me | www.geus.me | github.com/tdegeus/GooseFEM
================================================================================================= */
#include <Eigen/Eigen>
#include <pybind11/pybind11.h>
#include <pybind11/eigen.h>
#include <pyxtensor/pyxtensor.hpp>
#include "../include/GooseFEM/GooseFEM.h"
// =================================================================================================
namespace py = pybind11;
// =================================================================================================
void init_MeshQuad4(py::module& m)
{
py::class_<GooseFEM::Mesh::Quad4::Regular>(m, "Regular")
.def(py::init<size_t,size_t,double>(),
"Regular mesh: 'nx' pixels in horizontal direction, 'ny' in vertical direction, edge size 'h'",
py::arg("nx"),
py::arg("ny"),
py::arg("h")=1.)
.def("coor",
&GooseFEM::Mesh::Quad4::Regular::coor)
.def("conn",
&GooseFEM::Mesh::Quad4::Regular::conn)
.def("nelem",
&GooseFEM::Mesh::Quad4::Regular::nelem)
.def("nnode",
&GooseFEM::Mesh::Quad4::Regular::nnode)
.def("nne",
&GooseFEM::Mesh::Quad4::Regular::nne)
.def("ndim",
&GooseFEM::Mesh::Quad4::Regular::ndim)
.def("nelx",
&GooseFEM::Mesh::Quad4::Regular::nelx)
.def("nely",
&GooseFEM::Mesh::Quad4::Regular::nely)
.def("h",
&GooseFEM::Mesh::Quad4::Regular::h)
.def("nodesBottomEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesBottomEdge)
.def("nodesTopEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesTopEdge)
.def("nodesLeftEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesLeftEdge)
.def("nodesRightEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesRightEdge)
.def("nodesBottomOpenEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesBottomOpenEdge)
.def("nodesTopOpenEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesTopOpenEdge)
.def("nodesLeftOpenEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesLeftOpenEdge)
.def("nodesRightOpenEdge",
&GooseFEM::Mesh::Quad4::Regular::nodesRightOpenEdge)
.def("nodesBottomLeftCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesBottomLeftCorner)
.def("nodesBottomRightCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesBottomRightCorner)
.def("nodesTopLeftCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesTopLeftCorner)
.def("nodesTopRightCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesTopRightCorner)
.def("nodesLeftBottomCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesLeftBottomCorner)
.def("nodesLeftTopCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesLeftTopCorner)
.def("nodesRightBottomCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesRightBottomCorner)
.def("nodesRightTopCorner",
&GooseFEM::Mesh::Quad4::Regular::nodesRightTopCorner)
.def("dofs",
&GooseFEM::Mesh::Quad4::Regular::dofs)
.def("nodesPeriodic",
&GooseFEM::Mesh::Quad4::Regular::nodesPeriodic)
.def("nodesOrigin",
&GooseFEM::Mesh::Quad4::Regular::nodesOrigin)
.def("dofsPeriodic",
&GooseFEM::Mesh::Quad4::Regular::dofsPeriodic)
.def("elementMatrix",
&GooseFEM::Mesh::Quad4::Regular::elementMatrix)
.def("__repr__",
[](const GooseFEM::Mesh::Quad4::Regular&){ return "<GooseFEM.Mesh.Quad4.Regular>"; });
// -------------------------------------------------------------------------------------------------
py::class_<GooseFEM::Mesh::Quad4::FineLayer>(m, "FineLayer")
.def(py::init<size_t,size_t,double,size_t>(),
"FineLayer mesh: 'nx' pixels in horizontal direction (length 'Lx'), idem in vertical direction",
py::arg("nx"),
py::arg("ny"),
py::arg("h")=1.,
py::arg("nfine")=1)
.def("coor",
&GooseFEM::Mesh::Quad4::FineLayer::coor)
.def("conn",
&GooseFEM::Mesh::Quad4::FineLayer::conn)
.def("nelem",
&GooseFEM::Mesh::Quad4::FineLayer::nelem)
.def("nnode",
&GooseFEM::Mesh::Quad4::FineLayer::nnode)
.def("nne",
&GooseFEM::Mesh::Quad4::FineLayer::nne)
.def("ndim",
&GooseFEM::Mesh::Quad4::FineLayer::ndim)
.def("nelx",
&GooseFEM::Mesh::Quad4::FineLayer::nelx)
.def("nely",
&GooseFEM::Mesh::Quad4::FineLayer::nely)
.def("h",
&GooseFEM::Mesh::Quad4::FineLayer::h)
.def("elementsMiddleLayer",
&GooseFEM::Mesh::Quad4::FineLayer::elementsMiddleLayer)
.def("nodesBottomEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesBottomEdge)
.def("nodesTopEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesTopEdge)
.def("nodesLeftEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesLeftEdge)
.def("nodesRightEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesRightEdge)
.def("nodesBottomOpenEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesBottomOpenEdge)
.def("nodesTopOpenEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesTopOpenEdge)
.def("nodesLeftOpenEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesLeftOpenEdge)
.def("nodesRightOpenEdge",
&GooseFEM::Mesh::Quad4::FineLayer::nodesRightOpenEdge)
.def("nodesBottomLeftCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesBottomLeftCorner)
.def("nodesBottomRightCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesBottomRightCorner)
.def("nodesTopLeftCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesTopLeftCorner)
.def("nodesTopRightCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesTopRightCorner)
.def("nodesLeftBottomCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesLeftBottomCorner)
.def("nodesLeftTopCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesLeftTopCorner)
.def("nodesRightBottomCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesRightBottomCorner)
.def("nodesRightTopCorner",
&GooseFEM::Mesh::Quad4::FineLayer::nodesRightTopCorner)
.def("dofs",
&GooseFEM::Mesh::Quad4::FineLayer::dofs)
.def("nodesPeriodic",
&GooseFEM::Mesh::Quad4::FineLayer::nodesPeriodic)
.def("nodesOrigin",
&GooseFEM::Mesh::Quad4::FineLayer::nodesOrigin)
.def("dofsPeriodic",
&GooseFEM::Mesh::Quad4::FineLayer::dofsPeriodic)
.def("__repr__",
[](const GooseFEM::Mesh::Quad4::FineLayer&){ return "<GooseFEM.Mesh.Quad4.FineLayer>"; }
);
}
// =================================================================================================
void init_MeshQuad4Map(py::module& m)
{
// -------------------------------------------------------------------------------------------------
py::class_<GooseFEM::Mesh::Quad4::Map::RefineRegular>(m, "RefineRegular")
.def(py::init<const GooseFEM::Mesh::Quad4::Regular &, size_t, size_t>(),
"Refine a regular mesh",
py::arg("mesh"),
py::arg("nx"),
py::arg("ny"))
.def("getCoarseMesh",
&GooseFEM::Mesh::Quad4::Map::RefineRegular::getCoarseMesh)
.def("getFineMesh",
&GooseFEM::Mesh::Quad4::Map::RefineRegular::getFineMesh)
.def("getMap",
&GooseFEM::Mesh::Quad4::Map::RefineRegular::getMap)
.def("mapToCoarse",
py::overload_cast<const xt::xtensor<double,1>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToCoarse, py::const_))
.def("mapToCoarse",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToCoarse, py::const_))
.def("mapToCoarse",
py::overload_cast<const xt::xtensor<double,4>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToCoarse, py::const_))
.def("mapToFine",
py::overload_cast<const xt::xtensor<double,1>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToFine, py::const_))
.def("mapToFine",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToFine, py::const_))
.def("mapToFine",
py::overload_cast<const xt::xtensor<double,4>&>(
&GooseFEM::Mesh::Quad4::Map::RefineRegular::mapToFine, py::const_))
.def("__repr__",
[](const GooseFEM::Mesh::Quad4::Map::RefineRegular &){
return "<GooseFEM.Mesh.Quad4.Map.RefineRegular>"; });
// -------------------------------------------------------------------------------------------------
py::class_<GooseFEM::Mesh::Quad4::Map::FineLayer2Regular>(m, "FineLayer2Regular")
.def(py::init<const GooseFEM::Mesh::Quad4::FineLayer &>(),
"Map a FineLayer mesh to a Regular mesh",
py::arg("mesh"))
.def("getRegularMesh",
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::getRegularMesh)
.def("getFineLayerMesh",
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::getFineLayerMesh)
.def("getMap",
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::getMap)
.def("getMapFraction",
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::getMapFraction)
.def("mapToRegular",
py::overload_cast<const xt::xtensor<double,1>&>(
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::mapToRegular, py::const_))
.def("mapToRegular",
py::overload_cast<const xt::xtensor<double,2>&>(
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::mapToRegular, py::const_))
.def("mapToRegular",
py::overload_cast<const xt::xtensor<double,4>&>(
&GooseFEM::Mesh::Quad4::Map::FineLayer2Regular::mapToRegular, py::const_))
.def("__repr__",
[](const GooseFEM::Mesh::Quad4::Map::FineLayer2Regular &){
return "<GooseFEM.Mesh.Quad4.Map.FineLayer2Regular>"; });
// -------------------------------------------------------------------------------------------------
}
// =================================================================================================
c4science · Help