(4a -> 4) Add menu to assign an Instance to an InstanceGroup

sleap/gui/app.py

@@ -86,7 +86,6 @@
 from sleap.skeleton import Skeleton
 from sleap.util import parse_uri_path
 
-
 logger = getLogger(__name__)
 
 
@@ -810,9 +809,33 @@ def new_instance_menu_action():
             self.commands.deleteFrameLimitPredictions,
         )
 
-        ### Tracks Menu ###
+        ### Sessions Menu ###
+
+        sessionsMenu = self.menuBar().addMenu("Sessions")
+        self.session_menu = sessionsMenu.addMenu("Set Session Instance")
+        add_menu_check_item(
+                sessionsMenu, "sync session", "Synchronize Session Settings"
+            ).setToolTip(
+                "If enabled, changes to the session settings will be synchronized across all instances."
+            )
+        add_menu_item(
+            sessionsMenu,
+            "clear session",
+            "Clear Current Session",
+            self.commands.clearCurrentSession,
+        )
+
+        tracksMenu.addSeparator()
 
         tracksMenu = self.menuBar().addMenu("Tracks")
+        self.inst_groups_menu = sessionsMenu.addMenu("Set Instance Group")
+        add_menu_item(
+            self.inst_groups_menu,
+            "set group",
+            "Set Instance Group",
+        self.commands.setInstanceGroup 
+        )
+
         self.track_menu = tracksMenu.addMenu("Set Instance Track")
         add_menu_check_item(
             tracksMenu, "propagate track labels", "Propagate Track Labels"
@@ -826,6 +849,8 @@ def new_instance_menu_action():
             "Transpose Instance Tracks",
             self.commands.transposeInstance,
         )
+
+
 
         tracksMenu.addSeparator()
 
@@ -1120,6 +1145,35 @@ def _update_gui_state(self):
 
         # Update menus
 
+        def _update_sessions_menu(self, frame_idx):
+            """Update the instance groups menu based on the selected frame index in the session."""
+            session = self.state.get("session")
+            if session is None:
+                return 
+
+
+            frame_idx = self.state["frame_idx"]
+            frame_group = session.frame_groups.get(frame_idx, None)
+            if frame_group is not None:
+                for inst_group_ind, instance_group in enumerate(frame_group.instance_groups):
+                    if inst_group_ind < 9:
+                        key_command = Qt.SHIFT + Qt.Key_0 + inst_group_ind + 1
+                    else:
+                        key_command = ""
+
+                    self.inst_groups_menu.addAction(
+                        instance_group.name,
+                        lambda x=instance_group: self.commands.setInstanceGroup(x),
+                        shortcut=key_command
+                    )
+
+            self.inst_groups_menu.addAction(
+                "New Instance Group",
+                self.commands.addInstanceGroup,
+                shortcut=Qt.SHIFT + Qt.Key_0
+            )
+
+        self.inst_groups_menu.setEnabled(has_selected_instance)
         self.track_menu.setEnabled(has_selected_instance)
         self.delete_tracks_menu.setEnabled(has_tracks)
         self._menu_actions["clear selection"].setEnabled(has_selected_instance)
@@ -1772,4 +1826,4 @@ def main(args: Optional[list] = None, labels: Optional[Labels] = None):
     else:
         app.exec_()
 
-    pass
+    pass

sleap/gui/commands.py

@@ -593,6 +593,11 @@ def setInstanceTrack(self, new_track: "Track"):
         """Sets track for selected instance."""
         self.execute(SetSelectedInstanceTrack, new_track=new_track)
 
+    def setInstanceGroup(self, instance_group: Optional["InstanceGroup"]):
+        """Sets the instance group for selected instance."""
+        self.execute(SetSelectedInstanceGroup, instance_group=instance_group)
+
+
     def deleteTrack(self, track: "Track"):
         """Delete a track and remove from all instances."""
         self.execute(DeleteTrack, track=track)
@@ -2693,6 +2698,23 @@ def ask_and_do(context: CommandContext, params: dict):
         if DeleteDialog(context).exec_():
             context.signal_update([UpdateTopic.project_instances])
 
+class AddInstanceGroup(EditCommand):
+    @staticmethod
+    def do_action(context, params):
+        frame_idx = context.state["frame_idx"]
+        session = context.state["session"]
+
+        if frame_idx in session.frame_groups:
+            frame_group = session.frame_groups[frame_idx]
+        else:
+            print(f"No frame group found for frame index {frame_idx}. Unable to add instance group.")
+            return
+
+        new_instance_group = frame_group.add_instance_group(instance_group=None)
+        if new_instance_group:
+            print(f"New empty instance group added to frame group at frame {frame_idx}.")
+        else:
+            print("Failed to add a new instance group.")
 
 class AddTrack(EditCommand):
     topics = [UpdateTopic.tracks]
@@ -2709,6 +2731,27 @@ def do_action(context: CommandContext, params: dict):
 
         context.execute(SetSelectedInstanceTrack, new_track=new_track)
 
+class SetSelectedInstanceGroup(EditCommand):
+    @staticmethod
+    def do_action(context, params):
+        selected_instance = context.state["selected_instance"]
+        frame_idx = context.state["frame_idx"]
+        video = context.state["video"]
+        session = context.state["session"]
+
+        frame_group = session.frame_groups.get(frame_idx, None)
+        if frame_group is None:
+            frame_group = session.new_frame_group(frame_idx=frame_idx)
+
+        camera = session.get_camera(video=video)
+
+        instance_group = params.get("instance_group")
+
+        if selected_instance and frame_group and instance_group:
+            frame_group.add_instance(instance=selected_instance, camera=camera, instance_group=instance_group)
+            print(f"Instance {selected_instance} added to group {instance_group} at frame {frame_idx}.")
+        else:
+            print("Error: Missing required components to add instance to group.")
 
 class SetSelectedInstanceTrack(EditCommand):
     topics = [UpdateTopic.tracks]
@@ -3360,6 +3403,7 @@ def add_nodes_from_template(
     def add_force_directed_nodes(
         cls, context, instance, visible, center_point: QtCore.QPoint = None
     ):
+
         import networkx as nx
 
         center_point = center_point or context.app.player.getVisibleRect().center()
@@ -3602,6 +3646,23 @@ def open_website(url: str):
     QtGui.QDesktopServices.openUrl(QtCore.QUrl(url))
 
 
+class AddInstanceGroup(AppCommand):
+    @staticmethod
+    def do_action(context: "CommandContext", params: dict):
+        group_name = params.get("group_name", "New Group")  
+        new_group = InstanceGroup(name=group_name)
+        context.app.instance_groups.append(new_group)
+        context.app.refresh_ui()
+        print(f"Added new instance group: {new_group.name}")
+
+    @staticmethod
+    def ask(context: "CommandContext", params: dict):
+        group_name = context.ui.prompt_user_input("Enter the name for the new instance group:")
+        if group_name:
+            params["group_name"] = group_name
+            return True
+        return False  
+
 class OpenWebsite(AppCommand):
     @staticmethod
     def do_action(context: CommandContext, params: dict):