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

get_referenced_elements to support IfcExternalInformation

Browse Source
This commit is contained in:
Andrej730
2024-08-26 17:10:07 +05:00
parent 868454c011
commit 1c3cf44122
2 changed files with 29 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/element.py
+10 -4
View File
@@ -1256,13 +1256,18 @@ REFERENCE_TYPES: dict[str, ReferenceData] = {
),
"IfcDocumentReference": ReferenceData("DocumentRefForObjects", "IfcRelAssociatesDocument", "RelatingDocument"),
"IfcLibraryReference": ReferenceData("LibraryRefForObjects", "IfcRelAssociatesLibrary", "RelatingLibrary"),
"IfcClassification": ReferenceData(
"ClassificationForObjects", "IfcRelAssociatesClassification", "RelatingClassification"
),
"IfcDocumentInformation": ReferenceData("DocumentInfoForObjects", "IfcRelAssociatesDocument", "RelatingDocument"),
"IfcLibraryInformation": ReferenceData("LibraryInfoForObjects", "IfcRelAssociatesLibrary", "RelatingLibrary"),
}
def get_referenced_elements(reference: ifcopenshell.entity_instance) -> set[ifcopenshell.entity_instance]:
"""Get all elements with assigned `reference`
:param reference: IfcExternalReference subtype reference
:param reference: IfcExternalReference/IfcExternalInformation subtype reference
:type reference: ifcopenshell.entity_instance
:return: The elements with assigned `reference`
:rtype: set[ifcopenshell.entity_instance]
@@ -1287,9 +1292,10 @@ def get_referenced_elements(reference: ifcopenshell.entity_instance) -> set[ifco
related_objects.update(rel.RelatedObjects)
else:
# IfcExternalReference
for external_rel in reference.ExternalReferenceForResources:
related_objects.update(external_rel.RelatedResourceObjects)
if reference.is_a("IfcExternalReference"):
# IfcExternalReference
for external_rel in reference.ExternalReferenceForResources:
related_objects.update(external_rel.RelatedResourceObjects)
reference_data = REFERENCE_TYPES.get(ifc_class)
if reference_data:
src/ifcopenshell-python/test/util/test_element.py
+19
View File
@@ -823,6 +823,7 @@ class TestGetReferencedElements(test.bootstrap.IFC4):
name="Foobar",
classification=result,
)
assert reference
assert subject.get_referenced_elements(reference) == set(elements)
def test_get_elements_referenced_by_library_reference(self):
@@ -843,6 +844,24 @@ class TestGetReferencedElements(test.bootstrap.IFC4):
ifcopenshell.api.document.assign_document(self.file, document=reference, products=elements)
assert subject.get_referenced_elements(reference) == set(elements)
def test_get_elements_referenced_by_document_info(self):
project = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcProject")
info = ifcopenshell.api.document.add_information(self.file)
assert subject.get_referenced_elements(info) == set([project])
def test_get_elements_referenced_by_classification(self):
project = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcProject")
info = ifcopenshell.api.classification.add_classification(self.file, classification="Name")
assert subject.get_referenced_elements(info) == set([project])
def test_get_elements_referenced_by_library(self):
info = ifcopenshell.api.library.add_library(self.file, name="Name")
element = ifcopenshell.api.root.create_entity(self.file, ifc_class="IfcWall")
rel = self.file.create_entity("IfcRelAssociatesLibrary")
rel.RelatingLibrary = info
rel.RelatedObjects = [element]
assert subject.get_referenced_elements(info) == set([element])
class TestGetReferencedElementsIFC2X3(test.bootstrap.IFC2X3, TestGetReferencedElements):
pass