90 lines
3.7 KiB
Python
90 lines
3.7 KiB
Python
# IfcOpenShell - IFC toolkit and geometry engine
|
|
# Copyright (C) 2025 Dion Moult <dion@thinkmoult.com>
|
|
#
|
|
# This file is part of IfcOpenShell.
|
|
#
|
|
# IfcOpenShell is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# IfcOpenShell is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with IfcOpenShell. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from typing import Optional
|
|
|
|
import ifcopenshell
|
|
|
|
|
|
def add_shape_aspect(
|
|
file: ifcopenshell.file,
|
|
name: str,
|
|
items: list[ifcopenshell.entity_instance],
|
|
representation: ifcopenshell.entity_instance,
|
|
part_of_product: ifcopenshell.entity_instance,
|
|
description: Optional[str] = None,
|
|
) -> ifcopenshell.entity_instance:
|
|
"""Adds a shape aspect to items that are part of a representation and product
|
|
|
|
Existing shape aspects will be reused where possible. If the items already
|
|
belong to another shape aspect with a different name, this relationship
|
|
will be purged.
|
|
|
|
Warning: it is not possible to add a shape aspect to types (i.e.
|
|
IfcRepresentationMap) in IFC2X3.
|
|
|
|
:param name: The name of the shape aspect. This is case sensitive.
|
|
:param items: IfcRepresentationItems that will be assigned to this aspect.
|
|
:param representation: The IfcShapeRepresentation that the items are in.
|
|
:param part_of_product: The IfcRepresentationMap or
|
|
IfcProductDefinitionShape that the representation is in.
|
|
:param description: A description to set for the shape aspect. It's usually
|
|
not necessary.
|
|
:return: The IfcShapeAspect
|
|
"""
|
|
result = None
|
|
items_set = set(items)
|
|
for aspect in part_of_product.HasShapeAspects or []:
|
|
if aspect.Name == name:
|
|
for aspect_rep in aspect.ShapeRepresentations:
|
|
if aspect_rep.ContextOfItems == representation.ContextOfItems:
|
|
aspect.Description = description
|
|
aspect_rep.Items = tuple(set(aspect_rep.Items) | items_set)
|
|
result = aspect
|
|
if not result:
|
|
aspect_rep = file.createIfcShapeRepresentation(
|
|
ContextOfItems=representation.ContextOfItems,
|
|
RepresentationIdentifier=representation.RepresentationIdentifier,
|
|
RepresentationType=representation.RepresentationType,
|
|
Items=items,
|
|
)
|
|
aspect.ShapeRepresentations += (aspect_rep,)
|
|
result = aspect
|
|
else:
|
|
for aspect_rep in aspect.ShapeRepresentations:
|
|
if aspect_rep.ContextOfItems != representation.ContextOfItems:
|
|
continue
|
|
if set(aspect_rep.Items) & items_set:
|
|
if new_items := set(aspect_rep.Items) - items_set:
|
|
aspect_rep.Items = tuple(new_items)
|
|
else:
|
|
file.remove(aspect_rep)
|
|
if not aspect.ShapeRepresentations:
|
|
file.remove(aspect)
|
|
|
|
if result:
|
|
return result
|
|
|
|
aspect_rep = file.createIfcShapeRepresentation(
|
|
ContextOfItems=representation.ContextOfItems,
|
|
RepresentationIdentifier=representation.RepresentationIdentifier,
|
|
RepresentationType=representation.RepresentationType,
|
|
Items=items,
|
|
)
|
|
return file.createIfcShapeAspect((aspect_rep,), name, description, True, part_of_product)
|