Page MenuHomec4science
Files F121034413

bind_py_material.cc
No OneTemporary
Actions

File Metadata

Created
Tue, Jul 8, 09:42

bind_py_material.cc
View Options

/**
* @file bind_py_material.cc
*
* @author Till Junge <till.junge@epfl.ch>
*
* @date 09 Jan 2018
*
* @brief python bindings for µSpectre's materials
*
* Copyright © 2018 Till Junge
*
* µSpectre is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation, either version 3, or (at
* your option) any later version.
*
* µSpectre is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with µSpectre; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* * Boston, MA 02111-1307, USA.
*
* Additional permission under GNU GPL version 3 section 7
*
* If you modify this Program, or any covered work, by linking or combining it
* with proprietary FFT implementations or numerical libraries, containing parts
* covered by the terms of those libraries' licenses, the licensors of this
* Program grant you additional permission to convey the resulting work.
*/
#include "common/common.hh"
#include "materials/material_base.hh"
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <pybind11/eigen.h>
#include <sstream>
#include <string>
using muSpectre::Dim_t;
using pybind11::literals::operator""_a;
namespace py = pybind11;
/* ---------------------------------------------------------------------- */
template <Dim_t Dim>
void add_material_linear_elastic_generic1_helper(py::module & mod);
/* ---------------------------------------------------------------------- */
template <Dim_t Dim>
void add_material_linear_elastic_generic2_helper(py::module & mod);
/**
* python binding for the optionally objective form of Hooke's law
*/
template <Dim_t dim>
void add_material_linear_elastic1_helper(py::module & mod);
template <Dim_t dim>
void add_material_linear_elastic2_helper(py::module & mod);
template <Dim_t dim>
void add_material_linear_elastic3_helper(py::module & mod);
template <Dim_t dim>
void add_material_linear_elastic4_helper(py::module & mod);
template <Dim_t Dim>
void add_material_hyper_elasto_plastic1_helper(py::module & mod);
/* ---------------------------------------------------------------------- */
template <Dim_t Dim>
class PyMaterialBase : public muSpectre::MaterialBase<Dim, Dim> {
public:
/* Inherit the constructors */
using Parent = muSpectre::MaterialBase<Dim, Dim>;
using Parent::Parent;
/* Trampoline (need one for each virtual function) */
void save_history_variables() override {
PYBIND11_OVERLOAD_PURE(void, // Return type
Parent, // Parent class
save_history_variables); // Name of function in C++
// (must match Python name)
}
/* Trampoline (need one for each virtual function) */
void initialise() override {
PYBIND11_OVERLOAD_PURE(
void, // Return type
Parent, // Parent class
initialise); // Name of function in C++ (must match Python name)
}
void compute_stresses(const typename Parent::StrainField_t & F,
typename Parent::StressField_t & P,
muSpectre::Formulation form) override {
PYBIND11_OVERLOAD_PURE(
void, // Return type
Parent, // Parent class
compute_stresses, // Name of function in C++ (must match Python name)
F, P, form);
}
void compute_stresses_tangent(const typename Parent::StrainField_t & F,
typename Parent::StressField_t & P,
typename Parent::TangentField_t & K,
muSpectre::Formulation form) override {
PYBIND11_OVERLOAD_PURE(
void, /* Return type */
Parent, /* Parent class */
compute_stresses, /* Name of function in C++ (must match Python name) */
F, P, K, form);
}
};
template <Dim_t dim>
void add_material_helper(py::module & mod) {
std::stringstream name_stream{};
name_stream << "MaterialBase_" << dim << "d";
std::string name{name_stream.str()};
using Material = muSpectre::MaterialBase<dim, dim>;
using MaterialTrampoline = PyMaterialBase<dim>;
using FC_t = muSpectre::LocalFieldCollection<dim>;
using FCBase_t = muSpectre::FieldCollectionBase<dim, FC_t>;
py::class_<Material, MaterialTrampoline /* <--- trampoline*/>(mod,
name.c_str())
.def(py::init<std::string>())
.def("save_history_variables", &Material::save_history_variables)
.def("list_fields", &Material::list_fields)
.def("get_real_field", &Material::get_real_field, "field_name"_a,
py::return_value_policy::reference_internal)
.def("size", &Material::size)
.def("add_pixel",
[](Material & mat, muSpectre::Ccoord_t<dim> pix) {
mat.add_pixel(pix);
},
"pixel"_a)
.def_property_readonly("collection",
[](Material & material) -> FCBase_t & {
return material.get_collection();
},
"returns the field collection containing internal "
"fields of this material");
add_material_linear_elastic1_helper<dim>(mod);
add_material_linear_elastic2_helper<dim>(mod);
add_material_linear_elastic3_helper<dim>(mod);
add_material_linear_elastic4_helper<dim>(mod);
add_material_hyper_elasto_plastic1_helper<dim>(mod);
add_material_linear_elastic_generic1_helper<dim>(mod);
add_material_linear_elastic_generic2_helper<dim>(mod);
}
void add_material(py::module & mod) {
auto material{mod.def_submodule("material")};
material.doc() = "bindings for constitutive laws";
add_material_helper<muSpectre::twoD>(material);
add_material_helper<muSpectre::threeD>(material);
}

Event Timeline

c4science · Help