mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-19 14:41:25 +00:00
New utility to get IFC all elements that are relevant in Brickschema
This commit is contained in:
@@ -17,4 +17,16 @@ def get_brick_type(element):
|
|||||||
result = ifc4_to_brick_map.get(f"{element.is_a()}.{predefined_type}", None)
|
result = ifc4_to_brick_map.get(f"{element.is_a()}.{predefined_type}", None)
|
||||||
if not result:
|
if not result:
|
||||||
result = ifc4_to_brick_map.get(element.is_a(), None)
|
result = ifc4_to_brick_map.get(element.is_a(), None)
|
||||||
return result
|
if result:
|
||||||
|
return f"https://brickschema.org/schema/Brick#{result}"
|
||||||
|
|
||||||
|
|
||||||
|
def get_brick_elements(ifc_file):
|
||||||
|
classes = {c.split(".")[0] for c in ifc4_to_brick_map.keys()}
|
||||||
|
elements = []
|
||||||
|
for ifc_class in classes:
|
||||||
|
try:
|
||||||
|
elements += ifc_file.by_type(ifc_class)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
return elements
|
||||||
|
|||||||
@@ -6,6 +6,13 @@ import ifcopenshell.util.brick as subject
|
|||||||
class TestGetBrickTypeIFC4(test.bootstrap.IFC4):
|
class TestGetBrickTypeIFC4(test.bootstrap.IFC4):
|
||||||
def test_run(self):
|
def test_run(self):
|
||||||
element = self.file.createIfcAirTerminalBox()
|
element = self.file.createIfcAirTerminalBox()
|
||||||
assert subject.get_brick_type(element) == "TerminalUnit"
|
assert subject.get_brick_type(element) == "https://brickschema.org/schema/Brick#TerminalUnit"
|
||||||
element.PredefinedType = "CONSTANTFLOW"
|
element.PredefinedType = "CONSTANTFLOW"
|
||||||
assert subject.get_brick_type(element) == "CAV"
|
assert subject.get_brick_type(element) == "https://brickschema.org/schema/Brick#CAV"
|
||||||
|
|
||||||
|
|
||||||
|
class TestGetBrickElementsIFC4(test.bootstrap.IFC4):
|
||||||
|
def test_run(self):
|
||||||
|
element = self.file.createIfcAirTerminalBox()
|
||||||
|
self.file.createIfcWall()
|
||||||
|
assert subject.get_brick_elements(self.file) == [element]
|
||||||
|
|||||||
Reference in New Issue
Block a user