Fixed arch bug in bim views

yorikvanhavre · Mar 15, 2024 · 1e96402 · 1e96402
1 parent 9e2add8
commit 1e96402
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/BimViews.py b/BimViews.py
@@ -183,7 +183,7 @@ def update(self, retrigger=True):
                 for obj in FreeCAD.ActiveDocument.Objects:
                     t = Draft.getType(obj)
                     if obj and (t in ["Building", "BuildingPart", "IfcBuilding", "IfcBuildingStorey"]):
-                        if t in ["Building", "IfcBuilding"]:
+                        if t in ["Building", "IfcBuilding"] or getattr(obj, "IfcType", "") == "Building":
                             building, _ = getTreeViewItem(obj)
                             subObjs = obj.Group
                             # find every levels belongs to the building
@@ -207,8 +207,8 @@ def update(self, retrigger=True):
                             treeViewItems.append(building)
                             lvHold.clear()
 
-                        if t in ["Building Storey", "IfcBuildingStorey"]:
-                            if Draft.getType(getParent(obj)) in ["Building", "IfcBuilding"]:
+                        if t in ["Building Storey", "IfcBuildingStorey"] or getattr(obj, "IfcType", "") == "Building Storey":
+                            if Draft.getType(getParent(obj)) in ["Building", "IfcBuilding"] or getattr(getParent(obj), "IfcType", "") == "Building":
                                 continue
                             lv, lvH = getTreeViewItem(obj)
                             subObjs = obj.Group