scbc_repos/ssat_sdk/MenuTree3/TFocus.py

# -*- coding:utf-8 -*-
from ssat_sdk import getMenuTree3SelectedPExcelPath

from ssat_sdk.picture.RGB import RGBColor
from ssat_sdk.picture.color_space import CIEluvCaculator
from ssat_sdk.picture.feature_detect import FeatureDetect
from ssat_sdk.picture.image_util import *

import os
import sys
import time
import numpy as np
import cv2 as cv
import json
from ssat_sdk.MenuTree3.TConfig import TConfig
from ssat_sdk.MenuTree3.TExcelParser import CExcelParser

g_level = ['First', 'Second', 'Third', 'Fourth', 'Fifth', 'Sixth',
           'Seventh', 'Eighth', 'Ninth', 'Tenth', 'Eleventh', 'Twelfth']


class TFocus():
    def __init__(self, project_path, uitRunner=None, screen=[1920, 1080], icon_shape=[1920, 1080]):
        self.params_path = project_path
        self.rgbColor = RGBColor()
        self.CIE = CIEluvCaculator()
        self.feature_detect = FeatureDetect()
        self.xlsparser = uitRunner.uitPathManage
        # icon分辨率width*height;
        self.icon_shape = icon_shape
        # 当前屏幕分辨率;
        self.screen = screen
        # 比率;
        self.rate = self.icon_shape[0] / self.screen[0]

    def findFocusByIcon(self, src_pic, icon_pic,
                        dcfg={"offset": 20, "minPeri": 0, "maxPeri": 0, "minArea": 0, "maxArea": 0, "morphology": []}):
        iconImg = cv.imread(icon_pic)
        # 获取小图平均bgr值;
        bgr = self.rgbColor.getAvgBGR(iconImg)
        print "findFocusByIcon,bgr:%s, src_pic=%s, icon_pic=%s, dcfg=%s" % (
            str(bgr), src_pic, icon_pic, str(dcfg))
        if dcfg.has_key("morphology"):
            morphology = dcfg["morphology"]
        else:
            morphology = []
        return self.feature_detect.findCurrentFoucsByColor(
            src_pic,
            [],
            bgr,
            dcfg["offset"],
            dcfg["minPeri"],
            dcfg["maxPeri"],
            dcfg["minArea"],
            dcfg["maxArea"],
            morphology)

    # 文本颜色焦点区域查找;
    def findFramelessBoxByIcon(self, src_pic, icon_pic, offset=20):
        iconImg = cv.imread(icon_pic)
        # 获取小图平均bgr值;
        bgr = self.rgbColor.getAvgBGR(iconImg)
        print "findFramelessBoxByIcon,bgr:%s, src_pic=%s, icon_pic=%s, offset=%s" % (
            str(bgr), src_pic, icon_pic, str(offset))
        return self.feature_detect.findFramelessFocusByColor(src_pic, [], bgr, offset)

    '''
        在传入的图片上，查找指定小图片bgr颜色框矩形区域，返回矩形区域的坐标
        :param bigPic:图片路径
        :param level:excel层次
        :param first_parent:大部分情况下是根菜单；少部分情况是当前option的parent；
        :return rectArea：矩形区域和区域图像内容
        '''

    def findRectByIcon(self, bigPic, level, first_parent, option=""):
        print "level:", level
        print "first_parent:", first_parent
        if option.__len__() == 0:
            icon_path = os.path.join(self.params_path, "icon\\", str(
                first_parent) + "." + str(level) + ".png")
        else:
            icon_path = os.path.join(self.params_path, "icon\\", str(
                first_parent) + "." + str(level) + "_" + option + ".png")
        print "icon_path_option=", icon_path
        if os.path.exists(icon_path) is False:
            icon_path = os.path.join(self.params_path, "icon\\", str(
                first_parent) + "." + str(level) + ".png")
            print "icon_path=", icon_path
            if os.path.exists(icon_path) is False:
                icon_path = os.path.join(self.params_path, "icon\\", str(
                    first_parent) + "." + str('First') + ".png")
                print "icon_path_First=", icon_path
                if os.path.exists(icon_path) is False:
                    return False, []
        # 获取配置文件值;
        dcfg = {"offset": 20, "minPeri": 0,
                "maxPeri": 0, "minArea": 0, "maxArea": 0}
        ini_path = os.path.join(self.params_path, "menutree.ini")
        if os.path.exists(ini_path) is True:
            tconfig = TConfig(ini_path)
            if tconfig.has_option(level, first_parent):
                dcfg = tconfig.get_dict(tconfig.get_value(level, first_parent))

        dcfg['minPeri'] = self.rate * dcfg['minPeri']
        dcfg['maxPeri'] = self.rate * dcfg['maxPeri']
        dcfg['minArea'] = pow(self.rate, 2) * dcfg['minArea']
        dcfg['maxArea'] = pow(self.rate, 2) * dcfg['maxArea']
        if "morphology" not in dcfg:
            dcfg["morphology"] = []
        return self.findFocusByIcon(bigPic, icon_path, dcfg)

    '''
    获取option配置信息;
    '''

    def getOptionConfig(self, cur_option, is_value_sheet=False):
        # 获取Option的others字段信息;
        if self.xlsparser is None:
            self.xlsparser = CExcelParser(getMenuTree3SelectedPExcelPath())
            # 读取excel并解析出内容;
            self.xlsparser.read_excel()
        # 获取option的路径表;
        pp = self.xlsparser.get_option_paths(cur_option)
        if pp.__len__() == 0:
            print 'getOptionConfig.get_option_paths(%s) is null' % (cur_option)
            return False, {"icon_path": "", "dcfg": {}, "dir_path": ""}

        # 获取first_parent;
        first_parent = pp[g_level[0]]['parent']
        # 当前option所在level;
        cur_level = g_level[pp.__len__() - 1]
        # 当前父路径;
        cur_parent = pp[cur_level]['parent']
        if is_value_sheet is True:
            cur_level = 'value'
        # print first_parent, cur_level, cur_parent, cur_option

        # 获取配置文件值;
        dcfg = {}
        ini_path = os.path.join(self.params_path, "menutree.ini")
        tconfig = TConfig(ini_path)
        # 首先找当前路径图;
        icon_path = os.path.join(self.params_path, "icon\\", str(
            cur_parent) + "." + str(cur_level) + "_" + cur_option + ".png")
        icon_dir_path = os.path.join(self.params_path, "icon\\", str(
            cur_parent) + "." + str(cur_level) + "_" + cur_option + ".dir.png")
        dcfg = tconfig.get_value_dict(cur_level, cur_parent + '.' + cur_option)
        print u"trying icon:%s"%icon_path
        print u"trying dcfg:get_value_dict(%s, %s)"%(cur_level, cur_parent + '.' + cur_option)
        # icon路径;
        if os.path.exists(icon_path) is False:
            print u'图标优先级0:[cur_parent].[cur_level]_[cur_option].png=%s is null' % (
                icon_path)
            # 当前次路径图;
            icon_path = os.path.join(self.params_path, "icon\\", str(
                cur_parent) + "." + str(cur_level) + ".png")
            icon_dir_path = os.path.join(self.params_path, "icon\\", str(
                cur_parent) + "." + str(cur_level) + ".dir.png")
            dcfg = tconfig.get_value_dict(cur_level, cur_parent)
            print u"trying icon:%s" % icon_path
            print u"trying dcfg:get_value_dict(%s, %s)" % (cur_level, cur_parent)
            if os.path.exists(icon_path) is False:
                print u'图标优先级1：[cur_parent].[cur_level].png=%s is null' % (
                    icon_path)
                # first层;
                icon_path = os.path.join(self.params_path, "icon\\", str(
                    first_parent) + "." + str(cur_level) + "_" + cur_option + ".png")
                icon_dir_path = os.path.join(self.params_path, "icon\\", str(
                    first_parent) + "." + str(cur_level) + "_" + cur_option + ".dir.png")
                dcfg = tconfig.get_value_dict(
                    cur_level, first_parent + '.' + cur_option)
                print u"trying icon:%s" % icon_path
                print u"trying dcfg:get_value_dict(%s, %s)" % (cur_level, first_parent + '.' + cur_option)
                if os.path.exists(icon_path) is False:
                    print u'图标优先级2：[first_parent].[cur_level]_[cur_option].png=%s is null' % (
                        icon_path)
                    # first层次路径图;
                    icon_path = os.path.join(self.params_path, "icon\\", str(
                        first_parent) + "." + str(cur_level) + ".png")
                    icon_dir_path = os.path.join(self.params_path, "icon\\", str(
                        first_parent) + "." + str(cur_level) + ".dir.png")
                    dcfg = tconfig.get_value_dict(cur_level, first_parent)
                    print u"trying icon:%s" % icon_path
                    print u"trying dcfg:get_value_dict(%s, %s)" % (cur_level, first_parent)
                    if os.path.exists(icon_path) is False:
                        print u'图标优先级3：[first_parent].[cur_level].png =%s is null' % (
                            icon_path)
                # endif;
            # endif;
        # endif;

        if dcfg.__len__() == 0:
            dcfg = {"offset": 20, "minPeri": 0,
                    "maxPeri": 0, "minArea": 0, "maxArea": 0, "morphology": []}
            print u"get dcfg fail!Using default para!"

        if os.path.exists(icon_path) is False:
            print 'getOptionConfig:current use icon_path=%s is null' % (
                icon_path)
            return False, {"icon_path": "", "dcfg": {}, "dir_path": ""}
        print u"实际使用的icon图片路径icon_path：%s, 实际使用INI配置：%s" % (
            str(icon_path), str(dcfg))

        # 返回路径;
        return True, {"icon_path": icon_path, "dcfg": dcfg, "dir_path": icon_dir_path}

    def findRectByIcon2(self, src_pic, cur_option, is_value_sheet=False):
        # 获取参数;
        result, opcfg = self.getOptionConfig(cur_option, is_value_sheet)
        if result is False:
            return False, []

        icon_path = opcfg['icon_path']
        dcfg = opcfg['dcfg']
        dcfg['minPeri'] = self.rate * dcfg['minPeri']
        dcfg['maxPeri'] = self.rate * dcfg['maxPeri']
        dcfg['minArea'] = pow(self.rate, 2) * dcfg['minArea']
        dcfg['maxArea'] = pow(self.rate, 2) * dcfg['maxArea']
        print u"实际使用的聚焦参数字典dcfg:%s" % str(dcfg)
        if "morphology" not in dcfg:
            dcfg["morphology"] = []

        return self.findFocusByIcon(src_pic, icon_path, dcfg)


def rgb2hsv(r, g, b):
    r, g, b = r / 255.0, g / 255.0, b / 255.0
    mx = max(r, g, b)
    mn = min(r, g, b)
    df = mx - mn
    # h值
    if mx == mn:
        h = 0
    elif mx == r and g >= b:
        h = (60 * (g - b) / df) % 360
    elif mx == r and g < b:
        h = (60 * ((g - b) / df) + 360) % 360
    elif mx == g:
        h = (60 * ((b - r) / df) + 120) % 360
    elif mx == b:
        h = (60 * ((r - g) / df) + 240) % 360
    # s值
    if mx == 0:
        s = 0
    else:
        s = df / mx * 255
    # v值
    v = mx * 255
    # 返回;
    return h, s, v


if __name__ == "__main__":
    # print '\n\n====>\n', pow(10, 2)
    tFocus = TFocus(r"D:\SAT\resource\MenuTree\MS6586\ISDB")
    tFocus.getOptionConfig('backlight', True)
    tFocus.findRectByIcon2('', 'backlight')  # lock
    if 0:
        bigPic_2 = r"D:\CH01.JPG"
        icon = r"D:\SAT\resource\MenuTree\mi\AM950_IND_V27N\result.png"
        result, contourRect = tFocus.findRectByIcon(bigPic_2, 'Third', 'lock')
        print "__name__,rect:", result, contourRect
        if result is True:
            saveCropPic(bigPic_2, icon, contourRect)