Files
pqAutomationApp/pqAutomationApp.py
2026-04-21 10:48:15 +08:00

1120 lines
44 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
import ttkbootstrap as ttk
import tkinter as tk
from tkinter import messagebox, filedialog
import sys
import threading
import time
import os
import datetime
import re
import traceback
import numpy as np
import matplotlib.pyplot as plt
from app_version import APP_NAME, APP_VERSION, get_app_title
from drivers.UCD323_Function import UCDController
from app.pq.pq_config import PQConfig
from app.pq.pq_result import PQResult
from app.views.pq_debug_panel import PQDebugPanel
from app.export import (
save_result_images as _save_result_images_impl,
export_excel_report as _export_excel_report_impl,
EXCEL_EXPORT_CONFIG as _EXCEL_EXPORT_CONFIG,
)
from app.views.panels import custom_template_panel as _ctp
from app.views.panels import side_panels as _sp
from app.views.panels import cct_panel as _ccp
from app.views.panels import main_layout as _main
# Step 0/1 重构:资源工具和纯算法已迁移到 app/ 包,这里重新导入以保持
# 对原函数名/方法名的向后兼容(老代码内部仍用 self.calculate_* 调用)。
from app.resources import (
backgroud_style_set,
get_resource_path,
load_icon,
)
from app.tests.color_accuracy import (
calculate_color_accuracy as _calc_color_accuracy,
calculate_delta_e_2000 as _calc_delta_e_2000,
get_accuracy_color_standards as _get_accuracy_color_standards,
)
from app.tests.eotf import calculate_pq_curve as _calc_pq_curve
from app.tests.gamma import calculate_gamma as _calc_gamma
from app.tests.gamut import calculate_gamut_coverage as _calc_gamut_coverage
from app.plots.plot_accuracy import plot_accuracy as _plot_accuracy
from app.plots.plot_cct import plot_cct as _plot_cct
from app.plots.plot_contrast import plot_contrast as _plot_contrast
from app.plots.plot_eotf import plot_eotf as _plot_eotf
from app.plots.plot_gamma import plot_gamma as _plot_gamma
from app.plots.plot_gamut import plot_gamut as _plot_gamut
from app.views.chart_frame import (
clear_chart as _cf_clear_chart,
create_result_chart_frame as _cf_create_result_chart_frame,
init_accuracy_chart as _cf_init_accuracy_chart,
init_cct_chart as _cf_init_cct_chart,
init_contrast_chart as _cf_init_contrast_chart,
init_eotf_chart as _cf_init_eotf_chart,
init_gamma_chart as _cf_init_gamma_chart,
init_gamut_chart as _cf_init_gamut_chart,
on_chart_tab_changed as _cf_on_chart_tab_changed,
update_chart_tabs_state as _cf_update_chart_tabs_state,
)
from app.config_io import (
clear_config_file as _cfg_clear_config_file,
get_config_path as _cfg_get_config_path,
load_pq_config as _cfg_load_pq_config,
save_pq_config as _cfg_save_pq_config,
)
from app.tests.local_dimming import (
clear_ld_records as _ld_clear_ld_records,
measure_ld_luminance as _ld_measure_ld_luminance,
save_local_dimming_results as _ld_save_local_dimming_results,
send_ld_window as _ld_send_ld_window,
start_local_dimming_test as _ld_start_local_dimming_test,
stop_local_dimming_test as _ld_stop_local_dimming_test,
update_ld_results as _ld_update_ld_results,
)
from app.device.connection import (
check_com_connections as _dev_check_com_connections,
check_port_connection as _dev_check_port_connection,
disconnect_com_connections as _dev_disconnect_com_connections,
enable_com_widgets as _dev_enable_com_widgets,
get_available_com_ports as _dev_get_available_com_ports,
get_available_ucd_ports as _dev_get_available_ucd_ports,
refresh_com_ports as _dev_refresh_com_ports,
update_connection_indicator as _dev_update_connection_indicator,
)
from app.runner.test_runner import (
get_current_test_result as _run_get_current_test_result,
new_pq_results as _run_new_pq_results,
on_custom_template_test_completed as _run_on_custom_template_test_completed,
on_test_completed as _run_on_test_completed,
on_test_error as _run_on_test_error,
run_custom_sdr_test as _run_run_custom_sdr_test,
run_hdr_movie_test as _run_run_hdr_movie_test,
run_screen_module_test as _run_run_screen_module_test,
run_sdr_movie_test as _run_run_sdr_movie_test,
run_test as _run_run_test,
send_fix_pattern as _run_send_fix_pattern,
test_cct as _run_test_cct,
test_color_accuracy as _run_test_color_accuracy,
test_contrast as _run_test_contrast,
test_custom_sdr as _run_test_custom_sdr,
test_eotf as _run_test_eotf,
test_gamma as _run_test_gamma,
test_gamut as _run_test_gamut,
)
plt.rcParams["font.family"] = ["sans-serif"]
plt.rcParams["font.sans-serif"] = ["Microsoft YaHei"]
class PQAutomationApp:
def __init__(self, root):
self.root = root
self.root.title(get_app_title())
self.root.geometry("900x650")
self.root.minsize(900, 650)
self.root.protocol("WM_DELETE_WINDOW", self.on_closing)
self.app_name = APP_NAME
self.app_version = APP_VERSION
self.config_cleared = False
# 初始化设备连接状态
self.ca = None # CA410色度计
self.ucd = UCDController() # 信号发生器
# 初始化测试状态
self.testing = False
self.test_thread = None
# 采集节奏参数:默认在稳定性与速度之间取平衡,可按现场情况再微调。
self.pattern_settle_time = 0.4
self.pattern_progress_log_step = 5
# 创建主框架
self.main_frame = ttk.Frame(root)
self.main_frame.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)
backgroud_style_set()
# 创建配置对象
self.config = PQConfig()
self.results = None
# 加载上次保存的设置
self.config_file = self.get_config_path()
self.load_pq_config()
# 如果加载的配置不是屏模组,强制切换为屏模组
if self.config.current_test_type != "screen_module":
self.config.set_current_test_type("screen_module")
# 初始化侧边栏功能显示状态 - 使用统一的页面管理
self.current_panel = None # 当前显示的面板名称
self.panels = {} # 存储所有面板的信息
self.log_visible = False
# 创建左侧面板
self.left_frame = ttk.Frame(self.main_frame, width=180)
self.left_frame.pack(side=tk.LEFT, fill=tk.Y, padx=0, pady=5)
self.left_frame.pack_propagate(False)
# 创建左侧导航栏
self.sidebar_frame = ttk.Frame(self.left_frame, bootstyle="primary")
self.sidebar_frame.pack(fill=tk.BOTH, expand=True, padx=0, pady=5)
# self.sidebar_frame.pack_propagate(False)
# 创建右侧内容区域
self.content_frame = ttk.Frame(self.main_frame)
self.content_frame.pack(
side=tk.RIGHT, fill=tk.BOTH, expand=True, padx=5, pady=5
)
# 创建右侧内容区域的上中下三个分区
self.control_frame_top = ttk.Frame(self.content_frame)
self.control_frame_top.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
self.control_frame_middle = ttk.Frame(self.content_frame)
self.control_frame_middle.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
self.control_frame_bottom = ttk.Frame(self.content_frame)
self.control_frame_bottom.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
# 创建右上角悬浮配置框
self.create_floating_config_panel()
# 创建右侧结果显示区域
self.result_frame = ttk.LabelFrame(self.control_frame_middle, text="测试结果")
self.result_frame.pack(
side=tk.BOTTOM, fill=tk.BOTH, expand=True, padx=0, pady=5
)
# 创建日志显示区域
self.create_log_panel()
# 创建 Local Dimming 面板
self.create_local_dimming_panel()
# 创建测试类型选择区域
self.create_test_type_frame()
# 创建操作按钮区域
self.create_operation_frame()
# 创建结果图表区域
self.create_result_chart_frame()
# 创建客户模板结果显示区域(黑底表格)
self.create_custom_template_result_panel()
# 在所有控件创建完成后,统一初始化测试类型
self.root.after(100, self.initialize_default_test_type)
# 状态栏
self.status_var = tk.StringVar(value="就绪")
self.status_bar = ttk.Label(
root, textvariable=self.status_var, relief=tk.SUNKEN, anchor=tk.W
)
self.status_bar.pack(side=tk.BOTTOM, fill=tk.X)
def _dispatch_ui(self, fn, *args, **kwargs):
"""把 ``fn(*args, **kwargs)`` 调度到 Tk 主线程执行。
统一替代散落各处的 ``self.root.after(0, lambda: ...)`` 写法:
- 自动捕获异常并记录日志,避免工作线程静默丢失 UI 更新失败;
- 参数用位置/关键字传入,绕开 ``lambda`` 闭包捕获变量的常见坑;
- 允许在 UI 销毁(如关闭窗口)后安全失败。
"""
def _runner():
try:
fn(*args, **kwargs)
except Exception as exc:
log = getattr(self, "log_gui", None)
if log is not None:
try:
log.log(f"UI 调度异常: {exc}")
except Exception:
pass
try:
self.root.after(0, _runner)
except Exception:
pass
def initialize_default_test_type(self):
"""初始化默认测试类型(在所有控件创建完成后调用)"""
try:
# 强制切换到屏模组
self.change_test_type("screen_module")
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 默认测试类型已设置为屏模组")
except Exception as e:
if hasattr(self, "log_gui"):
self.log_gui.log(f"初始化默认测试类型失败: {str(e)}")
get_config_path = _cfg_get_config_path
load_pq_config = _cfg_load_pq_config
save_pq_config = _cfg_save_pq_config
init_gamut_chart = _cf_init_gamut_chart
init_gamma_chart = _cf_init_gamma_chart
init_eotf_chart = _cf_init_eotf_chart
init_cct_chart = _cf_init_cct_chart
init_contrast_chart = _cf_init_contrast_chart
init_accuracy_chart = _cf_init_accuracy_chart
clear_chart = _cf_clear_chart
create_result_chart_frame = _cf_create_result_chart_frame
on_chart_tab_changed = _cf_on_chart_tab_changed
create_floating_config_panel = _main.create_floating_config_panel
create_test_items_content = _main.create_test_items_content
create_signal_format_content = _main.create_signal_format_content
create_connection_content = _main.create_connection_content
create_operation_frame = _main.create_operation_frame
create_test_type_frame = _main.create_test_type_frame
update_config_info_display = _main.update_config_info_display
on_screen_module_timing_changed = _main.on_screen_module_timing_changed
create_cct_params_frame = _ccp.create_cct_params_frame
_get_cct_var_dict = _ccp._get_cct_var_dict
_parse_cct_float = _ccp._parse_cct_float
_save_cct_params_for = _ccp._save_cct_params_for
_handle_cct_focus_out = _ccp._handle_cct_focus_out
on_sdr_cct_param_focus_out = _ccp.on_sdr_cct_param_focus_out
save_sdr_cct_params = _ccp.save_sdr_cct_params
on_hdr_cct_param_focus_out = _ccp.on_hdr_cct_param_focus_out
save_hdr_cct_params = _ccp.save_hdr_cct_params
recalculate_cct = _ccp.recalculate_cct
recalculate_gamut = _ccp.recalculate_gamut
on_cct_param_focus_out = _ccp.on_cct_param_focus_out
save_cct_params = _ccp.save_cct_params
reload_cct_params = _ccp.reload_cct_params
toggle_cct_params_frame = _ccp.toggle_cct_params_frame
on_screen_gamut_ref_changed = _ccp.on_screen_gamut_ref_changed
on_sdr_gamut_ref_changed = _ccp.on_sdr_gamut_ref_changed
on_hdr_gamut_ref_changed = _ccp.on_hdr_gamut_ref_changed
get_available_ucd_ports = _dev_get_available_ucd_ports
get_available_com_ports = _dev_get_available_com_ports
refresh_com_ports = _dev_refresh_com_ports
check_com_connections = _dev_check_com_connections
update_connection_indicator = _dev_update_connection_indicator
check_port_connection = _dev_check_port_connection
enable_com_widgets = _dev_enable_com_widgets
disconnect_com_connections = _dev_disconnect_com_connections
create_custom_template_result_panel = _ctp.create_custom_template_result_panel
show_custom_result_context_menu = _ctp.show_custom_result_context_menu
set_custom_result_table_locked = _ctp.set_custom_result_table_locked
start_custom_row_single_step = _ctp.start_custom_row_single_step
_clear_custom_result_row = _ctp._clear_custom_result_row
_run_custom_row_single_step = _ctp._run_custom_row_single_step
_update_custom_result_row = _ctp._update_custom_result_row
copy_custom_result_table = _ctp.copy_custom_result_table
fill_custom_result_test_data = _ctp.fill_custom_result_test_data
clear_custom_template_results = _ctp.clear_custom_template_results
auto_expand_custom_result_view = _ctp.auto_expand_custom_result_view
append_custom_template_result = _ctp.append_custom_template_result
start_custom_template_test = _ctp.start_custom_template_test
update_custom_button_visibility = _ctp.update_custom_button_visibility
create_log_panel = _sp.create_log_panel
create_local_dimming_panel = _sp.create_local_dimming_panel
toggle_local_dimming_panel = _sp.toggle_local_dimming_panel
toggle_log_panel = _sp.toggle_log_panel
update_sidebar_selection = _sp.update_sidebar_selection
# ---- 单步调试面板(统一实现,委托到 side_panels 模块) ----
_toggle_debug_panel = _sp._toggle_debug_panel
toggle_screen_debug_panel = _sp.toggle_screen_debug_panel
toggle_sdr_debug_panel = _sp.toggle_sdr_debug_panel
toggle_hdr_debug_panel = _sp.toggle_hdr_debug_panel
clear_config_file = _cfg_clear_config_file
start_local_dimming_test = _ld_start_local_dimming_test
update_ld_results = _ld_update_ld_results
stop_local_dimming_test = _ld_stop_local_dimming_test
send_ld_window = _ld_send_ld_window
measure_ld_luminance = _ld_measure_ld_luminance
clear_ld_records = _ld_clear_ld_records
save_local_dimming_results = _ld_save_local_dimming_results
def update_test_items(self):
"""根据当前测试类型更新测试项目复选框"""
# 先隐藏所有测试项目框架
for config in self.test_items.values():
config["frame"].pack_forget()
current_test_type = self.config.current_test_type
self.test_vars = {}
if current_test_type in self.test_items:
config = self.test_items[current_test_type]
frame = config["frame"]
frame.pack(fill=tk.X, padx=5, pady=5)
# 添加测试类型标签
type_label = ttk.Label(
frame,
text=self.get_test_type_display_name(current_test_type),
style="primary.TLabel",
)
type_label.grid(row=0, column=0, columnspan=2, sticky=tk.W, padx=5, pady=3)
# 从配置中读取保存的选择状态
saved_test_items = self.config.current_test_types[current_test_type].get(
"test_items", []
)
# 添加复选框
for i, (text, var_name) in enumerate(config["items"]):
# 修改:根据配置决定是否勾选
# 如果配置中有该测试项,则勾选;否则不勾选
is_checked = var_name in saved_test_items
var = tk.BooleanVar(value=is_checked)
self.test_vars[f"{current_test_type}_{var_name}"] = var
ttk.Checkbutton(
frame,
text=text,
variable=var,
bootstyle="round-toggle",
command=self.update_config_and_tabs,
).grid(row=i // 2 + 1, column=i % 2, sticky=tk.W, padx=10, pady=5)
# 只有在 chart_notebook 已创建后才更新状态
if hasattr(self, "chart_notebook"):
self.update_chart_tabs_state()
# 更新色度参数框的显示状态
if hasattr(self, "cct_params_frame"):
self.toggle_cct_params_frame()
def get_test_type_display_name(self, test_type):
"""获取测试类型的显示名称"""
display_names = {
"screen_module": "屏模组性能测试",
"sdr_movie": "SDR Movie测试",
"hdr_movie": "HDR Movie测试",
}
return display_names.get(test_type, test_type)
def register_panel(self, panel_name, frame, button, visible_attr):
"""注册一个面板到管理系统"""
self.panels[panel_name] = {
"frame": frame,
"button": button,
"visible_attr": visible_attr,
}
def show_panel(self, panel_name):
"""显示指定面板,隐藏其他所有面板"""
if panel_name not in self.panels:
return
# 如果当前面板就是要显示的面板,则隐藏它
if self.current_panel == panel_name:
self.hide_all_panels()
return
# 隐藏所有面板
self.hide_all_panels()
# 显示指定面板
panel_info = self.panels[panel_name]
# 隐藏主内容区域
self.control_frame_top.pack_forget()
self.control_frame_middle.pack_forget()
self.control_frame_bottom.pack_forget()
# 显示目标面板
panel_info["frame"].pack(side=tk.TOP, fill=tk.BOTH, expand=True, padx=5, pady=5)
# 更新按钮样式
if panel_info["button"]:
panel_info["button"].configure(style="SidebarSelected.TButton")
# 更新状态
setattr(self, panel_info["visible_attr"], True)
self.current_panel = panel_name
def hide_all_panels(self):
"""隐藏所有面板,显示主内容区域"""
# 隐藏所有注册的面板
for panel_name, panel_info in self.panels.items():
panel_info["frame"].pack_forget()
if panel_info["button"]:
panel_info["button"].configure(style="Sidebar.TButton")
setattr(self, panel_info["visible_attr"], False)
# 显示主内容区域
self.control_frame_top.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
self.control_frame_middle.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
self.control_frame_bottom.pack(
side=tk.TOP, fill=tk.BOTH, expand=True, padx=0, pady=5
)
self.current_panel = None
def change_test_type(self, test_type):
"""切换测试类型"""
# 切换测试类型时,自动隐藏日志面板和 Local Dimming 面板
if self.current_panel in ("log", "local_dimming"):
self.hide_all_panels()
# 先保存当前测试类型的色度参数
if hasattr(self, "cct_x_ideal_var"):
try:
current_type = self.config.current_test_type
if current_type == "screen_module":
self.save_cct_params()
elif current_type == "sdr_movie":
self.save_sdr_cct_params()
elif current_type == "hdr_movie":
if hasattr(self, "save_hdr_cct_params"):
self.save_hdr_cct_params()
except Exception as e:
if hasattr(self, "log_gui"):
self.log_gui.log(f"保存参数失败: {str(e)}")
# 更新测试类型
self.test_type_var.set(test_type)
if hasattr(self, "config") and hasattr(self.config, "set_current_test_type"):
success = self.config.set_current_test_type(test_type)
if not success and hasattr(self, "log_gui"):
self.log_gui.log(f"切换测试类型失败: {test_type}")
# 更新测试项目和侧边栏
self.update_test_items()
self.update_sidebar_selection()
self.on_test_type_change()
# ========== ✅ 1. 切换信号格式 Tab ==========
if hasattr(self, "signal_tabs"):
try:
# 定义测试类型与信号格式 Tab 的映射
tab_mapping = {
"screen_module": 0, # 屏模组测试
"sdr_movie": 1, # SDR测试
"hdr_movie": 2, # HDR
}
target_tab = tab_mapping.get(test_type, 0)
# 先启用所有 Tab
for i in range(3):
self.signal_tabs.tab(i, state="normal")
# 切换到目标 Tab
self.signal_tabs.select(target_tab)
# 强制刷新显示
self.signal_tabs.update()
self.root.update_idletasks()
# 强制显示对应的 Frame
if target_tab == 0:
self.screen_module_signal_frame.tkraise()
elif target_tab == 1:
self.sdr_signal_frame.tkraise()
elif target_tab == 2:
self.hdr_signal_frame.tkraise()
# 禁用其他 Tab
for i in range(3):
if i != target_tab:
self.signal_tabs.tab(i, state="disabled")
# 日志记录
if hasattr(self, "log_gui"):
tab_names = ["屏模组测试", "SDR测试", "HDR"]
self.log_gui.log(f"✓ 已切换到 {tab_names[target_tab]} 信号格式")
except Exception as e:
if hasattr(self, "log_gui"):
self.log_gui.log(f"切换信号格式失败: {str(e)}")
else:
if hasattr(self, "log_gui"):
self.log_gui.log("⚠️ signal_tabs 尚未创建")
# ========== 2. 动态切换 Gamma/EOTF Tab ==========
if hasattr(self, "chart_notebook"):
try:
current_tabs = list(self.chart_notebook.tabs())
# 获取当前 Tab 的索引
gamma_tab_id = str(self.gamma_chart_frame)
eotf_tab_id = str(self.eotf_chart_frame)
if test_type == "hdr_movie":
# ========== HDR 测试:移除 Gamma添加 EOTF ==========
# 1. 如果 Gamma Tab 存在,移除它
if gamma_tab_id in current_tabs:
gamma_index = current_tabs.index(gamma_tab_id)
self.chart_notebook.forget(gamma_index)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已隐藏 Gamma 曲线 Tab")
# 2. 如果 EOTF Tab 不存在,添加它(在色域图之后)
if eotf_tab_id not in current_tabs:
self.chart_notebook.insert(
1, self.eotf_chart_frame, text="EOTF 曲线"
)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已显示 EOTF 曲线 Tab")
else:
# ========== SDR/屏模组测试:移除 EOTF添加 Gamma ==========
# 1. 如果 EOTF Tab 存在,移除它
if eotf_tab_id in current_tabs:
eotf_index = current_tabs.index(eotf_tab_id)
self.chart_notebook.forget(eotf_index)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已隐藏 EOTF 曲线 Tab")
# 2. 如果 Gamma Tab 不存在,添加它(在色域图之后)
if gamma_tab_id not in current_tabs:
self.chart_notebook.insert(
1, self.gamma_chart_frame, text="Gamma 曲线"
)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已显示 Gamma 曲线 Tab")
# ========== 3. 仅在 SDR 测试显示客户模板结果 Tab ==========
custom_tab_id = str(self.custom_template_tab_frame)
current_tabs = list(self.chart_notebook.tabs())
if test_type == "sdr_movie":
if custom_tab_id not in current_tabs:
self.chart_notebook.add(
self.custom_template_tab_frame,
text="客户模板结果显示",
)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已显示客户模板结果 Tab")
else:
if custom_tab_id in current_tabs:
self.chart_notebook.forget(self.custom_template_tab_frame)
if hasattr(self, "log_gui"):
self.log_gui.log("✓ 已隐藏客户模板结果 Tab")
# 刷新显示
self.chart_notebook.update_idletasks()
except Exception as e:
if hasattr(self, "log_gui"):
self.log_gui.log(f"切换 Gamma/EOTF Tab 失败: {str(e)}")
def on_test_type_change(self):
"""根据测试类型更新内容区域"""
test_type = self.test_type_var.get()
# 获取当前测试类型的配置
if hasattr(self, "config") and hasattr(self.config, "get_current_config"):
current_config = self.config.get_current_config()
# 更新配置信息显示
self.update_config_info_display()
# SDR 选中时显示客户模版按钮
self.update_custom_button_visibility()
def start_test(self):
"""开始测试"""
# 检查设备连接状态
if self.ca is None or self.ucd is None:
messagebox.showerror("错误", "请先连接CA410和信号发生器")
return
# 检查是否已经在测试中
if self.testing:
messagebox.showinfo("提示", "测试已在进行中")
return
# ✅ 禁用并隐藏单步调试
if hasattr(self, "debug_panel"):
self.debug_panel.disable_all_debug()
self.log_gui.log("✓ 单步调试已禁用")
if hasattr(self, "debug_container"):
self.debug_container.pack_forget()
self.log_gui.log("✓ 单步调试面板已隐藏")
# 获取测试类型和测试项目
test_type = self.test_type_var.get()
test_items = self.get_selected_test_items()
if not test_items:
messagebox.showinfo("提示", "请至少选择一个测试项目")
return
# 自动收起配置项
if hasattr(self, "config_panel_frame"):
try:
if self.config_panel_frame.winfo_viewable():
self.config_panel_frame.btn.invoke()
self.root.update_idletasks()
time.sleep(0.2)
except:
pass
# 禁用配置项按钮
try:
self.config_panel_frame.btn.configure(state="disabled")
except:
pass
# ✅ 新增:禁用色域参考标准下拉框
try:
if hasattr(self, "screen_gamut_combo"):
self.screen_gamut_combo.configure(state="disabled")
if hasattr(self, "sdr_gamut_combo"):
self.sdr_gamut_combo.configure(state="disabled")
if hasattr(self, "hdr_gamut_combo"):
self.hdr_gamut_combo.configure(state="disabled")
except Exception as e:
self.log_gui.log(f"禁用色域参考标准失败: {str(e)}")
# 隐藏所有重新计算按钮
if hasattr(self, "recalc_cct_btn"):
try:
self.recalc_cct_btn.grid_remove()
except:
pass
if hasattr(self, "sdr_recalc_cct_btn"):
try:
self.sdr_recalc_cct_btn.grid_remove()
except:
pass
if hasattr(self, "hdr_recalc_cct_btn"):
try:
self.hdr_recalc_cct_btn.grid_remove()
except:
pass
# 更新UI状态
self.testing = True
self.start_btn.config(state=tk.DISABLED)
self.stop_btn.config(state=tk.NORMAL)
self.save_btn.config(state=tk.DISABLED)
self.clear_config_btn.config(state=tk.DISABLED)
self.status_var.set("测试进行中...")
# 清空日志和图表
self.log_gui.clear_log()
self.clear_chart()
# 根据测试类型显示不同提示
if test_type == "screen_module":
# 屏模组测试:提示 byPass All PQ
message = f"开始屏模组性能测试,请 byPass All PQ"
elif test_type == "sdr_movie":
# SDR测试提示设置正确图像模式
message = f"开始 SDR Movie 测试,请设置正确的图像模式"
elif test_type == "hdr_movie":
# HDR测试提示设置正确图像模式
message = f"开始 HDR Movie 测试,请设置正确的图像模式"
else:
message = f"开始{self.get_test_type_name(test_type)}测试"
confirm = messagebox.askyesno("确认测试", message)
if not confirm:
self.testing = False
self.start_btn.config(state=tk.NORMAL)
self.stop_btn.config(state=tk.DISABLED)
self.clear_config_btn.config(state=tk.NORMAL)
self.status_var.set("测试已取消")
# 恢复配置项按钮
if hasattr(self, "config_panel_frame"):
try:
self.config_panel_frame.btn.configure(state="normal")
except:
pass
return
# 在新线程中执行测试
self.test_thread = threading.Thread(
target=self.run_test, args=(test_type, test_items)
)
self.test_thread.daemon = True
self.test_thread.start()
def stop_test(self):
"""停止测试 - 放弃本次所有数据(完全集成版)"""
if not self.testing:
return
# ========== 1. 添加确认对话框 ==========
confirm = messagebox.askyesno(
"确认停止测试",
"测试正在进行中,确定要停止吗?\n\n⚠️ 停止后将放弃本次测试的所有数据,无法保存。",
icon="warning",
)
if not confirm:
self.log_gui.log("用户取消停止操作")
return
# ========== 2. 立即设置停止标志 ==========
self.testing = False # ← 关键:先设置标志,让测试线程停止
self.log_gui.log("=" * 50)
self.log_gui.log("⚠️ 正在停止测试...")
self.log_gui.log("=" * 50)
# ========== 3. 立即更新UI状态让用户感知到停止==========
self.stop_btn.config(state=tk.DISABLED)
self.status_var.set("正在停止测试,请稍候...")
self.root.update() # 立即刷新界面
# ========== 4. 等待测试线程结束 ==========
if self.test_thread and self.test_thread.is_alive():
self.log_gui.log("等待测试线程结束...")
# 等待最多5秒
for i in range(50): # 50 * 0.1秒 = 5秒
if not self.test_thread.is_alive():
break
time.sleep(0.1)
self.root.update() # 保持界面响应
if self.test_thread.is_alive():
self.log_gui.log("⚠️ 测试线程未能正常结束,将在后台继续等待")
else:
self.log_gui.log("✓ 测试线程已结束")
# ========== 5. 延迟1秒后执行清理使用内部函数==========
def cleanup_and_finish():
"""清理数据并完成停止操作"""
# ========== 5.1 清理测试数据 ==========
try:
self.log_gui.log("清理测试数据...")
# 清空测试结果对象
if hasattr(self, "results"):
self.results = None
self.log_gui.log(" ✓ 测试结果对象已清空")
# 清空中间数据缓存
for attr in [
"gamut_results",
"gamma_results",
"cct_results",
"contrast_results",
"accuracy_results",
]:
if hasattr(self, attr):
setattr(self, attr, None)
self.log_gui.log(" ✓ 所有中间数据已清空")
except Exception as e:
self.log_gui.log(f"⚠️ 清理数据时出错: {str(e)}")
# ========== 5.2 清空图表显示 ==========
try:
self.clear_chart()
self.log_gui.log("✓ 图表已清空")
except Exception as e:
self.log_gui.log(f"⚠️ 清空图表时出错: {str(e)}")
try:
self.clear_custom_template_results()
self.log_gui.log("✓ 客户模板结果表格已清空")
except Exception as e:
self.log_gui.log(f"⚠️ 清空客户模板结果表格失败: {str(e)}")
# ========== 5.2.5 跳转到色域图Tab第一个Tab==========
try:
if hasattr(self, "chart_notebook"):
self.chart_notebook.select(self.gamut_chart_frame)
self.root.update_idletasks() # ← 刷新界面
self.log_gui.log("✓ 已跳转到色域图界面")
except Exception as e:
self.log_gui.log(f"⚠️ 跳转到色域图失败: {str(e)}")
# ========== 5.3 更新UI状态 ==========
self.set_custom_result_table_locked(False)
self.start_btn.config(state=tk.NORMAL)
self.stop_btn.config(state=tk.DISABLED)
self.save_btn.config(state=tk.DISABLED)
self.clear_config_btn.config(state=tk.NORMAL)
if hasattr(self, "custom_btn"):
self.custom_btn.config(state=tk.NORMAL)
self.status_var.set("测试已停止 - 数据已清空")
self.log_gui.log("✓ UI状态已更新")
# ========== 5.4 恢复配置项按钮 ==========
if hasattr(self, "config_panel_frame"):
try:
self.config_panel_frame.btn.configure(state="normal")
self.log_gui.log("✓ 配置项已恢复")
except:
pass
# ========== 5.4.5 禁用色域参考标准下拉框 ==========
try:
if hasattr(self, "screen_gamut_combo"):
self.screen_gamut_combo.configure(state="disabled")
if hasattr(self, "sdr_gamut_combo"):
self.sdr_gamut_combo.configure(state="disabled")
if hasattr(self, "hdr_gamut_combo"):
self.hdr_gamut_combo.configure(state="disabled")
self.log_gui.log("✓ 色域参考标准已禁用")
except Exception as e:
self.log_gui.log(f"禁用色域参考标准失败: {str(e)}")
# ========== 5.5 隐藏所有重新计算按钮 ==========
try:
button_hidden_count = 0
for btn_attr in [
"recalc_cct_btn",
"sdr_recalc_cct_btn",
"hdr_recalc_cct_btn",
"recalc_gamut_btn", # ✅ 新增
"sdr_recalc_gamut_btn", # ✅ 新增
"hdr_recalc_gamut_btn", # ✅ 新增
]:
if hasattr(self, btn_attr):
try:
getattr(self, btn_attr).grid_remove()
button_hidden_count += 1
except:
pass
if button_hidden_count > 0:
self.log_gui.log(f"✓ 已隐藏 {button_hidden_count} 个重新计算按钮")
except Exception as e:
self.log_gui.log(f"⚠️ 隐藏按钮时出错: {str(e)}")
# ========== 5.6 禁用并隐藏单步调试 ==========
if hasattr(self, "debug_panel"):
try:
self.debug_panel.disable_all_debug()
self.log_gui.log("✓ 单步调试已禁用")
except Exception as e:
self.log_gui.log(f"⚠️ 禁用单步调试失败: {str(e)}")
# ✅ 隐藏调试面板
if hasattr(self, "debug_container"):
try:
self.debug_container.pack_forget()
self.log_gui.log("✓ 单步调试面板已隐藏")
except Exception as e:
self.log_gui.log(f"⚠️ 隐藏调试面板失败: {str(e)}")
# ========== 5.7 最终日志 ==========
self.log_gui.log("=" * 50)
self.log_gui.log("✓ 测试已停止,所有数据已清空")
self.log_gui.log("=" * 50)
# ========== 5.8 显示提示信息 ==========
messagebox.showinfo(
"测试已停止",
"测试已停止,本次测试数据已清空。\n\n可以重新开始新的测试。",
)
# ========== 延迟1秒后执行清理 ==========
self.root.after(1000, cleanup_and_finish)
# ==================== 保存测试结果 ====================
def save_results(self):
"""保存测试结果(图片 + Excel。实现委派给 app.export。"""
save_dir = filedialog.askdirectory(title="选择保存测试结果的目录")
if not save_dir:
return
try:
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
test_type = self.get_test_type_name(self.test_type_var.get())
result_dir = os.path.join(save_dir, f"{test_type}_{timestamp}")
os.makedirs(result_dir, exist_ok=True)
current_test_type = self.test_type_var.get()
selected_items = self.get_selected_test_items()
log = self.log_gui.log
log(f"保存测试类型: {current_test_type}")
log(f"已选测试项: {selected_items}")
# 1) 图片
_save_result_images_impl(
result_dir, current_test_type, selected_items,
lambda attr: getattr(self, attr, None),
log,
)
# 2) Excel
if (current_test_type in _EXCEL_EXPORT_CONFIG
and hasattr(self, "results") and self.results):
_export_excel_report_impl(
result_dir, current_test_type, selected_items,
self.results, log,
)
# 3) 成功提示
log("=" * 50)
log(f"✅ 测试结果已保存到目录: {result_dir}")
log("=" * 50)
messagebox.showinfo("成功", f"测试结果已保存到目录:\n{result_dir}")
except Exception as e:
self.log_gui.log(f"❌ 保存测试结果失败: {str(e)}")
import traceback
self.log_gui.log(traceback.format_exc())
messagebox.showerror("错误", f"保存测试结果失败: {str(e)}")
new_pq_results = _run_new_pq_results
run_test = _run_run_test
run_screen_module_test = _run_run_screen_module_test
run_custom_sdr_test = _run_run_custom_sdr_test
run_sdr_movie_test = _run_run_sdr_movie_test
run_hdr_movie_test = _run_run_hdr_movie_test
send_fix_pattern = _run_send_fix_pattern
test_custom_sdr = _run_test_custom_sdr
test_gamut = _run_test_gamut
test_gamma = _run_test_gamma
test_eotf = _run_test_eotf
test_cct = _run_test_cct
test_contrast = _run_test_contrast
test_color_accuracy = _run_test_color_accuracy
calculate_delta_e_2000 = staticmethod(_calc_delta_e_2000)
get_accuracy_color_standards = staticmethod(_get_accuracy_color_standards)
calculate_gamut_coverage = staticmethod(_calc_gamut_coverage)
calculate_gamma = staticmethod(_calc_gamma)
calculate_color_accuracy = staticmethod(_calc_color_accuracy)
calculate_pq_curve = staticmethod(_calc_pq_curve)
plot_gamut = _plot_gamut
plot_gamma = _plot_gamma
plot_eotf = _plot_eotf
plot_cct = _plot_cct
plot_contrast = _plot_contrast
plot_accuracy = _plot_accuracy
on_test_completed = _run_on_test_completed
on_custom_template_test_completed = _run_on_custom_template_test_completed
get_current_test_result = _run_get_current_test_result
on_test_error = _run_on_test_error
update_chart_tabs_state = _cf_update_chart_tabs_state
def get_test_type_name(self, test_type):
"""获取测试类型的显示名称"""
if test_type == "screen_module":
return "屏模组性能测试"
elif test_type == "sdr_movie":
return "SDR Movie测试"
elif test_type == "hdr_movie":
return "HDR Movie测试"
return test_type
def get_selected_test_items(self):
"""获取当前选中的测试项"""
selected_items = []
for var_name, var in self.test_vars.items():
if var.get():
selected_items.append(var_name.split("_")[-1])
return selected_items
def update_config(self, event=None):
"""更新配置"""
try:
self.config.set_device_config(
self.ca_com_var.get(),
self.ucd_list_var.get(),
self.ca_channel_var.get(),
)
# 保存当前选中的测试项到配置
self.config.set_current_test_items(self.get_selected_test_items())
# 待修改为三种测试类型的timing值
self.config.set_current_timing(self.screen_module_timing_var.get())
# 自动保存配置到文件
self.save_pq_config()
except Exception as e:
self.log_gui.log(f"更新配置失败: {str(e)}")
def update_config_and_tabs(self):
"""更新配置并同步Tab状态"""
self.update_config()
self.update_chart_tabs_state()
# 根据当前测试类型保存对应参数
current_test_type = self.config.current_test_type
selected_items = self.get_selected_test_items()
if current_test_type == "screen_module":
if "cct" in selected_items:
self.save_cct_params()
elif current_test_type == "sdr_movie":
if "cct" in selected_items:
self.save_sdr_cct_params()
elif current_test_type == "hdr_movie":
if "cct" in selected_items:
if hasattr(self, "save_hdr_cct_params"):
self.save_hdr_cct_params()
# 控制参数框的显示
self.toggle_cct_params_frame()
def on_closing(self):
"""窗口关闭时的处理"""
try:
# ✅ 检查是否清理了配置
if not self.config_cleared:
# 保存配置
self.save_pq_config()
else:
print("配置已清理,不再保存")
# 断开设备连接
if self.ucd.status:
self.ucd.close()
if self.ca is not None:
self.ca.close()
# 关闭窗口
self.root.destroy()
except Exception as e:
print(f"关闭窗口时出错: {str(e)}")
self.root.destroy()
def main():
try:
# root = tk.Tk()
root = ttk.Window(themename="yeti")
app = PQAutomationApp(root)
root.mainloop()
except Exception as e:
print("程序发生错误:", e)
traceback.print_exc()
finally:
pass
if __name__ == "__main__":
main()