From 345de6a942ce0b8f469ace58e95b689378adc2d4 Mon Sep 17 00:00:00 2001
From: zhangsheng <zhangs1125@sina.com>
Date: Tue, 8 Apr 2025 16:47:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=9B=BA=E4=BB=B6=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 mainwindow.py | 198 +++++++++++++++++++++++++++++++++-----------------
 1 file changed, 133 insertions(+), 65 deletions(-)

diff --git a/mainwindow.py b/mainwindow.py
index 6fb1b88..6df91a8 100644
--- a/mainwindow.py
+++ b/mainwindow.py
@@ -1,14 +1,16 @@
 import sys
 import socket
+import os
 import numpy as np
 from scipy.fft import fft, fftfreq
 import matplotlib.pyplot as plt
 from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas, NavigationToolbar2QT
-from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QPushButton, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QFrame, QGridLayout, QStatusBar, QSizePolicy
+from PyQt5.QtWidgets import QApplication, QMainWindow, QWidget, QPushButton, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QFrame, QGridLayout, QStatusBar, QSizePolicy, QFileDialog
 from PyQt5.QtCore import Qt
 import matplotlib
 import struct
 import time
+from typing import Tuple, Optional
 # 启用高DPI支持
 QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
 QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)
@@ -19,6 +21,8 @@ matplotlib.use('Qt5Agg')
 plt.rcParams['font.sans-serif'] = ['SimHei']  # Windows 中文字体
 plt.rcParams['axes.unicode_minus'] = False  # 解决负号 '-' 显示问题
 
+HEADER_MAGIC = bytes([0xAA, 0x55, 0xAA])
+HEADER_SIZE = 6  # PackgeHead除去data字段的大小(3+1+1+1)
 
 class MatplotlibCanvas(FigureCanvas):
     """ 用于在 Qt 界面中嵌入 Matplotlib 绘图 """
@@ -81,7 +85,9 @@ class SocketClientApp(QMainWindow):
         self.port_input = QLineEdit(self)
         self.connect_button = QPushButton("连接")
         self.get_data_button = QPushButton("获取数据")
+        self.upgrade_button = QPushButton("更新固件")
         self.get_data_button.setEnabled(False)  # 初始状态不可点击
+        self.upgrade_button.setEnabled(False)  # 初始状态不可点击
         self.ip_input.setText("192.168.0.200")
         self.port_input.setText("12345")
 
@@ -114,6 +120,7 @@ class SocketClientApp(QMainWindow):
         button_layout = QHBoxLayout()
         button_layout.addWidget(self.connect_button)
         button_layout.addWidget(self.get_data_button)
+        button_layout.addWidget(self.upgrade_button)
         main_layout.addLayout(button_layout)
 
         # 添加分割线
@@ -153,10 +160,17 @@ class SocketClientApp(QMainWindow):
         # 事件绑定
         self.connect_button.clicked.connect(self.connect_to_server)
         self.get_data_button.clicked.connect(self.on_button_clicked)
+        self.upgrade_button.clicked.connect(self.on_button_upgrade)
 
         # 设置布局策略，确保控件大小随窗口调整
         self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
 
+    def calculate_crc(self,data: bytes) -> int:
+        """计算数据的累加和CRC"""
+        crc = 0
+        for byte in data:
+            crc += byte
+        return crc & 0xFF  # 只保留最低字节
     def connect_to_server(self):
         """ 连接到服务器 """
         ip = self.ip_input.text()
@@ -164,6 +178,7 @@ class SocketClientApp(QMainWindow):
 
         if not ip or not port:
             self.status_bar.showMessage("状态: 请输入有效的 IP 和端口号")
+            
             return
 
         try:
@@ -175,86 +190,92 @@ class SocketClientApp(QMainWindow):
             self.socket.connect((ip, port))
 
             self.get_data_button.setEnabled(True)  # 连接成功，启用获取数据按钮
+            self.upgrade_button.setEnabled(True)
             self.status_bar.showMessage(f"状态: 连接成功! 服务器 {ip}:{port}")
 
         except Exception as e:
             self.status_bar.showMessage(f"状态: 连接失败 - {str(e)}")
 
-    def receive_data(self):
+    def parse_package_head(self,data: bytes) -> Tuple[Optional[dict], Optional[bytes]]:
+        """
+        解析包头部
+        返回: (header_dict, remaining_data)
+        """
+        if len(data) < HEADER_SIZE:
+            return None, None
+
+        # 检查魔数
+        if data[:3] != HEADER_MAGIC:
+            print(f"Invalid header magic: {data[:3].hex()}")
+            return None, None
+
+        # 解析头部
+        cmd, version, result_code = struct.unpack_from('<BBB', data, 3)
+        return {
+            'cmd': cmd,
+            'version': version,
+            'result_code': result_code
+        }, data[HEADER_SIZE:]
+
+    def receive_data(self,length: int):
+        """
+        从socket接收指定长度的数据
+        """
+        data = bytearray()
+        while len(data) < length:
+            chunk = self.socket.recv(length - len(data))
+            if not chunk:
+                raise ConnectionError("Connection closed")
+            data.extend(chunk)
+        print(f"data size {len(data)}")
+        return bytes(data)
+
+    def process_packet(self):
         """ 循环接收数据，直到接收完整 """
-        received_data = b""  # 初始化接收数据缓存
-        complete_data = b""  # 用于存储拼接后的完整数据
-        buffer_size = 0
-        # 记录开始时间
-        start_time = time.time()
         try:
-            while True:
-                try:
-                    # 每次接收 1024 字节的数据
-                    chunk =  self.socket.recv(1460)
-                    buffer_size += len(chunk)
-                    print(f"len partf {len(chunk)},buffer_size {buffer_size}")
-                    if not chunk:
-                        break  # 如果没有数据，退出循环
-                    received_data += chunk
-                except socket.timeout:
-                    print(f"接收数据超时（{2} 秒），退出接收循环")
-                    break  # 超时后退出接收循环
-            # 记录结束时间
-            end_time = time.time()
+            # 1. 接收包头
+            start_time = time.time()
+            header_data = self.receive_data(6)
+            header, _ = self.parse_package_head(header_data)
+            if not header:
+                print("Invalid package header")
+                return None
 
-            # 计算并打印执行时间
-            execution_time = end_time - start_time
-            print(f"代码执行时间: {execution_time} 秒")
-            # 循环查找并解析帧头
-            while True:
-                # 查找帧头
-                frame_start = received_data.find(b'\xAA\x55\xAA')
-                if frame_start == -1:
-                    break  # 没有找到帧头，退出内层循环，等待更多数据
-
-                # 丢弃帧头之前的数据（可能是无效数据）
-                received_data = received_data[frame_start:]
-
-                # 检查数据是否足够解析长度字段
-                if len(received_data) < 7:
-                    break  # 数据不完整，退出内层循环，等待更多数据
-
-                try:
-                    # 解析数据长度字段
-                    data_length = struct.unpack('<I', received_data[3:7])[0]
-                except Exception as e:
-                    print(f"解析长度字段失败: {e}")
-                    received_data = received_data[1:]  # 丢弃无效字节，继续查找帧头
-                    continue
-
-                # 检查数据是否足够解析完整帧
-                print(f"解析长度字段 {len(received_data)}")
-                if len(received_data) < 7 + data_length:
-                    break  # 数据不完整，退出内层循环，等待更多数据
-
-                # 提取实际数据
-                actual_data = received_data[7:7 + data_length]
-                complete_data += actual_data  # 将数据拼接到完整数据中
-                # 丢弃已处理的数据
-                received_data = received_data[7 + data_length:]
+            # 2. 根据命令类型处理数据
+            if header['cmd'] == 0x01:
+                # 先接收WaveDataRsp的固定部分(5字节)
+                wave_header = self.receive_data(5)
+                channel_id = wave_header[0]
+                sampling_rate = struct.unpack('<i', wave_header[1:5])[0]
+                print(f"channel_id {channel_id}")
+                print(f"sampling_rate {sampling_rate}")
+                # 计算需要接收的数据长度
+                data_length = sampling_rate * 4
+                wave_data = self.receive_data(data_length)
+                end_time = time.time()
+                # 计算并打印执行时间
+                execution_time = end_time - start_time
+                print(f"代码执行时间: {execution_time} 秒")
+                # 合并成完整的数据包
+                # full_packet = header_data + wave_header + wave_data
+                return wave_data
 
+            else:
+                print(f"Unsupported command: {header['cmd']}")
+                return None
+        except ConnectionError as e:
+            print(f"Connection error: {e}")
+            return None
         except Exception as e:
-            print(f"发生错误: {e}")
-        print(f" 数据: {len(complete_data)}")
-        # 打印前100个字节的数据（以16进制方式）
-        print("前100个字节的数据（16进制表示）：")
-        for i in range(min(100, len(complete_data))):  # 确保不超过数据长度
-            print(f"{complete_data[i]:02X}", end=" ")  # 每个字节以16进制格式打印
-        print()  # 换行
-        return complete_data
+            print(f"Error processing packet: {e}")
+            return None
 
     def on_button_clicked(self):
         """ 获取数据并绘制 """
         try:
             self.status_bar.showMessage("状态: 正在获取数据...")
             self.socket.sendall(bytes([0xAA, 0x55, 0xAA,0x01]))  # 发送数据
-            data = self.receive_data()  # 接收所有数据
+            data = self.process_packet()  # 接收所有数据
             data = np.frombuffer(data, dtype=np.int32)  # 根据实际数据格式转换
             for i in range(min(100, len(data))):  # 确保不超过数据长度
                 print(f"{data[i]:1f}", end=" ")
@@ -269,6 +290,53 @@ class SocketClientApp(QMainWindow):
         except Exception as e:
             self.status_bar.showMessage(f"状态: 错误 - {str(e)}")
 
+    def on_button_upgrade(self):
+        """打开文件选择对话框"""
+        file_path, _ = QFileDialog.getOpenFileName(
+            self, "选择升级文件", "", "二进制文件 (*.bin);;所有文件 (*)"
+        )
+
+        if file_path:
+            file_name = os.path.basename(file_path)
+            self.status_bar.showMessage(f"已选择: {file_name}")
+        try:
+            print("Starting upgrade process...")
+
+            if not os.path.exists(file_path):
+                raise FileNotFoundError(f"Upgrade file {file_path} not found")
+
+            with open(file_path, "rb") as f:
+                package_data = f.read()
+
+            if not package_data:
+                raise ValueError("Upgrade file is empty")
+
+            print(f"Read upgrade package, size: {len(package_data)} bytes")
+
+
+            upgrade_len = len(package_data)
+            crc = self.calculate_crc(package_data)
+            upgrade_req = struct.pack("<IB", upgrade_len, crc) + package_data
+
+
+            header_magic = bytes([0xAA, 0x55, 0xAA])
+            cmd = 0x05
+            version = 1
+            result_code = 0
+
+            package_head = struct.pack("<3sBBB", header_magic, cmd, version, result_code)
+            full_packet = package_head + upgrade_req
+
+            print("Full packet prepared:")
+            print(f"Header magic: {package_head[:3].hex()}")
+            print(f"Command: {cmd}, Version: {version}")
+            print(f"Upgrade length: {upgrade_len}, CRC: {crc}")
+            print(f"Total packet size: {len(full_packet)} bytes")
+
+            self.socket.sendall(full_packet)
+            print("Upgrade packet ready to send (commented out actual send code)")
+        except Exception as e:
+            print(f"Upgrade failed: {str(e)}")
 
 if __name__ == '__main__':
     app = QApplication(sys.argv)