Column profiles now auto update when you change the material profile

src/ifcopenshell-python/ifcopenshell/api/geometry/add_representation.py

@@ -30,9 +30,9 @@ class Usecase:
             #  IfcExtrudedAreaSolid/IfcCircleProfileDef
             #  IfcExtrudedAreaSolid/IfcArbitraryClosedProfileDef
             #  IfcExtrudedAreaSolid/IfcArbitraryProfileDefWithVoids
-            #  IfcExtrudedAreaSolid/IfcMaterialProfileSet
+            #  IfcExtrudedAreaSolid/IfcMaterialProfileSetUsage
             "ifc_representation_class": None,  # Whether to cast a mesh into a particular class
-            "profile_set": None,  # The material profile set if the extrusion requires it
+            "profile_set_usage": None,  # The material profile set if the extrusion requires it
         }
         self.ifc_vertices = []
         for key, value in settings.items():
@@ -220,7 +220,7 @@ class Usecase:
             return self.create_arbitrary_extrusion_representation()
         elif self.settings["ifc_representation_class"] == "IfcExtrudedAreaSolid/IfcArbitraryProfileDefWithVoids":
             return self.create_arbitrary_void_extrusion_representation()
-        elif self.settings["ifc_representation_class"] == "IfcExtrudedAreaSolid/IfcMaterialProfileSet":
+        elif self.settings["ifc_representation_class"] == "IfcExtrudedAreaSolid/IfcMaterialProfileSetUsage":
             return self.create_material_profile_set_extrusion_representation()
         return self.create_mesh_representation()
 
@@ -410,10 +410,8 @@ class Usecase:
         )
 
     def create_material_profile_set_extrusion_representation(self):
-        profile_def = (
-            self.settings["profile_set"].CompositeProfile
-            or self.settings["profile_set"].MaterialProfiles[0].Profile
-        )
+        profile_set = self.settings["profile_set_usage"].ForProfileSet
+        profile_def = profile_set.CompositeProfile or profile_set.MaterialProfiles[0].Profile
         position = None
         if self.file.schema == "IFC2X3":
             position = self.file.createIfcAxis2Placement3D(

src/ifcopenshell-python/ifcopenshell/api/material/assign_profile.py

@@ -1,3 +1,6 @@
+import ifcopenshell.util.representation
+
+
 class Usecase:
     def __init__(self, file, **settings):
         self.file = file
@@ -6,9 +9,28 @@ class Usecase:
             self.settings[key] = value
 
     def execute(self):
-        if (
-            self.settings["material_profile"].Profile
-            and len(self.file.get_inverse(self.settings["material_profile"].Profile)) == 1
-        ):
-            self.file.remove(self.settings["material_profile"].Profile)
+        old_profile = self.settings["material_profile"].Profile
         self.settings["material_profile"].Profile = self.settings["profile"]
+        for profile_set in self.settings["material_profile"].ToMaterialProfileSet:
+            for inverse in self.file.get_inverse(profile_set):
+                if not inverse.is_a("IfcMaterialProfileSetUsage"):
+                    continue
+                if self.file.schema == "IFC2X3":
+                    for rel in self.file.get_inverse(inverse):
+                        if not rel.is_a("IfcRelAssociatesMaterial"):
+                            continue
+                        for element in rel.RelatedObjects:
+                            self.change_profile(element)
+                else:
+                    for rel in inverse.AssociatedTo:
+                        for element in rel.RelatedObjects:
+                            self.change_profile(element)
+
+        if old_profile and len(self.file.get_inverse(old_profile)) == 0:
+            self.file.remove(old_profile)
+
+    def change_profile(self, element):
+        representation = ifcopenshell.util.representation.get_representation(element, "Model", "Body", "MODEL_VIEW")
+        for subelement in self.file.traverse(representation):
+            if subelement.is_a("IfcSweptAreaSolid"):
+                subelement.SweptArea = self.settings["profile"]

src/ifcopenshell-python/ifcopenshell/util/element.py

@@ -62,14 +62,15 @@ def get_type(element):
                 return relationship.RelatingType
 
 
-def get_material(element):
+def get_material(element, should_skip_usage=False):
     if hasattr(element, "HasAssociations") and element.HasAssociations:
         for relationship in element.HasAssociations:
             if relationship.is_a("IfcRelAssociatesMaterial"):
-                if relationship.RelatingMaterial.is_a("IfcMaterialLayerSetUsage"):
-                    return relationship.RelatingMaterial.ForLayerSet
-                elif relationship.RelatingMaterial.is_a("IfcMaterialProfileSetUsage"):
-                    return relationship.RelatingMaterial.ForProfileSet
+                if should_skip_usage:
+                    if relationship.RelatingMaterial.is_a("IfcMaterialLayerSetUsage"):
+                        return relationship.RelatingMaterial.ForLayerSet
+                    elif relationship.RelatingMaterial.is_a("IfcMaterialProfileSetUsage"):
+                        return relationship.RelatingMaterial.ForProfileSet
                 return relationship.RelatingMaterial
     relating_type = get_type(element)
     if hasattr(relating_type, "HasAssociations") and relating_type.HasAssociations:

src/ifcopenshell-python/ifcopenshell/util/selector.py

@@ -221,7 +221,7 @@ class Selector:
             key = ".".join(key.split(".")[1:])
         elif "." in key and key.split(".")[0] == "material":
             try:
-                element = ifcopenshell.util.element.get_material(element)
+                element = ifcopenshell.util.element.get_material(element, should_skip_usage=True)
                 if not element:
                     return None
             except: