Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-10 09:48:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

util.representation.get_representations_iter

Browse Source
This commit is contained in:
Andrej730
2024-09-03 16:24:14 +05:00
parent b957505bfc
commit bec1d913c5
2 changed files with 19 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/representation.py
+17 -9
View File
@@ -20,7 +20,7 @@ import numpy as np
import numpy.typing as npt
import ifcopenshell
import ifcopenshell.util.placement
from typing import Optional, Union, TypedDict, Literal
from typing import Optional, Union, TypedDict, Literal, Iterator
CONTEXT_TYPE = Literal["Model", "Plan", "NotDefined"]
@@ -112,6 +112,19 @@ def is_representation_of_context(
return representation.ContextOfItems.ContextType == context
def get_representations_iter(element: ifcopenshell.entity_instance) -> Iterator[ifcopenshell.entity_instance]:
"""Get an iterator with element's IfcShapeRepresentations.
:param element: An IfcProduct or IfcTypeProduct
"""
if element.is_a("IfcProduct") and (rep := element.Representation):
for r in rep.Representations:
yield r
elif element.is_a("IfcTypeProduct") and (maps := element.RepresentationMaps):
for r in maps:
yield r.MappedRepresentation
def get_representation(
element: ifcopenshell.entity_instance,
context: Union[ifcopenshell.entity_instance, CONTEXT_TYPE],
@@ -126,14 +139,9 @@ def get_representation(
:param target_view: A TargetView string, or any if left blank.
:return: The first IfcShapeRepresentation matching the criteria.
"""
if element.is_a("IfcProduct") and element.Representation:
for r in element.Representation.Representations:
if is_representation_of_context(r, context, subcontext, target_view):
return r
elif element.is_a("IfcTypeProduct") and element.RepresentationMaps:
for r in element.RepresentationMaps:
if is_representation_of_context(r.MappedRepresentation, context, subcontext, target_view):
return r.MappedRepresentation
for r in get_representations_iter(element): # type: ignore
if is_representation_of_context(r, context, subcontext, target_view):
return r
def resolve_representation(representation: ifcopenshell.entity_instance) -> ifcopenshell.entity_instance: