Add Color by Instance Group preference saving

sleap/gui/app.py

@@ -224,6 +224,7 @@ def closeEvent(self, event):
         prefs["color predicted"] = self.state["color predicted"]
         prefs["trail shade"] = self.state["trail_shade"]
         prefs["share usage data"] = self.state["share usage data"]
+        prefs["color_by_instance_group"] = self.state["color_by_instance_group"]
 
         # Save preferences.
         prefs.save()
@@ -253,6 +254,11 @@ def closeEvent(self, event):
                 self.commands.saveProject()
                 # accept event (closes window)
                 event.accept()
+
+    def _save_color_by_instance_group(self, value):
+        """Update the color_by_instance_group preference and save it."""
+        prefs["color_by_instance_group"] = value
+        prefs.save()
 
     def dragEnterEvent(self, event):
         # TODO: Parse filenames and accept only if valid ext (or folder)
@@ -298,6 +304,8 @@ def labels(self, value):
     def _initialize_gui(self):
         """Creates menus, dock windows, starts timers to update gui state."""
 
+        self.state["color_by_instance_group"] = prefs["color_by_instance_group"]
+
         self._create_color_manager()
         self._create_video_player()
         self.statusBar()
@@ -648,6 +656,9 @@ def prev_vid():
         viewMenu.addSeparator()
         add_menu_check_item(viewMenu, "color predicted", "Color Predicted Instances")
 
+        add_menu_check_item(viewMenu, "color_by_instance_group", "Color by Instance Group")
+        self.state.connect("color_by_instance_group", lambda value: self._save_color_by_instance_group(value))
+
         add_submenu_choices(
             menu=viewMenu,
             title="Color Palette",

sleap/prefs.py

@@ -15,6 +15,7 @@ class Preferences(object):
         "medium step size": 10,
         "large step size": 100,
         "color predicted": False,
+        "color_by_instance_group": False,
         "propagate track labels": True,
         "palette": "standard",
         "bold lines": False,