Homec4science
Diffusion µSpectre (feat/optional_compilation_T1925) Browse

muSpectre/src/language_bindings/python/bind_py_system.cc
5107cf61d85ffeat/optional_compilati…

Last Change
Raw File

bind_py_system.cc
View Options

/**
* file bind_py_system.cc
*
* @author Till Junge <till.junge@epfl.ch>
*
* @date 09 Jan 2018
*
* @brief Python bindings for the system factory function
*
* @section LICENCE
*
* Copyright © 2018 Till Junge
*
* µSpectre is free software; you can redistribute it and/or
* modify it under the terms of the GNU 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 General Public License
* along with GNU Emacs; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#include "common/common.hh"
#include "common/ccoord_operations.hh"
#include "system/system_factory.hh"
#include "system/system_base.hh"
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
#include <sstream>
using namespace muSpectre;
namespace py=pybind11;
using namespace pybind11::literals;
/**
* the system factory is only bound for default template parameters
*/
template <Dim_t dim>
void add_system_factory_helper(py::module & mod) {
std::stringstream name {};
name << "SystemFactory" << dim << "d";
using Ccoord = Ccoord_t<dim>;
using Rcoord = Rcoord_t<dim>;
using Form = Formulation;
mod.def
(name.str().c_str(),
[](Ccoord res, Rcoord lens, Form form) {
return make_system(std::move(res), std::move(lens), std::move(form));
},
"resolutions"_a,
"lengths"_a=CcoordOps::get_cube<dim>(1.),
"formulation"_a=Formulation::finite_strain);
}
void add_system_factory(py::module & mod) {
add_system_factory_helper<twoD >(mod);
add_system_factory_helper<threeD>(mod);
}
/**
* SystemBase for which the material and spatial dimension are identical
*/
template <Dim_t dim>
void add_system_base_helper(py::module & mod) {
std::stringstream name{};
name << "SystemBase" << dim << 'd';
auto && name_str{name.str().c_str()};
using sys_t = SystemBase<dim, dim>;
py::class_<sys_t>(mod, name_str)
.def("__len__", &sys_t::size)
.def("__iter__", [](sys_t & s) {
return py::make_iterator(s.begin(), s.end());
})
.def("initialise", &sys_t::initialise, "flags"_a=FFT_PlanFlags::estimate);
}
void add_system_base(py::module & mod) {
add_system_base_helper<twoD> (mod);
add_system_base_helper<threeD>(mod);
}
PYBIND11_PLUGIN(system) {
py::module::import("common");
py::module mod("system", "bindings for systems and system factories");
add_system_factory(mod);
add_system_base(mod);
return mod.ptr();
}
c4science · Help