pqAutomationApp/app/plots/plot_gamut.py

"""色域图（Gamut）绘制 - Calman 风格。

架构：**图像渲染层** 与 **基础数据/框架层** 分离
------------------------------------------------
- 图像渲染层（重）：CIE 1931 / 1976 谱迹色域底图。
  由 `app.plots.gamut_background` 通过 colour-science 离屏渲染一次，
  结果以 numpy RGBA 数组缓存在内存与磁盘（settings/cache/），后续直接
  `ax.imshow(bg, extent=bbox)` 复用 → 主线程绘制开销可忽略。
- 基础数据/框架层（轻）：参考色域三角形、实测色域三角形、顶点标签、
  覆盖率信息框等矢量元素，每次绘制都在真实色度坐标系上重画。

视觉风格：参照 Calman colorspace 显示：
- 当前选中的参考标准：亮色实线 + 顶点空心方框；
- 其他参考标准：半透明虚线（便于对比，不喧宾夺主）；
- 实测色域：红色粗边 + 淡红填充 + 顶点圆点 + 浮动坐标标签；
- 右下角白底红字覆盖率信息框。
"""

from matplotlib.patches import Polygon
import numpy as np
from matplotlib.patches import PathPatch
from matplotlib.path import Path

import algorithm.pq_algorithm as pq_algorithm
from app.plots.gamut_background import (
    get_cie1931_background,
    get_cie1976_background,
)

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from pqAutomationApp import PQAutomationApp


# ============ 参考色域定义（CIE 1931 xy）============
_REF_GAMUTS_XY = {
    "BT.601":  [(0.6300, 0.3400), (0.3100, 0.5950), (0.1550, 0.0700)],
    "BT.709":  [(0.6400, 0.3300), (0.3000, 0.6000), (0.1500, 0.0600)],
    "DCI-P3":  [(0.6800, 0.3200), (0.2650, 0.6900), (0.1500, 0.0600)],
    "BT.2020": [(0.7080, 0.2920), (0.1700, 0.7970), (0.1310, 0.0460)],
}

_REF_COLORS = {
    "BT.601":  "#FBD985",
    "BT.709":  "#FFFFFF",
    "DCI-P3":  "#6AA2F7",
    "BT.2020": "#73FC9C",
}


# ============================================================
# 坐标转换
# ============================================================

def _xy_to_uv(x, y):
    """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


def _ref_gamut_uv(name):
    return [_xy_to_uv(x, y) for x, y in _REF_GAMUTS_XY[name]]


# ============================================================
# 数据/框架层绘制原语
# ============================================================

def _draw_reference_triangle(ax, vertices, color, *, is_current, label):
    """参考色域三角形"""
    xs = [p[0] for p in vertices] + [vertices[0][0]]
    ys = [p[1] for p in vertices] + [vertices[0][1]]

    if is_current:
        poly = Polygon(
            vertices, closed=True,
            facecolor=(1, 1, 1, 0.18),  # 半透明白
            edgecolor=color, linewidth=2.2, zorder=8,
        )
        ax.add_patch(poly)
        ax.plot(
            xs, ys,
            color=color, linewidth=2.2, linestyle="-",
            label=label, zorder=9,
        )
        ax.scatter(
            xs[:-1], ys[:-1],
            s=60, facecolors="none", edgecolors=color,
            linewidths=2, marker="s", zorder=10,
        )
    else:
        ax.plot(
            xs, ys,
            color=color, linewidth=1.2, linestyle="--",
            alpha=0.55, label=label, zorder=5,
        )


def _draw_measured_triangle(ax, vertices, *, uv_space=False):
    xs = [p[0] for p in vertices] + [vertices[0][0]]
    ys = [p[1] for p in vertices] + [vertices[0][1]]

    # 半透明红色填充
    poly = Polygon(
        vertices, closed=True,
        facecolor=(1.0, 0.1, 0.1, 0),
        edgecolor="#FF2A2A",
        linewidth=2.8, zorder=12,
        joinstyle="round"
    )
    ax.add_patch(poly)

    # 顶点（白边红心）
    ax.scatter(
        xs[:-1], ys[:-1],
        s=60, facecolors="#FF2A2A",
        edgecolors="white", linewidths=1.5,
        marker="o", zorder=13,
    )
    # for (x, y), name in zip(vertices, label_prefix):
    #     dx, dy = x - cx, y - cy
    #     norm = max(1e-6, (dx * dx + dy * dy) ** 0.5)
    #     ox = dx / norm * offset_pix
    #     oy = dy / norm * offset_pix

    #     ax.annotate(
    #         f"{name} ({x:.3f}, {y:.3f})",
    #         xy=(x, y),
    #         xytext=(ox, oy),
    #         textcoords="offset points",
    #         fontsize=8.5, color="white", fontweight="bold",
    #         ha="center", va="center",
    #         bbox=dict(
    #             boxstyle="round,pad=0.35",
    #             facecolor="#D81B1B",
    #             edgecolor="white",
    #             linewidth=1.2,
    #             alpha=0.95,
    #         ),
    #         arrowprops=dict(
    #             arrowstyle="-", color="#FF1F1F", lw=1.2, alpha=0.9,
    #         ),
    #         zorder=12,
    #         clip_on=False,
    #     )


def _draw_coverage_box(ax, x_pos, y_pos, current_ref, coverage, *, dark_mode):
    text_color = "#FFF" if dark_mode else "#111"
    box_face = "#111" if dark_mode else "#FFFFFF"
    box_edge = "#FFF" if dark_mode else "#333"
    ax.text(
        x_pos, y_pos,
        f"{current_ref}\n覆盖率: {coverage:.1f}%",
        ha="right", va="bottom",
        fontsize=11, fontweight="bold",
        color=text_color,
        bbox=dict(
            boxstyle="round,pad=0.38",
            facecolor=box_face,
            edgecolor=box_edge,
            linewidth=1.7,
            alpha=0.98,
        ),
        zorder=30,
    )

def _style_axes(ax, *, title, xlabel, ylabel, xlim, ylim, dark_mode):
    text = "#F4F6F8" if dark_mode else "#111"
    grid = "#444" if dark_mode else "#B8BDC3"
    spine_color = "#888" if dark_mode else "#666"
    ax.set_facecolor("#000" if dark_mode else "#FFFFFF")
    ax.set_title(title, fontsize=12, fontweight="bold", color=text, pad=8)
    ax.set_xlabel(xlabel, fontsize=10, color=text)
    ax.set_ylabel(ylabel, fontsize=10, color=text)
    ax.set_xlim(*xlim)
    ax.set_ylim(*ylim)
    ax.set_aspect("equal", adjustable="datalim")
    ax.grid(True, linestyle=":", linewidth=0.7, color=grid, alpha=0.32)
    ax.tick_params(axis="both", labelsize=9, colors=text)
    for spine in ax.spines.values():
        spine.set_color(spine_color)
        spine.set_linewidth(0.8)
    ax.set_clip_on(False)


def _blit_background(ax, background, bbox):
    """渲染层贴底：将预渲染的谱迹底图贴到真实色度坐标。"""
    xmin, xmax, ymin, ymax = bbox
    ax.imshow(
        background,
        extent=(xmin, xmax, ymin, ymax),
        origin="upper",          # canvas.buffer_rgba 行 0 为顶部
        interpolation="bicubic",
        zorder=0,
        aspect="auto",           # 由 _style_axes 的 set_aspect("equal") 控制
    )


# ============================================================
# 主入口
# ============================================================

def plot_gamut(self: "PQAutomationApp", results, coverage, test_type):
    """绘制色域图（图像层 + 框架层分离架构）。"""

    try:
        from app.views.theme_manager import is_dark
        dark_mode = is_dark()
    except Exception:
        dark_mode = False

    ax_xy = self.gamut_ax_xy
    ax_uv = self.gamut_ax_uv

    ax_xy.clear()
    ax_uv.clear()
    # 全局背景跟随浅/深色主题
    self.gamut_fig.patch.set_facecolor("#0D1014" if dark_mode else "#FFFFFF")

    # ========== 读取用户选择的参考标准 ==========
    if test_type == "screen_module":
        current_ref = self.screen_gamut_ref_var.get()
    elif test_type == "sdr_movie":
        current_ref = self.sdr_gamut_ref_var.get()
    elif test_type == "hdr_movie":
        current_ref = self.hdr_gamut_ref_var.get()
    else:
        current_ref = "DCI-P3"

    if current_ref not in _REF_GAMUTS_XY:
        self.log_gui.log(f"未知参考标准 '{current_ref}'，使用 DCI-P3", level="error")
        current_ref = "DCI-P3"

    # ========== 重新计算 xy 覆盖率 ==========
    xy_coverage = coverage
    uv_coverage = 0.0
    measured_xy = None

    if len(results) >= 3:
        measured_xy = [(float(r[0]), float(r[1])) for r in results[:3]]
        try:
            if current_ref == "BT.2020":
                _, xy_coverage = pq_algorithm.calculate_gamut_coverage_BT2020(measured_xy)
            elif current_ref == "BT.709":
                _, xy_coverage = pq_algorithm.calculate_gamut_coverage_BT709(measured_xy)
            elif current_ref == "DCI-P3":
                _, xy_coverage = pq_algorithm.calculate_gamut_coverage_DCIP3(measured_xy)
            elif current_ref == "BT.601":
                _, xy_coverage = pq_algorithm.calculate_gamut_coverage_BT601(measured_xy)
            self.log_gui.log(
                f"XY 空间覆盖率（{current_ref}）: {xy_coverage:.1f}%", level="success"
            )
        except Exception as e:
            self.log_gui.log(f"重新计算 XY 覆盖率失败: {str(e)}", level="error")
            xy_coverage = coverage

    # 需要叠加的次要参考色域
    other_refs = [
        r for r in _REF_GAMUTS_XY.keys()
        if r != current_ref
        and (r != "BT.601" or test_type == "sdr_movie")
    ]

    # ============================================================
    # 左图：CIE 1931 xy
    # ============================================================
    try:
        bg_xy, bbox_xy = get_cie1931_background()
        _blit_background(ax_xy, bg_xy, bbox_xy)
        _style_axes(
            ax_xy,
            title="CIE 1931 xy",
            xlabel="x", ylabel="y",
            xlim=(bbox_xy[0], bbox_xy[1]),
            ylim=(bbox_xy[2], bbox_xy[3]),
            dark_mode=dark_mode,
        )

        for ref_name in other_refs:
            _draw_reference_triangle(
                ax_xy, _REF_GAMUTS_XY[ref_name],
                _REF_COLORS[ref_name],
                is_current=False, label=ref_name,
            )
        _draw_reference_triangle(
            ax_xy, _REF_GAMUTS_XY[current_ref],
            _REF_COLORS[current_ref],
            is_current=True, label=f"{current_ref} (参考)",
        )

        if measured_xy is not None:
            r_xy, g_xy, b_xy = measured_xy
            self.log_gui.log(
                f"测量色域: R({r_xy[0]:.4f},{r_xy[1]:.4f}) "
                f"G({g_xy[0]:.4f},{g_xy[1]:.4f}) B({b_xy[0]:.4f},{b_xy[1]:.4f})",
                level="info",
            )
            _draw_measured_triangle(ax_xy, measured_xy, uv_space=False)

        _draw_coverage_box(
            ax_xy, bbox_xy[1] - 0.02, bbox_xy[2] + 0.02, current_ref, xy_coverage,
            dark_mode=dark_mode,
        )

        # 暗化三角形外部区域（黑色半透明遮罩）
        x0, x1 = bbox_xy[0], bbox_xy[1]
        y0, y1 = bbox_xy[2], bbox_xy[3]
        # 多边形路径：外框+三角形
        verts = [
            (x0, y0), (x0, y1), (x1, y1), (x1, y0), (x0, y0),
            *_REF_GAMUTS_XY[current_ref], _REF_GAMUTS_XY[current_ref][0]
        ]
        codes = [Path.MOVETO] + [Path.LINETO]*3 + [Path.CLOSEPOLY]
        codes += [Path.MOVETO] + [Path.LINETO]*2 + [Path.CLOSEPOLY]
        path = Path(verts, codes)
        mask_face = (0, 0, 0, 0.65) if dark_mode else (1, 1, 1, 0.50)
        patch = PathPatch(path, facecolor=mask_face, lw=0, zorder=7)
        ax_xy.add_patch(patch)

        legend = ax_xy.legend(
            loc="upper right", fontsize=8.5,
            framealpha=0.0, edgecolor="#000", fancybox=True,
            labelcolor="#FFF" if dark_mode else "#111"
        )
        legend.set_zorder(200)
        legend.get_frame().set_facecolor("#000" if dark_mode else "#FFFFFF")
        legend.get_frame().set_alpha(0.5 if dark_mode else 0.78)
        legend.get_frame().set_edgecolor("#FFF" if dark_mode else "#333")
        ax_xy.add_artist(legend)

    except Exception as e:
        self.log_gui.log(f"XY 图绘制失败: {str(e)}", level="error")
        import traceback
        self.log_gui.log(traceback.format_exc(), level="error")

    # ============================================================
    # 右图：CIE 1976 u'v'
    # ============================================================
    try:
        bg_uv, bbox_uv = get_cie1976_background()
        _blit_background(ax_uv, bg_uv, bbox_uv)
        _style_axes(
            ax_uv,
            title="CIE 1976 u'v'",
            xlabel="u'", ylabel="v'",
            xlim=(bbox_uv[0], bbox_uv[1]),
            ylim=(bbox_uv[2], bbox_uv[3]),
            dark_mode=dark_mode,
        )

        measured_uv = None
        if measured_xy is not None:
            measured_uv = [_xy_to_uv(x, y) for x, y in measured_xy]
            try:
                uv_coverage = pq_algorithm.calculate_uv_gamut_coverage(
                    [list(uv) for uv in measured_uv], reference=current_ref
                )
                self.log_gui.log(
                    f"UV 空间覆盖率（{current_ref}）: {uv_coverage:.1f}%",
                    level="success",
                )
            except Exception as e:
                self.log_gui.log(f"计算 UV 覆盖率失败: {str(e)}", level="error")
                uv_coverage = 0.0

        for ref_name in other_refs:
            _draw_reference_triangle(
                ax_uv, _ref_gamut_uv(ref_name),
                _REF_COLORS[ref_name],
                is_current=False, label=ref_name,
            )
        _draw_reference_triangle(
            ax_uv, _ref_gamut_uv(current_ref),
            _REF_COLORS[current_ref],
            is_current=True, label=f"{current_ref} (参考)",
        )

        if measured_uv is not None:
            _draw_measured_triangle(ax_uv, measured_uv, uv_space=True)

        _draw_coverage_box(
            ax_uv, bbox_uv[1] - 0.015, bbox_uv[2] + 0.015, current_ref, uv_coverage,
            dark_mode=dark_mode,
        )

        u0, u1 = bbox_uv[0], bbox_uv[1]
        v0, v1 = bbox_uv[2], bbox_uv[3]
        verts = [
            (u0, v0), (u0, v1), (u1, v1), (u1, v0), (u0, v0),
            *_ref_gamut_uv(current_ref), _ref_gamut_uv(current_ref)[0]
        ]
        codes = [Path.MOVETO] + [Path.LINETO]*3 + [Path.CLOSEPOLY]
        codes += [Path.MOVETO] + [Path.LINETO]*2 + [Path.CLOSEPOLY]
        path = Path(verts, codes)
        mask_face = (0, 0, 0, 0.65) if dark_mode else (1, 1, 1, 0.50)
        patch = PathPatch(path, facecolor=mask_face, lw=0, zorder=7)
        ax_uv.add_patch(patch)               
        
        legend_uv = ax_uv.legend(
            loc="upper right", fontsize=8.5,
            framealpha=0.0, edgecolor="#000", fancybox=True,
            labelcolor="#FFF" if dark_mode else "#111"
        )
        legend_uv.set_zorder(200)
        legend_uv.get_frame().set_facecolor("#000" if dark_mode else "#FFFFFF")
        legend_uv.get_frame().set_alpha(0.72 if dark_mode else 0.82)
        legend_uv.get_frame().set_edgecolor("#FFF" if dark_mode else "#333")
        ax_uv.add_artist(legend_uv)

    except Exception as e:
        self.log_gui.log(f"UV 图绘制失败: {str(e)}", level="error")
        import traceback
        self.log_gui.log(traceback.format_exc(), level="error")

    # ========== 总标题 ==========
    test_type_name = self.get_test_type_name(test_type)
    self.gamut_fig.suptitle(
        f"{test_type_name} - 色域测试",
        fontsize=12, y=0.98, fontweight="bold",
    )

    self.gamut_canvas.draw()
    self.chart_notebook.select(self.gamut_chart_frame)

    # 同步工具栏按钮选中状态
    if hasattr(self, "sync_gamut_toolbar"):
        self.sync_gamut_toolbar()

    self.log_gui.log("色域图绘制完成", level="success")


class PlotGamutMixin:
    """由 tools/refactor_to_mixins.py 自动生成。
    把本模块的自由函数挂到 PQAutomationApp 上，便于 F12 跳转与类型推断。
    """
    plot_gamut = plot_gamut