Add CamerasTable to SessionsDock

sleap/gui/app.py

@@ -1094,7 +1094,8 @@ def _update_gui_state(self):
         has_selected_edge = self.state["selected_edge"] is not None
         has_selected_video = self.state["selected_video"] is not None
         has_video = self.state["video"] is not None
-
+        has_selected_camcorder = self.state["selected_camera_table"] is not None
+
         has_frame_range = bool(self.state["has_frame_range"])
         has_unsaved_changes = bool(self.state["has_changes"])
         has_videos = self.labels is not None and len(self.labels.videos) > 0
@@ -1148,6 +1149,7 @@ def _update_gui_state(self):
         self._buttons["show video"].setEnabled(has_selected_video)
         self._buttons["remove video"].setEnabled(has_video)
         self._buttons["delete instance"].setEnabled(has_selected_instance)
+        self._buttons["unlink video"].setEnabled(has_selected_camcorder)
         self.suggestions_dock.suggestions_form_widget.buttons[
             "generate_button"
         ].setEnabled(has_videos)
@@ -1236,6 +1238,12 @@ def _has_topic(topic_list):
         if _has_topic([UpdateTopic.frame, UpdateTopic.project_instances]):
             self.state["last_interacted_frame"] = self.state["labeled_frame"]
 
+        if _has_topic([UpdateTopic.sessions]):
+            self.sessions_dock.table["camera_table"].model().items = self.labels.sessions[0]
+
+
+
+
     def plotFrame(self, *args, **kwargs):
         """Plots (or replots) current frame."""
         if self.state["video"] is None:

sleap/gui/commands.py

@@ -82,6 +82,7 @@ class UpdateTopic(Enum):
     frame = 8
     project = 9
     project_instances = 10
+    sessions = 11
 
 
 class AppCommand:
@@ -655,6 +656,9 @@ def openPrereleaseVersion(self):
         """Open the current prerelease version."""
         self.execute(OpenPrereleaseVersion)
 
+    def unlink_video_from_camera(self):
+        """Unlinks video from a camera"""
+        self.execute(UnlinkVideo)
 
 # File Commands
 
@@ -3901,3 +3905,20 @@ def copy_to_clipboard(text: str):
     clipboard = QtWidgets.QApplication.clipboard()
     clipboard.clear(mode=clipboard.Clipboard)
     clipboard.setText(text, mode=clipboard.Clipboard)
+
+class UnlinkVideo(EditCommand):
+    topics = [UpdateTopic.sessions]
+
+    @staticmethod
+    def do_action(context: CommandContext, params: dict):
+        camcorder = context.state["selected_camera_table"]
+        recording_session = camcorder.sessions[0] # potentially buggy
+
+        video = camcorder.get_video(recording_session)
+
+        if video is not None and recording_session is not None:
+            recording_session.remove_video(video)
+
+
+
+

sleap/gui/dataviews.py

@@ -30,6 +30,7 @@
 from sleap.io.dataset import Labels
 from sleap.instance import LabeledFrame, Instance
 from sleap.skeleton import Skeleton
+from sleap.io.cameras import Camcorder, RecordingSession
 
 
 class GenericTableModel(QtCore.QAbstractTableModel):
@@ -652,3 +653,21 @@ def columnCount(self, parent):
     def flags(self, index: QtCore.QModelIndex):
         """Overrides Qt method, returns flags (editable etc)."""
         return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable
+
+class CamerasTableModel(GenericTableModel):
+    """
+    Table model for for multi-view, showing which cameras 
+    are assigned to which videos
+
+    Args: 
+        obj: 'RecordingSession' which has information of cameras 
+              and paired video
+    """
+
+    properties = ("camera", "video")
+
+    def object_to_items(self, obj: RecordingSession):
+        return obj.linked_cameras
+
+    def item_to_data(self, obj: RecordingSession, item: Camcorder):
+        return {'camera': item.name, 'video': obj.get_video(item).filename}

sleap/gui/widgets/docks.py

@@ -31,6 +31,7 @@
     SkeletonNodesTableModel,
     SuggestionsTableModel,
     VideosTableModel,
+    CamerasTableModel,
 )
 from sleap.gui.dialogs.formbuilder import YamlFormWidget
 from sleap.gui.widgets.views import CollapsibleWidget
@@ -574,9 +575,20 @@ def create_table_edit_buttons(self) -> QWidget:
 
 class SessionsDock(DockWidget):
     def __init__(self, main_window: Optional[QMainWindow]):
-        super().__init__(name="Sessions", main_window=main_window, model_type=None)
+        self.camera_model_type = CamerasTableModel
+        super().__init__(name="Sessions", main_window=main_window, model_type=[self.camera_model_type])
+
 
     def lay_everything_out(self) -> None:
+        if self.table is None:
+            self.create_tables()
+
+
+        self.wgt_layout.addWidget(self.camera_table)
+
+        video_unlink_button = self.create_video_unlink_button()
+        self.wgt_layout.addWidget(video_unlink_button)
+
         triangulation_options = self.create_triangulation_options()
         self.wgt_layout.addWidget(triangulation_options)
 
@@ -601,3 +613,34 @@ def create_triangulation_options(self) -> QWidget:
         hbw = QWidget()
         hbw.setLayout(hb)
         return hbw
+
+    def create_models(self) -> GenericTableModel:
+        main_window = self.main_window
+        self.camera_model = self.camera_model_type(
+            items=main_window.state["selected_session"], context=main_window.commands
+        )
+
+        return {"camera_model": self.camera_model}
+
+    def create_tables(self) -> GenericTableView:
+        if self.model is None:
+            self.create_models()
+
+        main_window = self.main_window
+        self.camera_table = GenericTableView(
+            state=main_window.state,
+            row_name="camera_table",
+            model=self.camera_model,
+        )
+
+        return {"camera_table": self.camera_table}
+
+    def create_video_unlink_button(self) -> QWidget:
+        main_window = self.main_window
+
+        hb = QHBoxLayout()
+        self.add_button(hb, "Unlink Video", main_window.commands.unlink_video_from_camera)
+
+        hbw = QWidget()
+        hbw.setLayout(hb)
+        return hbw

tests/gui/test_dataviews.py

@@ -2,6 +2,7 @@
 import pytestqt
 
 from sleap.gui.dataviews import *
+from sleap.io.cameras import RecordingSession
 
 
 def test_skeleton_nodes(qtbot, centered_pair_predictions):
@@ -84,3 +85,45 @@ def test_table_sort_string(qtbot):
     # Make sure we can sort with both numbers and strings (i.e., "")
     table.model().sort(0)
     table.model().sort(1)
+
+def test_camera_table(qtbot, multiview_min_session_labels):
+    table = CamerasTableModel(items=multiview_min_session_labels.sessions[0])
+
+    assert table.columnCount() == 2
+    assert table.rowCount() == 8
+
+    num_rows = table.rowCount()
+
+    table = GenericTableView(
+        row_name="instance",
+        is_sortable=True,
+        name_prefix="",
+        model=CamerasTableModel(items=multiview_min_session_labels.sessions[0]),
+    )
+
+    # Testing if all comcorders are presented in the correct row
+    camcorders = multiview_min_session_labels.sessions[0].linked_cameras
+
+    for i in range(num_rows):
+        table.selectRow(i)
+        assert table.getSelectedRowItem() == camcorders[i]
+        assert table.model().data(table.currentIndex()) == camcorders[i].name
+
+    # Testing if a comcorder change is reflected
+    idxs_to_remove = [1, 2, 7]
+    for idx in idxs_to_remove:
+        multiview_min_session_labels.sessions[0].remove_video(camcorders[idx].get_video(multiview_min_session_labels.sessions[0]))
+
+    removed_camcorder = [cam for i, cam in enumerate(camcorders) if i not in idxs_to_remove]
+
+    for i in range(num_rows-3):
+        table.selectRow(i)
+        assert table.getSelectedRowItem() == camcorders[i]
+        assert table.model().data(table.currentIndex()) == removed_camcorder[i].name
+
+
+
+
+
+
+

tests/gui/widgets/test_docks.py

@@ -4,12 +4,13 @@
 import pytest
 from sleap import Labels, Video
 from sleap.gui.app import MainWindow
-from sleap.gui.commands import OpenSkeleton
+from sleap.gui.commands import OpenSkeleton, UnlinkVideo
 from sleap.gui.widgets.docks import (
     InstancesDock,
     SuggestionsDock,
     VideosDock,
     SkeletonDock,
+    SessionsDock, 
 )
 
 
@@ -107,3 +108,39 @@ def test_instances_dock(qtbot):
     assert dock.name == "Instances"
     assert dock.main_window is main_window
     assert dock.wgt_layout is dock.widget().layout()
+
+def test_sessions_dock(qtbot, multiview_min_session_labels):
+    main_window = MainWindow()
+    dock = SessionsDock(main_window)
+
+    # Testing if dock object loads corretly
+    assert dock.name == "Sessions"
+    assert dock.main_window is main_window
+    assert dock.wgt_layout is dock.widget().layout()
+
+    # loading label file
+    main_window.commands.loadLabelsObject(multiview_min_session_labels)
+
+
+    # Testing if cameras_table is loaded correctly
+    camcorders = multiview_min_session_labels.sessions[0].linked_cameras
+
+    for i in range(8):
+        main_window.sessions_dock.camera_table.selectRow(i)
+        assert main_window.sessions_dock.camera_table.getSelectedRowItem() == camcorders[i]
+        assert main_window.sessions_dock.camera_table.model().data(main_window.sessions_dock.camera_table.currentIndex()) == camcorders[i].name
+
+    # Testing if unlink video command works for camera table
+    idxs_to_remove = [0, 4, 7]
+    to_remove_camcorders = [cam for i, cam in enumerate(camcorders) if i in idxs_to_remove]
+    leftover_camcorder = [cam for i, cam in enumerate(camcorders) if i not in idxs_to_remove]
+
+    for cam in to_remove_camcorders:
+        main_window.state["selected_camera_table"] = cam
+        main_window._buttons["unlink video"].click()
+
+
+    for i in range(5):
+        main_window.sessions_dock.camera_table.selectRow(i)
+        assert main_window.sessions_dock.camera_table.getSelectedRowItem() == leftover_camcorder[i]
+        assert main_window.sessions_dock.camera_table.model().data(main_window.sessions_dock.camera_table.currentIndex()) == leftover_camcorder[i].name