1.1.0版本
This commit is contained in:
xinzhu.yin
589
utils/UCD323_Function.py
Normal file
589
utils/UCD323_Function.py
Normal file
@@ -0,0 +1,589 @@
|
||||
# -*- coding: UTF-8 -*-
|
||||
import UniTAP
|
||||
import time
|
||||
import gc
|
||||
from utils.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_mode = None
|
||||
self.status = False
|
||||
|
||||
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
|
||||
|
||||
except Exception as role_error:
|
||||
self._close_device_object(temp_dev)
|
||||
raise role_error
|
||||
|
||||
self.timing_manager = self.role.hdtx.pg.timing_manager
|
||||
self.color_mode = UniTAP.ColorInfo()
|
||||
self.status = True
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self._force_cleanup()
|
||||
return False
|
||||
|
||||
def close(self):
|
||||
"""关闭设备"""
|
||||
try:
|
||||
if self.dev:
|
||||
self._close_device_object(self.dev)
|
||||
|
||||
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.lUniTAP = None
|
||||
|
||||
for i in range(3):
|
||||
gc.collect()
|
||||
|
||||
time.sleep(2.0)
|
||||
|
||||
self.lUniTAP = UniTAP.TsiLib()
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
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
|
||||
|
||||
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.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
|
||||
|
||||
except Exception as e:
|
||||
pass
|
||||
|
||||
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
|
||||
|
||||
if test_type == "hdr_movie":
|
||||
color_format_str = self.config.current_test_types[test_type]["color_format"]
|
||||
color_format = UCDEnum.ColorInfo.get_color_format(color_format_str)
|
||||
|
||||
if color_format:
|
||||
self.color_mode.color_format = color_format
|
||||
else:
|
||||
return False
|
||||
|
||||
else:
|
||||
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()
|
||||
self.role.hdtx.pg.apply()
|
||||
return True
|
||||
|
||||
def set_color_mode(self, cf, bpc, cm):
|
||||
"""设置颜色模式"""
|
||||
current_dynamic_range = None
|
||||
current_transfer_characteristic = None
|
||||
|
||||
if hasattr(self.color_mode, "dynamic_range"):
|
||||
current_dynamic_range = self.color_mode.dynamic_range
|
||||
|
||||
if hasattr(self.color_mode, "transfer_characteristic"):
|
||||
current_transfer_characteristic = self.color_mode.transfer_characteristic
|
||||
|
||||
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_mode.color_format = color_format
|
||||
self.color_mode.bpc = bpc
|
||||
self.color_mode.colorimetry = colorimetry
|
||||
|
||||
if current_dynamic_range is not None:
|
||||
self.color_mode.dynamic_range = current_dynamic_range
|
||||
|
||||
if current_transfer_characteristic is not None:
|
||||
self.color_mode.transfer_characteristic = current_transfer_characteristic
|
||||
|
||||
return True
|
||||
|
||||
def apply_video_mode(self):
|
||||
"""应用当前colormode和timing"""
|
||||
if self.current_timing:
|
||||
self.set_video_mode()
|
||||
return True
|
||||
return False
|
||||
|
||||
def set_video_mode(self):
|
||||
"""设置视频模式"""
|
||||
video_mode = UniTAP.VideoMode(
|
||||
timing=self.current_timing, color_info=self.color_mode
|
||||
)
|
||||
self.role.hdtx.pg.set_vm(vm=video_mode)
|
||||
return True
|
||||
|
||||
def set_pattern(self, pattern, pattern_params=None):
|
||||
"""设置pattern"""
|
||||
if self.current_timing:
|
||||
if (
|
||||
pattern == UCDEnum.VideoPatternInfo.VideoPatternParams.SolidColor
|
||||
or pattern == UCDEnum.VideoPatternInfo.VideoPatternParams.WhiteVStrips
|
||||
or pattern
|
||||
== UCDEnum.VideoPatternInfo.VideoPatternParams.GradientRGBStripes
|
||||
or pattern == UCDEnum.VideoPatternInfo.VideoPatternParams.MotionPattern
|
||||
or pattern == UCDEnum.VideoPatternInfo.VideoPatternParams.SquareWindow
|
||||
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:
|
||||
self.role.hdtx.pg.set_pattern(self.current_pattern)
|
||||
|
||||
if self.current_pattern_param:
|
||||
self.role.hdtx.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 set_sdr_format(
|
||||
self, color_space=None, gamma=None, data_range=None, bit_depth=None
|
||||
):
|
||||
"""设置SDR信号格式"""
|
||||
|
||||
def _get_colorimetry_from_color_space(color_space_name):
|
||||
"""将色彩空间名称转换为UniTAP colorimetry枚举值"""
|
||||
try:
|
||||
colorimetry_map = {
|
||||
"BT.709": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT709,
|
||||
"BT.601": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT601,
|
||||
"BT.2020": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT2020_RGB,
|
||||
}
|
||||
result = colorimetry_map.get(color_space_name)
|
||||
return result if result else UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT709
|
||||
except Exception as e:
|
||||
return UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT709
|
||||
|
||||
def _set_gamma_transfer_characteristic(gamma_value_str):
|
||||
"""设置Gamma传输特性"""
|
||||
try:
|
||||
gamma_value = float(gamma_value_str)
|
||||
|
||||
if abs(gamma_value - 2.2) < 0.1:
|
||||
self.color_mode.transfer_characteristic = (
|
||||
UniTAP.ColorInfo.TransferCharacteristic.TRANSFER_BT709
|
||||
)
|
||||
return True
|
||||
|
||||
elif abs(gamma_value - 2.4) < 0.1:
|
||||
if hasattr(
|
||||
UniTAP.ColorInfo.TransferCharacteristic, "TRANSFER_GAMMA24"
|
||||
):
|
||||
self.color_mode.transfer_characteristic = (
|
||||
UniTAP.ColorInfo.TransferCharacteristic.TRANSFER_GAMMA24
|
||||
)
|
||||
return True
|
||||
else:
|
||||
self.color_mode.transfer_characteristic = (
|
||||
UniTAP.ColorInfo.TransferCharacteristic.TRANSFER_BT709
|
||||
)
|
||||
return False
|
||||
|
||||
elif abs(gamma_value - 2.6) < 0.1:
|
||||
if hasattr(
|
||||
UniTAP.ColorInfo.TransferCharacteristic, "TRANSFER_GAMMA26"
|
||||
):
|
||||
self.color_mode.transfer_characteristic = (
|
||||
UniTAP.ColorInfo.TransferCharacteristic.TRANSFER_GAMMA26
|
||||
)
|
||||
return True
|
||||
else:
|
||||
self.color_mode.transfer_characteristic = (
|
||||
UniTAP.ColorInfo.TransferCharacteristic.TRANSFER_BT709
|
||||
)
|
||||
return False
|
||||
else:
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
return False
|
||||
|
||||
try:
|
||||
if color_space:
|
||||
colorimetry = _get_colorimetry_from_color_space(color_space)
|
||||
if colorimetry:
|
||||
self.color_mode.colorimetry = colorimetry
|
||||
|
||||
if gamma:
|
||||
_set_gamma_transfer_characteristic(gamma)
|
||||
|
||||
if data_range:
|
||||
if data_range == "Full":
|
||||
self.color_mode.dynamic_range = (
|
||||
UniTAP.ColorInfo.DynamicRange.DR_VESA
|
||||
)
|
||||
elif data_range == "Limited":
|
||||
self.color_mode.dynamic_range = UniTAP.ColorInfo.DynamicRange.DR_CTA
|
||||
|
||||
if bit_depth:
|
||||
bpc = UCDEnum.SignalFormat.BitDepth.get_bit_value(bit_depth)
|
||||
self.color_mode.bpc = bpc
|
||||
|
||||
if self.current_timing:
|
||||
self.set_video_mode()
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
return False
|
||||
|
||||
def set_hdr_format(
|
||||
self,
|
||||
color_space=None,
|
||||
data_range=None,
|
||||
bit_depth=None,
|
||||
max_cll=None,
|
||||
max_fall=None,
|
||||
):
|
||||
"""设置HDR信号格式"""
|
||||
try:
|
||||
if color_space:
|
||||
colorimetry = self._get_colorimetry_from_color_space(color_space)
|
||||
if colorimetry:
|
||||
self.color_mode.colorimetry = colorimetry
|
||||
|
||||
if data_range:
|
||||
if data_range == "Full":
|
||||
self.color_mode.dynamic_range = (
|
||||
UniTAP.ColorInfo.DynamicRange.DR_VESA
|
||||
)
|
||||
elif data_range == "Limited":
|
||||
self.color_mode.dynamic_range = UniTAP.ColorInfo.DynamicRange.DR_CTA
|
||||
|
||||
if bit_depth:
|
||||
bpc = UCDEnum.SignalFormat.BitDepth.get_bit_value(bit_depth)
|
||||
self.color_mode.bpc = bpc
|
||||
|
||||
if self.current_timing:
|
||||
self.set_video_mode()
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
return False
|
||||
|
||||
def _get_colorimetry_from_color_space(self, color_space):
|
||||
"""将色彩空间字符串转换为UniTAP.ColorInfo.Colorimetry"""
|
||||
colorimetry_map = {
|
||||
"BT.709": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT709,
|
||||
"BT.601": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT601,
|
||||
"BT.2020": UniTAP.ColorInfo.Colorimetry.CM_ITUR_BT2020_RGB,
|
||||
"DCI-P3": UniTAP.ColorInfo.Colorimetry.CM_DCI_P3,
|
||||
}
|
||||
return colorimetry_map.get(color_space)
|
||||
Reference in New Issue
Block a user