import numpy as np import socket import logging import matplotlib.pyplot as plt import cv2 def rec_socket(recv_sock: socket.socket, cmd_type: str, ack: bool) -> bool: if ack: cmd = 'A' + cmd_type else: cmd = 'D' + cmd_type while True: try: temp = recv_sock.recv(1) except ConnectionError as e: logging.error(f'连接出错, 错误代码:\n{e}') return False except TimeoutError as e: logging.error(f'超时了，错误代码: \n{e}') return False except Exception as e: logging.error(f'遇见未知错误，错误代码: \n{e}') return False if temp == b'\xaa': break # 获取报文长度 temp = b'' while len(temp) < 4: try: temp += recv_sock.recv(1) except Exception as e: logging.error(f'接收报文长度失败, 错误代码: \n{e}') return False try: data_len = int.from_bytes(temp, byteorder='big') except Exception as e: logging.error(f'转换失败,错误代码 \n{e}, \n报文内容\n{temp}') return False # 读取报文内容 temp = b'' while len(temp) < data_len: try: temp += recv_sock.recv(data_len) except Exception as e: logging.error(f'接收报文内容失败, 错误代码: \n{e}，\n报文内容\n{temp}') return False data = temp if cmd.strip().upper() != data[:4].decode('ascii').strip().upper(): logging.error(f'客户端接收指令错误,\n指令内容\n{data}') return False else: if cmd == 'DIM': print(data) # 进行数据校验 temp = b'' while len(temp) < 3: try: temp += recv_sock.recv(1) except Exception as e: logging.error(f'接收报文校验失败, 错误代码: \n{e}') return False if temp == b'\xff\xff\xbb': return True else: logging.error(f"接收了一个完美的只错了校验位的报文，\n data: {data}") return False # def main(): # socket_receive = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # socket_receive.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # socket_receive.bind(('127.0.0.1', 21123)) # socket_receive.listen(5) # socket_send = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # socket_send.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # socket_send.bind(('127.0.0.1', 21122)) # socket_send.listen(5) # print('等待连接') # socket_send_1, receive_addr_1 = socket_send.accept() # print("连接成功：", receive_addr_1) # # socket_send_2 = socket_send_1 # socket_send_2, receive_addr_2 = socket_receive.accept() # print("连接成功：", receive_addr_2) # while True: # cmd = input('请输入指令：').strip().upper() # if cmd == 'IM': # with open('data/newrawfile_ref.raw', 'rb') as f: # data = np.frombuffer(f.read(), dtype=np.float32).reshape(750, 288, 384) # data = data[:, [91, 92, 93, 94, 95, 96, 97, 98, 99, 100], :] # n_rows, n_bands, n_cols = data.shape[0], data.shape[1], data.shape[2] # print(f'n_rows：{n_rows}, n_bands：{n_bands}, n_cols：{n_cols}') # n_rows, n_cols, n_bands = [x.to_bytes(2, byteorder='big') for x in [n_rows, n_cols, n_bands]] # data = data.tobytes() # length = len(data) + 10 # print(f'length: {length}') # length = length.to_bytes(4, byteorder='big') # msg = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + n_rows + n_cols + n_bands + data + b'\xff\xff\xbb' # socket_send_1.send(msg) # print('发送成功') # result = socket_send_2.recv(5) # length = int.from_bytes(result[1:5], byteorder='big') # result = b'' # while len(result) < length: # result += socket_send_2.recv(length) # print(result) # data = result[4:length].decode() # print(data) def main(): socket_receive = socket.socket(socket.AF_INET, socket.SOCK_STREAM) socket_receive.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) socket_receive.bind(('127.0.0.1', 21123)) socket_receive.listen(5) socket_send = socket.socket(socket.AF_INET, socket.SOCK_STREAM) socket_send.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) socket_send.bind(('127.0.0.1', 21122)) socket_send.listen(5) print('等待连接') socket_send_1, receive_addr_1 = socket_send.accept() print("连接成功：", receive_addr_1) # socket_send_2 = socket_send_1 socket_send_2, receive_addr_2 = socket_receive.accept() print("连接成功：", receive_addr_2) while True: cmd = input().strip().upper() if cmd == 'IM': image_paths = [ r"D:\project\deep-learning-for-image-processing-master\data_set\test_image\1.jpg", r"D:\project\deep-learning-for-image-processing-master\data_set\test_image\2.jpg", r"D:\project\deep-learning-for-image-processing-master\data_set\test_image\3.jpg", r"D:\project\deep-learning-for-image-processing-master\data_set\test_image\4.jpg", r"D:\project\deep-learning-for-image-processing-master\data_set\test_image\5.jpg", ] # 连续发送五张图片 for image_path in image_paths: # 读取图片文件 img = cv2.imread(image_path) # img = cv2.imread(r"/Users/zhouchao/Library/CloudStorage/OneDrive-macrosolid/PycharmProjects/wood_color/data/data20220919/dark/rgb60.png") # img = cv2.imread(r"D:\project\Astragalin\data\02newrawfile_ref.png") img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = np.asarray(img, dtype=np.uint8) width = img.shape[0] height = img.shape[1] print(width, height) img_bytes = img.tobytes() length = len(img_bytes) + 8 print(length) length = length.to_bytes(4, byteorder='big') width = width.to_bytes(2, byteorder='big') height = height.to_bytes(2, byteorder='big') send_message = b'\xaa' + length + (' ' + cmd).upper().encode('ascii') + width + height + img_bytes + b'\xff\xff\xbb' socket_send_1.send(send_message) print('发送成功') result = socket_send_2.recv(10) print(result) if __name__ == '__main__': main()