From a44439655ce22213010d7c816c87d4ffe290d362 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bruno=20Perdig=C3=A3o?= <bruno@brunopo.com>
Date: Fri, 17 Jan 2025 16:23:40 -0300
Subject: [PATCH] Install Bonsai decorators from Viewport Overlays in
 `load_post`.

---
 src/bonsai/bonsai/bim/handler.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/bonsai/bonsai/bim/handler.py b/src/bonsai/bonsai/bim/handler.py
index 1f63ff18b8..b9194e84ed 100644
--- a/src/bonsai/bonsai/bim/handler.py
+++ b/src/bonsai/bonsai/bim/handler.py
@@ -30,6 +30,9 @@ from bonsai.bim.ifc import IfcStore
 from bonsai.bim.module.owner.prop import get_user_person, get_user_organisation
 from bonsai.bim.module.model.data import AuthoringData
 from bonsai.bim.module.model.workspace import LIST_OF_TOOLS, TOOLS_TO_CLASSES_MAP
+from bonsai.bim.module.aggregate.decorator import AggregateDecorator
+from bonsai.bim.module.georeference.decorator import GeoreferenceDecorator
+from bonsai.bim.module.model.decorator import WallAxisDecorator, SlabDirectionDecorator
 from mathutils import Vector
 from math import cos, degrees
 from typing import Union, Callable
@@ -316,3 +319,18 @@ def load_post(scene):
 
     if tool.Ifc.get() and bpy.data.is_saved:
         bpy.context.scene.BIMProperties.has_blend_warning = True
+
+    # Bonsai overlays
+    georeference_props = bpy.context.scene.BIMGeoreferenceProperties
+    aggregate_props = bpy.context.scene.BIMAggregateProperties
+    model_props = bpy.context.scene.BIMModelProperties
+    if aggregate_props.aggregate_decorator:
+        AggregateDecorator.install(bpy.context)
+    if georeference_props.should_visualise:
+        GeoreferenceDecorator.install(bpy.context)
+    if model_props.show_wall_axis:
+        WallAxisDecorator.install(bpy.context)
+    if model_props.show_slab_direction:
+        SlabDirectionDecorator.install(bpy.context)
+        
+