282 lines
9.5 KiB
Python
282 lines
9.5 KiB
Python
"""Local Dimming 测试逻辑（应用层）。
|
||
|
||
整合自原 drivers/local_dimming_test.py：窗口图片生成与测试主循环
|
||
直接落在本模块，UCD 通用操作下沉到 drivers.ucd_helpers。
|
||
"""
|
||
|
||
import atexit
|
||
import csv
|
||
import datetime
|
||
import os
|
||
import shutil
|
||
import sys
|
||
import threading
|
||
import time
|
||
import tkinter as tk
|
||
from tkinter import filedialog, messagebox
|
||
|
||
import numpy as np
|
||
from PIL import Image
|
||
|
||
from drivers.ucd_helpers import get_current_resolution, send_image_pattern
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# 模块级常量与窗口图片缓存
|
||
# --------------------------------------------------------------------------
|
||
|
||
DEFAULT_WINDOW_PERCENTAGES = [1, 2, 5, 10, 18, 25, 50, 75, 100]
|
||
|
||
_TEMP_DIR = None
|
||
_IMAGE_CACHE = {} # {(width, height, percentage): file_path}
|
||
|
||
|
||
def _cleanup_temp_dir():
|
||
global _TEMP_DIR
|
||
if _TEMP_DIR and os.path.exists(_TEMP_DIR):
|
||
try:
|
||
shutil.rmtree(_TEMP_DIR)
|
||
except Exception:
|
||
pass
|
||
_TEMP_DIR = None
|
||
_IMAGE_CACHE.clear()
|
||
|
||
|
||
def _get_temp_dir():
|
||
global _TEMP_DIR
|
||
if _TEMP_DIR is None:
|
||
if getattr(sys, "frozen", False):
|
||
base = os.path.dirname(sys.executable)
|
||
else:
|
||
base = os.getcwd()
|
||
_TEMP_DIR = os.path.join(base, "temp_local_dimming")
|
||
os.makedirs(_TEMP_DIR, exist_ok=True)
|
||
atexit.register(_cleanup_temp_dir)
|
||
return _TEMP_DIR
|
||
|
||
|
||
def _make_window_image_array(width, height, percentage):
|
||
"""生成黑底+居中白窗的 numpy 图像，保持屏幕比例。"""
|
||
image = np.zeros((height, width, 3), dtype=np.uint8)
|
||
if percentage >= 100:
|
||
ww, wh = width, height
|
||
else:
|
||
scale = (percentage / 100.0) ** 0.5
|
||
ww = int(width * scale)
|
||
wh = int(height * scale)
|
||
x1 = (width - ww) // 2
|
||
y1 = (height - wh) // 2
|
||
image[y1:y1 + wh, x1:x1 + ww] = 255
|
||
return image
|
||
|
||
|
||
def _ensure_window_image(width, height, percentage):
|
||
"""生成或复用缓存的窗口 PNG 文件，返回路径。"""
|
||
key = (width, height, percentage)
|
||
cached = _IMAGE_CACHE.get(key)
|
||
if cached and os.path.exists(cached):
|
||
return cached
|
||
arr = _make_window_image_array(width, height, percentage)
|
||
fname = f"window_{width}x{height}_{percentage:03d}percent.png"
|
||
path = os.path.join(_get_temp_dir(), fname)
|
||
Image.fromarray(arr, mode="RGB").save(path, format="PNG")
|
||
_IMAGE_CACHE[key] = path
|
||
return path
|
||
|
||
|
||
# --------------------------------------------------------------------------
|
||
# GUI 入口（绑定为 PQAutomationApp 方法）
|
||
# --------------------------------------------------------------------------
|
||
|
||
def start_local_dimming_test(self):
|
||
"""开始 Local Dimming 测试。"""
|
||
if not self.ca or not self.ucd.status:
|
||
messagebox.showerror("错误", "请先连接 CA410 和 UCD323")
|
||
return
|
||
|
||
self.ld_start_btn.config(state=tk.DISABLED)
|
||
self.ld_stop_btn.config(state=tk.NORMAL)
|
||
self.ld_save_btn.config(state=tk.DISABLED)
|
||
|
||
for item in self.ld_tree.get_children():
|
||
self.ld_tree.delete(item)
|
||
|
||
wait_time = float(self.ld_wait_time_var.get())
|
||
stop_event = threading.Event()
|
||
self.ld_stop_event = stop_event
|
||
|
||
def worker():
|
||
log = self.log_gui.log
|
||
log("=" * 60, level="separator")
|
||
log("开始 Local Dimming 测试", level="info")
|
||
log("=" * 60, level="separator")
|
||
|
||
width, height = get_current_resolution(self.ucd)
|
||
total = len(DEFAULT_WINDOW_PERCENTAGES)
|
||
log(f" 分辨率: {width}x{height}", level="info")
|
||
log(f" 测试窗口: {DEFAULT_WINDOW_PERCENTAGES}", level="info")
|
||
log(f" 等待时间: {wait_time} 秒", level="info")
|
||
|
||
results = []
|
||
for i, percentage in enumerate(DEFAULT_WINDOW_PERCENTAGES, 1):
|
||
if stop_event.is_set():
|
||
log("测试已停止", level="error")
|
||
break
|
||
|
||
log(f"[{i}/{total}] 测试 {percentage}% 窗口...", level="info")
|
||
try:
|
||
image_path = _ensure_window_image(width, height, percentage)
|
||
except Exception as e:
|
||
log(f" 图像生成失败: {e}", level="error")
|
||
continue
|
||
|
||
if not send_image_pattern(self.ucd, image_path):
|
||
log(f" {percentage}% 窗口发送失败，跳过", level="error")
|
||
continue
|
||
|
||
log(f"等待 {wait_time} 秒...", level="info")
|
||
time.sleep(wait_time)
|
||
|
||
try:
|
||
x, y, lv, _X, _Y, _Z = self.ca.readAllDisplay()
|
||
except Exception as e:
|
||
log(f" 采集亮度异常: {e}", level="error")
|
||
continue
|
||
if lv is None:
|
||
log(f" {percentage}% 窗口采集失败", level="error")
|
||
continue
|
||
|
||
log(f"采集亮度: {lv:.2f} cd/m²", level="info")
|
||
results.append((percentage, x, y, lv, _X, _Y, _Z))
|
||
|
||
log("=" * 60, level="separator")
|
||
log(f"Local Dimming 测试完成 ({len(results)}/{total})", level="success")
|
||
log("=" * 60, level="separator")
|
||
|
||
self.ld_test_results = results
|
||
self._dispatch_ui(self.update_ld_results, results)
|
||
self._dispatch_ui(self.ld_start_btn.config, state=tk.NORMAL)
|
||
self._dispatch_ui(self.ld_stop_btn.config, state=tk.DISABLED)
|
||
self._dispatch_ui(self.ld_save_btn.config, state=tk.NORMAL)
|
||
|
||
threading.Thread(target=worker, daemon=True).start()
|
||
|
||
|
||
def update_ld_results(self, results):
|
||
"""把批量测试结果填入 Treeview。"""
|
||
for percentage, x, y, lv, _X, _Y, _Z in results:
|
||
self.ld_tree.insert(
|
||
"", tk.END,
|
||
values=(f"{percentage}%", f"{lv:.2f}", f"{x:.4f}", f"{y:.4f}"),
|
||
)
|
||
|
||
|
||
def stop_local_dimming_test(self):
|
||
"""请求停止当前 Local Dimming 测试。"""
|
||
ev = getattr(self, "ld_stop_event", None)
|
||
if ev:
|
||
ev.set()
|
||
|
||
|
||
def send_ld_window(self, percentage):
|
||
"""发送指定百分比的白色窗口（手动模式）。"""
|
||
if not self.ucd.status:
|
||
messagebox.showwarning("警告", "请先连接 UCD323 设备")
|
||
return
|
||
|
||
self.log_gui.log(f"🔆 发送 {percentage}% 窗口...", level="info")
|
||
self.current_ld_percentage = percentage
|
||
|
||
def send():
|
||
width, height = get_current_resolution(self.ucd)
|
||
try:
|
||
image_path = _ensure_window_image(width, height, percentage)
|
||
except Exception as e:
|
||
self._dispatch_ui(self.log_gui.log, f"图像生成失败: {e}")
|
||
return
|
||
ok = send_image_pattern(self.ucd, image_path)
|
||
msg = (
|
||
f"{percentage}% 窗口已发送" if ok
|
||
else f"{percentage}% 窗口发送失败"
|
||
)
|
||
self._dispatch_ui(self.log_gui.log, msg)
|
||
|
||
threading.Thread(target=send, daemon=True).start()
|
||
|
||
|
||
def measure_ld_luminance(self):
|
||
"""测量当前显示的亮度并追加一行到 Treeview。"""
|
||
if not self.ca:
|
||
messagebox.showwarning("警告", "请先连接 CA410 色度计")
|
||
return
|
||
if self.current_ld_percentage is None:
|
||
messagebox.showinfo("提示", "请先发送一个窗口图案")
|
||
return
|
||
|
||
self.log_gui.log("📏 正在采集亮度...", level="info")
|
||
|
||
def measure():
|
||
try:
|
||
x, y, lv, _X, _Y, _Z = self.ca.readAllDisplay()
|
||
except Exception as e:
|
||
self._dispatch_ui(self.log_gui.log, f"采集异常: {str(e)}")
|
||
return
|
||
if lv is None:
|
||
self._dispatch_ui(self.log_gui.log, "采集失败")
|
||
return
|
||
timestamp = datetime.datetime.now().strftime("%H:%M:%S")
|
||
self._dispatch_ui(
|
||
self.ld_result_label.config,
|
||
text=f"亮度: {lv:.2f} cd/m² | x: {x:.4f} | y: {y:.4f}",
|
||
)
|
||
self._dispatch_ui(
|
||
self.ld_tree.insert, "", tk.END,
|
||
values=(
|
||
f"{self.current_ld_percentage}%",
|
||
f"{lv:.2f}", f"{x:.4f}", f"{y:.4f}", timestamp,
|
||
),
|
||
)
|
||
self._dispatch_ui(self.log_gui.log, f"采集完成: {lv:.2f} cd/m²")
|
||
|
||
threading.Thread(target=measure, daemon=True).start()
|
||
|
||
|
||
def clear_ld_records(self):
|
||
"""清空 Treeview 中的测试记录。"""
|
||
for item in self.ld_tree.get_children():
|
||
self.ld_tree.delete(item)
|
||
self.ld_result_label.config(text="亮度: -- cd/m² | x: -- | y: --")
|
||
self.current_ld_percentage = None
|
||
self.log_gui.log("测试记录已清空", level="info")
|
||
|
||
|
||
def save_local_dimming_results(self):
|
||
"""把 Treeview 中的全部记录导出为 CSV。"""
|
||
if len(self.ld_tree.get_children()) == 0:
|
||
messagebox.showinfo("提示", "没有可保存的数据")
|
||
return
|
||
|
||
default_name = (
|
||
f"LocalDimming_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
|
||
)
|
||
save_path = filedialog.asksaveasfilename(
|
||
title="保存测试结果",
|
||
initialfile=default_name,
|
||
defaultextension=".csv",
|
||
filetypes=[("CSV 文件", "*.csv"), ("所有文件", "*.*")],
|
||
)
|
||
if not save_path:
|
||
return
|
||
|
||
try:
|
||
with open(save_path, "w", newline="", encoding="utf-8-sig") as f:
|
||
writer = csv.writer(f)
|
||
writer.writerow(["窗口百分比", "亮度 (cd/m²)", "x", "y", "时间"])
|
||
for item in self.ld_tree.get_children():
|
||
writer.writerow(self.ld_tree.item(item, "values"))
|
||
self.log_gui.log(f"测试结果已保存: {save_path}", level="success")
|
||
messagebox.showinfo("成功", f"测试结果已保存到:\n{save_path}")
|
||
except Exception as e:
|
||
self.log_gui.log(f"保存失败: {str(e)}", level="error")
|
||
messagebox.showerror("错误", f"保存失败: {str(e)}")
|