xiongwei/detect-gui
1
0
Fork 0
mirror of http://git.xinwangdao.com/cnnc-embedded-parts-detect/detect-gui.git synced 2025-06-24 13:14:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
1c996ee9df
detect-gui/widget/task_run.py
njdaoyehu 1c996ee9df fixed
2024-11-26 10:13:55 +08:00

332 lines
13 KiB
Python
Raw Blame History

import cv2
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QPixmap, QImage, QTextCursor
from PyQt5.QtWidgets import QDialog, QWidget, QVBoxLayout, QHBoxLayout, QGridLayout, QScrollArea, QPushButton, \
QSpacerItem, QSizePolicy, QLabel, QMessageBox, QPlainTextEdit
from components.image_framework import MSG_LOCATION_RECORD, MSG_DETECTION_RECORD, get_msg_info
from core.context import AppContext
from widget.embed_item import EmbedItem
class TaskRunDialog(QDialog):
def __init__(self, param, parent=None):
super(TaskRunDialog, self).__init__(parent)
self.ratio = AppContext.get_ratio()
self.checkIndex = 1
self.param = param
self.setWindowFlags(Qt.WindowStaysOnTopHint)
self.setWindowFlags(Qt.FramelessWindowHint)
self.setWindowModality(Qt.ApplicationModal)
self.setAttribute(Qt.WA_DeleteOnClose)
self.setWindowTitle("检测窗口")
# self.showFullScreen()
self.bim_widget = None
self.check_widget = None
self.embed_widgets = []
# 广角镜头预览区
self.view_widget = None
self.view_label = None
self.picture_label = None
self.log_text = None
self.start_adjust_button = None
self.start_check_button = None
self.stop_check_button = None
self.stop_task_button = None
self.init_ui()
self.init_event()
def init_event(self):
AppContext.get_edge_context().get_component('image-framework').signals.on_image_detect_result.connect(
lambda data: self.image_result(data))
def image_result(self, msg):
qt_image = None
if msg.im2D is not None:
rgb_frame = cv2.cvtColor(msg.im2D, cv2.COLOR_BGR2RGB)
height, width, channels = rgb_frame.shape
bytes_per_line = channels * width
qt_image = QImage(rgb_frame.data, width, height, bytes_per_line, QImage.Format_RGB888)
if msg.msg_type == MSG_LOCATION_RECORD:
if qt_image is None:
return
try:
view_pixmap = QPixmap.fromImage(qt_image)
sc = view_pixmap.height() / 450
self.view_label.setFixedWidth(view_pixmap.width() / sc)
self.view_label.setPixmap(view_pixmap)
self.view_label.setGeometry((self.view_widget.width() - self.view_label.width()) / 2, 0, self.view_label.width(), self.view_label.height())
except Exception as e:
dd = 0
elif msg.msg_type == MSG_DETECTION_RECORD:
# 显示日志
self.log_text.appendPlainText(get_msg_info(msg.record.code).decode('utf-8'))
if msg.record.code < 0:
QMessageBox.warning(self, '错误信息', get_msg_info(msg.record.code).decode('utf-8'))
self.set_button_default()
AppContext.get_edge_context().get_component('image-framework').start_location()
elif msg.record.code == 1008:
self.start_adjust_button.setEnabled(False)
self.start_check_button.setEnabled(True)
self.stop_check_button.setEnabled(True)
self.stop_task_button.setEnabled(False)
elif msg.record.code == 1012:
QMessageBox.warning(self, '成功信息', "检测结束")
self.set_button_default()
AppContext.get_edge_context().get_component('image-framework').start_location()
else:
dddd = 0
if qt_image is None:
return
try:
view_pixmap = QPixmap.fromImage(qt_image)
self.picture_label.setPixmap(view_pixmap)
except Exception as e:
dd = 0
else:
ddd = 0
def showEvent(self, e):
AppContext.get_edge_context().get_component('image-framework').start_location()
def init_ui(self):
# bim
self.bim_widget = QWidget()
self.bim_widget.setObjectName("bimWidget")
self.bim_widget.setFixedWidth(3000)
# 选择区
self.check_widget = QWidget(self.bim_widget)
self.check_widget.setObjectName("checkWidget")
self.check_widget.setFixedSize(160, 160)
self.check_widget.move(90, 230)
# 预埋件
self.embed_widgets = []
for item in self.param:
embed_item = EmbedItem(self.bim_widget)
embed_item.setObjectName(item["code"])
embed_item.setItemParam(item)
self.embed_widgets.append(embed_item)
map_widget = QWidget()
legend_widget = QWidget()
legend_layout = QHBoxLayout()
legend_layout.setContentsMargins(0, 0, 0, 0)
legend_layout.setSpacing(0)
legend_widget.setLayout(legend_layout)
legend_check_widget = QWidget()
legend_check_widget.setObjectName("legendCheckWidget")
legend_std_widget = QWidget()
legend_std_widget.setObjectName("legendStdWidget")
legend_ok_widget = QWidget()
legend_ok_widget.setObjectName("legendOkWidget")
legend_ng_widget = QWidget()
legend_ng_widget.setObjectName("legendNgWidget")
legend_normal_widget = QWidget()
legend_normal_widget.setObjectName("legendNormalWidget")
legend_check_label = QLabel()
legend_check_label.setObjectName("legendCheckLabel")
legend_check_label.setText("检测区域")
legend_std_label = QLabel()
legend_std_label.setObjectName("legendStdLabel")
legend_std_label.setText("基准件")
legend_ok_label = QLabel()
legend_ok_label.setObjectName("legendOkLabel")
legend_ok_label.setText("合格件")
legend_ng_label = QLabel()
legend_ng_label.setObjectName("legendNgLabel")
legend_ng_label.setText("不合格件")
legend_normal_label = QLabel()
legend_normal_label.setObjectName("legendNormalLabel")
legend_normal_label.setText("待检测件")
legend_layout.addStretch(1)
legend_layout.addWidget(legend_check_widget)
legend_layout.addSpacing(5)
legend_layout.addWidget(legend_check_label)
legend_layout.addSpacing(40)
legend_layout.addWidget(legend_std_widget)
legend_layout.addSpacing(5)
legend_layout.addWidget(legend_std_label)
legend_layout.addSpacing(40)
legend_layout.addWidget(legend_ok_widget)
legend_layout.addSpacing(5)
legend_layout.addWidget(legend_ok_label)
legend_layout.addSpacing(40)
legend_layout.addWidget(legend_ng_widget)
legend_layout.addSpacing(5)
legend_layout.addWidget(legend_ng_label)
legend_layout.addSpacing(40)
legend_layout.addWidget(legend_normal_widget)
legend_layout.addSpacing(5)
legend_layout.addWidget(legend_normal_label)
legend_layout.addStretch(1)
content_widget = QWidget()
content_widget.setObjectName("contentWidget")
self.picture_label = QLabel()
self.picture_label.setObjectName("pictureWidget")
self.picture_label.setFixedWidth(930 * 450 / 700)
self.picture_label.setFixedHeight(450)
self.picture_label.setScaledContents(True)
self.picture_label.setStyleSheet("background: #222222;")
self.view_widget = QWidget()
self.view_widget.setObjectName("viewWidget")
self.view_widget.setFixedWidth(1920 * 450 / 1080)
self.view_widget.setFixedHeight(450)
self.view_label = QLabel(self.view_widget)
self.view_label.setObjectName("viewLabel")
self.view_label.setFixedHeight(450)
self.view_label.setScaledContents(True)
self.view_label.setStyleSheet("background: #222222;")
view_check_widget = QWidget(self.view_widget)
view_check_widget.setObjectName("viewCheckWidget")
view_check_widget.setGeometry(self.view_label.geometry().x() + 145, 170, 240, 240)
self.log_text = QPlainTextEdit()
self.log_text.setObjectName("logText")
self.log_text.setFixedHeight(450)
self.log_text.setReadOnly(True)
tool_widget = QWidget()
tool_widget.setObjectName("toolWidget")
tool_widget.setFixedWidth(150)
tool_widget.setFixedHeight(450)
tool_layout = QVBoxLayout()
tool_layout.setContentsMargins(0, 0, 0, 0)
tool_layout.setSpacing(0)
self.start_adjust_button = QPushButton()
self.start_adjust_button.setObjectName("startAdjustButton")
self.start_adjust_button.setText("开始校准")
self.start_adjust_button.setFixedHeight(self.ratio * 42)
self.start_adjust_button.clicked.connect(self.start_adjust)
self.start_check_button = QPushButton()
self.start_check_button.setObjectName("startCheckButton")
self.start_check_button.setEnabled(False)
self.start_check_button.setText("开始检测")
self.start_check_button.setFixedHeight(self.ratio * 42)
self.start_check_button.clicked.connect(self.start_check)
self.stop_check_button = QPushButton()
self.stop_check_button.setObjectName("stopCheckButton")
self.stop_check_button.setEnabled(False)
self.stop_check_button.setText("停止检测")
self.stop_check_button.setFixedHeight(self.ratio * 42)
self.stop_check_button.clicked.connect(self.stop_check)
self.stop_task_button = QPushButton()
self.stop_task_button.setObjectName("stopTaskButton")
self.stop_task_button.setText("结束任务")
self.stop_task_button.setFixedHeight(self.ratio * 42)
self.stop_task_button.clicked.connect(self.stop_task)
tool_widget.setLayout(tool_layout)
spacer = QSpacerItem(10, 10, QSizePolicy.Minimum, QSizePolicy.Expanding)
tool_layout.addItem(spacer)
tool_layout.addWidget(self.start_adjust_button)
tool_layout.addSpacing(20)
tool_layout.addWidget(self.start_check_button)
tool_layout.addSpacing(20)
tool_layout.addWidget(self.stop_check_button)
tool_layout.addSpacing(20)
tool_layout.addWidget(self.stop_task_button)
tool_layout.addSpacing(20)
content_layout = QHBoxLayout()
content_layout.setContentsMargins(0, 0, 0, 20)
content_layout.setSpacing(0)
content_layout.addWidget(self.view_widget)
content_layout.addSpacing(20)
content_layout.addWidget(self.log_text)
content_layout.addSpacing(20)
content_layout.addWidget(self.picture_label)
content_layout.addSpacing(20)
content_layout.addWidget(tool_widget)
content_widget.setLayout(content_layout)
layout = QVBoxLayout()
layout.setContentsMargins(20, 20, 20, 20)
layout.setSpacing(0)
layout.addWidget(map_widget)
layout.addSpacing(5)
layout.addWidget(legend_widget)
layout.addSpacing(20)
layout.addWidget(content_widget)
self.setLayout(layout)
map_area = QScrollArea()
map_area.setObjectName("mapArea")
map_area.setWidgetResizable(True)
map_area.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
map_area.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
map_widget.setLayout(QGridLayout())
map_widget.layout().setContentsMargins(0, 0, 0, 0)
map_widget.layout().setSpacing(0)
map_widget.layout().addWidget(map_area)
map_area.setWidget(self.bim_widget)
self.set_button_default()
# 校准
def start_adjust(self):
self.start_adjust_button.setEnabled(False)
self.start_check_button.setEnabled(False)
self.stop_check_button.setEnabled(False)
self.stop_task_button.setEnabled(False)
self.log_text.setPlainText("")
self.picture_label.setPixmap(QPixmap(""))
result = AppContext.get_edge_context().get_component('image-framework').start_adjust()
if result.code < 0:
self.set_button_default()
QMessageBox.warning(self, '校准启动失败', get_msg_info(msg.record.code).decode('utf-8'))
def start_check(self):
self.start_adjust_button.setEnabled(False)
self.start_check_button.setEnabled(False)
self.stop_check_button.setEnabled(False)
self.stop_task_button.setEnabled(False)
result = AppContext.get_edge_context().get_component('image-framework').start_check()
if result.code < 0:
self.set_button_default()
QMessageBox.warning(self, '检测启动失败', get_msg_info(msg.record.code).decode('utf-8'))
def stop_check(self):
self.start_adjust_button.setEnabled(False)
self.start_check_button.setEnabled(False)
self.stop_check_button.setEnabled(False)
self.stop_task_button.setEnabled(False)
result = AppContext.get_edge_context().get_component('image-framework').stop_check()
if result.code < 0:
QMessageBox.warning(self, '检测停止失败', get_msg_info(msg.record.code).decode('utf-8'))
self.set_button_default()
def stop_task(self):
AppContext.get_edge_context().get_component('image-framework').stop_all()
self.reject()
def set_button_default(self):
self.start_adjust_button.setEnabled(True)
self.start_check_button.setEnabled(False)
self.stop_check_button.setEnabled(False)
self.stop_task_button.setEnabled(True)
Reference in New Issue View Git Blame Copy Permalink