pqAutomationApp/app/plots/plot_accuracy.py

"""色准测试结果绘制。

布局：
- 左侧：大尺寸 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):
    """左侧仅保留大条形图。"""
    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=7)
    ax.invert_yaxis()

    x_max = max(15.0, max(delta_e_values) * 1.15)
    ax.set_xlim(0, x_max)
    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):
    """绘制 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=11, fontweight="bold",
                 color="#111", pad=4)
    ax.set_xlabel("u'", fontsize=9, color="#222", labelpad=1)
    ax.set_ylabel("v'", fontsize=9, color="#222", labelpad=1)
    ax.tick_params(axis="both", labelsize=8, 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=8,
        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):
    """底部结果条"""
    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=20, fontweight="normal", color="#111111",
        transform=ax.transAxes,
    )
    ax.text(
        0.52, 0.50,
        f"Max dE2000: {mx:.2f}",
        ha="left", va="center",
        fontsize=20, 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()

    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=11, y=0.975, fontweight="bold", color="#111")

    gs = fig.add_gridspec(
        2, 2,
        width_ratios=[1.12, 1.0],
        height_ratios=[4.0, 0.62],
        left=0.08, right=0.985,
        top=0.91, bottom=0.06,
        wspace=0.14, hspace=0.10,
    )

    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)

    try:
        standards = get_accuracy_color_standards(test_type)
    except Exception:
        standards = {}
    _draw_uv_diagram(ax_uv, color_patches, measurements, standards)
    _draw_result_judgement(ax_judge, accuracy_data)

    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