[GeoMechanicsApplication] Add support to give list of modelpart names to "ApplyFinalStressesOfPreviousStageToInitialState" process (#13907)

markelov208 · web-flow · commit 757588215044 · 2025-10-23T19:01:45.000+02:00
* added model_part_name_list

* corrected ApplyExcavationProcess Factory

* replaced "rest" with "use_input_model_part"

* extracted ProcessUtilities::CheckAndReturnModelPartNames

* added unit tests for ProcessUtilities::GetModelPartsFromSettings

* used a mdpa file name

* removed unused auto&amp; r_model_part =

* added ApplyExcavationProcess::Info()

* added parametrized gtests

* reverted use_input_model_part

* fix code smells

* removed unused using

* added modelpart name list

* removed unused auto&amp; model_part =

* used a single line definition for parameters

* response to review
diff --git a/applications/GeoMechanicsApplication/custom_processes/apply_final_stresses_of_previous_stage_to_initial_state.cpp b/applications/GeoMechanicsApplication/custom_processes/apply_final_stresses_of_previous_stage_to_initial_state.cpp
@@ -11,6 +11,8 @@
 //
 
 #include "apply_final_stresses_of_previous_stage_to_initial_state.h"
+#include "containers/model.h"
+#include "custom_utilities/process_utilities.h"
 #include "includes/initial_state.h"
 #include "includes/kratos_parameters.h"
 #include "includes/model_part.h"
@@ -22,44 +24,51 @@
 namespace Kratos
 {
 
-ApplyFinalStressesOfPreviousStageToInitialState::ApplyFinalStressesOfPreviousStageToInitialState(ModelPart& rModelPart,
-                                                                                                 const Parameters&)
-    : mrModelPart(rModelPart)
+ApplyFinalStressesOfPreviousStageToInitialState::ApplyFinalStressesOfPreviousStageToInitialState(
+    Model& rModel, const Parameters& rProcessSettings)
 {
+    mrModelParts = ProcessUtilities::GetModelPartsFromSettings(
+        rModel, rProcessSettings, ApplyFinalStressesOfPreviousStageToInitialState::Info());
 }
 
 void ApplyFinalStressesOfPreviousStageToInitialState::ExecuteInitialize()
 {
-    block_for_each(mrModelPart.Elements(), [this](Element& rElement) {
-        std::vector<Vector> stresses_on_integration_points;
-        rElement.CalculateOnIntegrationPoints(PK2_STRESS_VECTOR, stresses_on_integration_points,
-                                              mrModelPart.GetProcessInfo());
-        if (stresses_on_integration_points.empty()) {
-            rElement.CalculateOnIntegrationPoints(
-                CAUCHY_STRESS_VECTOR, stresses_on_integration_points, mrModelPart.GetProcessInfo());
-        }
-        std::vector<ConstitutiveLaw::Pointer> constitutive_laws;
-        rElement.CalculateOnIntegrationPoints(CONSTITUTIVE_LAW, constitutive_laws, mrModelPart.GetProcessInfo());
+    for (const auto& r_model_part : mrModelParts) {
+        block_for_each(r_model_part.get().Elements(), [&r_model_part, this](Element& rElement) {
+            std::vector<Vector> stresses_on_integration_points;
+            rElement.CalculateOnIntegrationPoints(PK2_STRESS_VECTOR, stresses_on_integration_points,
+                                                  r_model_part.get().GetProcessInfo());
+            if (stresses_on_integration_points.empty()) {
+                rElement.CalculateOnIntegrationPoints(CAUCHY_STRESS_VECTOR, stresses_on_integration_points,
+                                                      r_model_part.get().GetProcessInfo());
+            }
+            std::vector<ConstitutiveLaw::Pointer> constitutive_laws;
+            rElement.CalculateOnIntegrationPoints(CONSTITUTIVE_LAW, constitutive_laws,
+                                                  r_model_part.get().GetProcessInfo());
 
-        CheckRetrievedElementData(constitutive_laws, stresses_on_integration_points, rElement.GetId());
-        mStressesByElementId[rElement.GetId()] = stresses_on_integration_points;
-    });
+            CheckRetrievedElementData(constitutive_laws, stresses_on_integration_points, rElement.GetId());
+            mStressesByElementId[rElement.GetId()] = stresses_on_integration_points;
+        });
+    }
 }
 
 void ApplyFinalStressesOfPreviousStageToInitialState::ExecuteBeforeSolutionLoop()
 {
-    block_for_each(mrModelPart.Elements(), [this](Element& rElement) {
-        std::vector<ConstitutiveLaw::Pointer> constitutive_laws;
-        rElement.CalculateOnIntegrationPoints(CONSTITUTIVE_LAW, constitutive_laws, mrModelPart.GetProcessInfo());
-        const auto stresses_on_integration_points = mStressesByElementId.at(rElement.GetId());
-        for (auto i = std::size_t{0}; i < constitutive_laws.size(); ++i) {
-            auto p_initial_state = make_intrusive<InitialState>();
-            p_initial_state->SetInitialStressVector(stresses_on_integration_points[i]);
-            p_initial_state->SetInitialStrainVector(ZeroVector{constitutive_laws[i]->GetStrainSize()});
-            constitutive_laws[i]->SetInitialState(p_initial_state);
-            constitutive_laws[i]->InitializeMaterial(rElement.GetProperties(), rElement.GetGeometry(), {});
-        }
-    });
+    for (const auto& r_model_part : mrModelParts) {
+        block_for_each(r_model_part.get().Elements(), [&r_model_part, this](Element& rElement) {
+            std::vector<ConstitutiveLaw::Pointer> constitutive_laws;
+            rElement.CalculateOnIntegrationPoints(CONSTITUTIVE_LAW, constitutive_laws,
+                                                  r_model_part.get().GetProcessInfo());
+            const auto stresses_on_integration_points = mStressesByElementId.at(rElement.GetId());
+            for (auto i = std::size_t{0}; i < constitutive_laws.size(); ++i) {
+                auto p_initial_state = make_intrusive<InitialState>();
+                p_initial_state->SetInitialStressVector(stresses_on_integration_points[i]);
+                p_initial_state->SetInitialStrainVector(ZeroVector{constitutive_laws[i]->GetStrainSize()});
+                constitutive_laws[i]->SetInitialState(p_initial_state);
+                constitutive_laws[i]->InitializeMaterial(rElement.GetProperties(), rElement.GetGeometry(), {});
+            }
+        });
+    }
     mStressesByElementId.clear();
 }
 
diff --git a/applications/GeoMechanicsApplication/custom_processes/apply_final_stresses_of_previous_stage_to_initial_state.h b/applications/GeoMechanicsApplication/custom_processes/apply_final_stresses_of_previous_stage_to_initial_state.h
@@ -26,7 +26,7 @@ class KRATOS_API(GEO_MECHANICS_APPLICATION) ApplyFinalStressesOfPreviousStageToI
 {
 public:
     KRATOS_CLASS_POINTER_DEFINITION(ApplyFinalStressesOfPreviousStageToInitialState);
-    ApplyFinalStressesOfPreviousStageToInitialState(ModelPart& rModelPart, const Parameters&);
+    ApplyFinalStressesOfPreviousStageToInitialState(Model& rModel, const Parameters& rProcessSettings);
     ~ApplyFinalStressesOfPreviousStageToInitialState() override = default;
 
     ApplyFinalStressesOfPreviousStageToInitialState(const ApplyFinalStressesOfPreviousStageToInitialState&) = delete;
@@ -37,7 +37,7 @@ class KRATOS_API(GEO_MECHANICS_APPLICATION) ApplyFinalStressesOfPreviousStageToI
     [[nodiscard]] std::string Info() const override;
 
 private:
-    ModelPart& mrModelPart;
+    std::vector<std::reference_wrapper<ModelPart>> mrModelParts;
 
     static void CheckRetrievedElementData(const std::vector<ConstitutiveLaw::Pointer>& rConstitutiveLaws,
                                           const std::vector<Vector>& rStressesOnIntegrationPoints,
diff --git a/applications/GeoMechanicsApplication/custom_python/add_custom_processes_to_python.cpp b/applications/GeoMechanicsApplication/custom_python/add_custom_processes_to_python.cpp
@@ -187,7 +187,7 @@ void AddCustomProcessesToPython(pybind11::module& m)
 
     py::class_<ApplyFinalStressesOfPreviousStageToInitialState, ApplyFinalStressesOfPreviousStageToInitialState::Pointer, Process>(
         m, "ApplyFinalStressesOfPreviousStageToInitialState")
-        .def(py::init<ModelPart&, const Parameters&>());
+        .def(py::init<Model&, const Parameters&>());
 
     py::class_<ApplyInitialUniformStressField, ApplyInitialUniformStressField::Pointer, Process>(
         m, "ApplyInitialUniformStressField")
diff --git a/applications/GeoMechanicsApplication/python_scripts/apply_final_stresses_of_previous_stage_to_initial_state.py b/applications/GeoMechanicsApplication/python_scripts/apply_final_stresses_of_previous_stage_to_initial_state.py
@@ -6,5 +6,4 @@ def Factory(settings, model):
     if not isinstance(settings, Core.Parameters):
         raise TypeError("expected input shall be a Parameters object, encapsulating a json string")
 
-    model_part = model[settings["Parameters"]["model_part_name"].GetString()]
-    return Geo.ApplyFinalStressesOfPreviousStageToInitialState(model_part, settings["Parameters"])
+    return Geo.ApplyFinalStressesOfPreviousStageToInitialState(model, settings["Parameters"])
diff --git a/applications/GeoMechanicsApplication/tests/cpp_tests/custom_processes/test_apply_final_stresses_of_previous_stage_to_initial_state.cpp b/applications/GeoMechanicsApplication/tests/cpp_tests/custom_processes/test_apply_final_stresses_of_previous_stage_to_initial_state.cpp
@@ -117,9 +117,9 @@ KRATOS_TEST_CASE_IN_SUITE(ApplyFinalStressesOfPreviousStageToInitialState_SetsIn
         PK2_STRESS_VECTOR, std::vector<Vector>(number_of_integration_points, initial_stress_vector),
         dummy_process_info);
 
-    const auto dummy_parameters = Parameters{};
+    const auto parameters = Parameters{R"({"model_part_name" : "MainModelPart"})"};
     ApplyFinalStressesOfPreviousStageToInitialState apply_final_stresses_of_previous_stage_to_initial_state(
-        model_part, dummy_parameters);
+        model, parameters);
 
     // This is also the order in which these functions are called in an analysis
     // which is why we emulate exactly this order here.
@@ -146,8 +146,8 @@ KRATOS_TEST_CASE_IN_SUITE(ApplyFinalStressesOfPreviousStageToInitialState_Throws
     Model model;
     auto& model_part = CreateModelPartWithAStubElement(model);
 
-    const auto                                      dummy_parameters = Parameters{};
-    ApplyFinalStressesOfPreviousStageToInitialState process(model_part, dummy_parameters);
+    const auto parameters = Parameters{R"({"model_part_name" : "MainModelPart"})"};
+    ApplyFinalStressesOfPreviousStageToInitialState process(model, parameters);
 
     const auto dummy_process_info          = ProcessInfo{};
     const auto empty_constitutive_law_list = std::vector<ConstitutiveLaw::Pointer>{};
@@ -162,10 +162,10 @@ KRATOS_TEST_CASE_IN_SUITE(ApplyFinalStressesOfPreviousStageToInitialState_Throws
                           KratosGeoMechanicsFastSuiteWithoutKernel)
 {
     Model model;
-    auto& model_part = CreateModelPartWithAStubElement(model);
+    CreateModelPartWithAStubElement(model);
 
-    const auto                                      dummy_parameters = Parameters{};
-    ApplyFinalStressesOfPreviousStageToInitialState process(model_part, dummy_parameters);
+    const auto parameters = Parameters{R"({"model_part_name_list" : ["MainModelPart"]})"};
+    ApplyFinalStressesOfPreviousStageToInitialState process(model, parameters);
 
     KRATOS_EXPECT_EXCEPTION_IS_THROWN(
         process.ExecuteInitialize(),
@@ -181,8 +181,8 @@ KRATOS_TEST_CASE_IN_SUITE(ApplyFinalStressesOfPreviousStageToInitialState_Throws
     model_part.Elements()[1].SetValuesOnIntegrationPoints(
         PK2_STRESS_VECTOR, std::vector<Vector>(2, ScalarVector(4, 1.0)), dummy_process_info);
 
-    const auto                                      dummy_parameters = Parameters{};
-    ApplyFinalStressesOfPreviousStageToInitialState process(model_part, dummy_parameters);
+    const auto parameters = Parameters{R"({"model_part_name_list" : ["MainModelPart"]})"};
+    ApplyFinalStressesOfPreviousStageToInitialState process(model, parameters);
 
     KRATOS_EXPECT_EXCEPTION_IS_THROWN(process.ExecuteInitialize(),
                                       "Number of retrieved stress vectors (2) does not match the "
diff --git a/applications/GeoMechanicsApplication/tests/cpp_tests/custom_utilities/test_process_utilities.cpp b/applications/GeoMechanicsApplication/tests/cpp_tests/custom_utilities/test_process_utilities.cpp
@@ -11,6 +11,7 @@
 //
 
 #include "custom_processes/apply_excavation_process.h"
+#include "custom_processes/apply_final_stresses_of_previous_stage_to_initial_state.h"
 #include "custom_processes/apply_k0_procedure_process.h"
 #include "custom_utilities/process_utilities.h"
 #include "testing/testing.h"
@@ -115,8 +116,12 @@ static const std::vector<NamedProcessFactory> kProcessFactories = {
      [](Model& rModel, const Parameters& rSettings) {
     return std::make_unique<Kratos::ApplyExcavationProcess>(rModel, rSettings);
 }},
-    {"ApplyK0ProcedureProcess", [](Model& rModel, const Parameters& rSettings) {
+    {"ApplyK0ProcedureProcess",
+     [](Model& rModel, const Parameters& rSettings) {
     return std::make_unique<Kratos::ApplyK0ProcedureProcess>(rModel, rSettings);
+}},
+    {"ApplyFinalStressesOfPreviousStageToInitialState", [](Model& rModel, const Parameters& rSettings) {
+    return std::make_unique<Kratos::ApplyFinalStressesOfPreviousStageToInitialState>(rModel, rSettings);
 }}};
 
 INSTANTIATE_TEST_SUITE_P(ProcessUtilitiesTests, ProcessWithModelPartsTest, ::testing::ValuesIn(kProcessFactories));
