328 lines
9.8 KiB
Python
328 lines
9.8 KiB
Python
"""色准测试结果绘制。
|
||
|
||
布局：
|
||
- 左侧：大尺寸 ColorChecker 条形图（每个条形使用对应颜色）。
|
||
- 右侧：CIE 1976 u'v' 色度图（目标点/实测点/偏移连线）。
|
||
"""
|
||
|
||
from typing import TYPE_CHECKING
|
||
|
||
from matplotlib.patches import Rectangle
|
||
from matplotlib.lines import Line2D
|
||
|
||
from app.plots.gamut_background import get_cie1976_background
|
||
from app.tests.color_accuracy import get_accuracy_color_standards
|
||
|
||
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 _grade_color(delta_e: float) -> str:
|
||
if delta_e < 3:
|
||
return "#1FAE45" # 绿
|
||
if delta_e < 5:
|
||
return "#E08A00" # 橙
|
||
return "#D81B1B" # 红
|
||
|
||
|
||
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):
|
||
"""左侧仅保留大条形图。"""
|
||
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]
|
||
edge_colors = [_grade_color(dE) for dE in delta_e_values]
|
||
|
||
ax.barh(
|
||
y_pos,
|
||
delta_e_values,
|
||
height=0.72,
|
||
color=bar_colors,
|
||
edgecolor=edge_colors,
|
||
linewidth=1.0,
|
||
zorder=3,
|
||
)
|
||
|
||
ax.set_yticks(y_pos)
|
||
ax.set_yticklabels(color_patches, fontsize=max(5, 7 * font_scale))
|
||
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))
|
||
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("#FFFFFF")
|
||
for spine in ax.spines.values():
|
||
spine.set_color("#9A9A9A")
|
||
spine.set_linewidth(0.9)
|
||
|
||
|
||
# ============================================================
|
||
# 子图：CIE 1976 u'v' 色度图（目标 vs 实测）
|
||
# ============================================================
|
||
|
||
def _draw_uv_diagram(ax, color_patches, measurements, standards, font_scale=1.0):
|
||
"""绘制 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="upper", 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)
|
||
|
||
ax.set_facecolor("#000")
|
||
ax.set_aspect("equal", adjustable="box")
|
||
ax.set_title("CIE 1976 u'v'", fontsize=max(8, 11 * font_scale), fontweight="bold",
|
||
color="#111", pad=4)
|
||
ax.set_xlabel("u'", fontsize=max(7, 9 * font_scale), color="#222", labelpad=1)
|
||
ax.set_ylabel("v'", fontsize=max(7, 9 * font_scale), color="#222", labelpad=1)
|
||
ax.tick_params(axis="both", labelsize=max(6, 8 * font_scale), colors="#222")
|
||
for sp in ax.spines.values():
|
||
sp.set_color("#666")
|
||
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 = _COLOR_MAP.get(name, "#FFFFFF")
|
||
|
||
# 目标点：仅空心方框（不填充标准颜色）
|
||
ax.scatter(
|
||
[s_u], [s_v],
|
||
s=56, marker="s",
|
||
facecolors="none", edgecolors="#FFFFFF",
|
||
linewidths=1.25, zorder=18,
|
||
)
|
||
# 实测点：白色外圈 + 内层圆点
|
||
ax.scatter(
|
||
[m_u], [m_v],
|
||
s=52, marker="o",
|
||
facecolors="none", edgecolors="#FFFFFF",
|
||
linewidths=1.0, zorder=19,
|
||
)
|
||
ax.scatter(
|
||
[m_u], [m_v],
|
||
s=24, marker="o",
|
||
facecolors=face, edgecolors="#111111",
|
||
linewidths=0.85, zorder=20,
|
||
)
|
||
|
||
legend_handles = [
|
||
Line2D([0], [0], marker="s", linestyle="none",
|
||
markerfacecolor="#CCCCCC", markeredgecolor="#FFFFFF",
|
||
markersize=7, label="目标 (Target)"),
|
||
Line2D([0], [0], marker="o", linestyle="none",
|
||
markerfacecolor="#CCCCCC", markeredgecolor="#000000",
|
||
markersize=7, label="实测 (Actual)"),
|
||
]
|
||
leg = ax.legend(
|
||
handles=legend_handles,
|
||
loc="lower right", fontsize=max(6, 8 * font_scale),
|
||
framealpha=0.88, labelcolor="#FFF",
|
||
)
|
||
if leg is not None:
|
||
leg.get_frame().set_facecolor("#111")
|
||
leg.get_frame().set_edgecolor("#FFF")
|
||
leg.set_zorder(50)
|
||
|
||
|
||
def _draw_result_judgement(ax, accuracy_data, font_scale=1.0):
|
||
"""底部结果条"""
|
||
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)
|
||
|
||
ax.add_patch(Rectangle(
|
||
(0.0, 0.10), 1.0, 0.80,
|
||
transform=ax.transAxes,
|
||
facecolor="#FFFFFF", edgecolor="#C6C6C6", 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="#111111",
|
||
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="#111111",
|
||
transform=ax.transAxes,
|
||
)
|
||
|
||
|
||
# ============================================================
|
||
# 主入口
|
||
# ============================================================
|
||
|
||
def plot_accuracy(self: "PQAutomationApp", accuracy_data, test_type):
|
||
"""绘制色准测试结果 - Calman 风格（色块 + CIE 1976 u'v' + 统计）。"""
|
||
|
||
fig = self.accuracy_fig
|
||
fig.clear()
|
||
|
||
# 根据当前画布像素尺寸动态缩放字体，避免窗口缩小时文字挤压重叠。
|
||
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}）"
|
||
|
||
fig.suptitle(
|
||
title,
|
||
fontsize=max(8, 11 * font_scale),
|
||
y=0.975,
|
||
fontweight="bold",
|
||
color="#111",
|
||
)
|
||
|
||
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)
|
||
|
||
try:
|
||
standards = get_accuracy_color_standards(test_type)
|
||
except Exception:
|
||
standards = {}
|
||
|
||
_draw_uv_diagram(
|
||
ax_uv,
|
||
color_patches,
|
||
measurements,
|
||
standards,
|
||
font_scale=font_scale,
|
||
)
|
||
_draw_result_judgement(ax_judge, accuracy_data, font_scale=font_scale)
|
||
|
||
try:
|
||
self.update_accuracy_result_table(accuracy_data, standards)
|
||
except Exception:
|
||
pass
|
||
|
||
self.accuracy_canvas.draw()
|
||
self.chart_notebook.select(self.accuracy_chart_frame)
|
||
|
||
|
||
class PlotAccuracyMixin:
|
||
"""由 tools/refactor_to_mixins.py 自动生成。
|
||
把本模块的自由函数挂到 PQAutomationApp 上，便于 F12 跳转与类型推断。
|
||
"""
|
||
plot_accuracy = plot_accuracy
|