Skip to content

Commit

Permalink
define python holder type for ModelPart::SubModelPartsContainerType
Browse files Browse the repository at this point in the history
  • Loading branch information
matekelemen committed Mar 6, 2024
1 parent 50d711e commit 5623ccc
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions kratos/python/add_model_part_to_python.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -737,7 +737,7 @@ void AddModelPartToPython(pybind11::module& m)

namespace py = pybind11;

py::class_<typename ModelPart::SubModelPartsContainerType >(m, "SubModelPartsContainerType")
py::class_<ModelPart::SubModelPartsContainerType, ModelPart::SubModelPartsContainerType::Pointer>(m, "SubModelPartsContainerType")
.def("__iter__", [](typename ModelPart::SubModelPartsContainerType& self){ return py::make_iterator(self.begin(), self.end());}, py::keep_alive<0,1>())
;

Expand Down Expand Up @@ -917,8 +917,7 @@ void AddModelPartToPython(pybind11::module& m)
.def("GetParentModelPart", [](ModelPart& self) -> ModelPart& {return self.GetParentModelPart();}, py::return_value_policy::reference_internal)
.def("GetRootModelPart", [](ModelPart& self) -> ModelPart& {return self.GetRootModelPart();}, py::return_value_policy::reference_internal)
.def("GetModel", [](ModelPart& self) -> Model& {return self.GetModel();}, py::return_value_policy::reference_internal)
.def_property("SubModelParts", [](ModelPart& self){ return self.SubModelParts(); },
[](ModelPart& self, ModelPart::SubModelPartsContainerType& subs){ KRATOS_ERROR << "setting submodelparts is not allowed"; })
.def_property_readonly("SubModelParts", [](ModelPart& self) -> ModelPart::SubModelPartsContainerType::Pointer {return std::make_shared<ModelPart::SubModelPartsContainerType>(self.SubModelParts());})
.def_property_readonly("MasterSlaveConstraints", ModelPartGetMasterSlaveConstraints1)
.def("GetHistoricalVariablesNames", [](ModelPart& rModelPart) -> std::unordered_set<std::string> {
std::unordered_set<std::string> variable_names;
Expand Down

0 comments on commit 5623ccc

Please sign in to comment.