Skip to content
Draft
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 27 additions & 0 deletions src/capture_method/__init__.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,14 @@
import os
import sys
from collections import OrderedDict
from collections.abc import Callable
from dataclasses import dataclass
from enum import EnumMeta, StrEnum, auto, unique
from itertools import starmap
from typing import TYPE_CHECKING, Never, TypedDict, cast

from cv2.typing import MatLike
from PySide6 import QtCore
from typing_extensions import override

from capture_method.CaptureMethodBase import CaptureMethodBase
Expand Down Expand Up @@ -221,3 +224,27 @@ def get_camera_info(index: int, device_name: str):
for camera_info in starmap(get_camera_info, enumerate(named_video_inputs))
if camera_info is not None
]


class CaptureMethodSignal(QtCore.QObject):
"""Made to look like a `QtCore.SignalInstance`, but with safe `connect`/`disconnect` methods."""

__frame_signal = QtCore.Signal(object)

def subscribe_to_new_frame(self, slot: Callable[[MatLike | None], object]):
try:
return self.__frame_signal.connect(slot, QtCore.Qt.ConnectionType.UniqueConnection)
except RuntimeError:
pass

def unsubscribe_from_new_frame(self, slot: Callable[[MatLike | None], object]):
try:
self.__frame_signal.disconnect(slot)
except RuntimeError:
pass

def _push_new_frame_to_subscribers(self, frame: MatLike | None, /):
return self.__frame_signal.emit(frame)

def __init__(self):
super().__init__()