PyMesh/python/PyFEM.cpp at master · RealTimeTool/PyMesh · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <memory>

#include <pybind11/pybind11.h>
#include <pybind11/eigen.h>
#include <pybind11/stl.h>

#include <Assembler/Materials/Material.h>
#include <Assembler/FEAssembler.h>

namespace py = pybind11;
using namespace PyMesh;

void init_FEM(py::module& m) {
    py::class_<Material, std::shared_ptr<Material> >(m, "Material")
        .def_static("create", &Material::create)
        .def_static("create_isotropic", &Material::create_isotropic)
        .def_static("create_orthotropic", &Material::create_orthotropic)
        .def_static("create_periodic", &Material::create_periodic)
        .def_static("craete_element_wise_isotropic", &Material::create_element_wise_isotropic)
        .def_static("craete_element_wise_orthotropic", &Material::create_element_wise_orthotropic)
        .def_static("create_element_wise_symmetric", &Material::create_element_wise_symmetric)
        .def("get_material_tensor", &Material::get_material_tensor)
        .def("strain_to_stress", &Material::strain_to_stress)
        .def("get_density", py::overload_cast<>(&Material::get_density, py::const_), "Get denisity (uniform material only).")
        .def("get_density", py::overload_cast<VectorF>(&Material::get_density, py::const_), "Get density at specific point.")
        .def("get_dim", &Material::get_dim)
        .def("update", &Material::update) ;

    py::class_<FEAssembler, std::shared_ptr<FEAssembler> >(m, "FEAssembler")
        .def_static("create", &FEAssembler::create)
        .def_static("create_from_name", &FEAssembler::create_from_name)
        .def("assemble", &FEAssembler::assemble)
        .def("set_material", &FEAssembler::set_material);
}