First attempt at ripping out agnostic code into ifcopenshell.api namespace. See #1399.

2026-08-12 02:23:34 +00:00 · 2021-03-25 10:48:31 +11:00
parent 911d703f57
commit d77bc6bdaa
190 changed files with 388 additions and 437 deletions
@@ -0,0 +1,67 @@
+import ifcopenshell
+
+
+class Usecase:
+    def __init__(self, file, settings=None):
+        self.file = file
+        self.settings = {"product": None, "type": "IfcMaterial", "material": None}
+        for key, value in settings.items():
+            self.settings[key] = value
+
+    def execute(self):
+        if self.settings["type"] == "IfcMaterial":
+            self.assign_ifc_material()
+        elif self.settings["type"] == "IfcMaterialConstituentSet":
+            material_set = self.file.create_entity(self.settings["type"])
+            self.create_material_association(material_set)
+        elif self.settings["type"] == "IfcMaterialLayerSet":
+            material_set = self.file.create_entity(self.settings["type"])
+            self.create_material_association(material_set)
+        elif self.settings["type"] == "IfcMaterialLayerSetUsage":
+            material_set = self.file.create_entity("IfcMaterialLayerSet")
+            material_set_usage = self.create_layer_set_usage(material_set)
+            self.create_material_association(material_set_usage)
+        elif self.settings["type"] == "IfcMaterialProfileSet":
+            material_set = self.file.create_entity(self.settings["type"])
+            self.create_material_association(material_set)
+        elif self.settings["type"] == "IfcMaterialList":
+            material_set = self.file.create_entity(self.settings["type"])
+            material_set.Materials = [self.settings["material"]]
+            self.create_material_association(material_set)
+
+    def create_layer_set_usage(self, material_set):
+        return self.file.create_entity("IfcMaterialLayerSetUsage", **{
+            "ForLayerSet": material_set,
+            "LayerSetDirection": "AXIS2" if self.settings["product"].is_a("IfcWall") else "AXIS3",
+            "DirectionSense": "POSITIVE",
+            "OffsetFromReferenceLine": 0
+        })
+
+    def assign_ifc_material(self):
+        rel = self.get_rel_associates_material(self.settings["material"])
+        if not rel:
+            return self.create_material_association(self.settings["material"])
+        related_objects = list(rel.RelatedObjects)
+        related_objects.append(self.settings["product"])
+        rel.RelatedObjects = related_objects
+
+    def create_material_association(self, relating_material):
+        return self.file.create_entity(
+            "IfcRelAssociatesMaterial",
+            **{
+                "GlobalId": ifcopenshell.guid.new(),
+                "RelatedObjects": [self.settings["product"]],
+                "RelatingMaterial": relating_material,
+            }
+        )
+
+    def get_rel_associates_material(self, material):
+        if self.file.schema == "IFC2X3":
+            rel = [
+                r
+                for r in self.file.by_type("IfcRelAssociatesMaterial")
+                if r.RelatingMaterial == self.settings["material"]
+            ]
+            return rel[0] if rel else None
+        if self.settings["material"].AssociatedTo:
+            return self.settings["material"].AssociatedTo[0]