# -*- coding:utf-8 -*- import os from UIT_PathManage import UITPathManage from ssat_sdk.MenuTree3.BaseLog import CBaseLog from ssat_sdk.MenuTree3.TConfig import TConfig # 测试加头文件; from TData import CTData g_level = ['First', 'Second', 'Third', 'Fourth', 'Fifth', 'Sixth', 'Seventh', 'Eighth', 'Ninth', 'Tenth', 'Eleventh', 'Twelfth'] # 注意：所有不对外暴露的变量和函数需要私有化，以明确哪些接口和参数是对外的。 # 这样便于后期维护时，根据对外的变量和函数来做处理。 class COptionConfig(TConfig, CBaseLog): def __init__(self, menuTreeDir, pathManager): self.__pathManager = pathManager if self.__pathManager is None: self.error(u"路径参数错误：None") # 状态：False表示路径值失败; self.status = False # menuTree目录; self.__uiTreeDir = menuTreeDir # 判断是否为目录; if not os.path.isdir(self.__uiTreeDir): self.error(u"%s 不是menuTree目录" % self.__uiTreeDir) # 判断目录是否存在; if not os.path.exists(self.__uiTreeDir): self.error(u"menuTree（%s）目录不存在" % self.__uiTreeDir) # 判断配置文件是否存在; self.__configPath = os.path.join(self.__uiTreeDir, "menutree.ini") if not os.path.exists(self.__configPath): self.error(u"menutree配置文件不存在：%s" % self.__configPath) self.status = True # 初始化父类; TConfig.__init__(self, self.__configPath) # 获取超级密码; def getSuperPassword(self): return self.get_value("Password", "super") # 获取普通密码; def getOrdinaryPassword(self): return self.get_value("Password", "ordinary") # 获取父节点等待进入子节点时间; def getParentWaitTime(self, parent): optionName = parent + "WaitTime" try: waitTime = float(self.get_value("waitTime", optionName)) except Exception: waitTime = 1.0 self.error(u"获取%s界面的等待界面时间失败，使用默认值1.0s" % str(parent)) return waitTime # 获取阀值字典; def getThresholdDict(self, optionName): section = "Threshold" thresholdDict = {} value = self.get_value(section, optionName) self.info("value:%s,%s" % (value, type(value))) if not value: return thresholdDict value_str = str(value) self.info("value_str:%s,%s" % (value_str, type(value_str))) value_dict = eval(value_str) self.info("value_dict:%s,%s" % (value_dict, type(value_dict))) return value_dict # 获取Option的图片配置; def __getICONConfig(self, optionName, is_value_sheet=False): # 默认值; def_cfg = {"icon_path": "", "dcfg": {}, "dir_path": ""} paths = self.__pathManager.get_option_paths(optionName) # 判断路径节点是否空; if paths.__len__() == 0: self.error(u"当前【%s】的路径节点空，使用默认的配置值%s" % (optionName, def_cfg)) return False, def_cfg # 获取first parent; first_parent = paths[g_level[0]]['parent'] # 获取当前option的level; cur_level = g_level[paths.__len__() - 1] # 当前option的父节点名称; cur_parent = paths[cur_level]['parent'] if is_value_sheet is True: cur_level = "value" # option聚焦的首配图片路径; icon_path = os.path.join(self.__uiTreeDir, "icon\\", cur_parent + "." + cur_level + "_" + optionName + ".png") # option聚焦的dir首配图片路径; icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + "_" + optionName + ".dir.png") # 图片聚焦时的定位参数; opc_cfg = self.get_value_dict(cur_level, cur_parent + '.' + optionName) # 首配图片判断是否存在,不存在取用次配图片路径; if not os.path.exists(icon_path): # 使用次配图片，父级配图; self.warn(u"Option(%s)首配图片不存在：%s" % (optionName, icon_path)) icon_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + ".png") icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + cur_parent + "." + cur_level + ".dir.png") opc_cfg = self.get_value_dict(cur_level, cur_parent) self.warn(u"Option(%s)首配图片不存在，尝试使用次配图(%s)：%s，配置%s" % (optionName, cur_parent, icon_path, opc_cfg)) # 如果次配都不存在，使用顶层配图; if not os.path.exists(icon_path): self.warn(u"Option(%s)次配图片不存在：%s" % (optionName, icon_path)) # 使用顶层配图（first parent） icon_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + "_" + optionName + ".png") icon_dir_path = os.path.join(self.__uiTreeDir, "icon\\" + first_parent + "." + cur_level + "_" + optionName + ".dir.png") opc_cfg = self.get_value_dict(cur_level, first_parent) self.warn(u"Option(%s)次配图片不存在，尝试使用顶层配图(%s)：%s，配置%s" % (optionName, first_parent, icon_path, opc_cfg)) # 如果顶层配图不存在，退出; if not os.path.exists(icon_path): self.error(u"%s对应的顶层菜单配图不存在%s，使用默认配置%s" % (optionName, icon_path, def_cfg)) return False, def_cfg # endif # endif # endif if opc_cfg.__len__() == 0: opc_cfg = {"offset": 20, "minPeri": 0, "maxPeri": 0, "minArea": 0, "maxArea": 0, "morphology": []} return True, {"icon_path": icon_path, "dcfg": opc_cfg, "dir_path": icon_dir_path} # 获取Option的图片配置; def getOptionICONConfig(self, optionName): return self.__getICONConfig(optionName, False) # 获取Value的图片配置; def getValueICONConfig(self, valueName): return self.__getICONConfig(valueName, True) # 获取Option的OCR配置; def getOptionOCRConfig(self, optionName): ocr_dict = [] # 默认的ocr配置; def_orc = [{"lan": "ChinesePRC+English", "type": 4}, {"lan": "ChinesePRC+English", "type": 253}, {"lan": "ChinesePRC+English", "type": 10001}] paths = self.__pathManager.get_option_paths(optionName) # 判断路径节点是否空; if paths.__len__() == 0: self.error(u"当前【%s】的路径节点空，将使用默认的ocr配置%s" % (optionName, def_orc)) else: # 如果状态False,退出; if self.status is False: self.warn(u"配置文件(%s)不存在,%s使用默认的ocr配置%s" % (self.__configPath, optionName, def_orc)) else: # 读取指定的ocr配置信息; for i in range(g_level.index(paths.__len__() - 1), -1, -1): cur_parent = paths['parent'] first_parent = paths['first_parent'] cur_level = g_level[i] # 是否有当前option的ocr配置; if self.has_option(cur_level, optionName + '.ocr'): ocr_dict = self.get_dict(self.get_value(cur_level, optionName + '.ocr')) self.warn(u"%s使用自身的ocr配置%s" % (optionName, ocr_dict)) break # 如果option本身没有配置,获取其父节点的配置; if self.has_option(cur_level, cur_parent + '.ocr'): ocr_dict = self.get_dict(self.get_value(cur_level, cur_parent + '.ocr')) self.warn(u"%s使用父节点%s的ocr配置%s" % (optionName, cur_parent, ocr_dict)) break # 如果option父级没有配置,则获取顶层配置(first parent); if self.has_option(cur_level, first_parent + '.ocr'): ocr_dict = self.get_dict(self.get_value(cur_level, first_parent + '.ocr')) self.warn(u"%s使用顶层节点%s的ocr配置%s" % (optionName, first_parent, ocr_dict)) break # end-for # end-if # end-if if ocr_dict.__len__() == 0: ocr_dict = def_orc self.warn(u"无有效的ocr配置，将使用默认的ocr配置%s" % def_orc) self.info(u"%s使用的ocr配置=%s" % (optionName, ocr_dict)) return ocr_dict # 获取ICON源屏幕的分辨率配置值; # 该值用途：由于有些icon是在某些分辨率下截图的，但是同机芯项目可能使用同一套UI，只是分辨率不一样而已。 # 所以，根据当初的icon分辨率，可以计算出在别的分辨率下的icon区域大小; def getICONResolutionConfig(self): return self.get_value_dict('Screen', 'shape') if self.has_option('Screen', 'shape') else [1920, 1080] if __name__ == "__main__": tData = CTData() upath = UITPathManage(tData) opc = COptionConfig(r'D:\SAT\resource\MenuTree\RT2851\2851', upath) print opc.getOptionICONConfig("picture")