561 lines
21 KiB
Python
561 lines
21 KiB
Python
import sys
|
||
import serial.tools.list_ports
|
||
from PySide6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
|
||
QHBoxLayout, QComboBox, QPushButton, QLabel,
|
||
QLineEdit, QMessageBox)
|
||
from PySide6.QtCore import Qt, QTimer
|
||
from PySide6.QtGui import QIntValidator
|
||
from serial_recorder import SerialRecorder
|
||
import os
|
||
|
||
# 定义命令常量
|
||
CMD_ON = b'\x01\x00\x01\x02'
|
||
CMD_OFF = b'\x01\x00\x00\x01'
|
||
|
||
# CMD_DATA_ON = b'\xAA\xAA\xAA\xAA\x00\x00\x00\x00\x05\x00\x00\x00\x00\x00\x00\x00\xAD'
|
||
# CMD_DATA_OFF = b'\xAA\xAA\xAA\xAA\x01\x00\x00\x00\x05\x00\x00\x00\x00\x00\x00\x00\xAE'
|
||
|
||
# 智己项目 - 开关 9字节
|
||
CMD_DATA_ON = b'\x0A\x01\x77\x3B\x00\x00\x80\x3F\x7C'
|
||
CMD_DATA_OFF = b'\x0A\x01\x77\x3B\x00\x00\x00\x00\xBD'
|
||
|
||
class SerialUI(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
# self.setWindowTitle("Signal Collector V1.1.1 BUILD202503261405")
|
||
self.resize(600, 400)
|
||
|
||
# 设置无边框窗口
|
||
self.setWindowFlags(Qt.FramelessWindowHint)
|
||
|
||
# 设置应用样式
|
||
self.setStyleSheet("""
|
||
QMainWindow {
|
||
background-color: #2B2B2B;
|
||
border: none;
|
||
}
|
||
QWidget {
|
||
background-color: #2B2B2B;
|
||
color: #FFFFFF;
|
||
}
|
||
QLabel {
|
||
color: #CCCCCC;
|
||
font-size: 14px;
|
||
}
|
||
QComboBox {
|
||
background-color: #3C3F41;
|
||
border: 1px solid #646464;
|
||
border-radius: 3px;
|
||
color: #FFFFFF;
|
||
padding: 5px;
|
||
min-width: 100px;
|
||
}
|
||
QComboBox:hover {
|
||
border: 1px solid #FF8C00;
|
||
}
|
||
QComboBox::drop-down {
|
||
border: none;
|
||
padding-right: 8px;
|
||
}
|
||
QPushButton {
|
||
background-color: #3C3F41;
|
||
border: 1px solid #646464;
|
||
border-radius: 3px;
|
||
color: #FFFFFF;
|
||
padding: 5px 15px;
|
||
min-height: 25px;
|
||
}
|
||
QPushButton:hover {
|
||
background-color: #434749;
|
||
border: 1px solid #FF8C00;
|
||
}
|
||
QPushButton:pressed {
|
||
background-color: #333637;
|
||
}
|
||
QPushButton:checked {
|
||
background-color: #FF8C00;
|
||
border: 1px solid #FF8C00;
|
||
}
|
||
QLineEdit {
|
||
background-color: #3C3F41;
|
||
border: 1px solid #646464;
|
||
border-radius: 3px;
|
||
color: #FFFFFF;
|
||
padding: 5px;
|
||
min-width: 100px;
|
||
}
|
||
QLineEdit:focus {
|
||
border: 1px solid #FF8C00;
|
||
}
|
||
QLineEdit:disabled {
|
||
background-color: #2B2B2B;
|
||
color: #808080;
|
||
}
|
||
QStatusBar {
|
||
background-color: #3C3F41;
|
||
color: #CCCCCC;
|
||
}
|
||
QMenuBar {
|
||
background-color: #2B2B2B;
|
||
color: #FFFFFF;
|
||
}
|
||
QMenuBar::item {
|
||
background-color: #2B2B2B;
|
||
color: #FFFFFF;
|
||
}
|
||
QMenuBar::item:selected {
|
||
background-color: #3C3F41;
|
||
}
|
||
QMenu {
|
||
background-color: #2B2B2B;
|
||
color: #FFFFFF;
|
||
}
|
||
QMenu::item:selected {
|
||
background-color: #3C3F41;
|
||
}
|
||
#record_btn {
|
||
background-color: #FF8C00;
|
||
border: none;
|
||
font-weight: bold;
|
||
}
|
||
#record_btn:hover {
|
||
background-color: #FFA500;
|
||
}
|
||
#record_btn:checked {
|
||
background-color: #CD3333;
|
||
}
|
||
#titleBar {
|
||
background-color: #2B2B2B;
|
||
border-bottom: 1px solid #3C3F41;
|
||
min-height: 30px;
|
||
}
|
||
#titleLabel {
|
||
color: #FFFFFF;
|
||
font-size: 14px;
|
||
padding-left: 10px;
|
||
}
|
||
#closeButton {
|
||
background-color: transparent;
|
||
border: none;
|
||
color: #FFFFFF;
|
||
padding: 5px 10px;
|
||
}
|
||
#closeButton:hover {
|
||
background-color: #CD3333;
|
||
}
|
||
""")
|
||
|
||
# 创建中心部件和布局
|
||
central_widget = QWidget()
|
||
self.setCentralWidget(central_widget)
|
||
layout = QVBoxLayout(central_widget)
|
||
layout.setContentsMargins(0, 0, 0, 0) # 移除边距
|
||
|
||
# 创建自定义标题栏
|
||
title_bar = QWidget()
|
||
title_bar.setObjectName("titleBar")
|
||
title_bar_layout = QHBoxLayout(title_bar)
|
||
title_bar_layout.setContentsMargins(0, 0, 0, 0)
|
||
|
||
# 标题标签
|
||
title_label = QLabel("Signal Collector V1.1.2 BUILD202511141419")
|
||
title_label.setObjectName("titleLabel")
|
||
|
||
# 关闭按钮
|
||
close_button = QPushButton("×")
|
||
close_button.setObjectName("closeButton")
|
||
close_button.setFixedSize(30, 30)
|
||
close_button.clicked.connect(self.close)
|
||
|
||
title_bar_layout.addWidget(title_label)
|
||
title_bar_layout.addStretch()
|
||
title_bar_layout.addWidget(close_button)
|
||
|
||
# 添加标题栏到主布局
|
||
layout.addWidget(title_bar)
|
||
|
||
# 创建内容区域
|
||
content_widget = QWidget()
|
||
content_layout = QVBoxLayout(content_widget)
|
||
content_layout.setContentsMargins(20, 20, 20, 20)
|
||
|
||
# 串口选择区域
|
||
port_layout = QHBoxLayout()
|
||
port_label = QLabel("串口:")
|
||
self.port_combo = QComboBox()
|
||
self.refresh_btn = QPushButton("刷新")
|
||
self.refresh_btn.clicked.connect(self.refresh_ports)
|
||
port_layout.addWidget(port_label)
|
||
port_layout.addWidget(self.port_combo)
|
||
port_layout.addWidget(self.refresh_btn)
|
||
content_layout.addLayout(port_layout)
|
||
|
||
# 波特率选择区域
|
||
baud_layout = QHBoxLayout()
|
||
baud_label = QLabel("波特率:")
|
||
self.baud_input = QLineEdit()
|
||
self.baud_input.setText("4000000")
|
||
self.baud_input.setValidator(QIntValidator(1200, 4000000))
|
||
self.baud_input.setPlaceholderText("请输入波特率")
|
||
self.connect_btn = QPushButton("连接端口")
|
||
self.connect_btn.setCheckable(True)
|
||
self.connect_btn.clicked.connect(self.toggle_connection)
|
||
baud_layout.addWidget(baud_label)
|
||
baud_layout.addWidget(self.baud_input)
|
||
baud_layout.addWidget(self.connect_btn)
|
||
content_layout.addLayout(baud_layout)
|
||
|
||
# 通道数设置区域
|
||
channel_layout = QHBoxLayout()
|
||
channel_label = QLabel("通道数:")
|
||
self.channel_input = QLineEdit()
|
||
self.channel_input.setText("8")
|
||
self.channel_input.setValidator(QIntValidator(1, 32))
|
||
self.channel_input.setPlaceholderText("请输入通道数(1-32)")
|
||
self.apply_channel_btn = QPushButton("应用设置")
|
||
self.apply_channel_btn.clicked.connect(self.apply_channel_settings)
|
||
channel_layout.addWidget(channel_label)
|
||
channel_layout.addWidget(self.channel_input)
|
||
channel_layout.addWidget(self.apply_channel_btn)
|
||
content_layout.addLayout(channel_layout)
|
||
|
||
# 数据开关命令区域
|
||
data_cmd_layout = QHBoxLayout()
|
||
data_cmd_label = QLabel("数据开关:")
|
||
self.data_cmd_btn_on = QPushButton("ON")
|
||
self.data_cmd_btn_on.clicked.connect(self.on_data_cmd_btn_on)
|
||
self.data_cmd_btn_off = QPushButton("OFF")
|
||
self.data_cmd_btn_off.clicked.connect(self.on_data_cmd_btn_off)
|
||
|
||
data_cmd_layout.addWidget(data_cmd_label)
|
||
data_cmd_layout.addWidget(self.data_cmd_btn_on)
|
||
data_cmd_layout.addWidget(self.data_cmd_btn_off)
|
||
content_layout.addLayout(data_cmd_layout)
|
||
|
||
# 开关按钮区域
|
||
cmd_layout = QHBoxLayout()
|
||
cmd_label = QLabel("开关命令:")
|
||
self.cmd_btn_on = QPushButton("ON")
|
||
self.cmd_btn_on.clicked.connect(self.on_cmd_btn_on)
|
||
self.cmd_btn_off = QPushButton("OFF")
|
||
self.cmd_btn_off.clicked.connect(self.on_cmd_btn_off)
|
||
|
||
cmd_layout.addWidget(cmd_label)
|
||
cmd_layout.addWidget(self.cmd_btn_on)
|
||
cmd_layout.addWidget(self.cmd_btn_off)
|
||
content_layout.addLayout(cmd_layout)
|
||
|
||
# 录制控制区域
|
||
record_layout = QHBoxLayout()
|
||
self.record_btn = QPushButton("开始录制")
|
||
self.record_btn.setCheckable(True)
|
||
self.record_btn.clicked.connect(self.toggle_recording)
|
||
record_layout.addWidget(self.record_btn)
|
||
content_layout.addLayout(record_layout)
|
||
|
||
|
||
# 标签输入区域
|
||
tag_layout = QHBoxLayout()
|
||
self.tag_input = QLineEdit()
|
||
self.tag_input.setPlaceholderText("输入标签内容")
|
||
self.add_label_btn = QPushButton("增加标签")
|
||
self.add_label_btn.clicked.connect(self.add_tag)
|
||
tag_layout.addWidget(self.tag_input)
|
||
tag_layout.addWidget(self.add_label_btn)
|
||
content_layout.addLayout(tag_layout)
|
||
|
||
# 数据统计显示区域
|
||
stats_layout = QHBoxLayout()
|
||
self.stats_label = QLabel("总帧数: 0 | 有效帧: 0 | 损坏帧: 0")
|
||
self.stats_label.setAlignment(Qt.AlignCenter)
|
||
stats_layout.addWidget(self.stats_label)
|
||
content_layout.addLayout(stats_layout)
|
||
|
||
# 添加内容区域到主布局
|
||
layout.addWidget(content_widget)
|
||
|
||
# 添加状态栏
|
||
self.statusBar().showMessage("就绪")
|
||
|
||
# 初始化串口列表
|
||
self.refresh_ports()
|
||
|
||
# 初始化录制状态
|
||
self.is_recording = False
|
||
self.serial_recorder = None
|
||
|
||
# 设置对象名称以便样式表识别
|
||
self.record_btn.setObjectName("record_btn")
|
||
|
||
# 增加组件间距
|
||
content_layout.setSpacing(10)
|
||
|
||
# 为所有水平布局添加间距
|
||
for layout_item in [port_layout, baud_layout, channel_layout, data_cmd_layout, cmd_layout, record_layout, tag_layout]:
|
||
layout_item.setSpacing(10)
|
||
|
||
# 设置窗口最小尺寸
|
||
self.setMinimumSize(450, 250)
|
||
|
||
# 初始化时禁用录制按钮和命令按钮
|
||
self.record_btn.setEnabled(False)
|
||
self.cmd_btn_on.setEnabled(False)
|
||
self.cmd_btn_off.setEnabled(False)
|
||
self.data_cmd_btn_on.setEnabled(False)
|
||
self.data_cmd_btn_off.setEnabled(False)
|
||
|
||
# 创建记录保存目录
|
||
self.records_save_path = './records'
|
||
if not os.path.exists(self.records_save_path):
|
||
os.makedirs(self.records_save_path)
|
||
|
||
# 创建定时器用于更新统计信息
|
||
self.stats_timer = QTimer()
|
||
self.stats_timer.timeout.connect(self.update_stats_display)
|
||
self.stats_timer.start(1000) # 每秒更新一次
|
||
|
||
# 添加退出时的确认标志
|
||
self.exit_confirmed = False
|
||
|
||
# 添加鼠标事件处理
|
||
self.title_bar = title_bar
|
||
self.title_bar.mousePressEvent = self.mousePressEvent
|
||
self.title_bar.mouseMoveEvent = self.mouseMoveEvent
|
||
self.title_bar.mouseReleaseEvent = self.mouseReleaseEvent
|
||
|
||
# 窗口拖动相关变量
|
||
self.dragging = False
|
||
self.drag_position = None
|
||
|
||
def mousePressEvent(self, event):
|
||
"""鼠标按下事件"""
|
||
if event.button() == Qt.LeftButton and self.title_bar.rect().contains(event.position().toPoint()):
|
||
self.dragging = True
|
||
self.drag_position = event.globalPosition().toPoint() - self.frameGeometry().topLeft()
|
||
event.accept()
|
||
|
||
def mouseMoveEvent(self, event):
|
||
"""鼠标移动事件"""
|
||
if self.dragging:
|
||
self.move(event.globalPosition().toPoint() - self.drag_position)
|
||
event.accept()
|
||
|
||
def mouseReleaseEvent(self, event):
|
||
"""鼠标释放事件"""
|
||
if event.button() == Qt.LeftButton:
|
||
self.dragging = False
|
||
event.accept()
|
||
|
||
def on_data_cmd_btn_on(self):
|
||
"""发送数据ON命令"""
|
||
if self.serial_recorder and self.serial_recorder.serial:
|
||
try:
|
||
self.serial_recorder.serial.write(CMD_DATA_ON)
|
||
self.statusBar().showMessage("已发送数据ON命令")
|
||
except serial.serialutil.SerialTimeoutException:
|
||
QMessageBox.warning(self, "警告", "发送数据ON命令超时，请检查串口连接")
|
||
self.statusBar().showMessage("发送数据ON命令失败")
|
||
except Exception as e:
|
||
QMessageBox.warning(self, "错误", f"发送数据ON命令失败: {str(e)}")
|
||
self.statusBar().showMessage("发送数据ON命令失败")
|
||
|
||
def on_data_cmd_btn_off(self):
|
||
"""发送数据OFF命令"""
|
||
if self.serial_recorder and self.serial_recorder.serial:
|
||
try:
|
||
self.serial_recorder.serial.write(CMD_DATA_OFF)
|
||
self.statusBar().showMessage("已发送数据OFF命令")
|
||
except serial.serialutil.SerialTimeoutException:
|
||
QMessageBox.warning(self, "警告", "发送数据OFF命令超时，请检查串口连接")
|
||
self.statusBar().showMessage("发送数据OFF命令失败")
|
||
except Exception as e:
|
||
QMessageBox.warning(self, "错误", f"发送数据OFF命令失败: {str(e)}")
|
||
self.statusBar().showMessage("发送数据OFF命令失败")
|
||
|
||
def on_cmd_btn_on(self):
|
||
"""发送ON命令"""
|
||
if self.serial_recorder and self.serial_recorder.serial:
|
||
try:
|
||
self.serial_recorder.serial.write(CMD_ON)
|
||
self.statusBar().showMessage("已发送ON命令")
|
||
except serial.serialutil.SerialTimeoutException:
|
||
QMessageBox.warning(self, "警告", "发送ON命令超时，请检查串口连接")
|
||
self.statusBar().showMessage("发送ON命令失败")
|
||
except Exception as e:
|
||
QMessageBox.warning(self, "错误", f"发送ON命令失败: {str(e)}")
|
||
self.statusBar().showMessage("发送ON命令失败")
|
||
|
||
def on_cmd_btn_off(self):
|
||
"""发送OFF命令"""
|
||
if self.serial_recorder and self.serial_recorder.serial:
|
||
try:
|
||
self.serial_recorder.serial.write(CMD_OFF)
|
||
self.statusBar().showMessage("已发送OFF命令")
|
||
except serial.serialutil.SerialTimeoutException:
|
||
QMessageBox.warning(self, "警告", "发送OFF命令超时，请检查串口连接")
|
||
self.statusBar().showMessage("发送OFF命令失败")
|
||
except Exception as e:
|
||
QMessageBox.warning(self, "错误", f"发送OFF命令失败: {str(e)}")
|
||
self.statusBar().showMessage("发送OFF命令失败")
|
||
|
||
def update_stats_display(self):
|
||
"""更新统计信息显示"""
|
||
if self.serial_recorder:
|
||
stats = self.serial_recorder.frame_processor.get_statistics()
|
||
self.stats_label.setText(
|
||
f"总帧数: {stats['total_frames']} | "
|
||
f"有效帧: {stats['valid_frames']} | "
|
||
f"损坏帧: {stats['corrupt_frames']}"
|
||
)
|
||
|
||
def refresh_ports(self):
|
||
"""刷新可用串口列表"""
|
||
self.port_combo.clear()
|
||
ports = serial.tools.list_ports.comports()
|
||
|
||
if ports:
|
||
for port in ports:
|
||
# 格式化串口信息：端口 - 设备名称 (设备ID)
|
||
port_info = f"{port.device} - {port.description}"
|
||
if port.serial_number:
|
||
port_info += f" ({port.serial_number})"
|
||
self.port_combo.addItem(port_info, port.device)
|
||
else:
|
||
self.port_combo.addItem("未找到可用串口")
|
||
|
||
def toggle_connection(self):
|
||
"""切换串口连接状态"""
|
||
if not self.connect_btn.isChecked():
|
||
# 断开连接
|
||
if self.serial_recorder:
|
||
self.serial_recorder.stop()
|
||
self.serial_recorder.disconnect()
|
||
self.statusBar().showMessage("已断开连接")
|
||
self.record_btn.setEnabled(False)
|
||
self.cmd_btn_on.setEnabled(False)
|
||
self.cmd_btn_off.setEnabled(False)
|
||
self.data_cmd_btn_on.setEnabled(False)
|
||
self.data_cmd_btn_off.setEnabled(False)
|
||
self.connect_btn.setText("连接") # 更新按钮文字为"连接"
|
||
else:
|
||
# 建立连接
|
||
port = self.port_combo.currentData()
|
||
if not port:
|
||
QMessageBox.warning(self, "警告", "请选择串口")
|
||
self.connect_btn.setChecked(False)
|
||
return
|
||
|
||
try:
|
||
baudrate = int(self.baud_input.text())
|
||
num_channels = int(self.channel_input.text().strip())
|
||
if not 1 <= num_channels <= 32:
|
||
raise ValueError("通道数必须在1-32之间")
|
||
except ValueError as e:
|
||
QMessageBox.warning(self, "警告", str(e))
|
||
self.connect_btn.setChecked(False)
|
||
return
|
||
|
||
# 创建串口记录器实例
|
||
self.serial_recorder = SerialRecorder(
|
||
output_dir=self.records_save_path,
|
||
num_channels=num_channels
|
||
)
|
||
|
||
# 设置串口参数
|
||
self.serial_recorder.set_port(port, baudrate)
|
||
|
||
# 测试连接
|
||
if self.serial_recorder.test_connection():
|
||
self.statusBar().showMessage("已连接")
|
||
self.record_btn.setEnabled(True)
|
||
self.cmd_btn_on.setEnabled(True)
|
||
self.cmd_btn_off.setEnabled(True)
|
||
self.data_cmd_btn_on.setEnabled(True)
|
||
self.data_cmd_btn_off.setEnabled(True)
|
||
self.connect_btn.setText("断开") # 更新按钮文字为"断开"
|
||
else:
|
||
self.connect_btn.setChecked(False)
|
||
QMessageBox.warning(self, "错误", "连接失败")
|
||
|
||
def toggle_recording(self):
|
||
"""切换录制状态"""
|
||
if not self.record_btn.isChecked():
|
||
# 停止录制
|
||
if self.serial_recorder:
|
||
self.serial_recorder.stop()
|
||
self.is_recording = False
|
||
self.record_btn.setText("开始录制")
|
||
self.statusBar().showMessage("录制已停止")
|
||
else:
|
||
# 开始录制
|
||
if self.serial_recorder:
|
||
if self.serial_recorder.start():
|
||
self.is_recording = True
|
||
self.record_btn.setText("停止录制")
|
||
self.statusBar().showMessage("正在录制")
|
||
else:
|
||
self.record_btn.setChecked(False)
|
||
QMessageBox.warning(self, "错误", "启动录制失败")
|
||
|
||
def add_tag(self):
|
||
"""添加标签"""
|
||
tag = self.tag_input.text().strip()
|
||
if tag:
|
||
# TODO: 实现标签添加逻辑
|
||
self.tag_input.clear()
|
||
self.statusBar().showMessage(f"已添加标签: {tag}")
|
||
|
||
def apply_channel_settings(self):
|
||
"""应用通道数设置"""
|
||
try:
|
||
num_channels = int(self.channel_input.text().strip())
|
||
if not 1 <= num_channels <= 32:
|
||
raise ValueError("通道数必须在1-32之间")
|
||
|
||
if self.serial_recorder:
|
||
self.serial_recorder.set_channel_nums(num_channels)
|
||
self.statusBar().showMessage(f"已设置通道数为: {num_channels}")
|
||
else:
|
||
self.statusBar().showMessage("请先连接串口")
|
||
|
||
except ValueError as e:
|
||
QMessageBox.warning(self, "警告", str(e))
|
||
self.statusBar().showMessage("设置通道数失败")
|
||
except Exception as e:
|
||
QMessageBox.warning(self, "错误", f"设置通道数失败: {str(e)}")
|
||
self.statusBar().showMessage("设置通道数失败")
|
||
|
||
def closeEvent(self, event):
|
||
"""关闭事件处理"""
|
||
if not self.exit_confirmed:
|
||
reply = QMessageBox.question(
|
||
self, '确认退出',
|
||
'确定要退出程序吗？',
|
||
QMessageBox.Yes | QMessageBox.No,
|
||
QMessageBox.No
|
||
)
|
||
|
||
if reply == QMessageBox.Yes:
|
||
self.exit_confirmed = True
|
||
self.cleanup_resources()
|
||
event.accept()
|
||
else:
|
||
event.ignore()
|
||
else:
|
||
self.cleanup_resources()
|
||
event.accept()
|
||
|
||
def cleanup_resources(self):
|
||
"""清理资源"""
|
||
if self.serial_recorder:
|
||
if self.is_recording:
|
||
self.serial_recorder.stop() # 确保停止录制并保存数据
|
||
self.serial_recorder.disconnect()
|
||
self.stats_timer.stop()
|
||
|
||
if __name__ == '__main__':
|
||
app = QApplication(sys.argv)
|
||
window = SerialUI()
|
||
window.show()
|
||
sys.exit(app.exec())
|