413 lines
14 KiB
Python
413 lines
14 KiB
Python
"""UCD 服务层：SignalService（硬件编排）+ PatternService（测试发图）。"""
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
|
||
from app.ucd.domain import (
|
||
EventBus,
|
||
PatternSpec,
|
||
SignalFormat,
|
||
TimingSpec,
|
||
UcdState,
|
||
image_pattern,
|
||
solid_rgb_pattern,
|
||
)
|
||
from app.ucd.device import IUcdDevice
|
||
|
||
log = logging.getLogger(__name__)
|
||
|
||
|
||
# ─── SignalService ────────────────────────────────────────────────────────
|
||
|
||
|
||
class SignalService:
|
||
"""协调 SignalFormat / Timing / Pattern 的写入与提交。"""
|
||
|
||
def __init__(self, device: IUcdDevice, bus: EventBus):
|
||
self._dev = device
|
||
self._bus = bus
|
||
|
||
# -- 高层接口 ------------------------------------------------
|
||
|
||
def apply(
|
||
self,
|
||
*,
|
||
signal: SignalFormat,
|
||
timing: TimingSpec,
|
||
pattern: PatternSpec,
|
||
) -> bool:
|
||
"""一次性提交信号格式 + timing + 图案。
|
||
|
||
Returns:
|
||
``format_changed``——本次相对上一次 :meth:`apply` 是否变化。
|
||
"""
|
||
log.info(
|
||
"SignalService.apply signal=%s timing=%s pattern=%s",
|
||
signal,
|
||
timing,
|
||
pattern.kind.value,
|
||
)
|
||
changed = self._dev.configure(signal, timing)
|
||
self._dev.set_pattern(pattern)
|
||
self._dev.apply()
|
||
return changed
|
||
|
||
def send_pattern(self, pattern: PatternSpec) -> None:
|
||
"""在已 configure 的信号上仅更新图案后 apply。"""
|
||
log.info("SignalService.send_pattern pattern=%s", pattern.kind.value)
|
||
self._dev.set_pattern(pattern)
|
||
self._dev.apply()
|
||
|
||
def send_solid_rgb(self, rgb: tuple[int, int, int] | list[int]) -> None:
|
||
self.send_pattern(solid_rgb_pattern(rgb))
|
||
|
||
def send_image(self, path: str) -> None:
|
||
self.send_pattern(image_pattern(path))
|
||
|
||
def update_signal_format(
|
||
self,
|
||
*,
|
||
color_space: str,
|
||
output_format: str,
|
||
bit_depth: str,
|
||
data_range: str = "Full",
|
||
max_cll: int | None = None,
|
||
max_fall: int | None = None,
|
||
) -> bool:
|
||
"""仅将信号格式提交到 SDK（沿用上一次的 timing），不切换图案。
|
||
|
||
UI 字符串先经域层解析做参数校验；解析失败抛 :class:`UcdConfigError`。
|
||
"""
|
||
_ = build_signal_format(
|
||
color_space=color_space,
|
||
output_format=output_format,
|
||
bit_depth=bit_depth,
|
||
data_range=data_range,
|
||
)
|
||
return self._dev.apply_signal_format(
|
||
color_space=color_space,
|
||
color_format=output_format,
|
||
bit_depth=bit_depth,
|
||
data_range=data_range,
|
||
max_cll=max_cll,
|
||
max_fall=max_fall,
|
||
)
|
||
|
||
# -- 透传给上层的查询 ---------------------------------------
|
||
|
||
@property
|
||
def device(self) -> IUcdDevice:
|
||
return self._dev
|
||
|
||
def current_resolution(self) -> tuple[int, int]:
|
||
return self._dev.current_resolution()
|
||
|
||
@property
|
||
def is_connected(self) -> bool:
|
||
"""UCD 设备是否已打开。供 GUI 做前置校验。"""
|
||
return self._dev.state != UcdState.CLOSED
|
||
|
||
@property
|
||
def format_changed(self) -> bool:
|
||
"""最近一次视频模式提交是否相对上次发生变化。"""
|
||
return self._dev.format_changed
|
||
|
||
@property
|
||
def last_error(self) -> str | None:
|
||
return self._dev.last_error
|
||
|
||
def apply_config(self, config) -> bool:
|
||
"""按 :class:`PQConfig` 写入色彩 / Timing / 当前 Pattern（不 apply 输出）。"""
|
||
return bool(self._dev.set_ucd_params(config))
|
||
|
||
def send_pattern_params(self, params) -> bool:
|
||
"""以 ``params`` 更新当前 pattern 的参数并 apply。"""
|
||
return bool(self._dev.send_current_pattern_params(params))
|
||
|
||
def apply_and_run(self, config, pattern_params) -> bool:
|
||
"""``set_ucd_params`` + ``set_pattern`` + ``run`` 的复合操作。"""
|
||
return bool(self._dev.apply_config_and_run(config, pattern_params))
|
||
|
||
def stage_test_profile(
|
||
self,
|
||
config,
|
||
*,
|
||
color_space: str,
|
||
output_format: str,
|
||
bit_depth: str,
|
||
data_range: str = "Full",
|
||
max_cll: int | None = None,
|
||
max_fall: int | None = None,
|
||
) -> bool:
|
||
"""按 PQConfig stage 色彩/Timing/Pattern 类型，并提交 UI 覆盖的信号格式。
|
||
|
||
自动化测试在发图前调用；等价于 ``apply_config`` + ``update_signal_format``。
|
||
"""
|
||
if not self.apply_config(config):
|
||
return False
|
||
return self.update_signal_format(
|
||
color_space=color_space,
|
||
output_format=output_format,
|
||
bit_depth=bit_depth,
|
||
data_range=data_range,
|
||
max_cll=max_cll,
|
||
max_fall=max_fall,
|
||
)
|
||
|
||
|
||
__all__ = [
|
||
"SignalService",
|
||
"build_signal_format",
|
||
"build_signal_format_from_profile",
|
||
"build_timing",
|
||
"solid_rgb_pattern",
|
||
"image_pattern",
|
||
"SignalFormat",
|
||
"TimingSpec",
|
||
"PatternSpec",
|
||
"PatternKind",
|
||
"Colorimetry",
|
||
"DynamicRange",
|
||
"UcdError",
|
||
]
|
||
|
||
|
||
# --- PatternService ---
|
||
|
||
import copy
|
||
from dataclasses import dataclass
|
||
|
||
from app.data_range_converter import convert_pattern_params
|
||
from app.pq.pq_config import get_pattern
|
||
|
||
|
||
@dataclass
|
||
class PatternSession:
|
||
mode: str
|
||
test_type: str
|
||
active_config: object
|
||
pattern_params: list[list[int]]
|
||
total_patterns: int
|
||
display_names: list[str]
|
||
|
||
|
||
class PatternService:
|
||
def __init__(self, app):
|
||
self.app = app
|
||
|
||
def _build_apply_config_error(self, test_type):
|
||
timing = self.app.config.current_test_types.get(test_type, {}).get("timing", "-")
|
||
detail = ""
|
||
err = self.app.signal_service.last_error
|
||
if err:
|
||
detail = f", detail={err}"
|
||
return f"UCD profile apply_config failed for {test_type}, timing={timing}{detail}"
|
||
|
||
def _stage_profile(
|
||
self,
|
||
active_config,
|
||
*,
|
||
color_space: str,
|
||
output_format: str,
|
||
bit_depth: str,
|
||
data_range: str = "Full",
|
||
max_cll: int | None = None,
|
||
max_fall: int | None = None,
|
||
test_type: str,
|
||
log_details: bool = False,
|
||
log_title: str = "",
|
||
log_fields: list[tuple[str, str]] | None = None,
|
||
) -> None:
|
||
if log_details and log_title:
|
||
self._log("=" * 50, "separator")
|
||
self._log(log_title, "info")
|
||
self._log("=" * 50, "separator")
|
||
for label, value in log_fields or []:
|
||
self._log(f" {label}: {value}", "info")
|
||
|
||
if not self.app.signal_service.stage_test_profile(
|
||
active_config,
|
||
color_space=color_space,
|
||
output_format=output_format,
|
||
bit_depth=bit_depth,
|
||
data_range=data_range,
|
||
max_cll=max_cll,
|
||
max_fall=max_fall,
|
||
):
|
||
raise RuntimeError(self._build_apply_config_error(test_type))
|
||
|
||
if log_details:
|
||
self._log("信号格式设置成功", "success")
|
||
|
||
def prepare_session(self, mode, *, test_type=None, log_details=False):
|
||
test_type = test_type or self.app.config.current_test_type
|
||
if hasattr(self.app.config, "set_current_test_type"):
|
||
self.app.config.set_current_test_type(test_type)
|
||
if not self.app.config.set_current_pattern(mode):
|
||
raise ValueError(f"未知的图案模式: {mode}")
|
||
|
||
active_config = self.app.config
|
||
source_params = self._get_source_pattern_params(mode)
|
||
|
||
if test_type == "screen_module":
|
||
screen_cfg = self.app.config.current_test_types.get("screen_module", {})
|
||
color_space = (
|
||
self.app.screen_module_color_space_var.get()
|
||
if hasattr(self.app, "screen_module_color_space_var")
|
||
else screen_cfg.get("colorimetry", "sRGB")
|
||
)
|
||
data_range = (
|
||
self.app.screen_module_data_range_var.get()
|
||
if hasattr(self.app, "screen_module_data_range_var")
|
||
else screen_cfg.get("data_range", "Full")
|
||
)
|
||
bit_depth = (
|
||
self.app.screen_module_bit_depth_var.get()
|
||
if hasattr(self.app, "screen_module_bit_depth_var")
|
||
else f"{int(screen_cfg.get('bpc', 8))}bit"
|
||
)
|
||
output_format = (
|
||
self.app.screen_module_output_format_var.get()
|
||
if hasattr(self.app, "screen_module_output_format_var")
|
||
else screen_cfg.get("color_format", "RGB")
|
||
)
|
||
|
||
self._stage_profile(
|
||
active_config,
|
||
color_space=color_space,
|
||
data_range=data_range,
|
||
bit_depth=bit_depth,
|
||
output_format=output_format,
|
||
test_type=test_type,
|
||
log_details=log_details,
|
||
log_title="设置屏模组信号格式:",
|
||
log_fields=[
|
||
("色彩空间", color_space),
|
||
("色彩格式", output_format),
|
||
("数据范围", data_range),
|
||
("编码位深", bit_depth),
|
||
("Timing", self.app.config.current_test_types[test_type]["timing"]),
|
||
],
|
||
)
|
||
|
||
elif test_type == "sdr_movie":
|
||
data_range = self.app.sdr_data_range_var.get()
|
||
converted_params = convert_pattern_params(
|
||
source_params, data_range=data_range, verbose=False
|
||
)
|
||
active_config = self.app.config.get_temp_config_with_converted_params(
|
||
mode=mode, converted_params=converted_params
|
||
)
|
||
if hasattr(active_config, "set_current_test_type"):
|
||
active_config.set_current_test_type(test_type)
|
||
|
||
self._stage_profile(
|
||
active_config,
|
||
color_space=self.app.sdr_color_space_var.get(),
|
||
data_range=data_range,
|
||
bit_depth=self.app.sdr_bit_depth_var.get(),
|
||
output_format=self.app.sdr_output_format_var.get(),
|
||
test_type=test_type,
|
||
log_details=log_details,
|
||
log_title="设置 SDR 信号格式:",
|
||
log_fields=[
|
||
("色彩空间", self.app.sdr_color_space_var.get()),
|
||
("色彩格式", self.app.sdr_output_format_var.get()),
|
||
("Gamma", self.app.sdr_gamma_type_var.get()),
|
||
("数据范围", data_range),
|
||
("编码位深", self.app.sdr_bit_depth_var.get()),
|
||
],
|
||
)
|
||
if log_details:
|
||
self._log(f"图案参数已设置，共 {len(converted_params)} 个图案", "success")
|
||
|
||
elif test_type == "hdr_movie":
|
||
data_range = self.app.hdr_data_range_var.get()
|
||
converted_params = convert_pattern_params(
|
||
source_params, data_range=data_range, verbose=False
|
||
)
|
||
active_config = self.app.config.get_temp_config_with_converted_params(
|
||
mode=mode, converted_params=converted_params
|
||
)
|
||
if hasattr(active_config, "set_current_test_type"):
|
||
active_config.set_current_test_type(test_type)
|
||
|
||
self._stage_profile(
|
||
active_config,
|
||
color_space=self.app.hdr_color_space_var.get(),
|
||
data_range=data_range,
|
||
bit_depth=self.app.hdr_bit_depth_var.get(),
|
||
output_format=self.app.hdr_output_format_var.get(),
|
||
max_cll=self.app.hdr_maxcll_var.get(),
|
||
max_fall=self.app.hdr_maxfall_var.get(),
|
||
test_type=test_type,
|
||
log_details=log_details,
|
||
log_title="设置 HDR 信号格式:",
|
||
log_fields=[
|
||
("色彩空间", self.app.hdr_color_space_var.get()),
|
||
("色彩格式", self.app.hdr_output_format_var.get()),
|
||
("数据范围", data_range),
|
||
("编码位深", self.app.hdr_bit_depth_var.get()),
|
||
("MaxCLL", self.app.hdr_maxcll_var.get()),
|
||
("MaxFALL", self.app.hdr_maxfall_var.get()),
|
||
],
|
||
)
|
||
if log_details:
|
||
self._log(f"图案参数已设置，共 {len(converted_params)} 个图案", "success")
|
||
|
||
else:
|
||
raise ValueError(f"不支持的测试类型: {test_type}")
|
||
|
||
pattern_params = copy.deepcopy(active_config.current_pattern["pattern_params"])
|
||
return PatternSession(
|
||
mode=mode,
|
||
test_type=test_type,
|
||
active_config=active_config,
|
||
pattern_params=pattern_params,
|
||
total_patterns=len(pattern_params),
|
||
display_names=self._get_display_names(mode, len(pattern_params)),
|
||
)
|
||
|
||
def send_session_pattern(self, session, index):
|
||
if index < 0 or index >= session.total_patterns:
|
||
raise IndexError(f"pattern 索引越界: {index}")
|
||
|
||
pattern_param = session.pattern_params[index]
|
||
if not self.app.signal_service.send_pattern_params(pattern_param):
|
||
raise RuntimeError(f"发送 pattern 失败: {index}")
|
||
return pattern_param
|
||
|
||
def send_rgb(self, rgb, *, session=None, test_type=None):
|
||
active_session = session or self.prepare_session(
|
||
"rgb",
|
||
test_type=test_type,
|
||
log_details=False,
|
||
)
|
||
converted_rgb = self._convert_rgb_for_test_type(rgb, active_session.test_type)
|
||
self.app.signal_service.send_solid_rgb(converted_rgb)
|
||
return True
|
||
|
||
def _get_source_pattern_params(self, mode):
|
||
return copy.deepcopy(get_pattern(mode)["pattern_params"])
|
||
|
||
def _get_display_names(self, mode, total_patterns):
|
||
if mode == "accuracy":
|
||
return self.app.config.get_accuracy_color_names()
|
||
if mode == "custom" and hasattr(self.app.config, "get_temp_pattern_names"):
|
||
return self.app.config.get_temp_pattern_names()
|
||
return [f"P {index + 1}" for index in range(total_patterns)]
|
||
|
||
def _convert_rgb_for_test_type(self, rgb, test_type):
|
||
if test_type == "sdr_movie":
|
||
data_range = self.app.sdr_data_range_var.get()
|
||
elif test_type == "hdr_movie":
|
||
data_range = self.app.hdr_data_range_var.get()
|
||
else:
|
||
data_range = "Full"
|
||
|
||
return convert_pattern_params([list(rgb)], data_range=data_range, verbose=False)[0]
|
||
|
||
def _log(self, message, level):
|
||
if hasattr(self.app, "log_gui"):
|
||
self.app.log_gui.log(message, level=level)
|