Merge pull request #143 from KumarLabJax/select-all

add select all feature

Author: keithshep

src/jabs/behavior_search/behavior_search_util.py

@@ -193,7 +193,6 @@ def _gen_contig_true_intervals(
 
     match pred_query.search_kind:
         case PredictionSearchKind.POSITIVE_PREDICTION:
-            print("Searching for positive predictions...")
             crit_mask = animal_predictions == TrackLabels.Label.BEHAVIOR.value
         case PredictionSearchKind.NEGATIVE_PREDICTION:
             crit_mask = animal_predictions == TrackLabels.Label.NOT_BEHAVIOR.value

src/jabs/resources/docs/user_guide/user_guide.md

@@ -360,6 +360,10 @@ While not in select mode:
 
 - z, x, c: enter select mode
 
+In all modes:
+
+- ctrl+a (windows) / cmd+a (Mac): select all frames for the current timeline
+
 ### Other
 
 Switching subject:

src/jabs/ui/central_widget.py

@@ -423,26 +423,52 @@ def _start_selection(self, pressed):
         if pressed:
             self._controls.enable_label_buttons()
             self._selection_start = self._player_widget.current_frame()
+            self._selection_end = None
             self._stacked_timeline.start_selection(self._selection_start)
         else:
             self._controls.disable_label_buttons()
             self._stacked_timeline.clear_selection()
 
+    def select_all(self):
+        """Select all frames in the current video for the current identity and behavior."""
+        if not self._controls.select_button_is_checked and self._controls.select_button_enabled:
+            self._controls.toggle_select_button()
+
+        if self._controls.select_button_enabled:
+            num_frames = self._player_widget.num_frames()
+            if num_frames > 0:
+                self._controls.enable_label_buttons()
+                self._selection_start = 0
+                self._selection_end = num_frames - 1
+                self._stacked_timeline.start_selection(self._selection_start, self._selection_end)
+
+    @property
+    def _curr_selection_end(self):
+        """Get the end of the current selection.
+
+        If no selection end is set, return the current frame index.
+        """
+        return (
+            self._selection_end
+            if self._selection_end is not None
+            else self._player_widget.current_frame()
+        )
+
     def _label_behavior(self):
         """Apply behavior label to currently selected range of frames"""
-        start, end = sorted([self._selection_start, self._player_widget.current_frame()])
+        start, end = sorted([self._selection_start, self._curr_selection_end])
         self._get_label_track().label_behavior(start, end)
         self._label_button_common()
 
     def _label_not_behavior(self):
         """apply _not_ behavior label to currently selected range of frames"""
-        start, end = sorted([self._selection_start, self._player_widget.current_frame()])
+        start, end = sorted([self._selection_start, self._curr_selection_end])
         self._get_label_track().label_not_behavior(start, end)
         self._label_button_common()
 
     def _clear_behavior_label(self):
         """clear all behavior/not behavior labels from current selection"""
-        label_range = sorted([self._selection_start, self._player_widget.current_frame()])
+        label_range = sorted([self._selection_start, self._curr_selection_end])
         self._get_label_track().clear_labels(*label_range)
         self._label_button_common()
 

src/jabs/ui/main_window.py

@@ -229,7 +229,7 @@ def __init__(self, app_name: str, app_name_long: str, *args, **kwargs):
 
         # add behavior search
         self.behavior_search = QtGui.QAction("Search Behaviors", self)
-        self.behavior_search.setShortcut(QtGui.QKeySequence.Find)
+        self.behavior_search.setShortcut(QtGui.QKeySequence.StandardKey.Find)
         self.behavior_search.setStatusTip("Search for behaviors")
         self.behavior_search.setEnabled(False)
         self.behavior_search.triggered.connect(self._search_behaviors)
@@ -275,6 +275,12 @@ def __init__(self, app_name: str, app_name_long: str, *args, **kwargs):
         self.enable_segmentation_features.triggered.connect(self._toggle_segmentation_features)
         feature_menu.addAction(self.enable_segmentation_features)
 
+        # select all action
+        select_all_action = QtGui.QAction(self)
+        select_all_action.setShortcut(QtGui.QKeySequence.StandardKey.SelectAll)
+        select_all_action.triggered.connect(self._handle_select_all)
+        self.addAction(select_all_action)
+
         # playlist widget added to dock on left side of main window
         self.video_list = VideoListDockWidget(self)
         self.addDockWidget(Qt.DockWidgetArea.LeftDockWidgetArea, self.video_list)
@@ -684,3 +690,7 @@ def _on_label_overlay_mode_changed(self):
             self._central_widget.label_overlay_mode = PlayerWidget.LabelOverlay.LABEL
         elif self._label_overlay_preds.isChecked():
             self._central_widget.label_overlay_mode = PlayerWidget.LabelOverlay.PREDICTION
+
+    def _handle_select_all(self):
+        """Handle the Select All event"""
+        self._central_widget.select_all()

src/jabs/ui/stacked_timeline_widget/label_overview_widget/label_overview_widget.py

@@ -146,13 +146,15 @@ def reset(self) -> None:
         self._label_widget.reset()
         self._num_frames = 0
 
-    def start_selection(self, starting_frame: int) -> None:
+    def start_selection(self, starting_frame: int, ending_frame: int | None = None) -> None:
         """Start a selection from the given frame to the current frame.
 
         Args:
             starting_frame: The frame index where the selection begins.
+            ending_frame: Optional; the frame index where the selection ends. If None,
+                selection continues to current frame.
         """
-        self._label_widget.start_selection(starting_frame)
+        self._label_widget.start_selection(starting_frame, ending_frame)
 
     def clear_selection(self) -> None:
         """Clear the current selection.

src/jabs/ui/stacked_timeline_widget/label_overview_widget/manual_label_widget.py

@@ -70,6 +70,7 @@ def __init__(self, *args, **kwargs) -> None:
         # current position
         self._current_frame = 0
         self._selection_start = None
+        self._selection_end = None
 
         # information about the video needed to properly render widget
         self._num_frames = 0
@@ -230,7 +231,13 @@ def _draw_selection_overlay(self, painter: QPainter) -> None:
         painter.setBrush(self._selection_brush)
 
         # figure out the start and width of the selection rectangle
-        if self._selection_start < self._current_frame:
+        if self._selection_end is not None:
+            # we have an explicit end frame for the selection
+            selection_start = max(self._selection_start - start, 0)
+            selection_width = (
+                min(end, self._selection_end) - max(start, self._selection_start) + 1
+            ) * self._frame_width
+        elif self._selection_start < self._current_frame:
             # normal selection, start is lower than current frame
             selection_start = max(self._selection_start - start, 0)
             selection_width = (
@@ -328,22 +335,27 @@ def set_framerate(self, fps: int) -> None:
         """
         self._framerate = fps
 
-    def start_selection(self, start_frame: int) -> None:
+    def start_selection(self, start_frame: int, end_frame: int | None = None) -> None:
         """Begin highlighting a selection range on the label bar.
 
-        Sets the starting frame for the selection overlay, which will extend to the current frame.
+        Sets the starting frame for the selection overlay, which will extend to the current frame
+        unless an end frame is specified.
 
         Args:
             start_frame: The frame index where the selection begins.
+            end_frame: Optional; the frame index where the selection ends. If not provided,
+                the selection will extend to the current frame.
         """
         self._selection_start = start_frame
+        self._selection_end = end_frame
 
     def clear_selection(self) -> None:
         """Stop highlighting the selection range.
 
         Clears the selection overlay from the label bar.
         """
         self._selection_start = None
+        self._selection_end = None
 
     def reset(self) -> None:
         """Reset the widget to its initial state.
@@ -353,5 +365,6 @@ def reset(self) -> None:
         self._labels = None
         self._identity_mask = None
         self._selection_start = None
+        self._selection_end = None
         self._num_frames = 0
         self.update()

src/jabs/ui/stacked_timeline_widget/stacked_timeline_widget.py

@@ -48,12 +48,13 @@ def __init__(self, *args, **kwargs) -> None:
 
         self._active_identity_index = None
         self._selection_starting_frame = None
+        self._selection_ending_frame = None
         self._view_mode = self.ViewMode.LABELS_AND_PREDICTIONS
         self._identity_mode = self.IdentityMode.ACTIVE
         self._num_identities = 0
         self._num_frames = 0
         self._framerate = 0
-        self._label_overview_widgets = []
+        self._label_overview_widgets: list[LabelOverviewWidget] = []
         self._prediction_overview_widgets = []
         self._identity_frames = []
         self._frame_labels = FrameLabelsWidget(self)
@@ -230,7 +231,8 @@ def active_identity_index(self, value: int) -> None:
             # Transfer selection to new active widget if selection is active
             if selection_frame is not None:
                 self._label_overview_widgets[self._active_identity_index].start_selection(
-                    selection_frame
+                    selection_frame,
+                    self._selection_ending_frame,
                 )
 
             # Set active state or frame border depending on display mode
@@ -370,18 +372,24 @@ def set_predictions(
             widget.framerate = self.framerate
             widget.set_labels(predictions_list[i], probabilities_list[i])
 
-    def start_selection(self, starting_frame: int) -> None:
-        """Start a selection from the given frame on the active identity's widget.
+    def start_selection(self, starting_frame: int, ending_frame: int | None = None) -> None:
+        """Start a selection from the given frame(s) on the active identity's widget.
 
         Records the starting frame and initiates selection mode on the currently active identity.
+        If `ending_frame` is provided, it sets the selection range to include that frame as well.
+        If `ending_frame` is None, the selection continues to the current frame.
 
         Args:
             starting_frame: The frame index where the selection begins.
+            ending_frame: Optional; the frame index where the selection ends. If None,
+                selection continues to current frame.
         """
         if self._active_identity_index is not None:
             self._selection_starting_frame = starting_frame
+            self._selection_ending_frame = ending_frame
             self._label_overview_widgets[self._active_identity_index].start_selection(
-                starting_frame
+                starting_frame,
+                ending_frame,
             )
             self._label_overview_widgets[self._active_identity_index].update()
 
@@ -393,6 +401,7 @@ def clear_selection(self) -> None:
         if self._active_identity_index is not None:
             self._label_overview_widgets[self._active_identity_index].clear_selection()
             self._selection_starting_frame = None
+            self._selection_ending_frame = None
             self._label_overview_widgets[self._active_identity_index].update_labels()
 
     def reset(self) -> None: