From 559db259a3333293fe50f0c085f0c112c0dd6782 Mon Sep 17 00:00:00 2001
From: Dion Moult <dion@thinkmoult.com>
Date: Mon, 27 Jan 2025 17:14:13 +1100
Subject: [PATCH] See #6030. Add isolate, show, and hide operators to group
 manager.

---
 .../bonsai/bim/module/group/__init__.py       |  1 +
 .../bonsai/bim/module/group/operator.py       | 43 +++++++++++++++++--
 src/bonsai/bonsai/bim/module/group/ui.py      |  9 ++++
 3 files changed, 49 insertions(+), 4 deletions(-)

diff --git a/src/bonsai/bonsai/bim/module/group/__init__.py b/src/bonsai/bonsai/bim/module/group/__init__.py
index 8e69ee49d3..45a509c3a9 100644
--- a/src/bonsai/bonsai/bim/module/group/__init__.py
+++ b/src/bonsai/bonsai/bim/module/group/__init__.py
@@ -29,6 +29,7 @@ classes = (
     operator.LoadGroups,
     operator.RemoveGroup,
     operator.SelectGroupElements,
+    operator.SetGroupVisibility,
     operator.ToggleGroup,
     operator.UnassignGroup,
     prop.ExpandedGroups,
diff --git a/src/bonsai/bonsai/bim/module/group/operator.py b/src/bonsai/bonsai/bim/module/group/operator.py
index 4abf12f2ca..e991dcf4f1 100644
--- a/src/bonsai/bonsai/bim/module/group/operator.py
+++ b/src/bonsai/bonsai/bim/module/group/operator.py
@@ -211,10 +211,6 @@ class SelectGroupElements(bpy.types.Operator):
     group: bpy.props.IntProperty()
     is_recursive: bpy.props.BoolProperty(name="Is Recursive", default=True, options={"SKIP_SAVE"})
 
-    @classmethod
-    def poll(cls, context):
-        return bool(tool.Ifc.get() and context.active_object)
-
     def invoke(self, context, event):
         self.is_recursive = not event.alt
         return self.execute(context)
@@ -224,3 +220,42 @@ class SelectGroupElements(bpy.types.Operator):
             ifcopenshell.util.element.get_grouped_by(tool.Ifc.get().by_id(self.group), is_recursive=self.is_recursive)
         )
         return {"FINISHED"}
+
+
+class SetGroupVisibility(bpy.types.Operator):
+    bl_idname = "bim.set_group_visibility"
+    bl_label = "Set Group Visibility"
+    bl_options = {"REGISTER", "UNDO"}
+    group: bpy.props.IntProperty()
+    should_include_children: bpy.props.BoolProperty(name="Should Include Children", default=True, options={"SKIP_SAVE"})
+    mode: bpy.props.StringProperty(name="Mode")
+
+    @classmethod
+    def description(cls, context, operator):
+        if operator.mode == "HIDE":
+            return "Hides the selected group and all children.\n" + "ALT+CLICK to ignore children"
+        elif operator.mode == "SHOW":
+            return "Shows the selected group and all children.\n" + "ALT+CLICK to ignore children"
+        return "Isolate the selected group and all children.\n" + "ALT+CLICK to ignore children"
+
+    def invoke(self, context, event):
+        if event.type == "LEFTMOUSE" and event.alt:
+            self.should_include_children = False
+        return self.execute(context)
+
+    def execute(self, context):
+        if self.mode == "ISOLATE":
+            context_override = tool.Blender.get_viewport_context()
+            with context.temp_override(**context_override):
+                bpy.ops.object.hide_view_set(unselected=True)
+                bpy.ops.object.hide_view_set(unselected=False)
+            should_hide = False
+        else:
+            should_hide = self.mode == "HIDE"
+
+        group = tool.Ifc.get().by_id(self.group)
+        elements = ifcopenshell.util.element.get_grouped_by(group, is_recursive=self.should_include_children)
+        for element in elements:
+            if obj := tool.Ifc.get_object(element):
+                obj.hide_set(should_hide)
+        return {"FINISHED"}
diff --git a/src/bonsai/bonsai/bim/module/group/ui.py b/src/bonsai/bonsai/bim/module/group/ui.py
index dbd266a18f..88077f99eb 100644
--- a/src/bonsai/bonsai/bim/module/group/ui.py
+++ b/src/bonsai/bonsai/bim/module/group/ui.py
@@ -56,6 +56,15 @@ class BIM_PT_groups(Panel):
             row = self.layout.row(align=True)
             row.alignment = "RIGHT"
             row.operator("bim.select_group_elements", text="", icon="RESTRICT_SELECT_OFF").group = group_id
+            op = row.operator("bim.set_group_visibility", icon="FULLSCREEN_EXIT", text="")
+            op.mode = "ISOLATE"
+            op.group = group_id
+            op = row.operator("bim.set_group_visibility", icon="HIDE_OFF", text="")
+            op.mode = "SHOW"
+            op.group = group_id
+            op = row.operator("bim.set_group_visibility", icon="HIDE_ON", text="")
+            op.mode = "HIDE"
+            op.group = group_id
             row.operator("bim.assign_group", text="", icon="FOLDER_REDIRECT").group = group_id
             row.operator("bim.enable_editing_group", text="", icon="GREASEPENCIL").group = group_id
             row.operator("bim.add_group", text="", icon="ADD").group = group_id