lazylabel-gui 1.0.0__py3-none-any.whl

lazylabel/controls.py

@@ -0,0 +1,108 @@
+from PyQt6.QtWidgets import (
+    QWidget,
+    QVBoxLayout,
+    QPushButton,
+    QLabel,
+    QFrame,
+    QTableWidget,
+    QTreeView,
+    QAbstractItemView,
+    QHBoxLayout,
+    QComboBox,
+    QHeaderView,
+)
+from PyQt6.QtCore import Qt
+from .reorderable_class_table import ReorderableClassTable
+
+
+class ControlPanel(QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        layout = QVBoxLayout(self)
+        layout.setAlignment(Qt.AlignmentFlag.AlignTop)
+        self.mode_label = QLabel("Mode: Points")
+        font = self.mode_label.font()
+        font.setPointSize(14)
+        font.setBold(True)
+        self.mode_label.setFont(font)
+        layout.addWidget(self.mode_label)
+        self.btn_sam_mode = QPushButton("Point Mode (1)")
+        self.btn_polygon_mode = QPushButton("Polygon Mode (2)")
+        self.btn_selection_mode = QPushButton("Selection Mode (E)")
+        layout.addWidget(self.btn_sam_mode)
+        layout.addWidget(self.btn_polygon_mode)
+        layout.addWidget(self.btn_selection_mode)
+        layout.addSpacing(20)
+        line1 = QFrame()
+        line1.setFrameShape(QFrame.Shape.HLine)
+        layout.addWidget(line1)
+        layout.addSpacing(10)
+        self.btn_clear_points = QPushButton("Clear Clicks (C)")
+        layout.addWidget(self.btn_clear_points)
+        layout.addStretch()
+        self.device_label = QLabel("Device: Unknown")
+        layout.addWidget(self.device_label)
+        self.setFixedWidth(250)
+
+
+class RightPanel(QWidget):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        layout = QVBoxLayout(self)
+
+        # File Explorer
+        file_explorer_layout = QVBoxLayout()
+        self.btn_open_folder = QPushButton("Open Image Folder")
+        self.file_tree = QTreeView()
+        file_explorer_layout.addWidget(self.btn_open_folder)
+        file_explorer_layout.addWidget(self.file_tree)
+        layout.addLayout(file_explorer_layout)
+
+        # Status Label
+        self.status_label = QLabel("")
+        self.status_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        layout.addWidget(self.status_label)
+
+        # Segment Table
+        segment_layout = QVBoxLayout()
+        class_filter_layout = QHBoxLayout()
+        class_filter_layout.addWidget(QLabel("Filter Class:"))
+        self.class_filter_combo = QComboBox()
+        class_filter_layout.addWidget(self.class_filter_combo)
+        segment_layout.addLayout(class_filter_layout)
+
+        self.segment_table = QTableWidget()
+        self.segment_table.setColumnCount(3)
+        self.segment_table.setHorizontalHeaderLabels(["Index", "Class", "Type"])
+        self.segment_table.horizontalHeader().setSectionResizeMode(
+            QHeaderView.ResizeMode.Stretch
+        )
+        self.segment_table.setSelectionBehavior(
+            QAbstractItemView.SelectionBehavior.SelectRows
+        )
+        self.segment_table.setSortingEnabled(True)
+        segment_layout.addWidget(self.segment_table)
+
+        segment_action_layout = QHBoxLayout()
+        self.btn_merge_selection = QPushButton("Merge to Class")
+        self.btn_delete_selection = QPushButton("Delete")
+        segment_action_layout.addWidget(self.btn_merge_selection)
+        segment_action_layout.addWidget(self.btn_delete_selection)
+        segment_layout.addLayout(segment_action_layout)
+        layout.addLayout(segment_layout, 2)
+
+        # Class Table
+        class_layout = QVBoxLayout()
+        class_layout.addWidget(QLabel("Class Order:"))
+        self.class_table = ReorderableClassTable()
+        self.class_table.setColumnCount(1)
+        self.class_table.setHorizontalHeaderLabels(["Class ID"])
+        self.class_table.horizontalHeader().setSectionResizeMode(
+            QHeaderView.ResizeMode.Stretch
+        )
+        class_layout.addWidget(self.class_table)
+        self.btn_reassign_classes = QPushButton("Reassign Class IDs")
+        class_layout.addWidget(self.btn_reassign_classes)
+        layout.addLayout(class_layout, 1)
+
+        self.setFixedWidth(350)

lazylabel/custom_file_system_model.py

@@ -0,0 +1,57 @@
+import os
+from PyQt6.QtCore import Qt, QModelIndex, QDir
+from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
+
+
+class CustomFileSystemModel(QFileSystemModel):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setFilter(QDir.Filter.NoDotAndDotDot | QDir.Filter.Files)
+        self.setNameFilterDisables(False)
+        self.setNameFilters(["*.png", "*.jpg", "*.jpeg", "*.tiff"])
+        self.highlighted_path = None
+
+    def set_highlighted_path(self, path):
+        self.highlighted_path = path
+
+    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
+        return 2
+
+    def headerData(
+        self,
+        section: int,
+        orientation: Qt.Orientation,
+        role: int = Qt.ItemDataRole.DisplayRole,
+    ):
+        if (
+            orientation == Qt.Orientation.Horizontal
+            and role == Qt.ItemDataRole.DisplayRole
+        ):
+            if section == 0:
+                return "File Name"
+            if section == 1:
+                return "Mask"
+        return super().headerData(section, orientation, role)
+
+    def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
+        if not index.isValid():
+            return None
+
+        # Handle the temporary highlight for saving
+        if role == Qt.ItemDataRole.BackgroundRole:
+            filePath = self.filePath(index)
+            if filePath == self.highlighted_path:
+                return QBrush(QColor("yellow"))
+
+        if index.column() == 1:
+            if role == Qt.ItemDataRole.CheckStateRole:
+                filePath = self.filePath(index.siblingAtColumn(0))
+                mask_path = os.path.splitext(filePath)[0] + ".npz"
+                return (
+                    Qt.CheckState.Checked
+                    if os.path.exists(mask_path)
+                    else Qt.CheckState.Unchecked
+                )
+            return None
+
+        return super().data(index, role)

lazylabel/editable_vertex.py

@@ -0,0 +1,25 @@
+from PyQt6.QtWidgets import QGraphicsEllipseItem, QGraphicsItem
+from PyQt6.QtCore import Qt
+from PyQt6.QtGui import QBrush, QPen
+
+
+class EditableVertexItem(QGraphicsEllipseItem):
+    def __init__(self, main_window, segment_index, vertex_index, x, y, w, h):
+        super().__init__(x, y, w, h)
+        self.main_window = main_window
+        self.segment_index = segment_index
+        self.vertex_index = vertex_index
+
+        self.setZValue(200)
+        self.setBrush(QBrush(Qt.GlobalColor.cyan))
+        self.setPen(QPen(Qt.GlobalColor.white, 1))
+        self.setFlag(QGraphicsItem.GraphicsItemFlag.ItemIsMovable)
+        self.setFlag(QGraphicsItem.GraphicsItemFlag.ItemSendsGeometryChanges)
+
+    def itemChange(self, change, value):
+        if change == QGraphicsItem.GraphicsItemChange.ItemPositionHasChanged:
+            new_pos = value
+            self.main_window.update_vertex_pos(
+                self.segment_index, self.vertex_index, new_pos
+            )
+        return super().itemChange(change, value)

lazylabel/hoverable_polygon_item.py

@@ -0,0 +1,23 @@
+from PyQt6.QtWidgets import QGraphicsPolygonItem
+from PyQt6.QtGui import QBrush
+
+
+class HoverablePolygonItem(QGraphicsPolygonItem):
+    def __init__(self, polygon, parent=None):
+        super().__init__(polygon, parent)
+        self.setAcceptHoverEvents(True)
+        self.default_brush = QBrush()
+        self.hover_brush = QBrush()
+
+    def set_brushes(self, default_brush, hover_brush):
+        self.default_brush = default_brush
+        self.hover_brush = hover_brush
+        self.setBrush(self.default_brush)
+
+    def hoverEnterEvent(self, event):
+        self.setBrush(self.hover_brush)
+        super().hoverEnterEvent(event)
+
+    def hoverLeaveEvent(self, event):
+        self.setBrush(self.default_brush)
+        super().hoverLeaveEvent(event)