mirror of
https://github.com/NanjingForestryUniversity/supermachine--tomato-passion_fruit.git
synced 2025-11-09 23:03:58 +00:00
198 lines
7.9 KiB
Python
198 lines
7.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @Time : 2024/6/16 17:13
|
||
# @Author : TG
|
||
# @File : qt_test.py
|
||
# @Software: PyCharm
|
||
|
||
import sys
|
||
import os
|
||
from PyQt5.QtWidgets import QApplication, QMainWindow, QFileDialog, QLabel, QVBoxLayout, QWidget
|
||
from PyQt5.QtGui import QPixmap, QImage
|
||
import win32file
|
||
from PIL import Image
|
||
import numpy as np
|
||
import cv2
|
||
|
||
|
||
class MainWindow(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.setWindowTitle("Tomato Image Sender")
|
||
self.setGeometry(100, 100, 800, 600)
|
||
|
||
central_widget = QWidget()
|
||
self.setCentralWidget(central_widget)
|
||
|
||
layout = QVBoxLayout()
|
||
central_widget.setLayout(layout)
|
||
|
||
self.image_label = QLabel()
|
||
layout.addWidget(self.image_label)
|
||
|
||
self.rgb_send_name = r'\\.\pipe\rgb_receive' # 发送数据管道名对应 main.py 的接收数据管道名
|
||
self.rgb_receive_name = r'\\.\pipe\rgb_send' # 接收数据管道名对应 main.py 的发送数据管道名
|
||
self.spec_send_name = r'\\.\pipe\spec_receive' # 发送数据管道名对应 main.py 的接收数据管道名
|
||
|
||
# 连接main.py创建的命名管道
|
||
self.rgb_send = win32file.CreateFile(
|
||
self.rgb_send_name,
|
||
win32file.GENERIC_WRITE,
|
||
0,
|
||
None,
|
||
win32file.OPEN_EXISTING,
|
||
0,
|
||
None
|
||
)
|
||
|
||
self.rgb_receive = win32file.CreateFile(
|
||
self.rgb_receive_name,
|
||
win32file.GENERIC_READ,
|
||
0,
|
||
None,
|
||
win32file.OPEN_EXISTING,
|
||
0,
|
||
None
|
||
)
|
||
|
||
self.spec_send = win32file.CreateFile(
|
||
self.spec_send_name,
|
||
win32file.GENERIC_WRITE,
|
||
0,
|
||
None,
|
||
win32file.OPEN_EXISTING,
|
||
0,
|
||
None
|
||
)
|
||
|
||
def send_image_group(self, image_dir):
|
||
'''
|
||
发送图像数据
|
||
:param image_dir: bmp和raw文件所在文件夹
|
||
:return:
|
||
'''
|
||
rgb_files = [os.path.join(image_dir, f) for f in os.listdir(image_dir) if f.endswith(('.bmp'))][:5]
|
||
spec_files = [os.path.join(image_dir, f) for f in os.listdir(image_dir) if f.endswith('.raw')][:1]
|
||
|
||
self.send_YR()
|
||
for _ in range(20):
|
||
for image_path in rgb_files:
|
||
img = cv2.imread(image_path, cv2.IMREAD_COLOR)
|
||
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
||
img = np.asarray(img, dtype=np.uint8)
|
||
|
||
|
||
try:
|
||
# win32file.WriteFile(self.rgb_send, len(img_data).to_bytes(4, byteorder='big'))
|
||
height = img.shape[0]
|
||
width = img.shape[1]
|
||
height = height.to_bytes(2, byteorder='big')
|
||
width = width.to_bytes(2, byteorder='big')
|
||
img_data = img.tobytes()
|
||
length = (len(img_data) + 6).to_bytes(4, byteorder='big')
|
||
# cmd = 'TO'：测试番茄数据；cmd = 'PF'：测试百香果数据
|
||
cmd = 'PF'
|
||
data_send = length + cmd.upper().encode('ascii') + height + width + img_data
|
||
win32file.WriteFile(self.rgb_send, data_send)
|
||
print(f'发送的图像数据长度: {len(data_send)}')
|
||
except Exception as e:
|
||
print(f"数据发送失败. 错误原因: {e}")
|
||
|
||
if spec_files:
|
||
spec_file = spec_files[0]
|
||
with open(spec_file, 'rb') as f:
|
||
spec_data = f.read()
|
||
|
||
try:
|
||
# win32file.WriteFile(self.spec_send, len(spec_data).to_bytes(4, byteorder='big'))
|
||
# print(f"发送的光谱数据长度: {len(spec_data)}")
|
||
heigth = 30
|
||
weight = 30
|
||
bands = 224
|
||
heigth = heigth.to_bytes(2, byteorder='big')
|
||
weight = weight.to_bytes(2, byteorder='big')
|
||
bands = bands.to_bytes(2, byteorder='big')
|
||
length = (len(spec_data)+8).to_bytes(4, byteorder='big')
|
||
# cmd = 'TO'：测试番茄数据；cmd = 'PF'：测试百香果数据
|
||
cmd = 'PF'
|
||
data_send = length + cmd.upper().encode('ascii') + heigth + weight + bands + spec_data
|
||
win32file.WriteFile(self.spec_send, data_send)
|
||
print(f'发送的光谱数据长度: {len(data_send)}')
|
||
print(f'spec长度: {len(spec_data)}')
|
||
except Exception as e:
|
||
print(f"数据发送失败. 错误原因: {e}")
|
||
|
||
self.receive_result()
|
||
|
||
def send_YR(self):
|
||
'''
|
||
发送预热指令
|
||
:return:
|
||
'''
|
||
length = 2
|
||
length = length.to_bytes(4, byteorder='big')
|
||
cmd = 'YR'
|
||
data_send = length + cmd.upper().encode('ascii')
|
||
try:
|
||
win32file.WriteFile(self.rgb_send, data_send)
|
||
print("发送预热指令成功")
|
||
except Exception as e:
|
||
print(f"发送预热指令失败. 错误原因: {e}")
|
||
|
||
def receive_result(self):
|
||
try:
|
||
# 读取结果数据
|
||
# 读取4个字节的数据长度信息，并将其转换为整数
|
||
data_length = int.from_bytes(win32file.ReadFile(self.rgb_receive, 4)[1], byteorder='big')
|
||
print(f"应该接收到的数据长度: {data_length}")
|
||
# 根据读取到的数据长度，读取对应长度的数据
|
||
data = win32file.ReadFile(self.rgb_receive, data_length)[1]
|
||
print(f"实际接收到的数据长度: {len(data)}")
|
||
# 解析数据
|
||
cmd_result = data[:2].decode('ascii').strip().upper()
|
||
brix = (int.from_bytes(data[2:4], byteorder='big')) / 1000
|
||
green_percentage = int.from_bytes(data[4:5], byteorder='big')
|
||
diameter = int.from_bytes(data[5:7], byteorder='big')
|
||
weight = int.from_bytes(data[7:8], byteorder='big')
|
||
defect_num = int.from_bytes(data[8:10], byteorder='big')
|
||
total_defect_area = int.from_bytes(data[10:14], byteorder='big')
|
||
heigth = int.from_bytes(data[14:16], byteorder='big')
|
||
width = int.from_bytes(data[16:18], byteorder='big')
|
||
rp = data[18:]
|
||
img = np.frombuffer(rp, dtype=np.uint8).reshape(heigth, width, -1)
|
||
print(f"指令:{cmd_result}, 糖度值:{brix}, 绿色占比:{green_percentage}, 直径:{diameter}, "
|
||
f"预估重量:{weight}, 缺陷个数:{defect_num}, 缺陷面积:{total_defect_area}, 结果图的尺寸:{img.shape}")
|
||
|
||
|
||
# 显示结果图像
|
||
image = Image.fromarray(img)
|
||
qimage = QImage(image.tobytes(), image.size[0], image.size[1], QImage.Format_RGB888)
|
||
pixmap = QPixmap.fromImage(qimage)
|
||
self.image_label.setPixmap(pixmap)
|
||
|
||
except Exception as e:
|
||
print(f"数据接收失败. 错误原因: {e}")
|
||
|
||
def open_file_dialog(self):
|
||
directory_dialog = QFileDialog()
|
||
directory_dialog.setFileMode(QFileDialog.Directory)
|
||
if directory_dialog.exec_():
|
||
selected_directory = directory_dialog.selectedFiles()[0]
|
||
self.send_image_group(selected_directory)
|
||
|
||
if __name__ == "__main__":
|
||
'''
|
||
1. 创建Qt应用程序
|
||
2. 创建主窗口
|
||
3. 显示主窗口
|
||
4. 打开文件对话框
|
||
5. 进入Qt事件循环
|
||
'''
|
||
#运行main.py后，运行qt_test.py
|
||
#运行qt_test.py后，选择文件夹，自动读取文件夹下的bmp和raw文件，发送到main.py
|
||
#main.py接收到数据后，返回结果数据，qt_test.py接收到结果数据，显示图片
|
||
#为确保测试正确，测试文件夹下的文件数量应该为5个bmp文件和1个raw文件
|
||
app = QApplication(sys.argv)
|
||
main_window = MainWindow()
|
||
main_window.show()
|
||
main_window.open_file_dialog()
|
||
sys.exit(app.exec_()) |