Homec4science
Diffusion akantu (features/heat_transfer_along_interfaces) Browse

rAKA/python/py_dof_manager.cc
1d4a14f57340features/heat_transfer_…

Last Change
Raw File

py_dof_manager.cc
View Options

/*
* Copyright (©) 2018-2021 EPFL (Ecole Polytechnique Fédérale de Lausanne)
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* Akantu 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 of the License, or (at your option) any
* later version.
*
* Akantu 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 Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Akantu. If not, see <http://www.gnu.org/licenses/>.
*/
/* -------------------------------------------------------------------------- */
#include "py_dof_manager.hh"
#include "py_aka_array.hh"
#include "py_akantu_pybind11_compatibility.hh"
/* -------------------------------------------------------------------------- */
#include <dof_manager.hh>
#include <non_linear_solver.hh>
#include <solver_callback.hh>
#include <time_step_solver.hh>
/* -------------------------------------------------------------------------- */
#include <pybind11/operators.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
/* -------------------------------------------------------------------------- */
namespace py = pybind11;
/* -------------------------------------------------------------------------- */
namespace akantu {
namespace {
class PySolverCallback : public SolverCallback {
public:
using SolverCallback::SolverCallback;
/// get the type of matrix needed
MatrixType getMatrixType(const ID & matrix_id) const override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_PURE(MatrixType, SolverCallback, getMatrixType,
matrix_id);
}
/// callback to assemble a Matrix
void assembleMatrix(const ID & matrix_id) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_PURE(void, SolverCallback, assembleMatrix, matrix_id);
}
/// callback to assemble a lumped Matrix
void assembleLumpedMatrix(const ID & matrix_id) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_PURE(void, SolverCallback, assembleLumpedMatrix,
matrix_id);
}
/// callback to assemble the residual (rhs)
void assembleResidual() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE_PURE(void, SolverCallback, assembleResidual);
}
/// callback for the predictor (in case of dynamic simulation)
void predictor() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, SolverCallback, predictor);
}
/// callback for the corrector (in case of dynamic simulation)
void corrector() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, SolverCallback, corrector);
}
void beforeSolveStep() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, SolverCallback, beforeSolveStep);
}
void afterSolveStep(bool converged) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, SolverCallback, afterSolveStep, converged);
}
};
class PyInterceptSolverCallback : public InterceptSolverCallback {
public:
using InterceptSolverCallback::InterceptSolverCallback;
MatrixType getMatrixType(const ID & matrix_id) const override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(MatrixType, InterceptSolverCallback, getMatrixType,
matrix_id);
}
void assembleMatrix(const ID & matrix_id) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, assembleMatrix,
matrix_id);
}
/// callback to assemble a lumped Matrix
void assembleLumpedMatrix(const ID & matrix_id) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, assembleLumpedMatrix,
matrix_id);
}
void assembleResidual() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, assembleResidual);
}
void predictor() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, predictor);
}
void corrector() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, corrector);
}
void beforeSolveStep() override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, beforeSolveStep);
}
void afterSolveStep(bool converged) override {
// NOLINTNEXTLINE
PYBIND11_OVERRIDE(void, InterceptSolverCallback, afterSolveStep,
converged);
}
};
} // namespace
/* -------------------------------------------------------------------------- */
void register_dof_manager(py::module & mod) {
py::class_<DOFManager, std::shared_ptr<DOFManager>>(mod, "DOFManager")
.def("getMatrix", &DOFManager::getMatrix,
py::return_value_policy::reference)
.def(
"getNewMatrix",
[](DOFManager & self, const std::string & name,
const std::string & matrix_to_copy_id) -> decltype(auto) {
return self.getNewMatrix(name, matrix_to_copy_id);
},
py::return_value_policy::reference)
.def(
"getResidual",
[](DOFManager & self) -> decltype(auto) {
return self.getResidual();
},
py::return_value_policy::reference)
.def(
"getSolution",
[](DOFManager & self) -> decltype(auto) {
return self.getSolution();
},
py::return_value_policy::reference)
.def("getArrayPerDOFs", &DOFManager::getArrayPerDOFs)
.def(
"hasMatrix",
[](DOFManager & self, const ID & name) -> bool {
return self.hasMatrix(name);
},
py::arg("name"))
.def("assembleToResidual", &DOFManager::assembleToResidual,
py::arg("dof_id"), py::arg("array_to_assemble"),
py::arg("scale_factor") = 1.)
.def("assembleMatMulVectToGlobalArray",
&DOFManager::assembleMatMulVectToGlobalArray, py::arg("dof_id"),
py::arg("A_id"), py::arg("x"), py::arg("array"),
py::arg("scale_factor") = 1.)
.def("assembleToLumpedMatrix", &DOFManager::assembleToLumpedMatrix,
py::arg("dof_id"), py::arg("array_to_assemble"),
py::arg("lumped_mtx"), py::arg("scale_factor") = 1.)
.def("assemblePreassembledMatrix",
&DOFManager::assemblePreassembledMatrix, py::arg("matrix_id"),
py::arg("terms"))
.def("zeroResidual", &DOFManager::zeroResidual)
.def("globalToLocalEquationNumber",
&DOFManager::globalToLocalEquationNumber)
.def("localToGlobalEquationNumber",
&DOFManager::localToGlobalEquationNumber)
.def("getLocalEquationsNumbers", &DOFManager::getLocalEquationsNumbers)
.def("getGlobalBlockedDOFs", &DOFManager::getGlobalBlockedDOFs)
.def("updateGlobalBlockedDofs", &DOFManager::updateGlobalBlockedDofs)
.def("getTimeStepSolver", &DOFManager::getTimeStepSolver,
py::return_value_policy::reference)
.def(
"assembleElementalArrayLocalArray",
[](DOFManager & self, const Array<Real> & elementary_vect,
Array<Real> & array_assembeled, ElementType type,
GhostType ghost_type, Real scale_factor) {
self.assembleElementalArrayLocalArray(elementary_vect,
array_assembeled, type,
ghost_type, scale_factor);
},
py::arg("elementary_vect"), py::arg("array_assembeled"),
py::arg("type"), py::arg("ghost_type") = _not_ghost,
py::arg("scale_factor") = 1.)
.def(
"assembleToGlobalArray",
[](DOFManager & self, const ID & dof_id,
const Array<Real> & array_to_assemble, SolverVector & global_array,
Real scale_factor) {
self.assembleToGlobalArray(dof_id, array_to_assemble, global_array,
scale_factor);
},
py::arg("dof_id"), py::arg("array_to_assemble"),
py::arg("global_array"), py::arg("scale_factor") = 1.)
.def("getNewGlobalVector", &DOFManager::getNewGlobalVector,
py::return_value_policy::reference);
py::class_<NonLinearSolver>(mod, "NonLinearSolver")
.def(
"set",
[](NonLinearSolver & self, const std::string & id, const Real & val) {
if (id == "max_iterations") {
self.set(id, int(val));
} else {
self.set(id, val);
}
})
.def("set",
[](NonLinearSolver & self, const std::string & id,
const SolveConvergenceCriteria & val) { self.set(id, val); })
.def("getNbIterations",
[](NonLinearSolver & self) -> Int {
return self.get("nb_iterations");
})
.def("getConvergenceStatus",
[](NonLinearSolver & self) -> bool { return self.get("converged"); })
.def("getError",
[](NonLinearSolver & self) -> Real { return self.get("error"); });
py::class_<TimeStepSolver>(mod, "TimeStepSolver")
.def(
"assembleResidual",
[](TimeStepSolver & self, const ID & residual_part) {
self.assembleResidual(residual_part);
},
py::arg("residual_part"))
.def("assembleResidual",
[](TimeStepSolver & self) { self.assembleResidual(); })
.def(
"assembleResidual",
[](TimeStepSolver & self, SolverCallback & solver_callback,
const ID & residual_part) {
self.assembleResidual(solver_callback, residual_part);
},
py::arg("solver_callback"), py::arg("residual_part"))
.def(
"assembleResidual",
[](TimeStepSolver & self, SolverCallback & solver_callback) {
self.assembleResidual(solver_callback);
},
py::arg("solver_callback"))
.def("getIntegrationScheme", &TimeStepSolver::getIntegrationScheme);
py::class_<SolverCallback, PySolverCallback>(mod, "SolverCallback")
.def(py::init_alias<DOFManager &>())
.def("getMatrixType", &SolverCallback::getMatrixType)
.def("assembleMatrix", &SolverCallback::assembleMatrix)
.def("assembleLumpedMatrix", &SolverCallback::assembleLumpedMatrix)
.def("assembleResidual",
[](SolverCallback & self) { self.assembleResidual(); })
.def("predictor", &SolverCallback::predictor)
.def("corrector", &SolverCallback::corrector)
.def("beforeSolveStep", &SolverCallback::beforeSolveStep)
.def("afterSolveStep", &SolverCallback::afterSolveStep)
.def_property_readonly("dof_manager", &SolverCallback::getSCDOFManager,
py::return_value_policy::reference);
py::class_<InterceptSolverCallback, SolverCallback,
PyInterceptSolverCallback>(mod, "InterceptSolverCallback")
.def(py::init_alias<SolverCallback &>());
}
} // namespace akantu
c4science · Help