(1->0) Add RecordingSession data structure and integrate with Labels

sleap/gui/app.py

@@ -467,6 +467,14 @@ def add_submenu_choices(menu, title, options, key):
             fileMenu, "replace videos", "Replace Videos...", self.commands.replaceVideo
         )
 
+        fileMenu.addSeparator()
+        add_menu_item(
+            fileMenu,
+            "add session",
+            "Add Recording Session...",
+            self.commands.addSession,
+        )
+
         fileMenu.addSeparator()
         add_menu_item(fileMenu, "save", "Save", self.commands.saveProject)
         add_menu_item(fileMenu, "save as", "Save As...", self.commands.saveProjectAs)

sleap/gui/commands.py

@@ -41,7 +41,8 @@ class which inherits from `AppCommand` (or a more specialized class such as
 import attr
 import cv2
 import numpy as np
-from qtpy import QtCore, QtGui, QtWidgets
+from qtpy import QtCore, QtWidgets, QtGui
+from qtpy.QtWidgets import QMessageBox, QProgressDialog
 
 from sleap.gui.dialogs.delete import DeleteDialog
 from sleap.gui.dialogs.filedialog import FileDialog
@@ -53,6 +54,7 @@ class which inherits from `AppCommand` (or a more specialized class such as
 from sleap.gui.state import GuiState
 from sleap.gui.suggestions import VideoFrameSuggestions
 from sleap.instance import Instance, LabeledFrame, Point, PredictedInstance, Track
+from sleap.io.cameras import RecordingSession
 from sleap.io.convert import default_analysis_filename
 from sleap.io.dataset import Labels
 from sleap.io.format.adaptor import Adaptor
@@ -432,6 +434,10 @@ def removeVideo(self):
         """Removes selected video from project."""
         self.execute(RemoveVideo)
 
+    def addSession(self):
+        """Shows gui for adding `RecordingSession`s to the project."""
+        self.execute(AddSession)
+
     def openSkeletonTemplate(self):
         """Shows gui for loading saved skeleton into project."""
         self.execute(OpenSkeleton, template=True)
@@ -1930,6 +1936,38 @@ def ask(context: CommandContext, params: dict) -> bool:
         return True
 
 
+class AddSession(EditCommand):
+    # topics = [UpdateTopic.session]
+
+    @staticmethod
+    def do_action(context: CommandContext, params: dict):
+
+        camera_calibration = params["camera_calibration"]
+        session = RecordingSession.load(filename=camera_calibration)
+
+        # Add session
+        context.labels.add_session(session)
+
+        # Load if no video currently loaded
+        if context.state["session"] is None:
+            context.state["session"] = session
+
+    @staticmethod
+    def ask(context: CommandContext, params: dict) -> bool:
+        """Shows gui for adding video to project."""
+        filters = ["Camera calibration (*.toml)"]
+        filename, selected_filter = FileDialog.open(
+            context.app,
+            dir=None,
+            caption="Select camera calibration...",
+            filter=";;".join(filters),
+        )
+
+        params["camera_calibration"] = filename
+
+        return len(filename) > 0
+
+
 class OpenSkeleton(EditCommand):
     topics = [UpdateTopic.skeleton]