From c59b6a3c1b83059fe784911de061df980f875e21 Mon Sep 17 00:00:00 2001
From: zhangsheng <zhangs1125@sina.com>
Date: Sat, 22 Feb 2025 13:55:58 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AC=AC=E4=B8=80=E6=AC=A1=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore      |   3 +
 mainwindow.py   | 264 ++++++++++++++++++++++++++++++++++++++++++++++++
 mainwindow.spec |  38 +++++++
 server.py       |  56 ++++++++++
 4 files changed, 361 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 mainwindow.py
 create mode 100644 mainwindow.spec
 create mode 100644 server.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..2118c21
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+build/
+.idea/
+dist/
diff --git a/mainwindow.py b/mainwindow.py
new file mode 100644
index 0000000..52b2f80
--- /dev/null
+++ b/mainwindow.py
@@ -0,0 +1,264 @@
+import sys
+import socket
+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.QtCore import Qt
+import matplotlib
+import struct
+import time
+# 启用高DPI支持
+QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
+QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)
+
+# 强制使用 Qt5Agg 后端
+matplotlib.use('Qt5Agg')
+
+plt.rcParams['font.sans-serif'] = ['SimHei']  # Windows 中文字体
+plt.rcParams['axes.unicode_minus'] = False  # 解决负号 '-' 显示问题
+
+
+class MatplotlibCanvas(FigureCanvas):
+    """ 用于在 Qt 界面中嵌入 Matplotlib 绘图 """
+    def __init__(self, parent=None):
+        self.fig, self.axs = plt.subplots(2, 1, figsize=(10, 5))  # 2 个子图（时域 + 频域）
+        self.fig.subplots_adjust(hspace=0.6)  # 增大时域图和频域图的间距
+        super().__init__(self.fig)
+        self.setParent(parent)
+        self.init_plot()
+
+    def init_plot(self):
+        """ 初始化默认图像（占位提示） """
+        self.axs[0].set_title("时域信号", fontsize=12, fontweight='bold')
+        self.axs[0].set_xlabel("采样点")
+        self.axs[0].set_ylabel("幅度")
+        self.axs[0].text(0.5, 0.5, "暂无数据", fontsize=12, ha='center', va='center', transform=self.axs[0].transAxes)
+
+        self.axs[1].set_title("频域信号", fontsize=12, fontweight='bold')
+        self.axs[1].set_xlabel("频率 (Hz)")
+        self.axs[1].set_ylabel("幅度")
+        self.axs[1].text(0.5, 0.5, "暂无数据", fontsize=12, ha='center', va='center', transform=self.axs[1].transAxes)
+
+        self.draw()  # 更新绘图
+
+    def plot_data(self, data):
+        """ 绘制时域和频域图 """
+        self.axs[0].clear()
+        self.axs[1].clear()
+
+        # 时域信号
+        self.axs[0].plot(data, color='blue')
+        self.axs[0].set_title("时域信号", fontsize=12, fontweight='bold')
+        self.axs[0].set_xlabel("采样点")
+        self.axs[0].set_ylabel("幅度")
+
+        # 频域信号
+        N = len(data)
+        T = 1.0 / 96000  # 假设采样率为 96000 Hz
+        yf = fft(data)
+        xf = fftfreq(N, T)[:N // 2]
+
+        self.axs[1].plot(xf, 2.0 / N * np.abs(yf[:N // 2]), color='red')
+        self.axs[1].set_title("频域信号", fontsize=12, fontweight='bold')
+        self.axs[1].set_xlabel("频率 (Hz)")
+        self.axs[1].set_ylabel("幅度")
+
+        self.draw()  # 更新绘图
+
+
+class SocketClientApp(QMainWindow):
+    def __init__(self):
+        super().__init__()
+        self.setWindowTitle("Socket Client & Data Plotter")
+        self.setGeometry(100, 100, 700, 600)  # 设置初始尺寸
+
+        # 组件初始化
+        self.ip_label = QLabel("IP 地址:")
+        self.port_label = QLabel("端口号:")
+        self.ip_input = QLineEdit(self)
+        self.port_input = QLineEdit(self)
+        self.connect_button = QPushButton("连接")
+        self.get_data_button = QPushButton("获取数据")
+        self.get_data_button.setEnabled(False)  # 初始状态不可点击
+        self.ip_input.setText("192.168.0.200")
+        self.port_input.setText("12345")
+
+        # 预留绘图区域
+        self.canvas = MatplotlibCanvas(self)
+        self.toolbar = NavigationToolbar2QT(self.canvas, self)  # 添加工具栏（放大、缩小、拖拽）
+
+        # 预留特征值显示区域
+        self.feature_label = QLabel("特征值:")
+        self.acceleration_label = QLabel("加速度: -")
+        self.velocity_label = QLabel("速度: -")
+
+        # 设置分割线
+        self.line = QFrame()
+        self.line.setFrameShape(QFrame.HLine)
+        self.line.setFrameShadow(QFrame.Sunken)
+
+        # 布局管理
+        main_layout = QVBoxLayout()
+
+        # 输入区域（IP & 端口）
+        input_layout = QHBoxLayout()
+        input_layout.addWidget(self.ip_label)
+        input_layout.addWidget(self.ip_input)
+        input_layout.addWidget(self.port_label)
+        input_layout.addWidget(self.port_input)
+        main_layout.addLayout(input_layout)
+
+        # 按钮区域
+        button_layout = QHBoxLayout()
+        button_layout.addWidget(self.connect_button)
+        button_layout.addWidget(self.get_data_button)
+        main_layout.addLayout(button_layout)
+
+        # 添加分割线
+        main_layout.addWidget(self.line)
+
+        # 工具栏（放大、缩小、拖拽）
+        main_layout.addWidget(self.toolbar)
+
+        # 绘图 + 特征值区域
+        graph_layout = QVBoxLayout()
+        graph_layout.addWidget(self.canvas)
+
+        # 特征值显示区域
+        feature_layout = QGridLayout()
+        feature_layout.addWidget(self.feature_label, 0, 0)
+        feature_layout.addWidget(self.acceleration_label, 0, 1)
+        feature_layout.addWidget(self.velocity_label, 0, 2)
+        graph_layout.addLayout(feature_layout)
+
+        main_layout.addLayout(graph_layout)
+
+        # 增大特征值区域与状态栏的间距
+        main_layout.addSpacing(20)
+
+        # 创建状态栏
+        self.status_bar = QStatusBar()
+        self.setStatusBar(self.status_bar)
+
+        # 状态栏显示初始消息
+        self.status_bar.showMessage("状态: 请填写 IP 和端口号")
+
+        # 主窗口的中央部件
+        central_widget = QWidget()
+        central_widget.setLayout(main_layout)
+        self.setCentralWidget(central_widget)
+
+        # 事件绑定
+        self.connect_button.clicked.connect(self.connect_to_server)
+        self.get_data_button.clicked.connect(self.on_button_clicked)
+
+        # 设置布局策略，确保控件大小随窗口调整
+        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
+
+    def connect_to_server(self):
+        """ 连接到服务器 """
+        ip = self.ip_input.text()
+        port = self.port_input.text()
+
+        if not ip or not port:
+            self.status_bar.showMessage("状态: 请输入有效的 IP 和端口号")
+            return
+
+        try:
+            port = int(port)
+            self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            # 设置 keep-alive 选项
+            self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
+            self.socket.connect((ip, port))
+
+            self.get_data_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):
+        """ 循环接收数据，直到接收完整 """
+        received_data = b""  # 初始化接收数据缓存
+        complete_data = b""  # 用于存储拼接后的完整数据
+        buffer_size = 0
+        # 记录开始时间
+        start_time = time.time()
+        try:
+            while True:
+                # 每次接收 1024 字节的数据
+                chunk =  self.socket.recv(1000)
+                buffer_size += len(chunk)
+                print(f"len partf {len(chunk)},buffer_size {buffer_size}")
+                # 拼接数据
+                if buffer_size > 97000*4:
+                    break  # 如果没有数据，退出循环
+                received_data += chunk
+            # 记录结束时间
+            end_time = time.time()
+
+            # 计算并打印执行时间
+            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]
+                print(f"解析到的长度字段 {len(actual_data)}")
+                complete_data += actual_data  # 将数据拼接到完整数据中
+                print(f"解析到的长度字段2 {len(complete_data)}")
+                # 丢弃已处理的数据
+                received_data = received_data[7 + data_length:]
+                print(f"还剩下解析长度字段 {len(received_data)}")
+
+        except Exception as e:
+            print(f"发生错误: {e}")
+        print(f" 数据: {len(complete_data)}")
+        return complete_data
+
+    def on_button_clicked(self):
+        """ 获取数据并绘制 """
+        try:
+            self.status_bar.showMessage("状态: 正在获取数据...")
+            data = self.receive_data()  # 接收所有数据
+            data = np.frombuffer(data, dtype=np.int32)  # 根据实际数据格式转换
+            LSB_32BIT = (2.8 / (2 ** 31)) * ((750 + 287) / 287) * 1000
+            scaled_data = data * LSB_32BIT
+            self.canvas.plot_data(scaled_data)  # 在 Qt 界面中绘图
+            self.status_bar.showMessage("状态: 数据绘制完成")
+        except Exception as e:
+            self.status_bar.showMessage(f"状态: 错误 - {str(e)}")
+
+
+if __name__ == '__main__':
+    app = QApplication(sys.argv)
+    window = SocketClientApp()
+    window.show()
+    sys.exit(app.exec_())
diff --git a/mainwindow.spec b/mainwindow.spec
new file mode 100644
index 0000000..8a16908
--- /dev/null
+++ b/mainwindow.spec
@@ -0,0 +1,38 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+
+a = Analysis(
+    ['mainwindow.py'],
+    pathex=[],
+    binaries=[],
+    datas=[],
+    hiddenimports=[],
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=[],
+    noarchive=False,
+    optimize=0,
+)
+pyz = PYZ(a.pure)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    a.binaries,
+    a.datas,
+    [],
+    name='mainwindow',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    runtime_tmpdir=None,
+    console=True,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)
diff --git a/server.py b/server.py
new file mode 100644
index 0000000..164ac0d
--- /dev/null
+++ b/server.py
@@ -0,0 +1,56 @@
+import socket
+import numpy as np
+import time
+
+# 设置服务端IP和端口
+HOST = '0.0.0.0'  # 监听所有网络接口
+PORT = 12345  # 端口号
+
+
+def generate_data(sample_rate=96000, duration=1):
+    """生成模拟的采样数据"""
+    t = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
+    # 使用正弦波生成模拟数据，可以替换为其他类型的数据
+    data = np.sin(2 * np.pi * 1000 * t)  # 1kHz 正弦波
+    return data.astype(np.float32)
+
+
+def start_server():
+    """启动socket服务端"""
+    # 创建socket对象
+    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    server_socket.bind((HOST, PORT))
+    server_socket.listen(1)
+    print(f"服务器启动，等待客户端连接...")
+
+    # 等待客户端连接
+    client_socket, client_address = server_socket.accept()
+    print(f"客户端 {client_address} 已连接")
+
+    try:
+        while True:
+            # 生成96k采样的数据
+            data = generate_data(sample_rate=96000, duration=1)  # 生成 1 秒钟的采样数据
+            # 将数据转换为字节流
+            data_bytes = data.tobytes()
+
+            # 发送数据
+            client_socket.sendall(data_bytes)
+            print("发送数据给客户端...")
+
+
+            # 等待 1 秒后再发送下一组数据
+            time.sleep(1)
+
+    except Exception as e:
+        print(f"发生错误: {e}")
+
+    finally:
+        # 关闭连接
+        client_socket.close()
+        server_socket.close()
+        print("连接已关闭")
+
+
+if __name__ == '__main__':
+    start_server()