scbc_repos/ssat_sdk/device_manage/capturecard_manager.py

# -*- coding:utf-8 -*-
import os
import sys
import time
# from device_manager import *
from ssat_sdk.utils import LoggingUtil
from ssat_sdk import sat_environment
from ssat_sdk.utils import string_util
from ssat_sdk.service.service_manager import ServiceManager
from ssat_sdk.device_manage.ub530_manager import UB530Manager
import CD750SSDK
import UB530SDK
UB530 = 'TC-UB530'
UB530EX = 'TC-UB530EX'
CD750 = 'CD750'


class CCardManager():
    status = False
    exe_path = sys.prefix + r"/Tools/CD750/AVerCapSDKDemo.exe"
    sManager = ServiceManager()
    DEVICE_NAME = sat_environment.getCCard_Selected()
    print u"视频采集卡名称：", DEVICE_NAME

    def __init__(self):
        if string_util.strcmp(UB530, self.DEVICE_NAME):     # Python版本;
            self.ccard = UB530Manager()
        elif string_util.strcmp(UB530EX, self.DEVICE_NAME):  # VC++版本;
            CCardManager.exe_path = sys.prefix + r"/Tools/SATHelper/SATHelper.exe"
        else:
            CCardManager.exe_path = sys.prefix + r"/Tools/CD750/AVerCapSDKDemo.exe"

    def start(self):
        print "CCardManager ssdk"
        if string_util.strcmp(self.DEVICE_NAME, CD750):
            self.status = CD750SSDK.StartApp(CCardManager.exe_path)
            time.sleep(3)
            if self.status == True:
                LoggingUtil.printLog("打开CD750 AVerCapSDKDemo成功")
                # 连接设备
                CD750SSDK.ConnectDevice(0)
                CCardManager.status = True
                # 开始流操作
                time.sleep(1)
                CD750SSDK.StreamOption(True)
            else:
                LoggingUtil.printLog("打开CD750 AVerCapSDKDemo失败")
            return self.status
        elif string_util.strcmp(self.DEVICE_NAME, UB530):  # Python版本;
            self.sManager.startServiceThread("ub530")
            time.sleep(2)
            return self.sManager.getServiceStatus("ub530")
        elif string_util.strcmp(self.DEVICE_NAME, UB530EX):  # VC++版本;
            self.status = UB530SDK.StartApp(CCardManager.exe_path)
            time.sleep(3)
            if self.status == True:
                LoggingUtil.printLog("打开UB530 VideoCapture成功")
                # 连接设备
                UB530SDK.ConnectDevice()
                CCardManager.status = True
            else:
                LoggingUtil.printLog("打开UB530 VideoCapture失败")
            return self.status

    def isExeRunning(self):
        if string_util.strcmp(self.DEVICE_NAME, CD750):
            return CD750SSDK.IsAppRunning(CCardManager.exe_path)
        elif string_util.strcmp(self.DEVICE_NAME, UB530EX):  # VC++版本;
            return UB530SDK.IsAppRunning(CCardManager.exe_path)
        elif string_util.strcmp(self.DEVICE_NAME, UB530):  # Python版本;
            return self.sManager.getServiceStatus("ub530")

    # start app
    def startApp(self):
        self.status = CD750SSDK.StartApp(CCardManager.exe_path)
        return self.status

    # close app;
    def stopApp(self):
        self.status = not CD750SSDK.StopApp(CCardManager.exe_path)
        return self.status

    # connect device
    def connect(self, index=0):
        return CD750SSDK.ConnectDevice(index)

    # disconnect device
    def disconnect(self, index=0):
        return CD750SSDK.DisconnectDevice(index)

    # show app
    def showApp(self):
        CD750SSDK.ShowApp()

    # hide app
    def hideApp(self):
        CD750SSDK.HideApp()

    # stream option,开始/停止流操作
    def streamOption(self, enable=True):
        return CD750SSDK.StreamOption(enable)

    # 截图单张
    def captureSingleImage(self, picpath):
        return CD750SSDK.CaptureSingleImage(picpath, 3, True, 0, 0, 0, 0)

    # 按张数截图：picpath图片路径,count截图张数
    def captureImageByCount(self, count, picpath, prefix="CD750"):
        return CD750SSDK.CaptureImageByCount(count, picpath, prefix, 3, True, 0, 0, 0, 0)

    # 按时间截图：times持续时间,单位毫秒； perCount每秒截图的张数； picpath图片保存路径;
    def captureImageByTime(self, times, perCount, picpath, prefix="CD750"):
        if string_util.strcmp(CD750, self.DEVICE_NAME):
            return CD750SSDK.CaptureImageByTime(times, perCount, picpath, prefix, 3, True, 0, 0, 0, 0)
        elif string_util.strcmp(UB530EX, self.DEVICE_NAME):  # VC++版本;
            return UB530SDK.CaptureImageByTime(times, 0, picpath, prefix, 2)
        elif string_util.strcmp(UB530, self.DEVICE_NAME):  # Python版本;
            count = 0
            runTime = time.time()
            endWork = False
            lastIndex = 1
            times = times/1000
            while True:
                lastCount = lastIndex + perCount
                for i in range(lastIndex, lastCount):
                    path = "%s%s_%d.jpg" % (picpath, prefix, i)
                    if self.ccard.takePicture(path) == 0:
                        count += 1
                    time.sleep(0.2)  # 回调响应时间;
                    if (time.time() - runTime >= times):
                        endWork = True
                        break
                lastIndex += perCount
                if endWork:
                    break
            # print '实际生成数量%d,目标生成数量%d'%(count,perCount*times)
            return True if count == perCount*times else False

    # 抓单张图;
    def takePicture(self, picpath):
        status = -1
        if string_util.strcmp(CD750, self.DEVICE_NAME):
            status = CD750SSDK.CaptureSingleImage(picpath, 3, True, 0, 0, 0, 0)
            time.sleep(0.5)
        elif string_util.strcmp(UB530EX, self.DEVICE_NAME):  # VC++版本;
            status = UB530SDK.CaptureSingleImage(picpath, 2)
            count = 0
            while os.path.exists(picpath) is False:
                time.sleep(0.1) # 等待磁盘完成响应;
                if count == 5:
                    break
                count = count + 1
            # 超过5次认为截图失败;
            status = True if count < 5  else False
        elif string_util.strcmp(UB530, self.DEVICE_NAME):  # Python版本;
            status = self.ccard.takePicture(picpath)
            if status == 0:
                status = 1
            else:
                status = 0
            time.sleep(1)
        return status

    # 返回保存的文件列表;
    def getCaptureImageList(self, picpath):
        list = []
        if string_util.strcmp(CD750, self.DEVICE_NAME):
            list = CD750SSDK.GetCaptureImageList(picpath)
        elif string_util.strcmp(UB530EX, self.DEVICE_NAME):  # VC++版本;
            list = UB530SDK.GetCaptureImageList(picpath)
        elif string_util.strcmp(UB530, self.DEVICE_NAME):  # Python版本;
            list = []
            # 过滤掉多余的'/'
            newpath = ''
            for i in range(0, picpath.__len__(), 2):
                char1 = picpath[i]
                if i < picpath.__len__()-1:
                    char2 = picpath[i+1]
                    newpath += char1
                    if char1 == char2 and char1 == '/':
                        continue
                    newpath += char2
                else:
                    newpath += char1
            picpath = newpath.replace('/', '\\')
            index = picpath.rfind('\\')
            # 路径;
            dir = picpath[:index]
            # 文件名;
            name = picpath[index+1:]
            # 前缀;
            preFix = name[:name.rfind("_")]
            for dirpath, dirname, filenames in os.walk(dir):
                for filepath in filenames:
                    list.append(os.path.join(dirpath, filepath))
        # endif
        list.sort()
        return list

    # 停止抓图;
    def stopCaptureImage(self):
        CD750SSDK.StopCaptureImage()

    # 开始录制视频;
    def startCaptureAudio(self, audiopath):
        CD750SSDK.StartCaptureAudio(audiopath)

    # 停止录制视频;
    def stopCaptureAudio(self):
        CD750SSDK.StopCaptureAudio()

    # 异步录制视频;
    def asyStartCaptureAudio(self, keeptime, audiopath):
        CD750SSDK.AsyCaptureAudio(keeptime, audiopath)


if __name__ == "__main__":
    cardManager = CCardManager()
    print cardManager.start()
    # time.sleep(1)
    # print u"开始截图"
    time_start = time.time()
    cardManager.takePicture("D:\\SAT\\test1.png")
    time_end = time.time()
    print('totally cost', time_end-time_start)

    # print u"按时间截图"
    # cardManager.captureImageByTime(5000, 5, "D:/SAT/aaa/")
    # time.sleep(5)
    # print u"遍历设备"
    # list = CD750SSDK.GetCaptureImageList("D:/SAT/aaa/")
    # if list != None:
    #     for file in list:
    #         print file
    # time.sleep(2)
    # # 断开连接
    # print u"断开设备"
    # cardManager.disconnect(0)
    # time.sleep(3)
    # # 关闭app
    # print u"关闭程序（宿主程序需要管理员权限）"
    # cardManager.stopApp()