From 599089ad672b4ad9b8c40c08fe0f3f72007e8bef Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Sat, 14 Nov 2020 13:36:49 +1100
Subject: [PATCH] Now materials that aren't also used as a styled item can be
 exported

---
 .../blenderbim/bim/export_ifc.py              | 45 ++++++++++++-------
 1 file changed, 28 insertions(+), 17 deletions(-)

diff --git a/src/ifcblenderexport/blenderbim/bim/export_ifc.py b/src/ifcblenderexport/blenderbim/bim/export_ifc.py
index a58647e410..80ef2b6bbe 100644
--- a/src/ifcblenderexport/blenderbim/bim/export_ifc.py
+++ b/src/ifcblenderexport/blenderbim/bim/export_ifc.py
@@ -109,7 +109,7 @@ class IfcParser:
         self.constraints = self.get_constraints()
         self.load_representations()
         self.load_presentation_layer_assignments()
-        self.materials = self.get_materials()
+        self.get_materials()
         self.styled_items = self.get_styled_items()
         self.spatial_structure_elements = self.get_spatial_structure_elements()
         self.groups = self.get_groups()
@@ -1134,26 +1134,37 @@ class IfcParser:
         return name
 
     def get_materials(self):
-        results = {}
         if not self.ifc_export_settings.has_representations:
-            return results
-        for product in self.selected_products + self.type_products:
+            return
+        for product in self.selected_products + self.selected_types:
             obj = product["raw"]
-            if obj.data is None:
-                continue
+            if obj.BIMObjectProperties.material_type == "IfcMaterial" and obj.BIMObjectProperties.material:
+                self.get_material(obj.BIMObjectProperties.material)
+            elif obj.BIMObjectProperties.material_type == "IfcMaterialConstituentSet":
+                for constituent in obj.BIMObjectProperties.material_set.material_constituents:
+                    self.get_material(constituent.material)
+            elif obj.BIMObjectProperties.material_type == "IfcMaterialLayerSet":
+                for layer in obj.BIMObjectProperties.material_set.material_layers:
+                    self.get_material(layer.material)
+            elif obj.BIMObjectProperties.material_type == "IfcMaterialProfileSet":
+                for profile in obj.BIMObjectProperties.material_set.material_profiles:
+                    self.get_material(profile.material)
             for slot in obj.material_slots:
-                if slot.material is None:
+                if slot.material is None or slot.link == "OBJECT":
                     continue
-                if slot.material.name in results or slot.link == "OBJECT":
-                    continue
-                material = {
-                    "ifc": None,
-                    "raw": slot.material,
-                    "attributes": self.get_material_attributes(slot.material),
-                }
-                results[slot.material.name] = material
-                self.get_material_psets(material, slot.material)
-        return results
+                self.get_material(slot.material)
+
+    def get_material(self, material):
+        if material.name in self.materials:
+            return
+        data = {
+            "ifc": None,
+            "raw": material,
+            "attributes": self.get_material_attributes(material),
+        }
+        print(material.name)
+        self.materials[material.name] = data
+        self.get_material_psets(data, material)
 
     def get_material_attributes(self, material):
         attributes = {"Name": material.name}