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
|