initial setup, supports mesh hole fixing and bulk property assignment

Author: JoWolp

__init__.py

@@ -0,0 +1,43 @@
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTIBILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+bl_info = {
+    "name" : "BII Blender Tools",
+    "author" : "bii",
+    "description" : "",
+    "blender" : (2, 80, 0),
+    "version" : (0, 0, 1),
+    "location" : "",
+    "warning" : "",
+    "category" : "BII Tools"
+}
+
+from . import close_mesh_holes
+from . import bii_functions_panel
+from . import bulk_assign_ifc_class
+from . import bulk_material_dropdown
+
+def register():
+    close_mesh_holes.register()
+    bii_functions_panel.register()
+    bulk_assign_ifc_class.register()
+    bulk_material_dropdown.register()
+
+def unregister():
+    close_mesh_holes.unregister()
+    bii_functions_panel.unregister()
+    bulk_assign_ifc_class.unregister()
+    bulk_material_dropdown.unregister()
+
+if __name__ == "__main__":
+    register()

bii_functions_panel.py

@@ -0,0 +1,30 @@
+import bpy
+
+class BiiFunctionsPanel(bpy.types.Panel):
+    """Creates a Panel in the Object properties window"""
+    bl_label = "BII IFC Tools"
+    bl_idname = "OBJECT_PT_mesh_cleanup_tools"
+    bl_space_type = 'VIEW_3D'
+    bl_region_type = 'UI'
+    bl_category = 'BII Tools'
+
+    def draw(self, context):
+        layout = self.layout
+
+        # Add a short description of the panel
+        row1 = layout.row(align = True)
+        row1.alignment = 'EXPAND'
+        row1.label(text="Default", icon='INFO')
+
+        layout.operator("object.close_mesh_holes_operator")
+
+        layout.prop(context.scene, "bulk_material", text="Bulk Material")
+
+        layout.operator("object.set_ifc_class_for_bulk_operator")
+
+
+def register():
+    bpy.utils.register_class(BiiFunctionsPanel)
+
+def unregister():
+    bpy.utils.unregister_class(BiiFunctionsPanel)

bulk_assign_ifc_class.py

@@ -0,0 +1,82 @@
+import bpy
+import ifcopenshell
+import blenderbim
+from blenderbim.bim.ifc import IfcStore
+import blenderbim.tool as tool
+from blenderbim.bim.module.pset.data import Data as PsetData
+import bmesh
+
+def calculate_volume(obj):
+    bm = bmesh.new()
+    bm.from_mesh(obj.data)
+    volume = bm.calc_volume(signed=True)
+    bm.free()
+    return volume
+
+def calculate_height(obj):
+    # Example calculation - adjust as needed
+    z_coords = [v.co.z for v in obj.data.vertices]
+    height = max(z_coords) - min(z_coords)
+    return height
+
+def set_ifc_class_for_bulk(self, context, material):
+    # Get the active IFC file
+    ifc_file = IfcStore.get_file()
+    if not ifc_file:
+        print("No IFC file found. Ensure you're working in a BlenderBIM project.")
+        return    
+
+    # Create a new Pset template file and the Pset template for Bulk
+    pset_template = ifcopenshell.api.run("pset_template.add_pset_template", ifc_file, name="dProB_Bulk")
+    ifcopenshell.api.run("pset_template.add_prop_template", ifc_file, pset_template=pset_template, name="BulkMaterial", description="Limited to what dProB is able to interpret!")
+    ifcopenshell.api.run("pset_template.add_prop_template", ifc_file, pset_template=pset_template, name="BulkVolume", primary_measure_type="IfcVolumeMeasure")
+    ifcopenshell.api.run("pset_template.add_prop_template", ifc_file, pset_template=pset_template, name="BulkHeight", primary_measure_type="IfcLengthMeasure")
+    blenderbim.bim.handler.refresh_ui_data()
+    blenderbim.bim.schema.reload(tool.Ifc.get().schema)
+    
+    # Loop through all selected objects
+    for obj in bpy.context.selected_objects:
+        if obj.type == 'MESH':
+
+            # Set the active object
+            bpy.context.view_layer.objects.active = obj
+            # Assign the IFC class
+            bpy.ops.bim.assign_class(ifc_class="IfcBuilding")
+            # get the ifc object
+            ifc_obj = IfcStore.get_file().by_id(obj.BIMObjectProperties.ifc_definition_id)
+            # check if the object is valid
+            if not ifc_obj:
+                print(f"Object {obj.name} has no IFC object.")
+                continue
+            pset = ifcopenshell.api.run("pset.add_pset", ifc_file, product=ifc_obj, name="dProB_Bulk")
+
+            volume = calculate_volume(obj)
+            height = calculate_height(obj)
+
+            # Set the properties of the Pset
+            new_values = {
+                "BulkMaterial": material,            
+                "BulkVolume": volume, 
+                "BulkHeight": height
+            }
+            ifcopenshell.api.run("pset.edit_pset", ifc_file, pset=pset, properties=new_values)
+
+            print(f"Assigned IfcBuilding and custom Pset to {obj.name}.")
+            
+
+class SetIfcClassForBulkOperator(bpy.types.Operator):
+    """Set IFC Class of selected Objects for custom dProB-Bulk"""
+    bl_idname = "object.set_ifc_class_for_bulk_operator"
+    bl_label = "Set IFC Class for Bulk"
+
+    def execute(self, context):
+        material = context.scene.bulk_material        
+        set_ifc_class_for_bulk(self, context, material)
+        return {'FINISHED'}
+
+
+def register():
+    bpy.utils.register_class(SetIfcClassForBulkOperator)
+
+def unregister():
+    bpy.utils.unregister_class(SetIfcClassForBulkOperator)

bulk_material_dropdown.py

@@ -0,0 +1,23 @@
+import bpy
+
+def get_material_options(self, context):
+    return [
+        ("Gleisschotter (Verschmutzt)", "Gleisschotter (Verschmutzt)", "Description for Option 1"),
+        ("Gleisschotter (Neu)", "Gleisschotter (Neu)", "Description for Option 2"),
+        ("Asphalt", "Asphalt", "Description for Option 3"),
+        ("Frostschutzschicht", "Frostschutzschicht", "Description for Option 4"),
+        ("Gleisunterbau", "Gleisunterbau", "Description for Option 5"),
+        ("Beton", "Beton", "Description for Option 6"),
+        # Add more options as needed
+    ]
+
+def register():
+    bpy.types.Scene.bulk_material = bpy.props.EnumProperty(
+        name="Bulk Material",
+        items=get_material_options,
+        description="Select the material for the bulk",
+    )
+
+def unregister():
+    del bpy.types.Scene.bulk_material
+

close_mesh_holes.py

@@ -0,0 +1,37 @@
+import bpy
+
+def close_mesh_holes(self, context):
+    # Loop through all selected objects
+    for obj in bpy.context.selected_objects:
+        # Check if the object is a mesh
+        if obj.type == 'MESH':
+            # Switch to Object Mode (required to perform some operations)
+            bpy.ops.object.mode_set(mode='OBJECT')
+            # Make the current object the active object
+            bpy.context.view_layer.objects.active = obj
+            # Switch to Edit Mode
+            bpy.ops.object.mode_set(mode='EDIT')
+            # Select all vertices
+            bpy.ops.mesh.select_all(action='SELECT')
+            # Fill holes
+            # Here, sides=0 means it will fill holes of any size. Adjust as needed.
+            bpy.ops.mesh.fill_holes(sides=0)
+            # Switch back to Object Mode
+            bpy.ops.object.mode_set(mode='OBJECT')
+
+    # Deselect all to clean up selection after operation
+            
+class CloseMeshHolesOperator(bpy.types.Operator):
+    """Print Hello World"""
+    bl_idname = "object.close_mesh_holes_operator"
+    bl_label = "Close Selected Mesh Holes"
+
+    def execute(self, context):
+        close_mesh_holes(self, context)
+        return {'FINISHED'}
+
+def register():
+    bpy.utils.register_class(CloseMeshHolesOperator)
+
+def unregister():
+    bpy.utils.unregister_class(CloseMeshHolesOperator)