def Initialize(self): super().Initialize() self.co_sim_settings = self.project_parameters["co_sim_settings"] self.is_strong_coupling = self.co_sim_settings[ "is_strong_coupling"].GetBool() connection_settings = CoSimIO.InfoFromParameters( self.project_parameters["co_sim_settings"]["io_settings"]) info = CoSimIO.Connect(connection_settings) self.connection_name = info.GetString("connection_name") if info.GetInt( "connection_status") != CoSimIO.ConnectionStatus.Connected: raise Exception("Connecting failed!") self.communication_settings = self.co_sim_settings[ "communication_settings"] # Exporting meshes to CoSimulation for model_part_name in self.communication_settings[ "export_meshes"].GetStringArray(): info = CoSimIO.Info() info.SetString("connection_name", self.connection_name) info.SetString("identifier", model_part_name.replace(".", "-")) CoSimIO.ExportMesh(info, self.model[model_part_name])
def __ExportImportDataOnModelPart(self, model_part, var_export, var_import, data_location): p = self.__RunPythonInSubProcess("import_export_data") connection_settings = CoSimIO.Info() connection_settings.SetString("my_name", "ExpImp") connection_settings.SetString("connect_to", "impExp") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) connection_name = info.GetString("connection_name") self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) export_info = CoSimIO.Info() export_info.SetString("connection_name", connection_name) export_info.SetString("identifier", "data_exchange_1") CoSimIO.ExportData(export_info, model_part, var_export, data_location) import_info = CoSimIO.Info() import_info.SetString("connection_name", connection_name) import_info.SetString("identifier", "data_exchange_2") CoSimIO.ImportData(import_info, model_part, var_import, data_location) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", connection_name) info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected) self.__CheckSubProcess(p)
def test_Export_Import_Data_raw_values(self): connection_settings = CoSimIO.Info() connection_settings.SetString("connection_name", "im_exp_data") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) values = [1.0, 2.5, 3.3, -9.4] export_info = CoSimIO.Info() export_info.SetString("connection_name", "im_exp_data") export_info.SetString("identifier", "data_exchange_1") CoSimIO.ExportData(export_info, values) RunPythonInSubProcess("import_export_data") import_info = CoSimIO.Info() import_info.SetString("connection_name", "im_exp_data") import_info.SetString("identifier", "data_exchange_2") imported_values = CoSimIO.ImportData(import_info) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", "im_exp_data") info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected) # checking the values after disconnecting to avoid deadlock self.assertVectorAlmostEqual(KM.Vector(values), KM.Vector(imported_values))
def test_Export_Import_Mesh(self): model = KM.Model() model_part = model.CreateModelPart("for_test") model_part_returned = model.CreateModelPart("for_test_2") for i in range(15): model_part.CreateNewNode(i + 1, i * 1.1, 1.1**i, 0.0) props = model_part.CreateNewProperties(0) for i in range(9): # this leaves some hanging nodes model_part.CreateNewElement("Element2D2N", i + 1, [i + 1, i + 2], props) connection_settings = CoSimIO.Info() connection_settings.SetString("connection_name", "im_exp_mesh") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) export_info = CoSimIO.Info() export_info.SetString("connection_name", "im_exp_mesh") export_info.SetString("identifier", "mesh_exchange_1") CoSimIO.ExportMesh(export_info, model_part) RunPythonInSubProcess("import_export_mesh") import_info = CoSimIO.Info() import_info.SetString("connection_name", "im_exp_mesh") import_info.SetString("identifier", "mesh_exchange_2") CoSimIO.ImportMesh(import_info, model_part_returned) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", "im_exp_mesh") info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected) # checking the values after disconnecting to avoid deadlock self.assertEqual(model_part.NumberOfNodes(), model_part_returned.NumberOfNodes()) self.assertEqual(model_part.NumberOfElements(), model_part_returned.NumberOfElements()) for node_orig, node_exchanged in zip(model_part.Nodes, model_part_returned.Nodes): self.assertAlmostEqual(node_orig.X0, node_exchanged.X0) self.assertAlmostEqual(node_orig.Y0, node_exchanged.Y0) self.assertAlmostEqual(node_orig.Z0, node_exchanged.Z0) for elem_orig, elem_exchanged in zip(model_part.Elements, model_part_returned.Elements): self.assertEqual(len(elem_orig.GetNodes()), len(elem_exchanged.GetNodes())) for node_orig, node_exchanged in zip(elem_orig.GetNodes(), elem_exchanged.GetNodes()): self.assertAlmostEqual(node_orig.X0, node_exchanged.X0) self.assertAlmostEqual(node_orig.Y0, node_exchanged.Y0) self.assertAlmostEqual(node_orig.Z0, node_exchanged.Z0)
def __init__(self, settings, model, solver_name): super(KratosCoSimIO, self).__init__(settings, model, solver_name) connection_settings = CoSimIO.InfoFromParameters(self.settings) connection_settings.SetString("connection_name", solver_name) info = CoSimIO.Connect(connection_settings) if info.GetInt( "connection_status") != CoSimIO.ConnectionStatus.Connected: raise Exception("Connecting failed!")
def test_Connect_Disconnect(self): connection_settings = CoSimIO.Info() connection_settings.SetString("connection_name", "c_d_test") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) RunPythonInSubProcess("connect_disconnect") disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", "c_d_test") info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected)
def __init__(self, settings, model, solver_name): super().__init__(settings, model, solver_name) connect_to = self.settings["connect_to"].GetString() if connect_to == "": raise Exception('"connect_to" must be specified!') connection_settings = CoSimIO.InfoFromParameters(self.settings) connection_settings.SetString("my_name", solver_name) info = CoSimIO.Connect(connection_settings) if info.GetInt( "connection_status") != CoSimIO.ConnectionStatus.Connected: raise Exception("Connecting failed!") self.connection_name = info.GetString("connection_name")
def test_Connect_Disconnect(self): p = self.__RunPythonInSubProcess("connect_disconnect") connection_settings = CoSimIO.Info() connection_settings.SetString("my_name", "partner_a") connection_settings.SetString("connect_to", "partner_b") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) connection_name = info.GetString("connection_name") self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", connection_name) info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected) self.__CheckSubProcess(p)
def test_Export_Import_Data_raw_values(self): p = self.__RunPythonInSubProcess("import_export_data") connection_settings = CoSimIO.Info() connection_settings.SetString("my_name", "ExpImp") connection_settings.SetString("connect_to", "impExp") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) connection_name = info.GetString("connection_name") self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Connected) values = CoSimIO.DoubleVector([1.0, 2.5, 3.3, -9.4]) export_info = CoSimIO.Info() export_info.SetString("connection_name", connection_name) export_info.SetString("identifier", "data_exchange_1") CoSimIO.ExportData(export_info, values) import_info = CoSimIO.Info() import_info.SetString("connection_name", connection_name) import_info.SetString("identifier", "data_exchange_2") imported_values = CoSimIO.DoubleVector() CoSimIO.ImportData(import_info, imported_values) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", connection_name) info = CoSimIO.Disconnect(disconnect_settings) self.assertEqual(info.GetInt("connection_status"), CoSimIO.ConnectionStatus.Disconnected) # checking the values after disconnecting to avoid deadlock self.assertVectorAlmostEqual(values, imported_values) self.__CheckSubProcess(p)
def Initialize(self): super().Initialize() connection_settings = CoSimIO.InfoFromParameters( self.project_parameters["co_sim_settings"]["io_settings"]) info = CoSimIO.Connect(connection_settings) self.connection_name = info.GetString("connection_name") if info.GetInt( "connection_status") != CoSimIO.ConnectionStatus.Connected: raise Exception("Connecting failed!") ## specifying which fields belong to which identifier when im-/exporting data self.data_comm_settings = {} for field_settings in chain( self.project_parameters["co_sim_settings"] ["communication_settings"]["export_fields"], self.project_parameters["co_sim_settings"] ["communication_settings"]["import_fields"]): identifier = field_settings["identifier"].GetString() model_part_name = field_settings["model_part_name"].GetString() model_part = self.model[model_part_name] variable_name = field_settings["variable_name"].GetString() variable = KM.KratosGlobals.GetVariable(variable_name) self.data_comm_settings[identifier] = { "model_part": model_part, "variable": variable } # declaring in place the functions that are registered by the CoSimIO def _ImportData(info): identifier = info.GetString("identifier") model_part = self.data_comm_settings[identifier]["model_part"] variable = self.data_comm_settings[identifier]["variable"] info = CoSimIO.Info() info.SetString("connection_name", self.connection_name) info.SetString("identifier", identifier) CoSimIO.ImportData(info, model_part, variable, CoSimIO.DataLocation.NodeHistorical) return CoSimIO.Info() def _ExportData(info): identifier = info.GetString("identifier") model_part = self.data_comm_settings[identifier]["model_part"] variable = self.data_comm_settings[identifier]["variable"] info = CoSimIO.Info() info.SetString("connection_name", self.connection_name) info.SetString("identifier", identifier) CoSimIO.ExportData(info, model_part, variable, CoSimIO.DataLocation.NodeHistorical) return CoSimIO.Info() def _ExportMesh(info): model_part_name = info.GetString("identifier") export_info = CoSimIO.Info() export_info.SetString("connection_name", self.connection_name) export_info.SetString("identifier", model_part_name.replace(".", "-")) CoSimIO.ExportMesh(export_info, self.model[model_part_name]) return CoSimIO.Info() def _AdvanceInTime(info): current_time = info.GetDouble("current_time") self.time = self._GetSolver().AdvanceInTime(current_time) return CoSimIO.Info() def _InitializeSolutionStep(info): self.InitializeSolutionStep() return CoSimIO.Info() def _Predict(info): self._GetSolver().Predict() return CoSimIO.Info() def _SolveSolutionStep(info): self._GetSolver().SolveSolutionStep() return CoSimIO.Info() def _FinalizeSolutionStep(info): self.FinalizeSolutionStep() return CoSimIO.Info() def _OutputSolutionStep(info): self.OutputSolutionStep() return CoSimIO.Info() register_info = CoSimIO.Info() register_info.SetString("connection_name", self.connection_name) register_info.SetString("function_name", "ImportData") CoSimIO.Register(register_info, _ImportData) register_info.SetString("function_name", "ExportData") CoSimIO.Register(register_info, _ExportData) register_info.SetString("function_name", "ExportMesh") CoSimIO.Register(register_info, _ExportMesh) register_info.SetString("function_name", "AdvanceInTime") CoSimIO.Register(register_info, _AdvanceInTime) register_info.SetString("function_name", "InitializeSolutionStep") CoSimIO.Register(register_info, _InitializeSolutionStep) register_info.SetString("function_name", "Predict") CoSimIO.Register(register_info, _Predict) register_info.SetString("function_name", "SolveSolutionStep") CoSimIO.Register(register_info, _SolveSolutionStep) register_info.SetString("function_name", "FinalizeSolutionStep") CoSimIO.Register(register_info, _FinalizeSolutionStep) register_info.SetString("function_name", "OutputSolutionStep") CoSimIO.Register(register_info, _OutputSolutionStep)
from KratosMultiphysics.CoSimulationApplication import CoSimIO connection_settings = CoSimIO.Info() connection_settings.SetString("my_name", "impExp") connection_settings.SetString("connect_to", "ExpImp") connection_settings.SetInt("echo_level", 0) info = CoSimIO.Connect(connection_settings) connection_name = info.GetString("connection_name") if info.GetInt("connection_status") != CoSimIO.ConnectionStatus.Connected: raise Exception("Connecting failed") import_info = CoSimIO.Info() import_info.SetString("connection_name", connection_name) import_info.SetString("identifier", "data_exchange_1") imported_values = CoSimIO.DoubleVector() CoSimIO.ImportData(import_info, imported_values) # print(imported_values) export_info = CoSimIO.Info() export_info.SetString("connection_name", connection_name) export_info.SetString("identifier", "data_exchange_2") CoSimIO.ExportData(export_info, imported_values) disconnect_settings = CoSimIO.Info() disconnect_settings.SetString("connection_name", connection_name) info = CoSimIO.Disconnect(disconnect_settings) if info.GetInt("connection_status") != CoSimIO.ConnectionStatus.Disconnected: raise Exception("Disconnecting failed")