645 lines
25 KiB
Python
645 lines
25 KiB
Python
# coding:utf-8
|
||
from PySide6.QtCore import Qt, QTimer, Signal, QObject
|
||
from PySide6.QtGui import QImage, QPixmap, QPalette, QBrush, QCursor, QIcon, QPainter, QFont
|
||
from PySide6.QtWidgets import (
|
||
QWidget,
|
||
QVBoxLayout,
|
||
QHBoxLayout,
|
||
QLabel,
|
||
QApplication,
|
||
QFrame,
|
||
QPushButton,
|
||
QStackedWidget,
|
||
QSizePolicy
|
||
)
|
||
from typing import Optional
|
||
|
||
import cv2
|
||
import time
|
||
from threading import Thread, Lock
|
||
import sys
|
||
|
||
from .switch_button import SwitchButton
|
||
|
||
import resources.resources_rc
|
||
from utils.image_paths import ImagePaths
|
||
|
||
|
||
class VideoStream(QObject):
|
||
# 系统(信息)信号
|
||
info_signal = Signal(str, str)
|
||
|
||
# 警告信号（重试）: 发送摄像头名称和状态描述
|
||
warning_signal = Signal(str, str)
|
||
|
||
# 异常错误信号（超次数）
|
||
error_signal = Signal(str, str)
|
||
|
||
def __init__(self, rtsp_url, name="Stream", max_retry=3):
|
||
super().__init__()
|
||
self.rtsp_url = rtsp_url
|
||
self.name = name
|
||
self.cap = None
|
||
self.frame = None
|
||
self.timestamp = 0
|
||
self.lock = Lock()
|
||
self.running = False
|
||
self.thread = None
|
||
|
||
self.max_retry = max_retry
|
||
self.retry_count = 0
|
||
self.manual_stop = False # 是否停止自动重连
|
||
|
||
def start(self):
|
||
self.running = True
|
||
self.thread = Thread(target=self.update, daemon=True)
|
||
self.thread.start()
|
||
return self
|
||
|
||
def reset_retry(self):
|
||
with self.lock:
|
||
self.retry_count = 0
|
||
self.manual_stop = False
|
||
if self.cap:
|
||
self.cap.release()
|
||
self.cap = None
|
||
self.info_signal.emit(self.name, "已手动重置, 开始尝试连接...")
|
||
|
||
def _handle_max_retry(self, failure_type: str):
|
||
"""
|
||
处理超过最大重试次数的情况
|
||
failure_type: 失败类型（"连接" 或 "读取"）
|
||
"""
|
||
self.manual_stop = True
|
||
error_msg = f"{failure_type}摄像头失败, 已经超过最大次数（{self.max_retry}次），可能设备故障，请检查后手动点击重连"
|
||
print(f"[{self.name}] {error_msg}")
|
||
self.error_signal.emit(self.name, error_msg)
|
||
time.sleep(1)
|
||
|
||
def _handle_retry_delay(self):
|
||
"""计算并执行重试前的延迟（递增间隔）"""
|
||
sleep_time = min(0.5 + self.retry_count * 0.5, 5)
|
||
warning_msg = (
|
||
f"第{self.retry_count}次连接失败，将在{sleep_time:.1f}秒后再次尝试..."
|
||
)
|
||
self.warning_signal.emit(self.name, warning_msg)
|
||
print(f"[{self.name}] {warning_msg}")
|
||
time.sleep(sleep_time)
|
||
|
||
def update(self):
|
||
while self.running:
|
||
if self.manual_stop:
|
||
time.sleep(1)
|
||
continue
|
||
|
||
if self.cap is None or not self.cap.isOpened():
|
||
print(
|
||
f"[{self.name}]正在连接 RTSP(第{self.retry_count+1}次): {self.rtsp_url}"
|
||
)
|
||
info_msg = f"正在连接 RTSP(第{self.retry_count+1}次): {self.rtsp_url}"
|
||
self.info_signal.emit(self.name, info_msg)
|
||
try:
|
||
self.cap = cv2.VideoCapture(self.rtsp_url, cv2.CAP_FFMPEG)
|
||
if hasattr(cv2, "CAP_PROP_BUFFERSIZE"):
|
||
self.cap.set(cv2.CAP_PROP_BUFFERSIZE, 2)
|
||
if hasattr(cv2, "CAP_PROP_READ_TIMEOUT"):
|
||
self.cap.set(cv2.CAP_PROP_READ_TIMEOUT, 2000)
|
||
if hasattr(cv2, "CAP_PROP_TCP_NODELAY"):
|
||
self.cap.set(cv2.CAP_PROP_TCP_NODELAY, 1)
|
||
except Exception as ex:
|
||
print(f"[{self.name}]连接失败: {ex}")
|
||
warning_msg = f"连接失败: {str(ex)}，准备重试..."
|
||
self.warning_signal.emit(self.name, warning_msg)
|
||
if self.cap:
|
||
self.cap.release()
|
||
self.cap = None
|
||
|
||
if self.cap is None or not self.cap.isOpened():
|
||
self.retry_count += 1
|
||
if self.retry_count >= self.max_retry:
|
||
self._handle_max_retry("连接")
|
||
continue
|
||
self._handle_retry_delay()
|
||
continue
|
||
else:
|
||
self.retry_count = 0
|
||
self.info_signal.emit(self.name, "连接成功")
|
||
|
||
ret, frame = self.cap.read()
|
||
if ret:
|
||
with self.lock:
|
||
self.frame = frame.copy()
|
||
self.timestamp = time.time()
|
||
self.retry_count = 0
|
||
else:
|
||
print(f"[{self.name}] 读取失败，准备重连...")
|
||
warning_msg = "读取帧失败，准备重连..."
|
||
self.warning_signal.emit(self.name, warning_msg)
|
||
if self.cap:
|
||
self.cap.release()
|
||
self.cap = None
|
||
self.retry_count += 1
|
||
if self.retry_count >= self.max_retry:
|
||
self._handle_max_retry("读取")
|
||
continue
|
||
self._handle_retry_delay()
|
||
|
||
def read(self):
|
||
with self.lock:
|
||
if self.frame is not None and time.time() - self.timestamp < 1.5:
|
||
return self.frame.copy()
|
||
else:
|
||
return None
|
||
|
||
def stop(self):
|
||
self.running = False
|
||
if self.thread is not None:
|
||
self.thread.join(timeout=0.5)
|
||
if self.cap is not None:
|
||
self.cap.release()
|
||
print(f"[{self.name}] 视频流已停止")
|
||
|
||
|
||
# ====================== 摄像头框功能模块 ======================
|
||
class CameraModule(QWidget):
|
||
"""单个摄像头模块：原图显示"""
|
||
|
||
# 重置信号，用于通知需要重置连接 (重连)
|
||
reset_signal = Signal()
|
||
|
||
# 视频显示信号，用于通知是否显示视频（刷新视频帧）
|
||
# 发送摄像头名 和 是否显示
|
||
video_display_signal = Signal(str, bool)
|
||
|
||
def __init__(
|
||
self, camera_name="摄像头", rtsp_url="", need_rotate_180=True, show_ai = False, parent=None
|
||
):
|
||
super().__init__(parent)
|
||
self.setObjectName("cameraModule")
|
||
self.camera_name = camera_name
|
||
self.rtsp_url = rtsp_url
|
||
self.need_rotate_180 = need_rotate_180 # 画面是否需要旋转180度后显示
|
||
self.show_ai = show_ai # 是否需要展示ai（为True会多出AI显示按钮）
|
||
|
||
# 初始化AI显示相关变量为None
|
||
self.ai_display_group = None
|
||
self.ai_display_label = None
|
||
self.ai_display_switch = None
|
||
|
||
self.setup_ui()
|
||
|
||
def setup_ui(self):
|
||
|
||
layout = QVBoxLayout(self)
|
||
layout.setSpacing(0)
|
||
layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
# --- 0. 标题布局： 标题 + 显示标签及开关 ---
|
||
title_layout = QHBoxLayout()
|
||
title_layout.setContentsMargins(0, 0, 0, 4)
|
||
title_layout.setSpacing(0)
|
||
|
||
self.title_label = QLabel()
|
||
self.title_label.setAlignment(Qt.AlignLeft)
|
||
self.title_label.setText(f"{self.camera_name}视频")
|
||
# self.title_label.setFixedWidth(212)
|
||
self.title_label.setFixedSize(212, 21)
|
||
self.title_label.setObjectName("cameraTitleLabel")
|
||
# background-image: url({ImagePaths.VIDEO_TITLE_BACKGROUND});
|
||
# min-width: 126px;
|
||
self.title_label.setStyleSheet(
|
||
f"""
|
||
#cameraTitleLabel {{
|
||
font-size: 18px;
|
||
color: #16ffff;
|
||
background-image: url({ImagePaths.VIDEO_TITLE_BACKGROUND});
|
||
padding-left: 12px;
|
||
}}
|
||
"""
|
||
)
|
||
|
||
# 1、创建【显示标签+开关】组合容器
|
||
self.display_group = QWidget() # 容器：包裹标签和开关
|
||
display_group_layout = QHBoxLayout(self.display_group)
|
||
display_group_layout.setContentsMargins(0, 0, 0, 0) # 容器内边距为0
|
||
display_group_layout.setSpacing(0)
|
||
|
||
# 显示标签
|
||
self.display_label = QLabel("显示")
|
||
self.display_label.setObjectName("displayLabel")
|
||
# font-weight: bold;
|
||
self.display_label.setStyleSheet("""
|
||
#displayLabel {
|
||
font-size: 18px;
|
||
color: #16ffff;
|
||
margin: 0px;
|
||
padding: 0px;
|
||
}
|
||
""")
|
||
# 让标签垂直居中
|
||
self.display_label.setFixedWidth(40)
|
||
self.display_label.setAlignment(Qt.AlignVCenter | Qt.AlignLeft)
|
||
|
||
# 显示开关
|
||
self.display_switch = SwitchButton()
|
||
self.display_switch.setChecked(True)
|
||
self.display_switch.switched.connect(self.onDisplayButtonSwitched)
|
||
|
||
# 将显示标签和开关添加到组合容器布局
|
||
display_group_layout.addWidget(self.display_label)
|
||
display_group_layout.addWidget(self.display_switch, alignment=Qt.AlignLeft)
|
||
|
||
if self.show_ai: # 当需要显示AI时才生成
|
||
# 2、创建【AI算法标签+开关】组合容器
|
||
self.ai_display_group = QWidget() # 容器：包裹AI显示标签和开关
|
||
ai_display_group_layout = QHBoxLayout(self.ai_display_group)
|
||
ai_display_group_layout.setContentsMargins(0, 0, 0, 0) # 容器内边距为0
|
||
ai_display_group_layout.setSpacing(0)
|
||
|
||
# AI算法显示标签
|
||
self.ai_display_label = QLabel("AI算法")
|
||
self.ai_display_label.setObjectName("aiDisplayLabel")
|
||
# font-weight: bold;
|
||
self.ai_display_label.setStyleSheet("""
|
||
#aiDisplayLabel {
|
||
font-size: 18px;
|
||
color: #16ffff;
|
||
margin: 0px;
|
||
padding-left: 2px;
|
||
}
|
||
""")
|
||
# self.ai_display_label.setFixedWidth(40)
|
||
self.ai_display_label.setAlignment(Qt.AlignVCenter | Qt.AlignLeft)
|
||
|
||
# AI开关
|
||
self.ai_display_switch = SwitchButton()
|
||
self.ai_display_switch.setChecked(False)
|
||
# self.ai_display_switch.switched.connect(self.onDisplayButtonSwitched) # 在camera_controller中处理
|
||
|
||
# 将AI显示标签和开关添加到组合容器布局
|
||
ai_display_group_layout.addWidget(self.ai_display_label)
|
||
ai_display_group_layout.addWidget(self.ai_display_switch, alignment=Qt.AlignLeft)
|
||
|
||
# 添加到标题布局
|
||
title_layout.addWidget(self.title_label, alignment=Qt.AlignLeft)
|
||
title_layout.addWidget(self.display_group, alignment=Qt.AlignRight)
|
||
if self.show_ai: # 需要添加 AI显示控件
|
||
title_layout.addWidget(self.ai_display_group, alignment=Qt.AlignLeft)
|
||
self.title_label.setFixedSize(139, 21) # 调整 xxx视频标签的宽度
|
||
|
||
# 视频显示容器：使用堆叠布局，让重连按钮和视频标签分层显示
|
||
self.video_container = QWidget()
|
||
self.video_container.setObjectName("videoContainer")
|
||
self.video_container.setFixedSize(327, 199) # 需要同步修改下面的尺寸
|
||
# #011d6b #033474
|
||
# self.video_container.setStyleSheet(
|
||
# "background-color: #011d6b; border: none;"
|
||
# ) # 明确视频显示背景色
|
||
self.video_container.setStyleSheet(
|
||
f"""
|
||
#videoContainer {{
|
||
border-image: url({ImagePaths.VIDEO_FRAME_BACKGROUND});
|
||
}}
|
||
"""
|
||
)
|
||
video_layout = QVBoxLayout(self.video_container)
|
||
video_layout.setContentsMargins(6, 6, 5, 6)
|
||
|
||
# 原始图像显示
|
||
self.raw_label = QLabel()
|
||
self.raw_label.setAlignment(Qt.AlignCenter)
|
||
self.raw_label.setStyleSheet("border: none;") # 移除背景，使用容器背景
|
||
# self.raw_label.setStyleSheet("background-color: red;")
|
||
self.raw_label.setText(f"{self.camera_name}摄像头, 连接中...")
|
||
# self.raw_label.setFixedSize(327, 199)
|
||
# self.raw_label.setFixedSize(314, 187) # 需要根据视频框背景大小调整 ！！！
|
||
self.raw_label.setFixedSize(316, 187) # 需要根据视频框背景大小调整 ！！！
|
||
|
||
# 重置按钮：居中显示、透明背景
|
||
self.reset_button = QPushButton()
|
||
# self.reset_button.setFixedSize(327, 199)
|
||
# self.reset_button.setFixedSize(314, 187) # 同上
|
||
self.reset_button.setFixedSize(316, 187) # 同上
|
||
self.reset_button.setStyleSheet(
|
||
"""
|
||
QPushButton {
|
||
background-color: transparent;
|
||
border: none;
|
||
padding: 0px;
|
||
}
|
||
"""
|
||
)
|
||
self.reset_button.setCursor(QCursor(Qt.PointingHandCursor))
|
||
self.reset_button.clicked.connect(self.onRestButtonClicked)
|
||
|
||
# 加载视频重新连接图片
|
||
try:
|
||
pixmap = QPixmap(ImagePaths.INTERFACE_REFRESH)
|
||
if not pixmap.isNull():
|
||
self.reset_button.setIcon(QIcon(pixmap))
|
||
self.reset_button.setIconSize(pixmap.size())
|
||
else:
|
||
# 图片加载失败（文件不存在或格式错误）
|
||
self.reset_button.setText("点击重置连接")
|
||
except Exception as e:
|
||
print(f"加载复位图片失败: {e}")
|
||
self.reset_button.setText("点击重置连接")
|
||
|
||
# 堆叠布局：重连按钮和视频标签分层
|
||
self.stacked_widget = QStackedWidget()
|
||
self.stacked_widget.addWidget(self.raw_label)
|
||
self.stacked_widget.addWidget(self.reset_button)
|
||
video_layout.addWidget(self.stacked_widget, alignment=Qt.AlignCenter)
|
||
|
||
# 添加到主布局
|
||
layout.addLayout(title_layout)
|
||
layout.addWidget(self.video_container)
|
||
|
||
# 显示开关切换槽函数
|
||
def onDisplayButtonSwitched(self, state:bool):
|
||
# 显示开关打开，state 为 True; 显示开关关闭，state 为 False
|
||
if state:
|
||
self.stacked_widget.setCurrentWidget(self.raw_label) # 视频显示标签
|
||
self.stacked_widget.setHidden(False)
|
||
else:
|
||
self.stacked_widget.setHidden(True)
|
||
self.video_display_signal.emit(self.camera_name, state)
|
||
|
||
# 重连按钮点击槽函数
|
||
def onRestButtonClicked(self):
|
||
self.stacked_widget.setCurrentWidget(self.raw_label)
|
||
self.raw_label.setText(f"{self.camera_name}摄像头, 连接中...")
|
||
self.reset_signal.emit() # 发送重连信号
|
||
|
||
def showResetButton(self):
|
||
self.raw_label.setText("")
|
||
self.stacked_widget.setCurrentWidget(self.reset_button)
|
||
|
||
def update_raw_image(self, image: Optional[QImage]):
|
||
if image:
|
||
pixmap = QPixmap.fromImage(image)
|
||
self.raw_label.setPixmap(
|
||
pixmap.scaled(
|
||
self.raw_label.size(),
|
||
Qt.IgnoreAspectRatio, # 忽略比例，占满窗口 可选：Qt.KeepAspectRatio
|
||
Qt.SmoothTransformation,
|
||
)
|
||
)
|
||
self.raw_label.setText("")
|
||
else:
|
||
self.raw_label.setPixmap(QPixmap())
|
||
# 没有图片数据，stream流会自动重连 或 手动重连
|
||
self.raw_label.setText(f"{self.camera_name}摄像头, 连接中...")
|
||
|
||
# 视频显示区域 总的标题 (如 振捣视频)
|
||
class VideoTitleWidget(QWidget):
|
||
def __init__(self, video_name, parent=None):
|
||
super().__init__(parent)
|
||
self.setStyleSheet("background-color: transparent;") # 整体透明
|
||
self.setup_ui(video_name)
|
||
|
||
def setup_ui(self, video_name):
|
||
# 水平布局：文字 + 图标
|
||
layout = QHBoxLayout(self)
|
||
layout.setContentsMargins(0, 0, 0, 0) # 布局内边距为0
|
||
|
||
# 文字标签
|
||
self.text_label = QLabel(f"{video_name}视频")
|
||
self.text_label.setStyleSheet("""
|
||
font-family: "微软雅黑";
|
||
font-size: 20px;
|
||
font-weight: bold;
|
||
color: #16FFFF;
|
||
background-color: transparent;
|
||
""")
|
||
font = self.text_label.font()
|
||
font.setLetterSpacing(QFont.SpacingType.AbsoluteSpacing, 3) # 字间距3px（可按需加大）
|
||
self.text_label.setFont(font)
|
||
|
||
# 图标标签
|
||
self.icon_label = QLabel()
|
||
self.icon_label.setPixmap(QPixmap(ImagePaths.VIDEO_CAMERA_MARK)) # 需要换为实际的图片的路径
|
||
self.icon_label.setStyleSheet("background-color: transparent;")
|
||
self.icon_label.setFixedSize(23, 16)
|
||
self.icon_label.setScaledContents(True)
|
||
|
||
# 添加到布局
|
||
layout.addWidget(self.text_label, alignment=Qt.AlignCenter)
|
||
layout.addWidget(self.icon_label, alignment=Qt.AlignCenter)
|
||
|
||
|
||
class VibrationVideoWidget(QWidget):
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName("vibrationVideoWidget")
|
||
|
||
# 1. 加载背景图片（替换为你的图片路径）
|
||
self.background_image = QImage() # 初始化图片对象
|
||
# 注意：图片路径需正确，建议使用绝对路径或项目相对路径
|
||
self.background_path = ImagePaths.VIDEO_BACKGROUND # 你的背景图片路径
|
||
if not self.background_image.load(self.background_path):
|
||
# 图片加载失败时的容错处理（显示红色背景作为提示）
|
||
print(f"警告：无法加载背景图片 {self.background_path}")
|
||
self.background_image = QImage(388, 839, QImage.Format_RGB32)
|
||
self.background_image.fill(Qt.GlobalColor.red) # 加载失败时用红色填充
|
||
|
||
# 显示摄像头画面的模组
|
||
# 需要修改为相应的地址!!!
|
||
# 注：在camera_controller中设置地址url
|
||
self.cam1 = CameraModule("上位料斗")
|
||
self.cam2 = CameraModule("下位料斗", show_ai=False)
|
||
self.cam3 = CameraModule("模具车", need_rotate_180=False)
|
||
|
||
self.setup_ui()
|
||
self.connect_signals()
|
||
|
||
def set_camera_urls(self, cam1_url, cam2_url, cam3_url):
|
||
"""设置三个摄像头的RTSP URL
|
||
注意: 在CameraController中调用
|
||
"""
|
||
self.cam1.rtsp_url = cam1_url
|
||
self.cam2.rtsp_url = cam2_url
|
||
self.cam3.rtsp_url = cam3_url
|
||
|
||
def setup_ui(self):
|
||
# 视频widget样式
|
||
# self.setFixedSize(387, 720) # 宽387、高792
|
||
self.setFixedSize(388, 839) # 背景图片宽388、高879
|
||
|
||
# 1、总的标题
|
||
self.title_widget = VideoTitleWidget("振捣")
|
||
|
||
|
||
# 布局设置
|
||
# 2. 创建摄像头容器（专门存放cam1/cam2/cam3）
|
||
self.camera_container = QWidget() # 容器widget
|
||
self.camera_container.setFixedSize(350, 720) # 容器固定尺寸
|
||
# 给容器添加样式
|
||
self.camera_container.setStyleSheet("""
|
||
background-color: transparent;
|
||
border: none;
|
||
""")
|
||
|
||
# 3. 容器内部布局（管理三个摄像头模块）
|
||
container_layout = QVBoxLayout(self.camera_container)
|
||
container_layout.setContentsMargins(6, 6, 6, 6) # 容器内边距
|
||
container_layout.setSpacing(6) # 摄像头模块之间的间距
|
||
|
||
# 将三个摄像头添加到容器布局中
|
||
container_layout.addWidget(self.cam1, alignment=Qt.AlignCenter)
|
||
container_layout.addWidget(self.cam2, alignment=Qt.AlignCenter)
|
||
container_layout.addWidget(self.cam3, alignment=Qt.AlignCenter)
|
||
|
||
# 4. 主窗口布局（管理摄像头容器等其他部件）
|
||
main_layout = QVBoxLayout(self) # 主窗口的布局
|
||
main_layout.setContentsMargins(0, 0, 9, 30) # 主布局内边距为0
|
||
|
||
# 添加标题部件
|
||
main_layout.addWidget(self.title_widget, alignment=Qt.AlignCenter) # 标题居中
|
||
|
||
# 添加43px间距（标题与摄像头容器之间）
|
||
main_layout.addSpacing(16)
|
||
|
||
# 把摄像头容器添加到主布局中（作为一个整体）
|
||
main_layout.addWidget(self.camera_container, alignment=Qt.AlignBottom | Qt.AlignHCenter)
|
||
|
||
|
||
# 初始化视频流 (注意：设置好了camera_urls之后调用)
|
||
def init_streams(self):
|
||
url1 = self.cam1.rtsp_url
|
||
url2 = self.cam2.rtsp_url
|
||
url3 = self.cam3.rtsp_url
|
||
|
||
self.stream1 = VideoStream(url1, "Cam1").start()
|
||
self.stream2 = VideoStream(url2, "Cam2").start()
|
||
self.stream3 = VideoStream(url3, "Cam3").start()
|
||
|
||
self.timer1 = QTimer()
|
||
self.timer1.timeout.connect(self.update_frame1)
|
||
|
||
self.timer2 = QTimer()
|
||
self.timer2.timeout.connect(self.update_frame2)
|
||
|
||
self.timer3 = QTimer()
|
||
self.timer3.timeout.connect(self.update_frame3)
|
||
|
||
# 定时读取视频流
|
||
self.timer1.start(33)
|
||
self.timer2.start(33)
|
||
self.timer3.start(33)
|
||
|
||
def connect_signals(self):
|
||
# 流Stream相关的槽函数连接放在了 camera_controller之中
|
||
|
||
# 视频显示相关的槽函数连接
|
||
self.cam1.video_display_signal.connect(self.onVideoDisplay)
|
||
self.cam2.video_display_signal.connect(self.onVideoDisplay)
|
||
self.cam3.video_display_signal.connect(self.onVideoDisplay)
|
||
|
||
def onVideoDisplay(self, cam_name: str, state: bool):
|
||
# 1. 摄像头与 stream、timer 的映射关系
|
||
cam_mapping = {
|
||
"上位料斗": (self.stream1, self.timer1),
|
||
"下位料斗": (self.stream2, self.timer2),
|
||
"模具车": (self.stream3, self.timer3)
|
||
}
|
||
|
||
# 2. 获取当前摄像头对应的 stream 和 timer（处理无效名称的情况）
|
||
stream, timer = cam_mapping.get(cam_name, (None, None))
|
||
if not stream or not timer:
|
||
print(f"警告：未知摄像头名称 {cam_name}")
|
||
return
|
||
|
||
# 3. 根据 state 执行操作
|
||
if state: # 显示视频
|
||
stream.reset_retry() # 视频流重连
|
||
timer.start(33) # 定时器开始，显示视频帧
|
||
else: # 关闭视频
|
||
timer.stop() # 定时器停止
|
||
stream.manual_stop = True # 视频流停止读取
|
||
|
||
def onStreamError(self, stream_name):
|
||
if stream_name == "Cam1":
|
||
self.cam1.showResetButton()
|
||
elif stream_name == "Cam2":
|
||
self.cam2.showResetButton()
|
||
elif stream_name == "Cam3":
|
||
self.cam3.showResetButton()
|
||
|
||
def update_frame1(self):
|
||
frame = self.stream1.read()
|
||
if frame is not None:
|
||
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
||
if self.cam1.need_rotate_180:
|
||
frame_rgb = cv2.rotate(frame_rgb, cv2.ROTATE_180)
|
||
h, w, ch = frame_rgb.shape
|
||
q_img = QImage(frame_rgb.data, w, h, ch * w, QImage.Format_RGB888)
|
||
self.cam1.update_raw_image(q_img)
|
||
else:
|
||
self.cam1.update_raw_image(None)
|
||
|
||
def update_frame2(self):
|
||
frame = self.stream2.read()
|
||
if frame is not None:
|
||
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
||
if self.cam2.need_rotate_180:
|
||
frame_rgb = cv2.rotate(frame_rgb, cv2.ROTATE_180)
|
||
h, w, ch = frame_rgb.shape
|
||
q_img = QImage(frame_rgb.data, w, h, ch * w, QImage.Format_RGB888)
|
||
self.cam2.update_raw_image(q_img)
|
||
else:
|
||
self.cam2.update_raw_image(None)
|
||
|
||
def update_frame3(self):
|
||
frame = self.stream3.read()
|
||
if frame is not None:
|
||
frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
||
if self.cam3.need_rotate_180:
|
||
frame_rgb = cv2.rotate(frame_rgb, cv2.ROTATE_180)
|
||
h, w, ch = frame_rgb.shape
|
||
q_img = QImage(frame_rgb.data, w, h, ch * w, QImage.Format_RGB888)
|
||
self.cam3.update_raw_image(q_img)
|
||
else:
|
||
self.cam3.update_raw_image(None)
|
||
|
||
def paintEvent(self, event):
|
||
# 1. 先调用父类的paintEvent
|
||
super().paintEvent(event)
|
||
|
||
# 2. 创建画家对象，绘制背景图片
|
||
painter = QPainter(self)
|
||
painter.setRenderHint(QPainter.RenderHint.SmoothPixmapTransform) # 图片缩放平滑
|
||
|
||
# 3. 获取部件的实际尺寸（这里是387x720，与setup_ui中一致）
|
||
widget_rect = self.rect()
|
||
|
||
# 4. 绘制背景图片（缩放至部件大小）
|
||
# 可选缩放策略：
|
||
# - Qt.KeepAspectRatio：保持图片比例，可能有黑边
|
||
# - Qt.IgnoreAspectRatio：拉伸填充，可能变形
|
||
# - Qt.KeepAspectRatioByExpanding：按比例放大至覆盖部件，可能裁剪边缘
|
||
scaled_pixmap = QPixmap.fromImage(self.background_image).scaled(
|
||
widget_rect.size(),
|
||
Qt.KeepAspectRatio, # 推荐：保持比例，避免变形
|
||
Qt.SmoothTransformation # 平滑缩放
|
||
)
|
||
|
||
# 5. 计算图片绘制位置（居中显示，若有黑边则均匀分布）
|
||
x = (widget_rect.width() - scaled_pixmap.width()) // 2
|
||
y = (widget_rect.height() - scaled_pixmap.height()) // 2
|
||
painter.drawPixmap(x, y, scaled_pixmap)
|
||
|
||
# 6. 结束绘制
|
||
painter.end()
|
||
|
||
# ====================== 清理资源 ======================
|
||
def closeEvent(self, event):
|
||
self.hide()
|
||
self.timer1.stop()
|
||
self.timer2.stop()
|
||
self.timer3.stop()
|
||
self.stream1.stop()
|
||
self.stream2.stop()
|
||
self.stream3.stop()
|
||
super().closeEvent(event)
|