[RomApp] Testing (with) RomManager

applications/RomApplication/python_scripts/rom_manager.py

@@ -11,19 +11,26 @@
 
 class RomManager(object):
 
-    def __init__(self,project_parameters_name, general_rom_manager_parameters, CustomizeSimulation, UpdateProjectParameters,UpdateMaterialParametersFile):
+    def __init__(self,project_parameters_name="ProjectParameters.json", general_rom_manager_parameters=None, CustomizeSimulation=None, UpdateProjectParameters=None,UpdateMaterialParametersFile=None):
         #FIXME:
         # - Use a method (upcoming) for smothly retrieving solutions. In here we are using the RomBasisOutput process in order to store the solutions
         # - There is some redundancy between the methods that launch the simulations. Can we create a single method?
         # - We must use the Hyper-Reduced model part for the HROM simulations. (The one in which we have eliminated the non selected elements and conditions)
         # - Not yet paralellised with COMPSs
         self.project_parameters_name = project_parameters_name
-        self.general_rom_manager_parameters = general_rom_manager_parameters
-        self.rom_training_parameters = self._SetRomTrainingParameters()
-        self.hrom_training_parameters = self.SetHromTrainingParameters()
-        self.CustomizeSimulation = CustomizeSimulation
-        self.UpdateProjectParameters = UpdateProjectParameters
-        self.UpdateMaterialParametersFile = UpdateMaterialParametersFile
+        self._SetUpRomManagerParameters(general_rom_manager_parameters)
+        if CustomizeSimulation is None:
+            self.CustomizeSimulation = self.DefaultCustomizeSimulation
+        else:
+            self.CustomizeSimulation = CustomizeSimulation
+        if UpdateProjectParameters is None:
+            self.UpdateProjectParameters = self.DefaultUpdateProjectParameters
+        else:
+            self.UpdateProjectParameters = UpdateProjectParameters
+        if UpdateMaterialParametersFile is None:
+            self.UpdateMaterialParametersFile = self.DefaultUpdateMaterialParametersFile
+        else:
+            self.UpdateMaterialParametersFile = UpdateMaterialParametersFile
         self.SetUpQuantityOfInterestContainers()
 
 
@@ -693,6 +700,93 @@ def _StoreNoResults(self, parameters):
 
 
 
+    def _SetUpRomManagerParameters(self, input):
+
+        if input is None:
+            input = KratosMultiphysics.Parameters()
+        self.general_rom_manager_parameters = input
+
+
+        default_settings = KratosMultiphysics.Parameters("""{
+            "rom_stages_to_train" : ["ROM","HROM"],             // ["ROM","HROM"]
+            "rom_stages_to_test" : [],              // ["ROM","HROM"]
+            "paralellism" : null,                        // null, TODO: add "compss"
+            "projection_strategy": "galerkin",            // "lspg", "galerkin", "petrov_galerkin"
+            "assembling_strategy": "global",            // "global", "elemental"
+            "save_gid_output": false,                    // false, true #if true, it must exits previously in the ProjectParameters.json
+            "save_vtk_output": false,                    // false, true #if true, it must exits previously in the ProjectParameters.json
+            "output_name": "id",                         // "id" , "mu"
+            "ROM":{
+                "svd_truncation_tolerance": 1e-5,
+                "model_part_name": "Structure",                            // This changes depending on the simulation: Structure, FluidModelPart, ThermalPart #TODO: Idenfity it automatically
+                "nodal_unknowns": ["DISPLACEMENT_X","DISPLACEMENT_Y"],     // Main unknowns. Snapshots are taken from these
+                "rom_basis_output_format": "numpy",
+                "rom_basis_output_name": "RomParameters",
+                "rom_basis_output_folder": "rom_data",
+                "snapshots_control_type": "step",                          // "step", "time"
+                "snapshots_interval": 1,
+                "galerkin_rom_bns_settings": {
+                    "monotonicity_preserving": false
+                },
+                "lspg_rom_bns_settings": {
+                    "train_petrov_galerkin": false,
+                    "basis_strategy": "residuals",                        // 'residuals', 'jacobian', 'reactions'
+                    "include_phi": false,
+                    "svd_truncation_tolerance": 1e-12,
+                    "solving_technique": "normal_equations",              // 'normal_equations', 'qr_decomposition'
+                    "monotonicity_preserving": false
+                },
+                "petrov_galerkin_rom_bns_settings": {
+                    "monotonicity_preserving": false
+                }
+            },
+            "HROM":{
+                "element_selection_type": "empirical_cubature",
+                "element_selection_svd_truncation_tolerance": 0,
+                "create_hrom_visualization_model_part" : true,
+                "echo_level" : 0
+            }
+        }""")
+
+        self.general_rom_manager_parameters.RecursivelyValidateAndAssignDefaults(default_settings)
+
+        self.rom_training_parameters = self._SetRomTrainingParameters()
+        self.hrom_training_parameters = self.SetHromTrainingParameters()
+
+
+    def DefaultCustomizeSimulation(self, cls, global_model, parameters):
+        # Default function that does nothing special
+        class DefaultCustomSimulation(cls):
+            def __init__(self, model, project_parameters):
+                super().__init__(model, project_parameters)
+
+            def Initialize(self):
+                super().Initialize()
+
+            def FinalizeSolutionStep(self):
+                super().FinalizeSolutionStep()
+
+            def CustomMethod(self):
+                pass  # Do nothing special
+
+        return DefaultCustomSimulation(global_model, parameters)
+
+
+    def DefaultUpdateProjectParameters(self, parameters, mu=None):
+        return parameters
+
+    def DefaultUpdateMaterialParametersFile(self, material_parametrs_file_name=None, mu=None):
+        pass
+        # with open(material_parametrs_file_name, mode="r+") as f:
+        #     data = json.load(f)
+        #     #change the angles of 1st and 2nd layer
+        #     data["properties"][0]["Material"]["Variables"]["EULER_ANGLES"][0] = mu[0]
+        #     data["properties"][1]["Material"]["Variables"]["EULER_ANGLES"][0] = mu[1]
+        #     #write to file and save file
+        #     f.seek(0)
+        #     json.dump(data, f, indent=4)
+        #     f.truncate()
+
     def _SetRomTrainingParameters(self):
         defaults = self._GetDefaulRomBasisOutputParameters()
         defaults["Parameters"]["rom_manager"].SetBool(True)  # Set the flag to true when inside the RomManager to trigger particular behavior for multiple parameters

applications/RomApplication/tests/structural_static_test_files/HROM/rom_data/RomParameters.json

@@ -2,18 +2,28 @@
     "rom_manager": false,
     "train_hrom": false,
     "run_hrom": true,
-    "rom_format":"numpy",
+    "rom_format": "numpy",
     "projection_strategy": "galerkin",
     "rom_settings": {
         "nodal_unknowns": [
             "DISPLACEMENT_X",
             "DISPLACEMENT_Y"
         ],
         "number_of_rom_dofs": 2,
-        "rom_bns_settings": {}
+        "rom_bns_settings": {
+            "monotonicity_preserving": false
+        }
     },
     "hrom_settings": {
-        "hrom_format":"numpy"
+        "hrom_format": "numpy",
+        "element_selection_type": "empirical_cubature",
+        "element_selection_svd_truncation_tolerance": 0.0,
+        "create_hrom_visualization_model_part": true,
+        "echo_level": 0,
+        "include_condition_parents": false,
+        "initial_candidate_elements_model_part_list": [],
+        "initial_candidate_conditions_model_part_list": [],
+        "constraint_sum_weights": true
     },
     "nodal_modes": {
         "1": [
@@ -836,5 +846,6 @@
             "22": 47.114266621501415,
             "3": 0.885733378498582
         }
-    }
+    },
+    "assembling_strategy": "global"
 }

applications/RomApplication/tests/structural_static_test_files/ROM/rom_data/RomParameters.json

@@ -2,17 +2,28 @@
     "rom_manager": false,
     "train_hrom": false,
     "run_hrom": false,
-    "rom_format":"numpy",
+    "rom_format": "numpy",
     "projection_strategy": "galerkin",
     "rom_settings": {
         "nodal_unknowns": [
             "DISPLACEMENT_X",
             "DISPLACEMENT_Y"
         ],
         "number_of_rom_dofs": 2,
-        "rom_bns_settings": {}
+        "rom_bns_settings": {
+            "monotonicity_preserving": false
+        }
+    },
+    "hrom_settings": {
+        "element_selection_type": "empirical_cubature",
+        "element_selection_svd_truncation_tolerance": 0.0,
+        "create_hrom_visualization_model_part": true,
+        "echo_level": 0,
+        "include_condition_parents": false,
+        "initial_candidate_elements_model_part_list": [],
+        "initial_candidate_conditions_model_part_list": [],
+        "constraint_sum_weights": true
     },
-    "hrom_settings": {},
     "nodal_modes": {
         "1": [
             [
@@ -833,5 +844,6 @@
             "13": 0.054464904205207665,
             "22": 0.000573026145484772
         }
-    }
+    },
+    "assembling_strategy": "global"
 }

applications/RomApplication/tests/test_RomApplication.py

@@ -22,6 +22,7 @@
 from test_monotonicity_preserving_rom import TestMonotonicityPreservingRom
 from test_save_rom_coefficients_process import TestSaveRomCoefficientsProcess
 from test_hrom_training_utility_rom import TestHromTrainingUtilityRom
+from test_rom_manager import TestRomManager
 
 def AssembleTestSuites():
     ''' Populates the test suites to run.
@@ -57,6 +58,9 @@ def AssembleTestSuites():
     smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestThermalPGRom]))
     smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestStructuralPGRom]))
     smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestMonotonicityPreservingRom]))
+    smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestSaveRomCoefficientsProcess]))
+    smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestHromTrainingUtilityRom]))
+    smallSuite.addTests(KratosUnittest.TestLoader().loadTestsFromTestCases([TestRomManager]))
 
     # - testNightly
     nightlySuite = suites['nightly']