Homec4science
Diffusion akantu (features/30-implementing-a-checkpoint-restart-mechanism) Browse

rAKA/python/py_heat_transfer_model.cc
fdaa2dacbd0cfeatures/30-implementin…

Last Change
Raw File

py_heat_transfer_model.cc
View Options

/* -------------------------------------------------------------------------- */
#include "py_aka_array.hh"
/* -------------------------------------------------------------------------- */
#include <heat_transfer_model.hh>
#include <non_linear_solver.hh>
/* -------------------------------------------------------------------------- */
//#include <pybind11/operators.h>
#include <pybind11/pybind11.h>
//#include <pybind11/stl.h>
/* -------------------------------------------------------------------------- */
namespace py = pybind11;
/* -------------------------------------------------------------------------- */
namespace akantu {
/* -------------------------------------------------------------------------- */
#define def_deprecated(func_name, mesg) \
def(func_name, [](py::args, py::kwargs) { AKANTU_ERROR(mesg); })
#define def_function_nocopy(func_name) \
def(#func_name, \
[](HeatTransferModel & self) -> decltype(auto) { \
return self.func_name(); \
}, \
py::return_value_policy::reference)
#define def_function(func_name) \
def(#func_name, [](HeatTransferModel & self) -> decltype(auto) { \
return self.func_name(); \
})
/* -------------------------------------------------------------------------- */
void register_heat_transfer_model(py::module & mod) {
py::class_<HeatTransferModelOptions>(mod, "HeatTransferModelOptions")
.def(py::init<AnalysisMethod>(),
py::arg("analysis_method") = _explicit_lumped_mass);
py::class_<HeatTransferModel, Model>(mod, "HeatTransferModel",
py::multiple_inheritance())
.def(py::init<Mesh &, UInt, const ID &>(),
py::arg("mesh"), py::arg("spatial_dimension") = _all_dimensions,
py::arg("id") = "heat_transfer_model")
.def("initFull",
[](HeatTransferModel & self,
const HeatTransferModelOptions & options) {
self.initFull(options);
},
py::arg("_analysis_method") = HeatTransferModelOptions())
.def("initFull",
[](HeatTransferModel & self,
const AnalysisMethod & _analysis_method) {
self.initFull(HeatTransferModelOptions(_analysis_method));
},
py::arg("_analysis_method"))
.def("setTimeStep", &HeatTransferModel::setTimeStep, py::arg("time_step"),
py::arg("solver_id") = "")
.def_function(getStableTimeStep)
.def_function_nocopy(getTemperature)
.def_function_nocopy(getBlockedDOFs)
.def("getTemperatureGradient", &HeatTransferModel::getTemperatureGradient,
py::arg("el_type"), py::arg("ghost_type") = _not_ghost,
py::return_value_policy::reference)
.def("getKgradT", &HeatTransferModel::getKgradT, py::arg("el_type"),
py::arg("ghost_type") = _not_ghost,
py::return_value_policy::reference);
}
} // namespace akantu
c4science · Help