refactor: enhance camera rotation functionality and integrate with live view

This commit is contained in:
2025-10-13 20:54:04 +02:00
parent 4d00f83083
commit 0bc6c01e7e
3 changed files with 46 additions and 11 deletions

View File

@@ -8,7 +8,7 @@ from core.media import MediaRepository
from core.camera.camera_manager import CameraManager
from ui.widgets.color_list_widget import ColorListWidget
from ui.widgets.thumbnail_list_widget import ThumbnailListWidget
from ui.widgets.split_view_widget import SplitView, CameraPlaceholder
from ui.widgets.split_view_widget import SplitView, CameraPlaceholder, ViewWithOverlay
class MainController:
@@ -23,6 +23,7 @@ class MainController:
self.thumbnail_list: ThumbnailListWidget = view.thumbnail_widget
self.split_view: SplitView = view.preview_widget
self.welcome_view: CameraPlaceholder = self.split_view.widget_start
self.live_view: ViewWithOverlay = self.split_view.widget_live
self.photo_button: QPushButton = view.photo_button
self.record_button: QPushButton = view.record_button
@@ -48,10 +49,9 @@ class MainController:
# UI control signals
self.photo_button.clicked.connect(self.take_photo)
# self.record_button.clicked.connect(self.toggle_record) # Placeholder
self.welcome_view.camera_start_btn.clicked.connect(self.camera_detect)
# You will need a way to select a camera, e.g., a combobox.
# self.view.camera_combobox.currentIndexChanged.connect(self.on_camera_selected_in_ui)
self.live_view.rotateCW.connect(self.camera_manager.rotate_right)
self.live_view.rotateCCW.connect(self.camera_manager.rotate_left)
def load_colors(self) -> None:
"""Loads colors from the database and populates the list."""

View File

@@ -18,6 +18,7 @@ class CameraWorker(QObject):
self.fps = 15
self.is_streaming = False
self.is_connected = False
self._rotation_index = 0
self._camera_mutex = QMutex()
@Slot()
@@ -93,7 +94,16 @@ class CameraWorker(QObject):
self.error_occurred.emit(error_msg)
return
if frame is not None:
if frame is None:
return
if self._rotation_index == 1:
frame = cv2.rotate(frame, cv2.ROTATE_90_CLOCKWISE)
elif self._rotation_index == 2:
frame = cv2.rotate(frame, cv2.ROTATE_180)
elif self._rotation_index == 3:
frame = cv2.rotate(frame, cv2.ROTATE_90_COUNTERCLOCKWISE)
# Process the frame and emit it.
rgb_image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
h, w, ch = rgb_image.shape
@@ -101,6 +111,14 @@ class CameraWorker(QObject):
pixmap = QPixmap.fromImage(qimg)
self.frame_ready.emit(pixmap)
@Slot()
def rotate_left(self):
self._rotation_index = (self._rotation_index - 1) % 4
@Slot()
def rotate_right(self):
self._rotation_index = (self._rotation_index + 1) % 4
class CameraController(QObject):
@@ -115,6 +133,8 @@ class CameraController(QObject):
_stop_camera_requested = Signal()
_start_stream_requested = Signal()
_stop_stream_requested = Signal()
_rotate_left_requested = Signal()
_rotate_right_requested = Signal()
def __init__(self, parent: QObject | None = None) -> None:
super().__init__(parent)
@@ -135,6 +155,8 @@ class CameraController(QObject):
self._stop_camera_requested.connect(self._worker.stop_camera)
self._start_stream_requested.connect(self._worker.start_stream)
self._stop_stream_requested.connect(self._worker.stop_stream)
self._rotate_left_requested.connect(self._worker.rotate_left)
self._rotate_right_requested.connect(self._worker.rotate_right)
# Initialize worker when thread starts
self._thread.started.connect(self._worker.initialize_worker)
@@ -163,3 +185,10 @@ class CameraController(QObject):
def stop_stream(self) -> None:
self._stop_stream_requested.emit()
def rotate_left(self):
self._rotate_left_requested.emit()
def rotate_right(self):
self._rotate_right_requested.emit()

View File

@@ -115,6 +115,12 @@ class CameraManager(QObject):
def get_active_camera_info(self) -> dict | None:
return self._active_camera_info
def rotate_left(self):
self._camera_controller.rotate_left()
def rotate_right(self):
self._camera_controller.rotate_right()
def shutdown(self) -> None:
"""Zamyka kontroler kamery i jego wątek."""
self.stop_camera()