Switched to a Qt6 Model/View framework to store data once only

Author: SilvioGiancola

osl_visualizer/dialogs.py

@@ -0,0 +1,16 @@
+import os
+from PyQt6 import uic
+from PyQt6.QtWidgets import QDialog
+
+class ConfigDialog(QDialog):
+    """Configuration dialog for user settings."""
+    def __init__(self, parent=None, current_jump_before=5000):
+        super().__init__(parent)
+        uic.loadUi(os.path.join(os.path.dirname(__file__), "ui/configdialog.ui"), self)
+        self.jumpBeforeSpinBox.setValue(current_jump_before)
+        self.okButton.clicked.connect(self.accept)
+        self.cancelButton.clicked.connect(self.reject)
+
+    def get_jump_before(self):
+        """Return the currently set 'jump before annotation' value."""
+        return self.jumpBeforeSpinBox.value()

osl_visualizer/main.py

@@ -16,3 +16,4 @@
 
     viewer.show()
     sys.exit(app.exec())
+# 

osl_visualizer/models.py

@@ -0,0 +1,64 @@
+from PyQt6.QtCore import Qt, QAbstractListModel, QModelIndex
+from utils import ms_to_hms_ms
+
+class VideoListModel(QAbstractListModel):
+    def __init__(self, videos=None):
+        super().__init__()
+        self.videos = videos or []
+
+    def rowCount(self, parent=QModelIndex()):
+        return len(self.videos)
+
+    def data(self, index, role):
+        if not index.isValid() or not (0 <= index.row() < len(self.videos)):
+            return None
+        video = self.videos[index.row()]
+        if role == Qt.ItemDataRole.DisplayRole:
+            path = video.get("path", "unknown")
+            n_events = len(video.get("annotations", []))
+            return f"{path} ({n_events} events)"
+        if role == Qt.ItemDataRole.UserRole:
+            return video
+        return None
+
+    def set_videos(self, videos):
+        self.beginResetModel()
+        self.videos = videos or []
+        self.endResetModel()
+
+class AnnotationListModel(QAbstractListModel):
+    def __init__(self, annotations=None):
+        super().__init__()
+        self.annotations = annotations or []
+
+    def rowCount(self, parent=QModelIndex()):
+        return len(self.annotations)
+
+    def data(self, index, role):
+        if not index.isValid() or not (0 <= index.row() < len(self.annotations)):
+            return None
+        ann = self.annotations[index.row()]
+        if role == Qt.ItemDataRole.DisplayRole:
+            return f"[{ms_to_hms_ms(ann['position'])}] {ann['label']}"
+        if role == Qt.ItemDataRole.UserRole:
+            return ann
+        return None
+
+    def set_annotations(self, annotations):
+        self.beginResetModel()
+        self.annotations = annotations or []
+        self.endResetModel()
+
+    def add_annotation(self, annotation):
+        idx = 0
+        while idx < len(self.annotations) and annotation["position"] > self.annotations[idx]["position"]:
+            idx += 1
+        self.beginInsertRows(QModelIndex(), idx, idx)
+        self.annotations.insert(idx, annotation)
+        self.endInsertRows()
+        return idx
+
+    def remove_annotation(self, idx):
+        self.beginRemoveRows(QModelIndex(), idx, idx)
+        del self.annotations[idx]
+        self.endRemoveRows()

osl_visualizer/ui/mainwindow.ui

@@ -44,7 +44,7 @@
          </widget>
         </item>
         <item>
-         <widget class="QListWidget" name="videoListWidget"/>
+         <widget class="QListView" name="videoListView"/>
         </item>
        </layout>
       </widget>
@@ -113,7 +113,7 @@
          </widget>
         </item>
         <item>
-         <widget class="QListWidget" name="annotationListWidget"/>
+         <widget class="QListView" name="annotationListView"/>
         </item>
         <item>
          <widget class="QLabel" name="editLabel">