You can now view / add / edit / remove structural boundary conditions

2026-09-24 08:56:52 +00:00 · 2021-03-19 10:55:52 +11:00
parent 0f365cadbd
commit 8b0a7821b1
8 changed files with 267 additions and 3 deletions
@@ -1,5 +1,8 @@
 import bpy
 import json
+import blenderbim.bim.module.structural.add_structural_boundary_condition as add_structural_boundary_condition
+import blenderbim.bim.module.structural.edit_structural_boundary_condition as edit_structural_boundary_condition
+import blenderbim.bim.module.structural.remove_structural_boundary_condition as remove_structural_boundary_condition
 import blenderbim.bim.module.structural.add_structural_analysis_model as add_structural_analysis_model
 import blenderbim.bim.module.structural.edit_structural_analysis_model as edit_structural_analysis_model
 import blenderbim.bim.module.structural.remove_structural_analysis_model as remove_structural_analysis_model
@@ -9,6 +12,110 @@ from blenderbim.bim.ifc import IfcStore
 from blenderbim.bim.module.structural.data import Data


+class AddStructuralBoundaryCondition(bpy.types.Operator):
+    bl_idname = "bim.add_structural_boundary_condition"
+    bl_label = "Add Structural Boundary Condition"
+
+    def execute(self, context):
+        obj = context.active_object
+        file = IfcStore.get_file()
+        connection = file.by_id(obj.BIMObjectProperties.ifc_definition_id)
+        add_structural_boundary_condition.Usecase(file, {"connection": connection}).execute()
+        Data.load(connection.id())
+        return {"FINISHED"}
+
+
+class RemoveStructuralBoundaryCondition(bpy.types.Operator):
+    bl_idname = "bim.remove_structural_boundary_condition"
+    bl_label = "Enable Editing Structural Boundary Condition"
+
+    def execute(self, context):
+        obj = context.active_object
+        file = IfcStore.get_file()
+        connection = file.by_id(obj.BIMObjectProperties.ifc_definition_id)
+        remove_structural_boundary_condition.Usecase(file, {"connection": connection}).execute()
+        Data.load(connection.id())
+        return {"FINISHED"}
+
+
+class EnableEditingStructuralBoundaryCondition(bpy.types.Operator):
+    bl_idname = "bim.enable_editing_structural_boundary_condition"
+    bl_label = "Enable Editing Structural Boundary Condition"
+
+    def execute(self, context):
+        obj = context.active_object
+        oprops = obj.BIMObjectProperties
+        props = obj.BIMStructuralProperties
+        while len(props.connection_attributes) > 0:
+            props.connection_attributes.remove(0)
+
+        data = Data.connections[oprops.ifc_definition_id]
+
+        for attribute in IfcStore.get_schema().declaration_by_name(data["type"]).all_attributes():
+            value = data[attribute.name()]
+            data_type = str(attribute.type_of_attribute)
+            new = props.connection_attributes.add()
+            new.name = attribute.name()
+            new.is_null = value is None
+            new.is_optional = attribute.optional()
+            if "<select" in data_type:
+                enum_items = [s.name() for s in attribute.type_of_attribute().declared_type().select_list()]
+                new.enum_items = json.dumps(enum_items)
+            if isinstance(value, bool):
+                new.bool_value = False if new.is_null else data[attribute.name()]
+                new.data_type = "bool"
+                new.enum_value = "IfcBoolean"
+            elif isinstance(value, float):
+                new.float_value = 0.0 if new.is_null else data[attribute.name()]
+                new.data_type = "float"
+                new.enum_value = [i for i in enum_items if i != "IfcBoolean"][0]
+            elif "<string>" in data_type:
+                new.string_value = "" if new.is_null else data[attribute.name()]
+                new.data_type = "string"
+
+        props.is_editing_connection = True
+        return {"FINISHED"}
+
+
+class EditStructuralBoundaryCondition(bpy.types.Operator):
+    bl_idname = "bim.edit_structural_boundary_condition"
+    bl_label = "Edit Structural Boundary Condition"
+
+    def execute(self, context):
+        obj = context.active_object
+        oprops = obj.BIMObjectProperties
+        props = obj.BIMStructuralProperties
+
+        file = IfcStore.get_file()
+        connection = file.by_id(obj.BIMObjectProperties.ifc_definition_id)
+        condition = connection.AppliedCondition
+
+        attributes = {}
+        for attribute in props.connection_attributes:
+            if attribute.is_null:
+                attributes[attribute.name] = {"value": None, "type": "null"}
+            elif attribute.data_type == "string":
+                attributes[attribute.name] = {"value": attribute.string_value, "type": "string"}
+            elif attribute.enum_value == "IfcBoolean":
+                attributes[attribute.name] = {"value": attribute.bool_value, "type": attribute.enum_value}
+            else:
+                attributes[attribute.name] = {"value": attribute.float_value, "type": attribute.enum_value}
+
+        edit_structural_boundary_condition.Usecase(file, {"condition": condition, "attributes": attributes}).execute()
+        Data.load(connection.id())
+        bpy.ops.bim.disable_editing_structural_boundary_condition()
+        return {"FINISHED"}
+
+
+class DisableEditingStructuralBoundaryCondition(bpy.types.Operator):
+    bl_idname = "bim.disable_editing_structural_boundary_condition"
+    bl_label = "Disable Editing Structural Boundary Condition"
+
+    def execute(self, context):
+        context.active_object.BIMStructuralProperties.is_editing_connection = False
+        return {"FINISHED"}
+
+
 class LoadStructuralAnalysisModels(bpy.types.Operator):
    bl_idname = "bim.load_structural_analysis_models"
    bl_label = "Load Structural Analysis Models"