niels/color_mode_selector_update

src/napari_deeplabcut/_widgets.py

@@ -620,7 +620,7 @@ def __init__(self, napari_viewer):
         self._radio_group = self._form_mode_radio_buttons()
 
         # form color scheme display + color mode selector
-        self._color_mode_selector = self._form_color_mode_selector()
+        self._color_grp, self._color_mode_selector = self._form_color_mode_selector()
         self._display = ColorSchemeDisplay(parent=self)
         self._color_scheme_display = self._form_color_scheme_display(self.viewer)
         self._view_scheme_cb.toggled.connect(self._show_color_scheme)
@@ -849,7 +849,7 @@ def _func():
             self.color_mode = group.checkedButton().text()
 
         group.buttonClicked.connect(_func)
-        return group
+        return group_box, group
 
     def _form_color_scheme_display(self, viewer):
         self.viewer.layers.events.inserted.connect(self._update_color_scheme)
@@ -865,7 +865,10 @@ def to_hex(nparray):
             return res
 
         self._display.reset()
-        mode = "label" if self.color_mode == str(keypoints.ColorMode.BODYPART) else "id"
+        mode = "label"
+        if self.color_mode == str(keypoints.ColorMode.INDIVIDUAL):
+            mode = "id"
+
         for layer in self.viewer.layers:
             if isinstance(layer, Points) and layer.metadata:
                 self._display.update_color_scheme(
@@ -874,7 +877,6 @@ def to_hex(nparray):
                         for name, color in layer.metadata["face_color_cycles"][mode].items()
                     }
                 )
-                break
 
     def _remap_frame_indices(self, layer):
         if not self._images_meta.get("paths"):
@@ -1055,7 +1057,12 @@ def on_remove(self, event):
             self._trails = None
 
     def on_active_layer_change(self, event) -> None:
-        """Hides all KeypointsDropdownMenu that aren't for the selected layer"""
+        """Updates the GUI when the active layer changes
+            * Hides all KeypointsDropdownMenu that aren't for the selected layer
+            * Sets the visibility of the "Color mode" box to True if the selected layer
+                is a multi-animal one, or False otherwise
+        """
+        self._color_grp.setVisible(self._is_multianimal(event.value))
         menu_idx = -1
         if event.value is not None and isinstance(event.value, Points):
             menu_idx = self._layer_to_menu.get(event.value, -1)
@@ -1066,17 +1073,15 @@ def on_active_layer_change(self, event) -> None:
             else:
                 menu.setHidden(True)
 
-
     def _update_colormap(self, colormap_name):
-        for layer in self.viewer.layers:
+        for layer in self.viewer.layers.selection:
             if isinstance(layer, Points) and layer.metadata:
                 face_color_cycle_maps = build_color_cycles(
                     layer.metadata["header"], colormap_name,
                 )
                 layer.metadata["face_color_cycles"] = face_color_cycle_maps
-                if self.color_mode == keypoints.ColorMode.BODYPART:
-                    face_color_prop = "label"
-                else:
+                face_color_prop = "label"
+                if self.color_mode == str(keypoints.ColorMode.INDIVIDUAL):
                     face_color_prop = "id"
 
                 layer.face_color = face_color_prop
@@ -1090,7 +1095,11 @@ def cycle_through_label_modes(self, *args):
 
     @register_points_action("Change color mode")
     def cycle_through_color_modes(self, *args):
-        self.color_mode = next(keypoints.ColorMode)
+        if (
+            self._active_layer_is_multianimal()
+            or self.color_mode != str(keypoints.ColorMode.BODYPART)
+        ):
+            self.color_mode = next(keypoints.ColorMode)
 
     @property
     def label_mode(self):
@@ -1138,6 +1147,27 @@ def color_mode(self, mode: Union[str, keypoints.ColorMode]):
 
         self._update_color_scheme()
 
+    def _is_multianimal(self, layer) -> bool:
+        is_multi = False
+        if layer is not None and isinstance(layer, Points):
+            try:
+                header = layer.metadata.get("header")
+                if header is not None:
+                    ids = header.individuals
+                    is_multi = len(ids) > 0 and ids[0] != ""
+            except AttributeError:
+                pass
+
+        return is_multi
+
+    def _active_layer_is_multianimal(self) -> bool:
+        """Returns: whether the active layer is a multi-animal points layer"""
+        for layer in self.viewer.layers.selection:
+            if self._is_multianimal(layer):
+                return True
+
+        return False
+
 
 @Points.bind_key("E")
 def toggle_edge_color(layer):