#-*- coding: utf-8 -*-
"""
此工具用于抓取各种声音和图片样本
"""
import sys,os,time
reload(sys)
sys.setdefaultencoding('utf-8')

from TST.CaptureCard  import *
from TST.SourceGenInput import *
from ssat_sdk.tv_operator import *

rcu = TvOperator()

class SampleTool() :
    HDMItimingNums = (65, 66, 67, 68, 69, 70, 71, 72, 73, 74)
    patternNums = (11, 21, 22, 23, 37, 41, 44, 47, 54,
                   56, 81, 85, 88, 134, 215, 304, 355)


    def takeTimingPattern(self, destDir,tNums, pNums):  # 检查pattern
        cc = CaptureCard()
        sg = SourceGenInput()
        result = 0
        bol = False
        timingIndex = 0
        patIndex = 0
        for tNum in tNums:
            print "切换timing：", tNum
            sg.setTiming(tNum)
            for pNum in pNums:
                print "切换pattern：",pNum
                sg.setPattern(pNum)
                #确保退出info信息显示
                time.sleep(3)
                rcu.sendKey("EXIT")
                picPath = destDir + "/" + str(tNum) + "-" + str(pNum) +".jpg"
                print "抓取图片：",picPath
                cc.takePicture(picPath)
                patIndex = patIndex + 1
                if (patIndex == pNums.__len__()):
                    patIndex = 0
                    break
            timingIndex = timingIndex + 1
        cc.close()

SP_DIR = "D:/TST/test1/standard_pattern/"
if __name__ == "__main__":
    samTool = SampleTool()
    samTool.takeTimingPattern(SP_DIR, SampleTool.HDMItimingNums, SampleTool.patternNums)