225 lines
7.8 KiB
Python
225 lines
7.8 KiB
Python
import os
|
||
import platform
|
||
import tkinter
|
||
from tkinter import *
|
||
from HCNetSDK import *
|
||
from PlayCtrl import *
|
||
from time import sleep
|
||
|
||
# 登录的设备信息
|
||
DEV_IP = create_string_buffer(b'10.17.35.231')
|
||
DEV_PORT = 8000
|
||
DEV_USER_NAME = create_string_buffer(b'admin')
|
||
DEV_PASSWORD = create_string_buffer(b'abcd1234')
|
||
|
||
WINDOWS_FLAG = True
|
||
win = None # 预览窗口
|
||
funcRealDataCallBack_V30 = None # 实时预览回调函数，需要定义为全局的
|
||
|
||
PlayCtrl_Port = c_long(-1) # 播放句柄
|
||
Playctrldll = None # 播放库
|
||
FuncDecCB = None # 播放库解码回调函数，需要定义为全局的
|
||
|
||
# 获取当前系统环境
|
||
def GetPlatform():
|
||
sysstr = platform.system()
|
||
print('' + sysstr)
|
||
if sysstr != "Windows":
|
||
global WINDOWS_FLAG
|
||
WINDOWS_FLAG = False
|
||
|
||
# 设置SDK初始化依赖库路径
|
||
def SetSDKInitCfg():
|
||
# 设置HCNetSDKCom组件库和SSL库加载路径
|
||
# print(os.getcwd())
|
||
strPath = os.getcwd().encode('gbk')
|
||
sdk_ComPath = NET_DVR_LOCAL_SDK_PATH()
|
||
sdk_ComPath.sPath = strPath
|
||
Objdll.NET_DVR_SetSDKInitCfg(2, byref(sdk_ComPath))
|
||
if WINDOWS_FLAG:
|
||
Objdll.NET_DVR_SetSDKInitCfg(3, create_string_buffer(strPath + b'\libcrypto-1_1-x64.dll'))
|
||
Objdll.NET_DVR_SetSDKInitCfg(4, create_string_buffer(strPath + b'\libssl-1_1-x64.dll'))
|
||
else:
|
||
Objdll.NET_DVR_SetSDKInitCfg(3, create_string_buffer(strPath + b'\libcrypto.so.1.1'))
|
||
Objdll.NET_DVR_SetSDKInitCfg(4, create_string_buffer(strPath + b'\libssl.so.1.1'))
|
||
|
||
def LoginDev(Objdll):
|
||
# 登录注册设备
|
||
device_info = NET_DVR_DEVICEINFO_V30()
|
||
lUserId = Objdll.NET_DVR_Login_V30(DEV_IP, DEV_PORT, DEV_USER_NAME, DEV_PASSWORD, byref(device_info))
|
||
return (lUserId, device_info)
|
||
|
||
def DecCBFun(nPort, pBuf, nSize, pFrameInfo, nUser, nReserved2):
|
||
# 解码回调函数
|
||
if pFrameInfo.contents.nType == 3:
|
||
# 解码返回视频YUV数据，将YUV数据转成jpg图片保存到本地
|
||
# 如果有耗时处理，需要将解码数据拷贝到回调函数外面的其他线程里面处理，避免阻塞回调导致解码丢帧
|
||
sFileName = ('../../pic/test_stamp[%d].jpg'% pFrameInfo.contents.nStamp)
|
||
nWidth = pFrameInfo.contents.nWidth
|
||
nHeight = pFrameInfo.contents.nHeight
|
||
nType = pFrameInfo.contents.nType
|
||
dwFrameNum = pFrameInfo.contents.dwFrameNum
|
||
nStamp = pFrameInfo.contents.nStamp
|
||
print(nWidth, nHeight, nType, dwFrameNum, nStamp, sFileName)
|
||
|
||
lRet = Playctrldll.PlayM4_ConvertToJpegFile(pBuf, nSize, nWidth, nHeight, nType, c_char_p(sFileName.encode()))
|
||
if lRet == 0:
|
||
print('PlayM4_ConvertToJpegFile fail, error code is:', Playctrldll.PlayM4_GetLastError(nPort))
|
||
else:
|
||
print('PlayM4_ConvertToJpegFile success')
|
||
|
||
def RealDataCallBack_V30(lPlayHandle, dwDataType, pBuffer, dwBufSize, pUser):
|
||
# 码流回调函数
|
||
if dwDataType == NET_DVR_SYSHEAD:
|
||
# 设置流播放模式
|
||
Playctrldll.PlayM4_SetStreamOpenMode(PlayCtrl_Port, 0)
|
||
# 打开码流，送入40字节系统头数据
|
||
if Playctrldll.PlayM4_OpenStream(PlayCtrl_Port, pBuffer, dwBufSize, 1024*1024):
|
||
# 设置解码回调，可以返回解码后YUV视频数据
|
||
global FuncDecCB
|
||
FuncDecCB = DECCBFUNWIN(DecCBFun)
|
||
Playctrldll.PlayM4_SetDecCallBackExMend(PlayCtrl_Port, FuncDecCB, None, 0, None)
|
||
# 开始解码播放
|
||
if Playctrldll.PlayM4_Play(PlayCtrl_Port, cv.winfo_id()):
|
||
print(u'播放库播放成功')
|
||
else:
|
||
print(u'播放库播放失败')
|
||
else:
|
||
print(u'播放库打开流失败')
|
||
elif dwDataType == NET_DVR_STREAMDATA:
|
||
Playctrldll.PlayM4_InputData(PlayCtrl_Port, pBuffer, dwBufSize)
|
||
else:
|
||
print (u'其他数据,长度:', dwBufSize)
|
||
|
||
def OpenPreview(Objdll, lUserId, callbackFun):
|
||
'''
|
||
打开预览
|
||
'''
|
||
preview_info = NET_DVR_PREVIEWINFO()
|
||
preview_info.hPlayWnd = None
|
||
preview_info.lChannel = 1 # 通道号
|
||
preview_info.dwStreamType = 0 # 主码流
|
||
preview_info.dwLinkMode = 0 # TCP
|
||
preview_info.bBlocked = 1 # 阻塞取流
|
||
|
||
# 开始预览并且设置回调函数回调获取实时流数据
|
||
lRealPlayHandle = Objdll.NET_DVR_RealPlay_V40(lUserId, byref(preview_info), callbackFun, None)
|
||
return lRealPlayHandle
|
||
|
||
def InputData(fileMp4, Playctrldll):
|
||
while True:
|
||
pFileData = fileMp4.read(4096)
|
||
if pFileData is None:
|
||
break
|
||
|
||
if not Playctrldll.PlayM4_InputData(PlayCtrl_Port, pFileData, len(pFileData)):
|
||
break
|
||
|
||
if __name__ == '__main__':
|
||
# 创建窗口
|
||
win = tkinter.Tk()
|
||
#固定窗口大小
|
||
win.resizable(0, 0)
|
||
win.overrideredirect(True)
|
||
|
||
sw = win.winfo_screenwidth()
|
||
# 得到屏幕宽度
|
||
sh = win.winfo_screenheight()
|
||
# 得到屏幕高度
|
||
|
||
# 窗口宽高
|
||
ww = 512
|
||
wh = 384
|
||
x = (sw - ww) / 2
|
||
y = (sh - wh) / 2
|
||
win.geometry("%dx%d+%d+%d" % (ww, wh, x, y))
|
||
|
||
# 创建退出按键
|
||
b = Button(win, text='退出', command=win.quit)
|
||
b.pack()
|
||
# 创建一个Canvas，设置其背景色为白色
|
||
cv = tkinter.Canvas(win, bg='white', width=ww, height=wh)
|
||
cv.pack()
|
||
|
||
# 获取系统平台
|
||
GetPlatform()
|
||
|
||
# 加载库,先加载依赖库
|
||
if WINDOWS_FLAG:
|
||
os.chdir(r'./lib/win')
|
||
Objdll = ctypes.CDLL(r'./HCNetSDK.dll')
|
||
else:
|
||
os.chdir(r'./lib/linux')
|
||
Objdll = cdll.LoadLibrary(r'./libhcnetsdk.so')
|
||
|
||
SetSDKInitCfg() # 设置组件库和SSL库加载路径
|
||
|
||
# 初始化DLL
|
||
Objdll.NET_DVR_Init()
|
||
# 启用SDK写日志
|
||
Objdll.NET_DVR_SetLogToFile(3, bytes('C:\\SdkLog_Python\\', encoding="gbk"), False)
|
||
|
||
# 加载播放库
|
||
Playctrldll = ctypes.CDLL(r'./PlayCtrl.dll')
|
||
# 获取一个播放句柄
|
||
if not Playctrldll.PlayM4_GetPort(byref(PlayCtrl_Port)):
|
||
print(u'获取播放库句柄失败')
|
||
|
||
# 登录设备
|
||
(lUserId, device_info) = LoginDev(Objdll)
|
||
if lUserId < 0:
|
||
err = Objdll.NET_DVR_GetLastError()
|
||
print('Login device fail, error code is: %d' % Objdll.NET_DVR_GetLastError())
|
||
# 释放资源
|
||
Objdll.NET_DVR_Cleanup()
|
||
exit()
|
||
|
||
# 定义码流回调函数
|
||
funcRealDataCallBack_V30 = REALDATACALLBACK(RealDataCallBack_V30)
|
||
# 开启预览
|
||
lRealPlayHandle = OpenPreview(Objdll, lUserId, funcRealDataCallBack_V30)
|
||
if lRealPlayHandle < 0:
|
||
print ('Open preview fail, error code is: %d' % Objdll.NET_DVR_GetLastError())
|
||
# 登出设备
|
||
Objdll.NET_DVR_Logout(lUserId)
|
||
# 释放资源
|
||
Objdll.NET_DVR_Cleanup()
|
||
exit()
|
||
|
||
#show Windows
|
||
win.mainloop()
|
||
|
||
# 开始云台控制
|
||
lRet = Objdll.NET_DVR_PTZControl(lRealPlayHandle, PAN_LEFT, 0)
|
||
if lRet == 0:
|
||
print ('Start ptz control fail, error code is: %d' % Objdll.NET_DVR_GetLastError())
|
||
else:
|
||
print ('Start ptz control success')
|
||
|
||
# 转动一秒
|
||
sleep(1)
|
||
|
||
# 停止云台控制
|
||
lRet = Objdll.NET_DVR_PTZControl(lRealPlayHandle, PAN_LEFT, 1)
|
||
if lRet == 0:
|
||
print('Stop ptz control fail, error code is: %d' % Objdll.NET_DVR_GetLastError())
|
||
else:
|
||
print('Stop ptz control success')
|
||
|
||
# 关闭预览
|
||
Objdll.NET_DVR_StopRealPlay(lRealPlayHandle)
|
||
|
||
# 停止解码，释放播放库资源
|
||
if PlayCtrl_Port.value > -1:
|
||
Playctrldll.PlayM4_Stop(PlayCtrl_Port)
|
||
Playctrldll.PlayM4_CloseStream(PlayCtrl_Port)
|
||
Playctrldll.PlayM4_FreePort(PlayCtrl_Port)
|
||
PlayCtrl_Port = c_long(-1)
|
||
|
||
# 登出设备
|
||
Objdll.NET_DVR_Logout(lUserId)
|
||
|
||
# 释放资源
|
||
Objdll.NET_DVR_Cleanup()
|
||
|