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

The material manager now shows total elements used by a material

Browse Source
This commit is contained in:
Dion Moult
2022-03-22 12:50:10 +11:00
parent 0c8273eb70
commit 9b9f0058f9
6 changed files with 87 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/element.py
+23
View File
@@ -133,6 +133,29 @@ def get_material(element, should_skip_usage=False):
return get_material(relating_type, should_skip_usage)
def get_elements_by_material(ifc_file, material):
results = []
for inverse in ifc_file.get_inverse(material):
if inverse.is_a("IfcRelAssociatesMaterial"):
results.extend(inverse.RelatedObjects)
elif inverse.is_a("IfcMaterialLayer"):
for material_set in inverse.ToMaterialLayerSet:
results.extend(get_elements_by_material(ifc_file, material_set))
elif inverse.is_a("IfcMaterialProfile"):
for material_set in inverse.ToMaterialProfileSet:
results.extend(get_elements_by_material(ifc_file, material_set))
elif inverse.is_a("IfcMaterialConstituent"):
for material_set in inverse.ToMaterialConstituentSet:
results.extend(get_elements_by_material(ifc_file, material_set))
elif inverse.is_a("IfcMaterialLayerSetUsage"):
results.extend(get_elements_by_material(ifc_file, inverse))
elif inverse.is_a("IfcMaterialProfileSetUsage"):
results.extend(get_elements_by_material(ifc_file, inverse))
elif inverse.is_a("IfcMaterialList"):
results.extend(get_elements_by_material(ifc_file, inverse))
return results
def get_layers(ifc_file, element):
layers = []
representations = []
src/ifcopenshell-python/test/util/test_element.py
+54
View File
@@ -268,6 +268,60 @@ class TestGetMaterial(test.bootstrap.IFC4):
assert subject.get_material(element) == material
class TestGetElementsByMaterial(test.bootstrap.IFC4):
def test_getting_elements_of_a_material(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
material = ifcopenshell.api.run("material.add_material", self.file)
ifcopenshell.api.run("material.assign_material", self.file, product=element, material=material)
assert subject.get_elements_by_material(self.file, material) == [element]
def test_getting_elements_of_a_material_layer_set(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
element_type = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWallType")
ifcopenshell.api.run("type.assign_type", self.file, related_object=element, relating_type=element_type)
material = ifcopenshell.api.run("material.add_material", self.file)
material_set = ifcopenshell.api.run("material.add_material_set", self.file, set_type="IfcMaterialLayerSet")
ifcopenshell.api.run("material.add_layer", self.file, layer_set=material_set, material=material)
ifcopenshell.api.run("material.assign_material", self.file, product=element_type, material=material_set)
ifcopenshell.api.run("material.assign_material", self.file, product=element, type="IfcMaterialLayerSetUsage")
usage = self.file.by_type("IfcMaterialLayerSetUsage")[0]
assert set(subject.get_elements_by_material(self.file, material)) == {element, element_type}
assert set(subject.get_elements_by_material(self.file, material_set)) == {element, element_type}
assert set(subject.get_elements_by_material(self.file, usage)) == {element}
def test_getting_elements_of_a_material_profile_set(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
element_type = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWallType")
ifcopenshell.api.run("type.assign_type", self.file, related_object=element, relating_type=element_type)
material = ifcopenshell.api.run("material.add_material", self.file)
material_set = ifcopenshell.api.run("material.add_material_set", self.file, set_type="IfcMaterialProfileSet")
ifcopenshell.api.run("material.add_profile", self.file, profile_set=material_set, material=material)
ifcopenshell.api.run("material.assign_material", self.file, product=element_type, material=material_set)
ifcopenshell.api.run("material.assign_material", self.file, product=element, type="IfcMaterialProfileSetUsage")
usage = self.file.by_type("IfcMaterialProfileSetUsage")[0]
assert set(subject.get_elements_by_material(self.file, material)) == {element, element_type}
assert set(subject.get_elements_by_material(self.file, material_set)) == {element, element_type}
assert set(subject.get_elements_by_material(self.file, usage)) == {element}
def test_getting_elements_of_a_material_constituent_set(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
material = ifcopenshell.api.run("material.add_material", self.file)
material_set = ifcopenshell.api.run("material.add_material_set", self.file, set_type="IfcMaterialConstituentSet")
ifcopenshell.api.run("material.add_constituent", self.file, constituent_set=material_set, material=material)
ifcopenshell.api.run("material.assign_material", self.file, product=element, material=material_set)
assert set(subject.get_elements_by_material(self.file, material)) == {element}
assert set(subject.get_elements_by_material(self.file, material_set)) == {element}
def test_getting_elements_of_a_material_list(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")
material = ifcopenshell.api.run("material.add_material", self.file)
material_set = ifcopenshell.api.run("material.add_material_set", self.file, set_type="IfcMaterialList")
ifcopenshell.api.run("material.add_list_item", self.file, material_list=material_set, material=material)
ifcopenshell.api.run("material.assign_material", self.file, product=element, material=material_set)
assert set(subject.get_elements_by_material(self.file, material)) == {element}
assert set(subject.get_elements_by_material(self.file, material_set)) == {element}
class TestGetlayers(test.bootstrap.IFC4):
def test_getting_the_layer_of_a_product_representation(self):
element = ifcopenshell.api.run("root.create_entity", self.file, ifc_class="IfcWall")