586 lines
19 KiB
Python
586 lines
19 KiB
Python
# -*- coding: UTF-8 -*-
|
||
import UniTAP
|
||
import time
|
||
import gc
|
||
from drivers.UCD323_Enum import UCDEnum
|
||
|
||
|
||
class UCDController:
|
||
"""UCD323信号发生器控制类"""
|
||
|
||
def __init__(self):
|
||
self.lUniTAP = UniTAP.TsiLib()
|
||
self.dev = None
|
||
self.role = None
|
||
self.timing_manager = None
|
||
self.config = None
|
||
self.color_info = None
|
||
self.status = False
|
||
self.current_interface = "HDMI"
|
||
|
||
self.current_timing = None
|
||
self.current_pattern = None
|
||
self.current_pattern_param = None
|
||
self.current_pattern_params = None
|
||
self.current_pattern_index = 0
|
||
|
||
def search_device(self):
|
||
"""搜索可用设备"""
|
||
available_devices = self.lUniTAP.get_list_of_available_devices()
|
||
return available_devices if available_devices else []
|
||
|
||
def open(self, device_name):
|
||
"""打开设备"""
|
||
temp_dev = None
|
||
|
||
try:
|
||
if self.dev is not None or self.status:
|
||
self._force_cleanup()
|
||
|
||
device_id = int(device_name.split(":")[0])
|
||
temp_dev = self.lUniTAP.open(device_id)
|
||
|
||
try:
|
||
self.role = temp_dev.select_role(UniTAP.dev.UCD323.HDMISource)
|
||
self.dev = temp_dev
|
||
self.current_interface = "HDMI"
|
||
|
||
except Exception as role_error:
|
||
self._close_device_object(temp_dev)
|
||
raise role_error
|
||
|
||
pg, _ = self.get_tx_modules()
|
||
self.timing_manager = pg.timing_manager
|
||
self.color_info = UniTAP.ColorInfo()
|
||
self.status = True
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
self._force_cleanup()
|
||
return False
|
||
|
||
def _reset_state(self):
|
||
"""重置所有运行时状态（不关闭设备句柄）"""
|
||
self.dev = None
|
||
self.role = None
|
||
self.status = False
|
||
self.timing_manager = None
|
||
self.current_timing = None
|
||
self.current_pattern = None
|
||
self.current_pattern_param = None
|
||
self.current_pattern_params = None
|
||
self.current_pattern_index = 0
|
||
self.current_interface = "HDMI"
|
||
|
||
def close(self):
|
||
"""关闭设备"""
|
||
try:
|
||
if self.dev:
|
||
self._close_device_object(self.dev)
|
||
|
||
self._reset_state()
|
||
self.lUniTAP = None
|
||
|
||
for i in range(3):
|
||
gc.collect()
|
||
|
||
time.sleep(2.0)
|
||
|
||
self.lUniTAP = UniTAP.TsiLib()
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
self._reset_state()
|
||
|
||
try:
|
||
self.lUniTAP = None
|
||
gc.collect()
|
||
time.sleep(2.0)
|
||
self.lUniTAP = UniTAP.TsiLib()
|
||
except Exception as init_error:
|
||
pass
|
||
|
||
return False
|
||
|
||
def _close_device_object(self, dev_obj):
|
||
"""显式关闭设备对象"""
|
||
try:
|
||
if dev_obj is None:
|
||
return
|
||
|
||
if self.lUniTAP and hasattr(self.lUniTAP, "close"):
|
||
try:
|
||
self.lUniTAP.close(dev_obj)
|
||
except Exception as e:
|
||
pass
|
||
|
||
dev_obj = None
|
||
gc.collect()
|
||
time.sleep(1.0)
|
||
|
||
except Exception as e:
|
||
pass
|
||
|
||
def _force_cleanup(self):
|
||
"""强制清理所有状态"""
|
||
try:
|
||
if self.dev:
|
||
self._close_device_object(self.dev)
|
||
|
||
self._reset_state()
|
||
|
||
except Exception as e:
|
||
pass
|
||
|
||
def get_tx_modules(self):
|
||
"""根据当前接口返回 (pg, ag) 模块。"""
|
||
if not self.role:
|
||
raise RuntimeError("UCD 未打开，无法获取 TX 模块")
|
||
|
||
interface = getattr(self, "current_interface", None)
|
||
if interface in (None, "HDMI"):
|
||
return self.role.hdtx.pg, self.role.hdtx.ag
|
||
if interface in ("DP", "Type-C"):
|
||
return self.role.dptx.pg, self.role.dptx.ag
|
||
raise ValueError(f"不支持的接口类型: {interface}")
|
||
|
||
def _resolve_timing(self, pg=None):
|
||
"""优先从 current_timing 读取 timing，必要时回退到 TX 模块。"""
|
||
if self.current_timing is not None:
|
||
return self.current_timing
|
||
|
||
if pg is not None:
|
||
get_vm = getattr(pg, "get_vm", None)
|
||
if callable(get_vm):
|
||
try:
|
||
vm = get_vm()
|
||
return getattr(vm, "timing", None)
|
||
except Exception:
|
||
pass
|
||
|
||
return None
|
||
|
||
def get_current_resolution(self, default=(3840, 2160)):
|
||
"""从当前 timing 获取 (width, height)，失败时返回 default。"""
|
||
try:
|
||
pg, _ = self.get_tx_modules()
|
||
timing = self._resolve_timing(pg)
|
||
if timing and hasattr(timing, "h_active") and hasattr(timing, "v_active"):
|
||
return timing.h_active, timing.v_active
|
||
except Exception:
|
||
pass
|
||
|
||
return default
|
||
|
||
def _cleanup(self):
|
||
"""清理设备资源"""
|
||
try:
|
||
if self.dev:
|
||
self._close_device_object(self.dev)
|
||
self.dev = None
|
||
|
||
if hasattr(self.lUniTAP, "cleanup"):
|
||
self.lUniTAP.cleanup()
|
||
|
||
except Exception as e:
|
||
pass
|
||
|
||
def set_ucd_params(self, config):
|
||
"""设置UCD323参数"""
|
||
self.config = config
|
||
test_type = self.config.current_test_type
|
||
|
||
pg, _ = self.get_tx_modules()
|
||
self.timing_manager = pg.timing_manager
|
||
|
||
color_format = self.config.current_test_types[test_type]["color_format"]
|
||
bpc = self.config.current_test_types[test_type]["bpc"]
|
||
colorimetry = self.config.current_test_types[test_type]["colorimetry"]
|
||
|
||
if not self.set_color_mode(color_format, bpc, colorimetry):
|
||
return False
|
||
|
||
timing_str = self.config.current_test_types[test_type]["timing"]
|
||
self.set_timing_from_string(timing_str)
|
||
|
||
self.current_pattern_index = 0
|
||
pattern_mode = self.config.current_pattern["pattern_mode"]
|
||
pattern = UCDEnum.VideoPatternInfo.get_video_pattern(pattern_mode)
|
||
|
||
if pattern is None:
|
||
return False
|
||
|
||
self.current_pattern = pattern
|
||
self.current_pattern_params = self.config.current_pattern["pattern_params"]
|
||
|
||
return True
|
||
|
||
def run(self):
|
||
"""运行设备"""
|
||
self.apply_video_mode()
|
||
self.apply_pattern()
|
||
pg, _ = self.get_tx_modules()
|
||
pg.apply()
|
||
return True
|
||
|
||
def send_image_pattern(self, image_path):
|
||
"""发送图片 Pattern（依赖当前 timing/color_info 状态）。"""
|
||
if not self.status or not self.role:
|
||
return False
|
||
|
||
try:
|
||
pg, _ = self.get_tx_modules()
|
||
self.apply_video_mode()
|
||
pg.set_pattern(pattern=image_path)
|
||
pg.apply()
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def send_solid_rgb_pattern(self, rgb):
|
||
"""发送纯色 RGB Pattern（依赖当前 timing/color_info 状态）。"""
|
||
if not self.status or not self.role:
|
||
return False
|
||
|
||
try:
|
||
self.current_pattern = UCDEnum.VideoPatternInfo.get_video_pattern("solidcolor")
|
||
if self.current_pattern is None:
|
||
return False
|
||
|
||
return self.send_current_pattern_params(list(rgb))
|
||
except Exception:
|
||
return False
|
||
|
||
def send_current_pattern_params(self, pattern_params):
|
||
"""发送当前已配置的 pattern，并可附带当前 pattern 参数。"""
|
||
if not self.status or not self.role:
|
||
return False
|
||
|
||
try:
|
||
if self.current_pattern is None:
|
||
return False
|
||
|
||
if pattern_params is not None and not self.set_pattern(
|
||
self.current_pattern,
|
||
pattern_params,
|
||
):
|
||
return False
|
||
|
||
self.run()
|
||
return True
|
||
except Exception:
|
||
return False
|
||
|
||
def set_color_mode(self, cf, bpc, cm):
|
||
"""设置颜色模式"""
|
||
current_dynamic_range = self.color_info.dynamic_range
|
||
|
||
color_format = UCDEnum.ColorInfo.get_color_format(cf)
|
||
if color_format is None:
|
||
return False
|
||
|
||
if not isinstance(bpc, int) or bpc <= 0:
|
||
return False
|
||
|
||
colorimetry = UCDEnum.ColorInfo.get_colorimetry(cm)
|
||
if colorimetry is None:
|
||
return False
|
||
|
||
self.color_info.color_format = color_format
|
||
self.color_info.bpc = bpc
|
||
self.color_info.colorimetry = colorimetry
|
||
self.color_info.dynamic_range = current_dynamic_range
|
||
|
||
return True
|
||
|
||
def apply_video_mode(self):
|
||
"""应用当前 color_info 和 timing"""
|
||
if self.current_timing:
|
||
self.set_video_mode()
|
||
return True
|
||
return False
|
||
|
||
def set_video_mode(self):
|
||
"""设置视频模式"""
|
||
# 对比上次发出的配置，判断是否会触发电视重新锁定信号
|
||
current_config = (
|
||
self.current_timing,
|
||
self.color_info.color_format,
|
||
self.color_info.colorimetry,
|
||
self.color_info.dynamic_range,
|
||
self.color_info.bpc,
|
||
)
|
||
self.format_changed = (current_config != getattr(self, "_last_sent_config", None))
|
||
video_mode = UniTAP.VideoMode(
|
||
timing=self.current_timing, color_info=self.color_info
|
||
)
|
||
pg, _ = self.get_tx_modules()
|
||
pg.set_vm(vm=video_mode)
|
||
self._last_sent_config = current_config
|
||
return True
|
||
|
||
def set_pattern(self, pattern, pattern_params=None):
|
||
"""设置pattern"""
|
||
if self.current_timing:
|
||
needs_params = {
|
||
UCDEnum.VideoPatternInfo.VideoPatternParams.SolidColor,
|
||
UCDEnum.VideoPatternInfo.VideoPatternParams.WhiteVStrips,
|
||
UCDEnum.VideoPatternInfo.VideoPatternParams.GradientRGBStripes,
|
||
UCDEnum.VideoPatternInfo.VideoPatternParams.MotionPattern,
|
||
UCDEnum.VideoPatternInfo.VideoPatternParams.SquareWindow,
|
||
}
|
||
if pattern in needs_params and pattern_params:
|
||
self.set_pattern_params(pattern, pattern_params)
|
||
return True
|
||
return False
|
||
|
||
def set_next_pattern(self):
|
||
"""设置下一个pattern"""
|
||
if self.current_pattern_index < len(self.current_pattern_params):
|
||
p = self.current_pattern_params[self.current_pattern_index]
|
||
self.set_pattern(self.current_pattern, p)
|
||
self.current_pattern_index += 1
|
||
else:
|
||
error_msg = (
|
||
f"No more patterns to set. (已设置 {self.current_pattern_index} 个图案)"
|
||
)
|
||
raise IndexError(error_msg)
|
||
|
||
def set_pattern_params(self, pattern, pattern_params):
|
||
"""设置pattern参数"""
|
||
if pattern:
|
||
if pattern == UCDEnum.VideoPatternInfo.VideoPatternParams.SolidColor:
|
||
self.current_pattern_param = UniTAP.SolidColorParams(
|
||
first=pattern_params[0],
|
||
second=pattern_params[1],
|
||
third=pattern_params[2],
|
||
)
|
||
return True
|
||
return False
|
||
|
||
def apply_pattern(self):
|
||
"""应用当前pattern"""
|
||
if self.current_pattern:
|
||
pg, _ = self.get_tx_modules()
|
||
pg.set_pattern(self.current_pattern)
|
||
|
||
if self.current_pattern_param:
|
||
pg.set_pattern_params(self.current_pattern_param)
|
||
return True
|
||
return False
|
||
|
||
def search_timing(self, width, height, refresh_rate, resolution_type=None):
|
||
"""根据分辨率参数搜索合适的timing"""
|
||
if resolution_type:
|
||
resolution_type = resolution_type.lower()
|
||
standard = None
|
||
if resolution_type == "dmt":
|
||
standard = UniTAP.common.timing.Timing.Standard.SD_DMT
|
||
elif resolution_type == "cta":
|
||
standard = UniTAP.common.timing.Timing.Standard.SD_CTA
|
||
elif resolution_type == "cvt":
|
||
standard = UniTAP.common.timing.Timing.Standard.SD_CVT
|
||
|
||
timing = self.timing_manager.search(
|
||
h_active=width,
|
||
v_active=height,
|
||
f_rate=int(refresh_rate) * 1000,
|
||
standard=standard,
|
||
)
|
||
|
||
if timing:
|
||
return timing
|
||
else:
|
||
for res_type in ["dmt", "cta", "cvt", "ovt"]:
|
||
result = self.search_timing(width, height, refresh_rate, res_type)
|
||
if result:
|
||
return result
|
||
|
||
return None
|
||
|
||
def parse_formatted_timing(self, timing_str):
|
||
"""解析格式化的timing字符串"""
|
||
if not isinstance(timing_str, str):
|
||
raise ValueError("timing_str 必须是字符串")
|
||
|
||
s = " ".join(timing_str.strip().split())
|
||
s = s.replace(" x", "x").replace("x ", "x")
|
||
|
||
parts = s.split(" ", 1)
|
||
if len(parts) < 2:
|
||
raise ValueError(f"无法解析timing: {timing_str}")
|
||
type_str = parts[0].strip().upper()
|
||
rest = parts[1].strip()
|
||
|
||
if "@" not in rest:
|
||
raise ValueError(f"无法解析timing(缺少 @): {timing_str}")
|
||
left, right = [p.strip() for p in rest.split("@", 1)]
|
||
|
||
if "x" not in left:
|
||
raise ValueError(f"无法解析分辨率(缺少 x): {timing_str}")
|
||
wh = left.split("x")
|
||
if len(wh) != 2:
|
||
raise ValueError(f"无法解析分辨率: {timing_str}")
|
||
try:
|
||
width = int(wh[0])
|
||
height = int(wh[1])
|
||
except Exception:
|
||
raise ValueError(f"分辨率数字解析失败: {timing_str}")
|
||
|
||
hz_str = right.replace("Hz", "").replace("HZ", "").strip()
|
||
try:
|
||
refresh_rate = float(hz_str)
|
||
except Exception:
|
||
raise ValueError(f"刷新率解析失败: {timing_str}")
|
||
|
||
rtype_map = {
|
||
"DMT": "dmt",
|
||
"CTA": "cta",
|
||
"CVT": "cvt",
|
||
"OVT": "ovt",
|
||
}
|
||
if type_str not in rtype_map:
|
||
raise ValueError(f"未知的分辨率类型: {type_str}")
|
||
resolution_type = rtype_map[type_str]
|
||
|
||
def find_best_id_in_dict(res_map):
|
||
best_id, best_diff = None, float("inf")
|
||
for rid, info in res_map.items():
|
||
if info["width"] == width and info["height"] == height:
|
||
diff = abs(float(info["refresh_rate"]) - refresh_rate)
|
||
if diff < best_diff:
|
||
best_diff = diff
|
||
best_id = rid
|
||
return best_id if best_diff <= 1.0 else None
|
||
|
||
def find_best_id_in_list_map(res_map):
|
||
best_id, best_diff = None, float("inf")
|
||
for rid, infos in res_map.items():
|
||
for info in infos:
|
||
if info["width"] == width and info["height"] == height:
|
||
diff = abs(float(info["refresh_rate"]) - refresh_rate)
|
||
if diff < best_diff:
|
||
best_diff = diff
|
||
best_id = rid
|
||
return best_id if best_diff <= 1.0 else None
|
||
|
||
resolution_id = None
|
||
if resolution_type == "dmt":
|
||
resolution_id = find_best_id_in_dict(UCDEnum.TimingInfo.dmt_resolution_map)
|
||
elif resolution_type == "cta":
|
||
resolution_id = find_best_id_in_dict(UCDEnum.TimingInfo.cta_resolution_map)
|
||
elif resolution_type == "cvt":
|
||
resolution_id = find_best_id_in_list_map(
|
||
UCDEnum.TimingInfo.cvt_resolution_map
|
||
)
|
||
elif resolution_type == "ovt":
|
||
resolution_id = find_best_id_in_list_map(
|
||
UCDEnum.TimingInfo.ovt_resolution_map
|
||
)
|
||
|
||
result = {
|
||
"resolution_type": resolution_type,
|
||
"width": width,
|
||
"height": height,
|
||
"refresh_rate": refresh_rate,
|
||
"resolution_id": resolution_id,
|
||
}
|
||
|
||
return result
|
||
|
||
def set_timing_from_string(self, timing_str):
|
||
"""根据格式化timing字符串设置设备timing"""
|
||
spec = self.parse_formatted_timing(timing_str)
|
||
rtype = spec["resolution_type"]
|
||
width = spec["width"]
|
||
height = spec["height"]
|
||
fr = spec["refresh_rate"]
|
||
|
||
if rtype != "ovt":
|
||
timing = self.search_timing(width, height, fr, rtype)
|
||
if timing:
|
||
self.current_timing = timing
|
||
return True
|
||
|
||
return False
|
||
|
||
def set_timing_from_id(self, rtype, rid):
|
||
"""根据(type, id)设置设备timing"""
|
||
timing = None
|
||
if rtype.lower() == "dmt":
|
||
timing = self.timing_manager.get_dmt(rid)
|
||
elif rtype.lower() == "cta":
|
||
timing = self.timing_manager.get_cta(rid)
|
||
elif rtype.lower() == "cvt":
|
||
timing = self.timing_manager.get_cvt(rid)
|
||
else:
|
||
raise ValueError(f"不支持的分辨率类型: {rtype}")
|
||
|
||
if timing:
|
||
self.current_timing = timing
|
||
return True
|
||
else:
|
||
return False
|
||
|
||
def apply_signal_format(
|
||
self, color_space=None, data_range=None, bit_depth=None, color_format=None, **_
|
||
):
|
||
"""统一设置信号格式（color_format / colorimetry / dynamic_range / bpc）。
|
||
注：Gamma/EOTF 传输特性在 ColorInfo API 中不存在；
|
||
max_cll / max_fall 暂无对应 SDK 接口，通过 **_ 接收后忽略。
|
||
"""
|
||
try:
|
||
if color_format:
|
||
fmt_key = UCDEnum.SignalFormat.OutputFormat.get_format_key(color_format)
|
||
cf = UCDEnum.ColorInfo.get_color_format(fmt_key)
|
||
if cf is not None:
|
||
self.color_info.color_format = cf
|
||
|
||
if color_space:
|
||
colorimetry = self._get_colorimetry_from_color_space(color_space, color_format)
|
||
if colorimetry:
|
||
self.color_info.colorimetry = colorimetry
|
||
|
||
if data_range:
|
||
if data_range == "Full":
|
||
self.color_info.dynamic_range = UniTAP.ColorInfo.DynamicRange.DR_VESA
|
||
elif data_range == "Limited":
|
||
self.color_info.dynamic_range = UniTAP.ColorInfo.DynamicRange.DR_CTA
|
||
|
||
if bit_depth:
|
||
bpc = UCDEnum.SignalFormat.BitDepth.get_bit_value(bit_depth)
|
||
self.color_info.bpc = bpc
|
||
|
||
if self.current_timing:
|
||
self.set_video_mode()
|
||
|
||
return True
|
||
|
||
except Exception:
|
||
return False
|
||
|
||
# 向后兼容别名
|
||
set_sdr_format = apply_signal_format
|
||
set_hdr_format = apply_signal_format
|
||
|
||
def _get_colorimetry_from_color_space(self, color_space, color_format=None):
|
||
"""将色彩空间字符串转换为UniTAP.ColorInfo.Colorimetry。
|
||
BT.2020 在 YCbCr 输出时使用 CM_ITUR_BT2020_YCbCr，RGB 输出时使用 CM_ITUR_BT2020_RGB。
|
||
"""
|
||
is_ycbcr = UCDEnum.SignalFormat.OutputFormat.is_ycbcr(color_format)
|
||
bt2020_cm = (
|
||
UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT2020_YCbCr
|
||
if is_ycbcr
|
||
else UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT2020_RGB
|
||
)
|
||
colorimetry_map = {
|
||
"sRGB": UniTAP.ColorInfo.Colorimetry.CM_sRGB,
|
||
"BT.709": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT709,
|
||
"BT.601": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT601,
|
||
"BT.2020": bt2020_cm,
|
||
"DCI-P3": UniTAP.ColorInfo.Colorimetry.CM_DCI_P3,
|
||
}
|
||
return colorimetry_map.get(color_space)
|