Root.copy_representation() now also copies shape aspects of product types

This commit is contained in:
Robin Quint
2025-08-09 12:30:50 +02:00
committed by Ryan Schultz
parent faf489e9dc
commit 8b155fc989
+45 -22
View File
@@ -103,39 +103,62 @@ class Root(bonsai.core.tool.Root):
copied_entities=copied_entities,
)
if source.Representation.is_a("IfcProductDefinitionShape"):
old_to_new_items = {}
for repr_index, representation in enumerate(source.Representation.Representations):
for item_index, item in enumerate(representation.Items):
old_to_new_items[item.id()] = dest.Representation.Representations[repr_index].Items[item_index]
for aspect in source.Representation.HasShapeAspects:
new_reprs = []
for repr in aspect.ShapeRepresentations:
new_items = []
for old_item in repr.Items:
new_items.append(old_to_new_items[old_item.id()])
new_repr = ifcopenshell.util.element.copy(tool.Ifc.get(), repr)
new_repr.Items = new_items
new_reprs.append(new_repr)
new_aspect = ifcopenshell.util.element.copy(tool.Ifc.get(), aspect)
new_aspect.ShapeRepresentations = new_reprs
new_aspect.PartOfProductDefinitionShape = dest.Representation
cls.copy_shape_aspects(source.Representation, dest.Representation)
elif dest.is_a("IfcTypeProduct"):
if not source.RepresentationMaps:
return copied_entities
dest.RepresentationMaps = [
ifcopenshell.util.element.copy_deep(
new_maps = []
for m in source.RepresentationMaps:
new_map = ifcopenshell.util.element.copy_deep(
tool.Ifc.get(),
m,
exclude=["IfcGeometricRepresentationContext"],
exclude_callback=exclude_callback,
copied_entities=copied_entities,
)
for m in source.RepresentationMaps
]
cls.copy_shape_aspects(m, new_map)
new_maps.append(new_map)
dest.RepresentationMaps = new_maps
return copied_entities
@classmethod
def copy_shape_aspects(cls, source_repr: ifcopenshell.entity_instance, dest_repr: ifcopenshell.entity_instance):
if source_repr.is_a("IfcProductDefinitionShape"):
old_to_new_items = {}
for repr_index, representation in enumerate(source_repr.Representations):
for item_index, item in enumerate(representation.Items):
old_to_new_items[item.id()] = dest_repr.Representations[repr_index].Items[item_index]
for aspect in source_repr.HasShapeAspects:
new_reprs = []
for repr in aspect.ShapeRepresentations:
new_items = []
for old_item in repr.Items:
new_items.append(old_to_new_items[old_item.id()])
new_repr = ifcopenshell.util.element.copy(tool.Ifc.get(), repr)
new_repr.Items = new_items
new_reprs.append(new_repr)
new_aspect = ifcopenshell.util.element.copy(tool.Ifc.get(), aspect)
new_aspect.ShapeRepresentations = new_reprs
new_aspect.PartOfProductDefinitionShape = dest_repr
elif source_repr.is_a("IfcRepresentationMap"):
old_to_new_items = {}
for item_index, item in enumerate(source_repr.MappedRepresentation.Items):
old_to_new_items[item.id()] = dest_repr.MappedRepresentation.Items[item_index]
for aspect in source_repr.HasShapeAspects:
new_reprs = []
for repr in aspect.ShapeRepresentations:
new_items = []
for old_item in repr.Items:
new_items.append(old_to_new_items[old_item.id()])
new_repr = ifcopenshell.util.element.copy(tool.Ifc.get(), repr)
new_repr.Items = new_items
new_reprs.append(new_repr)
new_aspect = ifcopenshell.util.element.copy(tool.Ifc.get(), aspect)
new_aspect.ShapeRepresentations = new_reprs
new_aspect.PartOfProductDefinitionShape = dest_repr
@classmethod
def does_type_have_representations(cls, element: ifcopenshell.entity_instance) -> bool: