Improve UI to change spatial container to prevent accidentally selecting incorrect objects

src/ifcblenderexport/blenderbim/bim/module/spatial/data.py

@@ -3,13 +3,21 @@ from blenderbim.bim.ifc import IfcStore
 
 class Data:
     products = {}
+    spatial_elements = {}
 
     @classmethod
     def purge(cls):
         cls.products = {}
+        cls.spatial_elements = {}
 
     @classmethod
-    def load(cls, product_id):
+    def load(cls, product_id=None):
+        if product_id:
+            return cls.load_product(product_id)
+        cls.load_spatial_elements()
+
+    @classmethod
+    def load_product(cls, product_id):
         file = IfcStore.get_file()
         if not file:
             return
@@ -21,3 +29,27 @@ class Data:
             cls.products[product_id] = {"type": structure.is_a(), "Name": structure.Name, "id": int(structure.id())}
         else:
             cls.products[product_id] = {"type": None, "Name": None, "id": None}
+
+    @classmethod
+    def load_spatial_elements(cls):
+        file = IfcStore.get_file()
+        if not file:
+            return
+        cls.spatial_elements = {}
+        ifc_class = "IfcSpatialElement"
+        if file.schema == "IFC2X3":
+            ifc_class = "IfcSpatialStructureElement"
+        for element in file.by_type(ifc_class):
+            decomposes = None
+            if element.Decomposes:
+                decomposes = element.Decomposes[0].RelatingObject.id()
+            is_decomposed_by = []
+            if element.IsDecomposedBy:
+                for rel in element.IsDecomposedBy:
+                    is_decomposed_by.extend([e.id() for e in rel.RelatedObjects])
+            cls.spatial_elements[element.id()] = {
+                "Name": element.Name,
+                "LongName": element.LongName,
+                "Decomposes": decomposes,
+                "IsDecomposedBy": is_decomposed_by
+            }