Set and Get Model Type

RFEM/enums.py

@@ -698,6 +698,13 @@ class export_to_ifc_export_type(Enum):
 
     E_EXPORT_IFC4_REFERENCE_VIEW, E_EXPORT_IFC4_STRUCTURAL_ANALYSIS_VIEW = range(2)
 
+class ModelType(Enum):
+    '''
+    Model Type | Enum
+    '''
+    E_MODEL_TYPE_1D_X_3D, E_MODEL_TYPE_1D_X_AXIAL, E_MODEL_TYPE_2D_XY_3D, \
+    E_MODEL_TYPE_2D_XY_PLATE, E_MODEL_TYPE_2D_XZ_3D, E_MODEL_TYPE_2D_XZ_PLANE_STRAIN, \
+    E_MODEL_TYPE_2D_XZ_PLANE_STRESS, E_MODEL_TYPE_3D = range(8)
 class ModalSolutionMethod(Enum):
     '''
     Modal Analysis Settings Solution Method

RFEM/initModel.py

@@ -144,6 +144,8 @@
     cModel.service.reset()
 else:
     new = client.service.new_model('My Model') + 'wsdl'
+    # If a new model is created, I thought we can create an input list for the user to define the model type
+    # new = client.service.set_model_type('E_MODEL_TYPE_2D_XZ_PLANE_STRESS')
     cModel = Client(new, transport=trans)
 
 # Init client model
@@ -297,4 +299,31 @@ def ParseCSVResultsFromSelectedFileToDict(filePath: str):
 
 def ParseXMLResultsFromSelectedFileToDict(filePath: str):
 
-    return __parseXMLAsDictionary(filePath)
+    return __parseXMLAsDictionary(filePath)
+
+def SetModelType(model_type = ModelType.E_MODEL_TYPE_3D):
+
+    '''
+    This method sets the model type. The model type is E_MODEL_TYPE_3D by default.
+
+    Args:
+        model_type (enum): The available model types are listed below.
+            ModelType.E_MODEL_TYPE_1D_X_3D
+            ModelType.E_MODEL_TYPE_1D_X_AXIAL
+            ModelType.E_MODEL_TYPE_2D_XY_3D
+            ModelType.E_MODEL_TYPE_2D_XY_PLATE
+            ModelType.E_MODEL_TYPE_2D_XZ_3D
+            ModelType.E_MODEL_TYPE_2D_XZ_PLANE_STRAIN
+            ModelType.E_MODEL_TYPE_2D_XZ_PLANE_STRESS
+            ModelType.E_MODEL_TYPE_3D
+    '''
+
+    clientModel.service.set_model_type(model_type.name)
+
+def GetModelType():
+
+    '''
+    The method returns a string of the current model type.
+    '''
+
+    return clientModel.service.get_model_type()