81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
"""通用 UCD323/UCDController 辅助函数。
|
||
|
||
封装"按当前接口取 tx 模块"、"读取分辨率"、"发送图片 Pattern"等所有
|
||
测试模块共用的低层 UCD 操作，避免在多个业务模块中重复 if/else。
|
||
"""
|
||
|
||
import UniTAP
|
||
|
||
|
||
def get_tx_modules(ucd):
|
||
"""根据当前接口返回 (pg, ag) 模块。
|
||
|
||
兼容 UCD323Controller（多接口，含 current_interface 属性）
|
||
与老的 UCDController（仅 HDMI）。
|
||
"""
|
||
interface = getattr(ucd, "current_interface", None)
|
||
if interface in (None, "HDMI"):
|
||
return ucd.role.hdtx.pg, ucd.role.hdtx.ag
|
||
if interface in ("DP", "Type-C"):
|
||
return ucd.role.dptx.pg, ucd.role.dptx.ag
|
||
raise ValueError(f"不支持的接口类型: {interface}")
|
||
|
||
|
||
def get_current_resolution(ucd, default=(3840, 2160)):
|
||
"""从 UCD 当前 timing 获取 (width, height)，失败时返回 default。"""
|
||
try:
|
||
pg, _ = get_tx_modules(ucd)
|
||
vm = pg.get_vm()
|
||
timing = getattr(vm, "timing", None)
|
||
if timing and hasattr(timing, "h_active") and hasattr(timing, "v_active"):
|
||
return timing.h_active, timing.v_active
|
||
except Exception:
|
||
pass
|
||
|
||
timing = getattr(ucd, "current_timing", None)
|
||
if timing and hasattr(timing, "h_active") and hasattr(timing, "v_active"):
|
||
return timing.h_active, timing.v_active
|
||
|
||
return default
|
||
|
||
|
||
def send_image_pattern(ucd, image_path, *, bpc=8, color_format=None, colorimetry=None):
|
||
"""通过 UCD 发送一张本地图片作为显示 Pattern。
|
||
|
||
自动停止音频以避免蜂鸣声。颜色参数留空时使用 RGB / sRGB 默认值。
|
||
返回 True/False。
|
||
"""
|
||
if not getattr(ucd, "status", False):
|
||
return False
|
||
|
||
try:
|
||
pg, ag = get_tx_modules(ucd)
|
||
except Exception:
|
||
return False
|
||
|
||
try:
|
||
ag.stop_generate()
|
||
except Exception:
|
||
pass
|
||
|
||
color_mode = UniTAP.ColorInfo()
|
||
color_mode.color_format = color_format or UniTAP.ColorInfo.ColorFormat.CF_RGB
|
||
color_mode.bpc = bpc
|
||
color_mode.colorimetry = colorimetry or UniTAP.ColorInfo.Colorimetry.CM_sRGB
|
||
|
||
timing = None
|
||
try:
|
||
vm = pg.get_vm()
|
||
timing = getattr(vm, "timing", None)
|
||
except Exception:
|
||
pass
|
||
|
||
try:
|
||
if timing:
|
||
pg.set_vm(vm=UniTAP.VideoMode(timing=timing, color_info=color_mode))
|
||
pg.set_pattern(pattern=image_path)
|
||
pg.apply()
|
||
return True
|
||
except Exception:
|
||
return False
|