From fc05a447c0ff6b3a3c90b85778874aaf3195747b Mon Sep 17 00:00:00 2001
From: wrz1-zzzzz <914921336@qq.com>
Date: Wed, 11 Dec 2024 19:43:14 +0800
Subject: [PATCH] socket

---
 socket/socket.py  |  8 +++---
 socket/socket2.py | 70 +++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 74 insertions(+), 4 deletions(-)
 create mode 100644 socket/socket2.py

diff --git a/socket/socket.py b/socket/socket.py
index ab09692..f9102ad 100644
--- a/socket/socket.py
+++ b/socket/socket.py
@@ -59,7 +59,7 @@ m. 模式切换：测下一个喷阀还是重发？
 
     @staticmethod
     def cmd_padding(cmd):
-        return b'\xAA' + cmd + b'\xFF\xFF\xBB'
+        return b'\xAA' + cmd + b'\xFF\xFF\xBB'   #起始结束位置
 
     @staticmethod
     def param_cmd_parser(cmd, default_value, checker=None):
@@ -179,9 +179,9 @@ m. 模式切换：测下一个喷阀还是重发？
     def send(self, cmd: bytes) -> None:
         cmd = self.cmd_padding(cmd)
         print("我要 send 这个了:")
-        print(cmd.hex())
+        print(cmd.hex())  #16
         try:
-            self.c.send(cmd)
+            self.c.send(cmd)  #发送至客户端
         except Exception as e:
             print(f"发失败了, 这是我找到的错误信息:\n{e}")
             return
@@ -213,7 +213,7 @@ if __name__ == '__main__':
     import argparse
     parser = argparse.ArgumentParser(description='阀门测程序')
     parser.add_argument('-c', default=False, action='store_true', help='是否是开个客户端', required=False)
-    parser.add_argument('-m', default='192.168.10.8', help='指定master主机名')
+    parser.add_argument('-m', default='127.0.0.1', help='指定master主机名')
     parser.add_argument('-p', default=13452, help='指定端口')
     parser.add_argument('-cnt', default=64, help='指定端口')
     args = parser.parse_args()
diff --git a/socket/socket2.py b/socket/socket2.py
new file mode 100644
index 0000000..155af79
--- /dev/null
+++ b/socket/socket2.py
@@ -0,0 +1,70 @@
+import socket
+import time
+
+import cv2
+import numpy as np
+
+class VirtualValve:
+    def __init__(self, host, port, column_num=64):
+        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 声明socket类型，同时生成链接对象
+        self.client.connect((host, port))  # 建立一个链接，连接到本地的13452端口
+        self.column_num = column_num
+
+    @staticmethod
+    def decode_data_to_matrix(data, column_num):
+        """
+        将接收到的数据解码为矩阵形式
+        :param data: 原始字节数据
+        :param column_num: 列数，用于解码为矩阵
+        :return: 解码后的矩阵
+        """
+        try:
+            # 假设数据中包含字节打包的矩阵
+            unpacked_data = np.frombuffer(data, dtype=np.uint8)  # 转换为无符号8位整型数组
+            unpacked_bits = np.unpackbits(unpacked_data)  # 将字节解压为位数组
+
+            # 计算行数（确保数据长度是列数的倍数）
+            row_num = len(unpacked_bits) // column_num
+            if len(unpacked_bits) % column_num != 0:
+                print("数据长度不是列数的整数倍，可能存在错误。")
+                return None
+
+            # 转换为矩阵
+            matrix = unpacked_bits.reshape((row_num, column_num))
+            return matrix
+        except Exception as e:
+            print(f"解码数据为矩阵时发生错误: {e}")
+            return None
+
+    def run(self):
+        while True:
+            data = self.client.recv(4096)  # 接收信息，指定接收的大小
+            if not data:
+                break
+
+            print("原始报文:")
+            print(data.hex())  # 输出接收到的原始字节数据
+
+            # 尝试解码为矩阵形式
+            matrix = self.decode_data_to_matrix(data, self.column_num)
+            if matrix is not None:
+                print("转换后的矩阵:")
+                print(matrix)
+            else:
+                print("无法将数据转换为矩阵。")
+
+
+
+if __name__ == '__main__':
+    import argparse
+    parser = argparse.ArgumentParser(description='阀门测程序')
+    parser.add_argument('-c', default=True, action='store_true', help='是否是开个客户端', required=False)
+    parser.add_argument('-m', default='127.0.0.1', help='指定master主机名')
+    parser.add_argument('-p', default=13452, help='指定端口')
+    parser.add_argument('-cnt', default=64, help='指定端口')
+    args = parser.parse_args()
+    mask_size = (1024, args.cnt)  # size of cv (Width, Height)
+    if args.c:
+        print("运行客户机")
+        virtual_valve = VirtualValve(host=args.m, port=args.p)
+        virtual_valve.run()
\ No newline at end of file