Option to convert custom properties to ifc psets when assigning an IFC class (#6921)

Feature location - https://files.catbox.moe/cuvdan.png (assign class with alt+click modifier) See #6921 for an example of converted props
2026-08-28 15:53:00 +00:00 · 2025-07-24 07:12:22 +01:00
parent e50215ddce
commit 0a50a3cb24
2 changed files with 97 additions and 1 deletions
@@ -238,3 +238,77 @@ class TestReassignClass(NewFile):
        assert len(ifc_file.by_type("IfcSlab")) == 0
        assert len(ifc_file.by_type("IfcWallType")) == n_wall_types + 1
        assert len(ifc_file.by_type("IfcSlabType")) == n_slab_types - 1
+
+
+class TestAssignClass(NewFile):
+    def create_objects(self, context):
+        # Create blender cylinder
+        bpy.ops.mesh.primitive_cylinder_add(vertices=10, location=(0, 4, 0))
+        datablock_obj = bpy.data.objects["Cylinder"]
+
+        # Create blender cube w/ props
+        bpy.ops.mesh.primitive_cube_add(location=(0, 0, 0))
+        obj = bpy.data.objects["Cube"]
+        # Set all the custom properties on the obj
+        obj["01_float"] = 3.14159
+        obj["02_float_array"] = [3.14159, 1.61803, 2.71828]
+        obj["03_integer"] = 2
+        obj["04_integer_array"] = [2, 3, 5, 7]
+        obj["05_boolean"] = True
+        obj["06_boolean_array"] = [True, False]
+        obj["07_string"] = "Bonsai!"
+        # Data Block is not proper Pointer in UI. Probably doesn't matter.
+        obj["08_data_block"] = datablock_obj
+        # Python expressions can also be stored, i.e. dictionary
+        obj["09_python"] = {"test": 12}
+        # Or a list of things, i.e. objects: [bpy.data.objects['IfcBuildingElementProxy/Cube']]
+        obj["10_python"] = bpy.context.selected_objects
+
+        return obj, datablock_obj
+
+    def test_normal_assign_ifc_class(self):
+        # Setup project
+        tool.Project.get_project_props().template_file = "IFC4 Demo Template.ifc"
+        bpy.ops.bim.create_project()
+        ifc_file = tool.Ifc.get()
+        context = bpy.context
+        obj, datablock_obj = self.create_objects(context)
+
+        # Assign IfcClass
+        bpy.ops.bim.assign_class(ifc_class="IfcBuildingElementProxy", predefined_type="ELEMENT", userdefined_type="")
+        element = tool.Ifc.get_entity(obj)
+        assert element
+
+        # Get Psets
+        psets = ifcopenshell.util.element.get_psets(element, psets_only=True)
+        assert psets == {}
+
+    def test_alternative_assign_ifc_class(self):
+        # Setup project
+        tool.Project.get_project_props().template_file = "IFC4 Demo Template.ifc"
+        bpy.ops.bim.create_project()
+        ifc_file = tool.Ifc.get()
+        context = bpy.context
+        obj, datablock_obj = self.create_objects(context)
+
+        # Assign IfcClass
+        bpy.ops.bim.assign_class(
+            ifc_class="IfcBuildingElementProxy", predefined_type="ELEMENT", userdefined_type="", props_to_pset=True
+        )
+        element = tool.Ifc.get_entity(obj)
+        assert element
+
+        # Get Psets
+        psets = ifcopenshell.util.element.get_psets(element, psets_only=True)
+        assert "BBIM_ImportedBlenderProps" in psets
+        pset = psets["BBIM_ImportedBlenderProps"]
+        assert "01_float" in pset and type(pset["01_float"]) is float
+        assert "02_float_array.1" in pset and type(pset["02_float_array.1"]) is float
+        assert "03_integer" in pset and type(pset["03_integer"]) is int
+        assert "04_integer_array.1" in pset and type(pset["04_integer_array.1"]) is int
+        assert "05_boolean" in pset and type(pset["05_boolean"]) is bool
+        assert "06_boolean_array.1" in pset and type(pset["06_boolean_array.1"]) is bool
+        assert "07_string" in pset and type(pset["07_string"]) is str
+        assert "08_data_block" not in pset
+        assert "09_python" not in pset
+        assert "10_python" not in pset