266 lines
10 KiB
Python
266 lines
10 KiB
Python
 
|
"""Thin PyQt5 rendering shell for the V1 monitor and settings tabs.
|
|||
|
|
|
|||
|
|
This module only renders ``MonitorViewState`` and binds ``SettingsDraft``; it
|
|||
|
|
never runs Pose, tracking or fall decisions. All event facts arrive from the
|
|||
|
|
``FallPipeline`` worker in ``v1/app.py``. Because acceptance-critical logic lives
|
|||
|
|
in ``v1/view_model.py`` (Qt-free and unit-tested), this file is the only surface
|
|||
|
|
that requires a Windows + PyQt5 + display smoke test.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import Dict, Optional
|
|||
|
|
|
|||
|
|
import numpy as np
|
|||
|
|
from PyQt5 import QtCore, QtGui, QtWidgets
|
|||
|
|
|
|||
|
|
from v1.view_model import (
|
|||
|
|
FIELD_BOUNDS,
|
|||
|
|
DraftValidationError,
|
|||
|
|
MonitorViewState,
|
|||
|
|
SettingsDraft,
|
|||
|
|
StatusColor,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
COLOR_HEX: Dict[StatusColor, str] = {
|
|||
|
|
StatusColor.SUCCESS: "#15803D",
|
|||
|
|
StatusColor.CAUTION: "#B45309",
|
|||
|
|
StatusColor.CRITICAL: "#C62828",
|
|||
|
|
StatusColor.OFFLINE: "#64748B",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# Light Windows theme tokens (see docs/ui/silver-pose-ui-ux-spec.md).
|
|||
|
|
_QSS = """
|
|||
|
|
QWidget { background: #EAF1F8; color: #1E293B; font-family: 'Segoe UI Variable','Microsoft YaHei UI',sans-serif; }
|
|||
|
|
QTabWidget::pane { background: #FFFFFF; border: 1px solid #C8D4E3; }
|
|||
|
|
QTabBar::tab { padding: 8px 20px; background: #FFFFFF; color: #52657C; }
|
|||
|
|
QTabBar::tab:selected { color: #1E293B; border-bottom: 2px solid #2563EB; }
|
|||
|
|
QGroupBox { background: #FFFFFF; border: 1px solid #C8D4E3; border-radius: 8px; margin-top: 12px; }
|
|||
|
|
QPushButton { background: #FFFFFF; border: 1px solid #C8D4E3; border-radius: 8px; padding: 8px 14px; min-height: 40px; }
|
|||
|
|
QPushButton:focus { border: 2px solid #2563EB; }
|
|||
|
|
QLabel[role="primary-action"] { }
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
_FIELD_LABELS = {
|
|||
|
|
"keypoint_confidence_threshold": "关键点置信度阈值",
|
|||
|
|
"suspect_window_seconds": "快速下移到水平的最大间隔（秒）",
|
|||
|
|
"confirm_window_seconds": "确认窗口（秒，1–3）",
|
|||
|
|
"recovery_window_seconds": "恢复稳定窗口（秒）",
|
|||
|
|
"cooldown_seconds": "确认后恢复判断冷却（秒）",
|
|||
|
|
"model_confidence_threshold": "模型检测置信度阈值",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_FIELD_STEP = {
|
|||
|
|
"keypoint_confidence_threshold": 0.05,
|
|||
|
|
"suspect_window_seconds": 0.1,
|
|||
|
|
"confirm_window_seconds": 0.1,
|
|||
|
|
"recovery_window_seconds": 0.5,
|
|||
|
|
"cooldown_seconds": 1.0,
|
|||
|
|
"model_confidence_threshold": 0.05,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def bgr_to_qimage(image: np.ndarray) -> QtGui.QImage:
|
|||
|
|
"""Convert an OpenCV BGR frame to a QImage owning a contiguous copy."""
|
|||
|
|
|
|||
|
|
rgb = np.ascontiguousarray(image[:, :, ::-1])
|
|||
|
|
height, width, _ = rgb.shape
|
|||
|
|
return QtGui.QImage(
|
|||
|
|
rgb.data, width, height, 3 * width, QtGui.QImage.Format_RGB888
|
|||
|
|
).copy()
|
|||
|
|
|
|||
|
|
|
|||
|
|
class VideoView(QtWidgets.QWidget):
|
|||
|
|
"""Paint the latest frame plus box/skeleton/id/state overlays."""
|
|||
|
|
|
|||
|
|
def __init__(self, parent: Optional[QtWidgets.QWidget] = None) -> None:
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self.setMinimumSize(480, 270)
|
|||
|
|
self._image: Optional[QtGui.QImage] = None
|
|||
|
|
self._view: Optional[MonitorViewState] = None
|
|||
|
|
|
|||
|
|
def update_frame(self, image: Optional[np.ndarray], view: MonitorViewState) -> None:
|
|||
|
|
self._image = bgr_to_qimage(image) if image is not None else None
|
|||
|
|
self._view = view
|
|||
|
|
self.update()
|
|||
|
|
|
|||
|
|
def paintEvent(self, _event: QtGui.QPaintEvent) -> None:
|
|||
|
|
painter = QtGui.QPainter(self)
|
|||
|
|
painter.fillRect(self.rect(), QtGui.QColor("#0B1220"))
|
|||
|
|
if self._image is None:
|
|||
|
|
painter.setPen(QtGui.QColor("#EAF1F8"))
|
|||
|
|
text = self._view.status_text if self._view else "无画面"
|
|||
|
|
painter.drawText(self.rect(), QtCore.Qt.AlignCenter, text)
|
|||
|
|
painter.end()
|
|||
|
|
return
|
|||
|
|
target = self._fitted_rect(self._image)
|
|||
|
|
painter.drawImage(target, self._image)
|
|||
|
|
if self._view is not None:
|
|||
|
|
self._draw_overlays(painter, target, self._image)
|
|||
|
|
painter.end()
|
|||
|
|
|
|||
|
|
def _fitted_rect(self, image: QtGui.QImage) -> QtCore.QRect:
|
|||
|
|
widget = self.rect()
|
|||
|
|
scale = min(widget.width() / image.width(), widget.height() / image.height())
|
|||
|
|
width = int(image.width() * scale)
|
|||
|
|
height = int(image.height() * scale)
|
|||
|
|
left = widget.left() + (widget.width() - width) // 2
|
|||
|
|
top = widget.top() + (widget.height() - height) // 2
|
|||
|
|
return QtCore.QRect(left, top, width, height)
|
|||
|
|
|
|||
|
|
def _draw_overlays(
|
|||
|
|
self, painter: QtGui.QPainter, target: QtCore.QRect, image: QtGui.QImage
|
|||
|
|
) -> None:
|
|||
|
|
scale_x = target.width() / image.width()
|
|||
|
|
scale_y = target.height() / image.height()
|
|||
|
|
|
|||
|
|
def to_widget(x: float, y: float) -> QtCore.QPointF:
|
|||
|
|
return QtCore.QPointF(target.left() + x * scale_x, target.top() + y * scale_y)
|
|||
|
|
|
|||
|
|
for overlay in self._view.people:
|
|||
|
|
color = QtGui.QColor(COLOR_HEX[overlay.color])
|
|||
|
|
painter.setPen(QtGui.QPen(color, 3))
|
|||
|
|
left, top, right, bottom = overlay.box_xyxy
|
|||
|
|
painter.drawRect(
|
|||
|
|
QtCore.QRectF(to_widget(left, top), to_widget(right, bottom))
|
|||
|
|
)
|
|||
|
|
painter.setPen(QtGui.QPen(color, 2))
|
|||
|
|
for start, end in overlay.skeleton_segments:
|
|||
|
|
painter.drawLine(to_widget(start.x, start.y), to_widget(end.x, end.y))
|
|||
|
|
for point in overlay.keypoints:
|
|||
|
|
if point is not None:
|
|||
|
|
painter.drawEllipse(to_widget(point.x, point.y), 2.5, 2.5)
|
|||
|
|
painter.setPen(QtGui.QColor("#FFFFFF"))
|
|||
|
|
painter.drawText(to_widget(left, top - 6), overlay.label)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MonitorTab(QtWidgets.QWidget):
|
|||
|
|
"""Live view: video, connection status, recent events, start/stop."""
|
|||
|
|
|
|||
|
|
start_requested = QtCore.pyqtSignal()
|
|||
|
|
stop_requested = QtCore.pyqtSignal()
|
|||
|
|
|
|||
|
|
def __init__(self, parent: Optional[QtWidgets.QWidget] = None) -> None:
|
|||
|
|
super().__init__(parent)
|
|||
|
|
layout = QtWidgets.QVBoxLayout(self)
|
|||
|
|
self._status = QtWidgets.QLabel("未开始")
|
|||
|
|
self._status.setMinimumHeight(24)
|
|||
|
|
self.video = VideoView()
|
|||
|
|
self.events = QtWidgets.QListWidget()
|
|||
|
|
self.events.setMaximumHeight(140)
|
|||
|
|
self.start_button = QtWidgets.QPushButton("开始监控")
|
|||
|
|
self.stop_button = QtWidgets.QPushButton("停止监控")
|
|||
|
|
self.stop_button.setEnabled(False)
|
|||
|
|
self.start_button.clicked.connect(self.start_requested.emit)
|
|||
|
|
self.stop_button.clicked.connect(self.stop_requested.emit)
|
|||
|
|
|
|||
|
|
controls = QtWidgets.QHBoxLayout()
|
|||
|
|
controls.addWidget(self.start_button)
|
|||
|
|
controls.addWidget(self.stop_button)
|
|||
|
|
controls.addStretch(1)
|
|||
|
|
layout.addWidget(self._status)
|
|||
|
|
layout.addWidget(self.video, 1)
|
|||
|
|
layout.addWidget(QtWidgets.QLabel("最近事件"))
|
|||
|
|
layout.addWidget(self.events)
|
|||
|
|
layout.addLayout(controls)
|
|||
|
|
|
|||
|
|
def render_view(self, image, view: MonitorViewState) -> None:
|
|||
|
|
self.video.update_frame(image, view)
|
|||
|
|
color = COLOR_HEX[view.status_color]
|
|||
|
|
self._status.setText("连接状态：{0}".format(view.status_text))
|
|||
|
|
self._status.setStyleSheet("color: {0}; font-weight: 600;".format(color))
|
|||
|
|
for event in view.events:
|
|||
|
|
self.events.insertItem(
|
|||
|
|
0,
|
|||
|
|
"{0} · {1} · 延迟 {2:.2f}s".format(
|
|||
|
|
event.event_id, event.track_id, event.latency_seconds
|
|||
|
|
),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def set_running(self, running: bool) -> None:
|
|||
|
|
self.start_button.setEnabled(not running)
|
|||
|
|
self.stop_button.setEnabled(running)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class SettingsTab(QtWidgets.QWidget):
|
|||
|
|
"""Edit a non-sensitive draft; changes apply only on the next start."""
|
|||
|
|
|
|||
|
|
def __init__(
|
|||
|
|
self, draft: SettingsDraft, env_ready: bool, model_summary: str,
|
|||
|
|
parent: Optional[QtWidgets.QWidget] = None,
|
|||
|
|
) -> None:
|
|||
|
|
super().__init__(parent)
|
|||
|
|
self._draft = draft
|
|||
|
|
self._spins: Dict[str, QtWidgets.QDoubleSpinBox] = {}
|
|||
|
|
layout = QtWidgets.QVBoxLayout(self)
|
|||
|
|
|
|||
|
|
source_box = QtWidgets.QGroupBox("来源与模型（只读）")
|
|||
|
|
source_form = QtWidgets.QFormLayout(source_box)
|
|||
|
|
source_form.addRow("RTSP 环境变量", QtWidgets.QLabel("已就绪" if env_ready else "未就绪"))
|
|||
|
|
source_form.addRow("模型", QtWidgets.QLabel(model_summary))
|
|||
|
|
layout.addWidget(source_box)
|
|||
|
|
|
|||
|
|
params_box = QtWidgets.QGroupBox("事件参数草稿")
|
|||
|
|
form = QtWidgets.QFormLayout(params_box)
|
|||
|
|
values = draft.draft_values
|
|||
|
|
for key, (low, high) in FIELD_BOUNDS.items():
|
|||
|
|
spin = QtWidgets.QDoubleSpinBox()
|
|||
|
|
spin.setRange(low, high)
|
|||
|
|
spin.setSingleStep(_FIELD_STEP[key])
|
|||
|
|
spin.setDecimals(2)
|
|||
|
|
spin.setValue(values[key])
|
|||
|
|
spin.valueChanged.connect(lambda value, name=key: self._on_edit(name, value))
|
|||
|
|
self._spins[key] = spin
|
|||
|
|
form.addRow(_FIELD_LABELS[key], spin)
|
|||
|
|
layout.addWidget(params_box)
|
|||
|
|
|
|||
|
|
self.status = QtWidgets.QLabel("")
|
|||
|
|
self.save_button = QtWidgets.QPushButton("保存（下次启动生效）")
|
|||
|
|
self.reset_button = QtWidgets.QPushButton("重置")
|
|||
|
|
self.save_button.clicked.connect(self._on_save)
|
|||
|
|
self.reset_button.clicked.connect(self._on_reset)
|
|||
|
|
actions = QtWidgets.QHBoxLayout()
|
|||
|
|
actions.addWidget(self.save_button)
|
|||
|
|
actions.addWidget(self.reset_button)
|
|||
|
|
actions.addStretch(1)
|
|||
|
|
layout.addWidget(self.status)
|
|||
|
|
layout.addLayout(actions)
|
|||
|
|
layout.addStretch(1)
|
|||
|
|
|
|||
|
|
def _on_edit(self, key: str, value: float) -> None:
|
|||
|
|
try:
|
|||
|
|
self._draft.edit(key, value)
|
|||
|
|
except DraftValidationError as exc:
|
|||
|
|
self.status.setText("校验失败：{0}".format(exc))
|
|||
|
|
return
|
|||
|
|
self.status.setText("已修改，未保存" if self._draft.is_dirty else "")
|
|||
|
|
|
|||
|
|
def _on_save(self) -> None:
|
|||
|
|
self.status.setText(self._draft.save())
|
|||
|
|
|
|||
|
|
def _on_reset(self) -> None:
|
|||
|
|
self._draft.discard()
|
|||
|
|
for key, spin in self._spins.items():
|
|||
|
|
spin.blockSignals(True)
|
|||
|
|
spin.setValue(self._draft.draft_values[key])
|
|||
|
|
spin.blockSignals(False)
|
|||
|
|
self.status.setText("已重置为已保存值")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class MainWindow(QtWidgets.QMainWindow):
|
|||
|
|
"""Top dual-tab window: 实时监控 / 设置."""
|
|||
|
|
|
|||
|
|
def __init__(
|
|||
|
|
self, draft: SettingsDraft, env_ready: bool, model_summary: str, source_name: str
|
|||
|
|
) -> None:
|
|||
|
|
super().__init__()
|
|||
|
|
self.setWindowTitle("Silver Pose · {0}".format(source_name))
|
|||
|
|
self.resize(1280, 800)
|
|||
|
|
self.setStyleSheet(_QSS)
|
|||
|
|
self.monitor = MonitorTab()
|
|||
|
|
self.settings = SettingsTab(draft, env_ready, model_summary)
|
|||
|
|
tabs = QtWidgets.QTabWidget()
|
|||
|
|
tabs.setTabPosition(QtWidgets.QTabWidget.North)
|
|||
|
|
tabs.addTab(self.monitor, "实时监控")
|
|||
|
|
tabs.addTab(self.settings, "设置")
|
|||
|
|
self.setCentralWidget(tabs)
|