pqAutomationApp/app/plots/plot_accuracy.py

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

Step 2 重构：从 pqAutomationApp.PQAutomationApp.plot_accuracy 原样搬迁。
"""

from matplotlib.patches import Rectangle


def plot_accuracy(self, accuracy_data, test_type):
    """绘制色准测试结果 - 29色显示 - 简洁版布局（显示 Gamma）"""

    self.accuracy_ax.clear()
    self.accuracy_ax.set_xlim(0, 1)
    self.accuracy_ax.set_ylim(0, 1)
    self.accuracy_ax.axis("off")

    self.accuracy_fig.subplots_adjust(
        left=0.05,
        right=0.95,
        top=0.95,
        bottom=0.02,
    )

    # 获取色准数据
    color_patches = accuracy_data.get("color_patches", [])
    delta_e_values = accuracy_data.get("delta_e_values", [])
    avg_delta_e = accuracy_data.get("avg_delta_e", 0)
    max_delta_e = accuracy_data.get("max_delta_e", 0)
    min_delta_e = accuracy_data.get("min_delta_e", 0)
    excellent_count = accuracy_data.get("excellent_count", 0)
    good_count = accuracy_data.get("good_count", 0)
    poor_count = accuracy_data.get("poor_count", 0)

    # 获取 Gamma 值
    target_gamma = accuracy_data.get("target_gamma", 2.2)

    test_type_name = self.get_test_type_name(test_type)

    # ========== 标题（动态显示 Gamma）==========
    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:  # screen_module
        title = f"{test_type_name} - 色准测试（全 29色 | Gamma {target_gamma}）"

    self.accuracy_fig.suptitle(
        title,
        fontsize=11,
        y=0.98,
        fontweight="bold",
    )

    # ========== 29色：6行5列布局 ==========
    cols = 5
    rows = 6

    patch_width = 0.135
    patch_height = 0.085
    x_start = 0.08
    y_start = 0.90
    x_gap = 0.035
    y_gap = 0.050

    # ========== 绘制色块 ==========
    for i, (color_name, delta_e) in enumerate(zip(color_patches, delta_e_values)):
        row = i // cols
        col = i % cols

        x = x_start + col * (patch_width + x_gap)
        y = y_start - row * (patch_height + y_gap)

        # 颜色映射
        color_map = {
            # 灰阶
            "White": "#FFFFFF",
            "Gray 80": "#E6E6E6",
            "Gray 65": "#D1D1D1",
            "Gray 50": "#BABABA",
            "Gray 35": "#9E9E9E",
            # 饱和色
            "100% Red": "#FF0000",
            "100% Green": "#00FF00",
            "100% Blue": "#0000FF",
            "100% Cyan": "#00FFFF",
            "100% Magenta": "#FF00FF",
            "100% Yellow": "#FFFF00",
            # ColorChecker 颜色
            "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",
        }

        patch_color = color_map.get(color_name, "#808080")

        # ΔE 等级颜色
        if delta_e < 3:
            edge_color = "green"
        elif delta_e < 5:
            edge_color = "orange"
        else:
            edge_color = "red"

        # 绘制色块
        rect = Rectangle(
            (x, y),
            patch_width,
            patch_height,
            transform=self.accuracy_ax.transAxes,
            facecolor=patch_color,
            edgecolor=edge_color,
            linewidth=1.8,
        )
        self.accuracy_ax.add_patch(rect)

        # ========== 标注色块名称（上方）==========
        self.accuracy_ax.text(
            x + patch_width / 2,
            y + patch_height + 0.015,
            color_name,
            ha="center",
            va="bottom",
            fontsize=5.5,
            fontweight="bold",
            transform=self.accuracy_ax.transAxes,
            clip_on=False,
        )

        # ========== 标注 ΔE 值（中心）==========
        dark_colors = [
            "100% Red",
            "100% Green",
            "100% Blue",
            "Gray 35",
            "Dark Skin",
            "Foliage",
            "Purple",
            "Purplish Blue",
            "Blue (Legacy)",
            "Green (Legacy)",
            "Red (Legacy)",
            "Magenta (Legacy)",
            "Cyan (Legacy)",
        ]

        text_color = "white" if color_name in dark_colors else "black"

        self.accuracy_ax.text(
            x + patch_width / 2,
            y + patch_height / 2,
            f"ΔE\n{delta_e:.2f}",
            ha="center",
            va="center",
            fontsize=5.2,
            fontweight="bold",
            color=text_color,
            transform=self.accuracy_ax.transAxes,
            bbox=dict(
                boxstyle="round,pad=0.22",
                facecolor="white" if text_color == "black" else "black",
                alpha=0.75,
                edgecolor=edge_color,
                linewidth=1.0,
            ),
        )

    # ========== 统计信息卡片（只保留外框）==========
    card_width = 0.84
    card_height = 0.15
    card_x = 0.08
    card_y = 0.01

    info_card = Rectangle(
        (card_x, card_y),
        card_width,
        card_height,
        transform=self.accuracy_ax.transAxes,
        facecolor="#F0F0F0",
        edgecolor="black",
        linewidth=1.5,
    )
    self.accuracy_ax.add_patch(info_card)

    # ========== 标题（带说明）==========
    self.accuracy_ax.text(
        card_x + card_width / 2,
        card_y + card_height - 0.008,
        "色准统计（5灰阶 + 18 ColorChecker + 6饱和色 | ΔE 2000 标准）",
        ha="center",
        va="top",
        fontsize=7.5,
        fontweight="bold",
        transform=self.accuracy_ax.transAxes,
    )

    # ========== 统计内容（无内部框）==========
    stats_y = card_y + card_height * 0.55

    # 左侧：ΔE 统计
    left_x = card_x + 0.02
    stats_text = [
        f"平均 ΔE: {avg_delta_e:.2f}",
        f"最大 ΔE: {max_delta_e:.2f}",
        f"最小 ΔE: {min_delta_e:.2f}",
    ]

    for i, text in enumerate(stats_text):
        self.accuracy_ax.text(
            left_x,
            stats_y - i * 0.030,
            text,
            ha="left",
            va="center",
            fontsize=7,
            fontweight="bold",
            transform=self.accuracy_ax.transAxes,
        )

    # 中间：色块统计
    middle_x = card_x + card_width * 0.32

    self.accuracy_ax.text(
        middle_x,
        stats_y,
        f"优秀 (ΔE<3): {excellent_count} 个",
        ha="left",
        va="center",
        fontsize=7,
        color="green",
        fontweight="bold",
        transform=self.accuracy_ax.transAxes,
    )

    self.accuracy_ax.text(
        middle_x,
        stats_y - 0.030,
        f"良好 (3≤ΔE<5): {good_count} 个",
        ha="left",
        va="center",
        fontsize=7,
        color="orange",
        fontweight="bold",
        transform=self.accuracy_ax.transAxes,
    )

    self.accuracy_ax.text(
        middle_x,
        stats_y - 0.060,
        f"偏差 (ΔE≥5): {poor_count} 个",
        ha="left",
        va="center",
        fontsize=7,
        color="red",
        fontweight="bold",
        transform=self.accuracy_ax.transAxes,
    )

    # 右侧：总体评价
    right_x = card_x + card_width - 0.02

    if avg_delta_e < 2:
        grade = "专业级"
        grade_icon = "★★★"
        grade_color = "darkgreen"
    elif avg_delta_e < 3:
        grade = "优秀"
        grade_icon = "✓✓"
        grade_color = "green"
    elif avg_delta_e < 5:
        grade = "良好"
        grade_icon = "✓"
        grade_color = "orange"
    else:
        grade = "需要校准"
        grade_icon = "✗"
        grade_color = "red"

    self.accuracy_ax.text(
        right_x,
        stats_y + 0.020,
        "总体评价:",
        ha="right",
        va="bottom",
        fontsize=7,
        fontweight="bold",
        transform=self.accuracy_ax.transAxes,
    )

    self.accuracy_ax.text(
        right_x,
        stats_y - 0.025,
        f"{grade} {grade_icon}",
        ha="right",
        va="top",
        fontsize=11,
        fontweight="bold",
        color=grade_color,
        transform=self.accuracy_ax.transAxes,
    )

    self.accuracy_canvas.draw()
    self.chart_notebook.select(self.accuracy_chart_frame)