jiateng_ws/widgets/main_window.py

import os
import sys
import logging
import json
from datetime import datetime

# 导入PySide6
from PySide6.QtWidgets import QWidget, QMessageBox, QTableWidgetItem, QStackedWidget, QLabel
from PySide6.QtCore import Qt, QTimer
from PySide6.QtGui import QBrush, QColor

# 导入UI
from ui.main_window_ui import MainWindowUI

# 导入相机显示组件和设置组件
from widgets.camera_display_widget import CameraDisplayWidget
from widgets.camera_settings_widget import CameraSettingsWidget

# 导入检验配置管理器
from utils.inspection_config_manager import InspectionConfigManager


class MainWindow(MainWindowUI):
    """主窗口"""
    
    def __init__(self, user_id=None, user_name=None, corp_name=None, corp_id=None, position_id=None):
        super().__init__()
        self.user_id = user_id
        self.user_name = user_name
        self.corp_name = corp_name
        self.corp_id = corp_id
        self.position_id = position_id
        
        # 设置窗口标题
        if user_name and corp_name:
            self.setWindowTitle(f"腾智微丝产线包装系统 - {user_name} ({corp_name})")

        # 加载配置文件
        self.config = self.load_config()
        self.camera_enabled = self.config.get('camera', {}).get('enabled', False)
        
        # 初始化检验配置管理器
        self.inspection_manager = InspectionConfigManager.get_instance()
        
        # 只有在相机启用时创建相机显示组件
        if self.camera_enabled:
            # 创建相机显示组件并添加到上料区
            self.camera_display = CameraDisplayWidget()
            self.material_content_layout.addWidget(self.camera_display)
        else:
            # 在上料区添加占位标签
            self.material_placeholder = QLabel("相机功能已禁用")
            self.material_placeholder.setAlignment(Qt.AlignCenter)
            self.material_placeholder.setStyleSheet("color: #888888; background-color: #f0f0f0;")
            self.material_content_layout.addWidget(self.material_placeholder)
        
        # 为下料区添加占位标签，确保它保持为空
        self.output_placeholder = QLabel("下料区 - 暂无内容")
        self.output_placeholder.setAlignment(Qt.AlignCenter)
        self.output_placeholder.setStyleSheet("color: #888888; background-color: #f0f0f0;")
        self.output_content_layout.addWidget(self.output_placeholder)
        
        # 创建堆叠部件
        self.stacked_widget = QStackedWidget()
        self.stacked_widget.addWidget(self.central_widget)  # 主页面
        
        # 不在这里直接初始化相机设置组件
        # 延迟创建，保证创建的时候SettingsUI的所有控件都已经准备好
        self.camera_settings = None
        
        # 设置中央部件为堆叠部件
        self.setCentralWidget(self.stacked_widget)
        
        # 连接信号和槽
        self.connect_signals()
        
        # 默认显示主页面
        self.stacked_widget.setCurrentIndex(0)
        
        # 初始化数据
        self.initialize_data()
        
        # 配置检验列 - 使用检验配置管理器获取启用的列数和标题
        self.update_inspection_columns()
        
        logging.info(f"主窗口已创建，用户: {user_name}")
    
    def load_config(self):
        """加载配置文件"""
        config_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "config", "app_config.json")
        try:
            with open(config_path, 'r', encoding='utf-8') as f:
                config = json.load(f)
            logging.info(f"已加载配置文件: {config_path}")
            return config
        except Exception as e:
            logging.error(f"加载配置文件失败: {e}")
            return {}
    
    def connect_signals(self):
        # 连接菜单动作
        self.main_action.triggered.connect(self.show_main_page)
        self.settings_action.triggered.connect(self.show_settings_page)
        
        # 工程号输入框回车事件
        self.order_edit.returnPressed.connect(self.handle_order_enter)
        
        # 连接按钮事件
        self.input_button.clicked.connect(self.handle_input)
        self.output_button.clicked.connect(self.handle_output)
        self.start_button.clicked.connect(self.handle_start)
        self.stop_button.clicked.connect(self.handle_stop)
        
        # 只有在相机启用时连接相机信号
        if self.camera_enabled and hasattr(self, 'camera_display'):
            self.camera_display.signal_camera_status.connect(self.handle_camera_status)
    
    def initialize_data(self):
        """初始化界面数据"""
        # 设置订单和批号
        self.order_edit.setText("ORD-2025-001")
        self.tray_edit.setCurrentText("托盘1")
        
        # 初始化项目表格数据示例
        project_data = [
            ["100", "50", "200", "95%"],
            ["3000", "1500", "6000", "92%"],
            ["36000", "18000", "72000", "90%"],
            ["120000", "60000", "240000", "91%"]
        ]
        
        for row in range(4):
            for col in range(0, 4):  # 从第2列开始（跳过项目列）
                item = QTableWidgetItem(project_data[row][col])
                # item.setTextAlignment(Qt.AlignCenter)  # 设置文本居中对齐
                self.project_table.setItem(row, col, item)
        
        # 初始化任务表格数据示例 - 注意：现在表格有3行，第0行是一级标题，第1行是二级标题，第2行是数据行
        # 数据应该填充在第2行（索引为2）
        data = ["200", "95", "0", "0"]  # 分别对应：总生产数量、总生产公斤、已完成数量、已完成公斤
        for col, value in enumerate(data):
            item = QTableWidgetItem(value)
            item.setTextAlignment(Qt.AlignCenter)  # 设置文本居中对齐
            self.task_table.setItem(2, col, item)
    
    def update_inspection_columns(self):
        """更新检验列配置 - 使用检验配置管理器获取启用的列数和标题"""
        try:
            # 获取已启用的检验配置
            enabled_configs = self.inspection_manager.get_enabled_configs()
            
            # 获取启用的列数
            column_count = len(enabled_configs)
            if column_count == 0:
                # 如果没有启用的列，至少显示一列
                column_count = 1
                headers = ["检验项"]
            else:
                # 如果有启用的列，使用配置的标题
                headers = [config['display_name'] for config in enabled_configs]
            
            # 设置检验列
            self.set_inspection_columns(column_count, headers)
            logging.info(f"已更新检验列配置：{column_count}列, 标题: {headers}")
        except Exception as e:
            logging.error(f"更新检验列配置失败: {str(e)}")
            # 如果更新失败，使用默认配置
            self.set_inspection_columns(1, ["检验项"])
    
    def show_main_page(self):
        self.stacked_widget.setCurrentWidget(self.central_widget)
        
        # 更新检验列配置
        self.update_inspection_columns()
        
        # 只有在相机启用时处理相机显示
        if self.camera_enabled and hasattr(self, 'camera_display'):
            # 如果相机已连接，直接开始显示相机画面
            if self.camera_display.camera_manager.isOpen:
                if not self.camera_display.camera_manager.isGrabbing:
                    self.camera_display.start_display()
        
        logging.info("显示主页面")
    
    def show_settings_page(self):
        """显示设置页面"""
        # 延迟创建设置组件
        if not hasattr(self, 'settings_widget'):
            from widgets.settings_widget import SettingsWidget
            self.settings_widget = SettingsWidget(self)
            self.stacked_widget.addWidget(self.settings_widget)
        
        # 切换到设置页面
        self.stacked_widget.setCurrentWidget(self.settings_widget)
        logging.info("显示设置页面")
    
    def handle_input(self):
        """处理上料按钮点击事件"""
        logging.info("上料按钮被点击")
        
        QMessageBox.information(self, "操作提示", "开始上料操作")
        # 这里添加上料相关的业务逻辑
    
    def handle_output(self):
        """处理下料按钮点击事件"""
        logging.info("下料按钮被点击")
        QMessageBox.information(self, "操作提示", "开始下料操作")
        # 这里添加下料相关的业务逻辑
    
    def handle_start(self):
        """处理开始按钮点击事件"""
        logging.info("开始按钮被点击")
        
        # 只有在相机启用时处理相机显示
        if self.camera_enabled and hasattr(self, 'camera_display'):
            # 开始显示相机画面
            if self.camera_display.camera_manager.isOpen:
                self.camera_display.start_display()
        
        QMessageBox.information(self, "操作提示", "生产线已启动")
        # 这里添加启动生产线的业务逻辑
    
    def handle_stop(self):
        """处理暂停按钮点击事件"""
        logging.info("暂停按钮被点击")
        
        # 只有在相机启用时处理相机显示
        if self.camera_enabled and hasattr(self, 'camera_display'):
            # 停止显示相机画面
            self.camera_display.stop_display()
        
        QMessageBox.information(self, "操作提示", "生产线已暂停")
        # 这里添加暂停生产线的业务逻辑
    
    def handle_camera_status(self, is_connected, message):
        """处理相机状态变化"""
        if is_connected:
            logging.info("相机已连接并显示")
        else:
            logging.warning(f"相机显示问题: {message}")
    
    def handle_camera_connection(self, is_connected, message):
        """处理相机连接状态变化"""
        if is_connected:
            logging.info("相机已连接")
            # 如果当前在主页面，直接开始显示相机画面
            if self.stacked_widget.currentWidget() == self.central_widget:
                self.camera_display.start_display()
        else:
            if message:
                logging.warning(f"相机连接失败: {message}")
            else:
                logging.info("相机已断开")
            # 如果相机断开，确保停止显示
            self.camera_display.stop_display()
    
    def handle_camera_params_changed(self, exposure_time, gain, frame_rate):
        """处理相机参数变化"""
        logging.info(f"相机参数已更新: 曝光={exposure_time:.1f}μs, 增益={gain:.1f}dB, 帧率={frame_rate:.1f}fps")
        # 这里可以添加对相机参数变化的处理逻辑
    
    def handle_camera_error(self, error_msg):
        """处理相机错误"""
        logging.error(f"相机错误: {error_msg}")
        QMessageBox.warning(self, "相机错误", error_msg)
    
    def closeEvent(self, event):
        """窗口关闭事件"""
        # 只有在相机启用时处理相机关闭
        if self.camera_enabled and hasattr(self, 'camera_display'):
            # 停止相机显示
            self.camera_display.stop_display()
        
        # 接受关闭事件
        event.accept() 
    
    def handle_order_enter(self):
        """处理工程号输入框按下回车事件"""
        logging.info("工程号输入框按下回车事件")
        # 获取当前输入的工程号
        order_text = self.order_edit.text().strip()
        
        if order_text:
            logging.info(f"输入的工程号: {order_text}")
            QMessageBox.information(self, "工程号确认", f"您输入的工程号是: {order_text}")
            # 这里可以添加其他工程号处理逻辑
        else:
            logging.warning("工程号为空")
            QMessageBox.warning(self, "输入提示", "请输入有效的工程号")
        
        # 处理完后可以清除焦点，让输入框失去焦点
        self.central_widget.setFocus()