重构移动utils文件夹

drivers/ucd_helpers.py

@@ -0,0 +1,80 @@
+"""通用 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