"""色准测试结果绘制。 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)