"""色准测试结果绘制。 布局： - 左侧：大尺寸 ColorChecker 条形图（每个条形使用对应颜色）。 - 右侧：CIE 1976 u'v' 色度图（目标点/实测点/偏移连线）。 """ from typing import TYPE_CHECKING from matplotlib.patches import Rectangle from matplotlib.lines import Line2D from matplotlib.patches import Circle from app.plots.gamut_background import get_cie1976_background from app.tests.color_accuracy import get_accuracy_color_standards from app.pq.color_patch_map import get_patch_color from app.pq.color_patch_map import get_patch_color_from_xy if TYPE_CHECKING: from pqAutomationApp import PQAutomationApp # ============================================================ # 常量 # ============================================================ _COLOR_MAP = { "White": "#FFFFFF", "Gray 80": "#E6E6E6", "Gray 65": "#D1D1D1", "Gray 50": "#BABABA", "Gray 35": "#9E9E9E", "Black": "#000000", "Dark Skin": "#735242", "Light Skin": "#C29682", "Blue Sky": "#5E7A9C", "Foliage": "#596B42", "Blue Flower": "#8280B0", "Bluish Green": "#63BDA8", "Orange": "#D97829", "Purplish Blue": "#4A5CA3", "Moderate Red": "#C25461", "Purple": "#5C3D6B", "Yellow Green": "#9EBA40", "Orange Yellow": "#E6A12E", "Blue (Legacy)": "#333D96", "Green (Legacy)": "#479447", "Red (Legacy)": "#B0303B", "Yellow (Legacy)": "#EDC721", "Magenta (Legacy)": "#BA5491", "Cyan (Legacy)": "#0085A3", "100% Red": "#FF0000", "100% Green": "#00FF00", "100% Blue": "#0000FF", "100% Cyan": "#00FFFF", "100% Magenta": "#FF00FF", "100% Yellow": "#FFFF00", } def _xy_to_uv(x: float, y: float): """CIE 1931 xy → CIE 1976 u'v'""" denom = -2.0 * x + 12.0 * y + 3.0 if abs(denom) < 1e-10: return 0.0, 0.0 return (4.0 * x) / denom, (9.0 * y) / denom # ============================================================ # 子图：左侧 Calman 风格面板 # ============================================================ def _draw_left_panel(ax, color_patches, delta_e_values, font_scale=1.0, dark_mode=False): """左侧仅保留大条形图""" ax.clear() n = len(color_patches) if n == 0: ax.set_axis_off() return y_pos = list(range(n)) bar_colors = [_COLOR_MAP.get(name, "#888888") for name in color_patches] ax.barh( y_pos, delta_e_values, height=0.72, color=bar_colors, edgecolor="#202020", linewidth=0.5, zorder=3, ) text_color = "#F3F5F7" if dark_mode else "#111111" bg_color = "#0F1115" if dark_mode else "#FFFFFF" spine_color = "#8C8F94" if dark_mode else "#9A9A9A" ax.set_yticks(y_pos) ax.set_yticklabels(color_patches, fontsize=max(5, 7 * font_scale), color=text_color) ax.invert_yaxis() x_max = max(15.0, max(delta_e_values) * 1.15) ax.set_xlim(0, x_max) ax.tick_params(axis="x", labelsize=max(6, 8 * font_scale), colors=text_color) ax.grid(axis="x", linestyle="-", linewidth=0.6, alpha=0.3, zorder=0) ax.grid(axis="y", linestyle=":", linewidth=0.35, alpha=0.15, zorder=0) ax.set_facecolor(bg_color) for spine in ax.spines.values(): spine.set_color(spine_color) spine.set_linewidth(0.9) # ============================================================ # 子图：CIE 1976 u'v' 色度图（目标 vs 实测） # ============================================================ def _draw_uv_diagram(ax, color_patches, measurements, standards, font_scale=1.0, dark_mode=False): """绘制 CIE 1976 u'v' 上的色准对比。""" ax.clear() try: bg, bbox = get_cie1976_background() xmin, xmax, ymin, ymax = bbox ax.imshow( bg, extent=(xmin, xmax, ymin, ymax), origin="lower", interpolation="bicubic", zorder=0, aspect="auto", ) ax.set_xlim(xmin, xmax) ax.set_ylim(ymin, ymax) except Exception: ax.set_xlim(0.0, 0.65) ax.set_ylim(0.0, 0.60) text_color = "#F3F5F7" if dark_mode else "#111111" sub_text_color = "#D3D7DD" if dark_mode else "#222222" tick_color = "#D3D7DD" if dark_mode else "#222222" legend_label_color = "#FFF" if dark_mode else "#111" legend_bg = "#111" if dark_mode else "#FFFFFF" legend_edge = "#FFF" if dark_mode else "#333" outer_edge = "#FFFFFF" if dark_mode else "#222222" ax.set_facecolor("#000" if dark_mode else "#FFFFFF") ax.set_aspect("equal", adjustable="box") ax.set_title( "CIE 1976 u'v'", fontsize=max(8, 11 * font_scale), fontweight="bold", color=text_color, pad=4, ) ax.set_xlabel("u'", fontsize=max(7, 9 * font_scale), color=sub_text_color, labelpad=1) ax.set_ylabel("v'", fontsize=max(7, 9 * font_scale), color=sub_text_color, labelpad=1) ax.tick_params(axis="both", labelsize=max(6, 8 * font_scale), colors=tick_color) for sp in ax.spines.values(): sp.set_color(outer_edge) sp.set_linewidth(0.9) for name, meas in zip(color_patches, measurements): if meas is None or len(meas) < 2: continue mx, my = meas[0], meas[1] sxy = standards.get(name) if sxy is None: continue sx, sy = sxy m_u, m_v = _xy_to_uv(mx, my) s_u, s_v = _xy_to_uv(sx, sy) # face = get_patch_color_from_xy(name, (sx, sy)).strip().upper() face = _COLOR_MAP.get(name, "#888888") print(name, face) # 目标点（Target） 空心方框 ax.scatter( s_u, s_v, s=90, marker="s", facecolors="none", edgecolors=outer_edge, linewidths=1.6, zorder=18, ) # 实测点（Actual） 彩色实心 + 白色描边 ax.scatter( [m_u], [m_v], s=80, marker="o", color=face, edgecolors=outer_edge, linewidths=1.2, zorder=20, ) # # Δu'v' 偏差连线 # ax.plot( # [s_u, m_u], # [s_v, m_v], # color=face, # linewidth=1.0, # alpha=0.8, # zorder=15, # ) legend_handles = [ Line2D( [0], [0], marker="s", linestyle="none", markerfacecolor="none", markeredgecolor=outer_edge, markersize=9, markeredgewidth=1.4, label="目标 (Target)", ), Line2D( [0], [0], marker="o", linestyle="none", markerfacecolor="#AAAAAA", markeredgecolor=outer_edge, markersize=9, markeredgewidth=1.2, label="实测 (Actual)", ), ] leg = ax.legend( handles=legend_handles, loc="lower right", fontsize=max(6, 8 * font_scale), framealpha=0.9, labelcolor=legend_label_color, ) if leg: leg.get_frame().set_facecolor(legend_bg) leg.get_frame().set_edgecolor(legend_edge) leg.set_zorder(50) def _draw_result_judgement(ax, accuracy_data, font_scale=1.0, dark_mode=False): """底部结果条""" ax.clear() ax.set_xlim(0, 1) ax.set_ylim(0, 1) ax.axis("off") avg = accuracy_data.get("avg_delta_e", 0.0) mx = accuracy_data.get("max_delta_e", 0.0) panel_bg = "#1A1E24" if dark_mode else "#FFFFFF" panel_edge = "#4A5058" if dark_mode else "#C6C6C6" text_color = "#F3F5F7" if dark_mode else "#111111" ax.add_patch(Rectangle( (0.0, 0.10), 1.0, 0.80, transform=ax.transAxes, facecolor=panel_bg, edgecolor=panel_edge, linewidth=1.0, )) ax.text( 0.03, 0.50, f"Avg dE2000: {avg:.2f}", ha="left", va="center", fontsize=max(11, 20 * font_scale), fontweight="normal", color=text_color, transform=ax.transAxes, ) ax.text( 0.52, 0.50, f"Max dE2000: {mx:.2f}", ha="left", va="center", fontsize=max(11, 20 * font_scale), fontweight="normal", color=text_color, transform=ax.transAxes, ) # ============================================================ # 主入口 # ============================================================ def plot_accuracy(self: "PQAutomationApp", accuracy_data, test_type): """绘制色准测试结果 - Calman 风格（色块 + CIE 1976 u'v' + 统计）。""" fig = self.accuracy_fig fig.clear() try: from app.views.theme_manager import is_dark dark_mode = is_dark() except Exception: dark_mode = False fig.patch.set_facecolor("#1B1F24" if dark_mode else "#FFFFFF") # 根据当前画布像素尺寸动态缩放字体，避免窗口缩小时文字挤压重叠。 font_scale = 1.0 try: canvas_widget = self.accuracy_canvas.get_tk_widget() cw = max(1, int(canvas_widget.winfo_width())) ch = max(1, int(canvas_widget.winfo_height())) font_scale = min(cw / 1000.0, ch / 600.0) font_scale = max(0.60, min(1.0, font_scale)) except Exception: font_scale = 1.0 color_patches = accuracy_data.get("color_patches", []) or [] delta_e_values = accuracy_data.get("delta_e_values", []) or [] measurements = accuracy_data.get("color_measurements", []) or [] try: target_gamma = float(accuracy_data.get("target_gamma", 2.2)) except (TypeError, ValueError): target_gamma = 2.2 test_type_name = self.get_test_type_name(test_type) if test_type == "sdr_movie": title = f"{test_type_name} - 色准测试（全 29色 | Gamma {target_gamma}）" elif test_type == "hdr_movie": title = f"{test_type_name} - 色准测试（全 29色 | PQ EOTF）" else: title = f"{test_type_name} - 色准测试（全 29色 | Gamma {target_gamma}）" title_color = "#F3F5F7" if dark_mode else "#111" fig.suptitle( title, fontsize=max(8, 11 * font_scale), y=0.975, fontweight="bold", color=title_color, ) gs = fig.add_gridspec( 2, 2, width_ratios=[1.12, 1.0], height_ratios=[4.8, 0.48], left=0.08, right=0.985, top=0.92, bottom=0.05, wspace=0.14, hspace=0.08, ) ax_left = fig.add_subplot(gs[0, 0]) ax_uv = fig.add_subplot(gs[0, 1]) ax_judge = fig.add_subplot(gs[1, :]) # 兼容外部对 self.accuracy_ax 的引用 self.accuracy_ax = ax_judge _draw_left_panel( ax_left, color_patches, delta_e_values, font_scale=font_scale, dark_mode=dark_mode, ) try: standards = get_accuracy_color_standards(test_type) except Exception: standards = {} _draw_uv_diagram( ax_uv, color_patches, measurements, standards, font_scale=font_scale, dark_mode=dark_mode, ) _draw_result_judgement( ax_judge, accuracy_data, font_scale=font_scale, dark_mode=dark_mode, ) try: self.update_accuracy_result_table(accuracy_data, standards) except Exception: pass # Select the tab first so the canvas is visible and winfo_width/height # return real pixel dimensions before the figure is rendered. self.chart_notebook.select(self.accuracy_chart_frame) self.accuracy_canvas.get_tk_widget().update_idletasks() self.accuracy_canvas.draw() class PlotAccuracyMixin: """由 tools/refactor_to_mixins.py 自动生成。 把本模块的自由函数挂到 PQAutomationApp 上，便于 F12 跳转与类型推断。 """ plot_accuracy = plot_accuracy