From 977d814d39a0b09b85c57d940f65bb8543722683 Mon Sep 17 00:00:00 2001 From: Andrej730 Date: Tue, 11 Mar 2025 18:35:34 +0500 Subject: [PATCH] bim.reassign_class - fix issue in ifc2x3 when selected product class would be ignored E.g. you select IfcSlab and change it to IfcRoof - since there is not IfcRoofType in IFC2X3 it would figure the matching product type is IfcBeamType (which is confusing too but that's another subject) and change this slab's type object to IfcBeamType which consequently change IfcSlab to IfcBeam instead of IfcRoof that was selected originally. Noticed investigating #5918 --- src/bonsai/bonsai/bim/module/root/operator.py | 3 +++ .../ifcopenshell/api/root/reassign_class.py | 22 ++++++++++++++---- .../test/api/root/test_reassign_class.py | 23 ++++++++++++++++++- 3 files changed, 43 insertions(+), 5 deletions(-) diff --git a/src/bonsai/bonsai/bim/module/root/operator.py b/src/bonsai/bonsai/bim/module/root/operator.py index 4accda58d9..624eefc11d 100644 --- a/src/bonsai/bonsai/bim/module/root/operator.py +++ b/src/bonsai/bonsai/bim/module/root/operator.py @@ -157,6 +157,9 @@ class ReassignClass(bpy.types.Operator, tool.Ifc.Operator): product=element, ifc_class=ifc_class_, predefined_type=predefined_type, + # Provide occurrence class in all cases as it won't really matter + # for non-IfcTypeProducts. + occurrence_class=ifc_class, ) reassigned_elements.add(element) diff --git a/src/ifcopenshell-python/ifcopenshell/api/root/reassign_class.py b/src/ifcopenshell-python/ifcopenshell/api/root/reassign_class.py index a7588962dd..a4f4f09b93 100644 --- a/src/ifcopenshell-python/ifcopenshell/api/root/reassign_class.py +++ b/src/ifcopenshell-python/ifcopenshell/api/root/reassign_class.py @@ -35,6 +35,7 @@ def reassign_class( product: ifcopenshell.entity_instance, ifc_class: str = "IfcBuildingElementProxy", predefined_type: Optional[str] = None, + occurrence_class: Optional[str] = None, ) -> ifcopenshell.entity_instance: """Changes the class of a product @@ -59,6 +60,11 @@ def reassign_class( :param ifc_class: The new IFC class you want to change it to. :param predefined_type: In case you want to change the predefined type too. User defined types are also allowed, just type what you want. + :param occurrence_class: IFC class to assign to occurrences in case + if provided ``ifc_class`` is IfcTypeProduct. + If omitted, class will be deduced automatically from the type. + Only really needed in IFC2X3, since in IFC4+ there is no ambiguity on + what class to assign to occurrences. :return: The newly modified product. Example: @@ -76,15 +82,20 @@ def reassign_class( """ usecase = Usecase() usecase.file = file - return usecase.execute(product, ifc_class, predefined_type) + return usecase.execute(product, ifc_class, predefined_type, occurrence_class) class Usecase: file: ifcopenshell.file def execute( - self, product: ifcopenshell.entity_instance, ifc_class: str, predefined_type: Union[str, None] + self, + product: ifcopenshell.entity_instance, + ifc_class: str, + predefined_type: Union[str, None], + occurrence_class: Union[str, None], ) -> ifcopenshell.entity_instance: + self.occurrence_class = occurrence_class was_type_product_before = product.is_a("IfcTypeProduct") schema = ifcopenshell.schema_by_name(self.file.schema) is_type_product_after: bool @@ -154,7 +165,7 @@ class Usecase: for rep in representations: ifcopenshell.api.geometry.assign_representation(self.file, product=element, representation=rep) - # Keep IFC valid. + # Keep IFC valid (PlacementForShapeRepresentation). if switch_type == "type_to_occurrence" and representations: ifcopenshell.api.geometry.edit_object_placement(self.file, product=element) @@ -169,7 +180,10 @@ class Usecase: element = self.reassign_class(element, ifc_class, predefined_type) if element.is_a("IfcTypeProduct"): for occurrence in ifcopenshell.util.element.get_types(element): - ifc_class_ = ifcopenshell.util.type.get_applicable_entities(ifc_class, self.file.schema)[0] + if self.occurrence_class: + ifc_class_ = self.occurrence_class + else: + ifc_class_ = next(iter(ifcopenshell.util.type.get_applicable_entities(ifc_class, self.file.schema))) self.reassign_class(occurrence, ifc_class_, predefined_type) else: element_type = ifcopenshell.util.element.get_type(element) diff --git a/src/ifcopenshell-python/test/api/root/test_reassign_class.py b/src/ifcopenshell-python/test/api/root/test_reassign_class.py index f7b1024b6b..f8053e13b5 100644 --- a/src/ifcopenshell-python/test/api/root/test_reassign_class.py +++ b/src/ifcopenshell-python/test/api/root/test_reassign_class.py @@ -198,4 +198,25 @@ class TestReassignClass(test.bootstrap.IFC4): class TestReassignClassIFC2X3(test.bootstrap.IFC2X3, TestReassignClass): - pass + def test_providing_occurrence_class(self): + element_type = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWallType") + element1 = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWall") + ifcopenshell.api.type.assign_type(self.file, related_objects=[element1], relating_type=element_type) + element2 = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWall") + ifcopenshell.api.type.assign_type(self.file, related_objects=[element2], relating_type=element_type) + + new_element_type = ifcopenshell.api.root.reassign_class( + self.file, + product=element_type, + ifc_class="IfcBuildingElementProxyType", + occurrence_class="IfcRoof", + ) + assert new_element_type.is_a("IfcBuildingElementProxyType") + occurrences = ifcopenshell.util.element.get_types(new_element_type) + assert len(occurrences) == 2 + # Assign IfcRoof instead of IfcBuildingElementProxy. + assert all(o.is_a("IfcRoof") for o in occurrences) + + # original clases are gone + assert len(self.file.by_type("IfcWall")) == 0 + assert len(self.file.by_type("IfcWallType")) == 0