Python調用海康sdk操作熱成像設備獲取對應點溫度， 海康官方提供有Java版的sdk，遺憾的是里面提供的api比較舊了新版的api需要根據c++版的開發文檔自己寫對應的Python接口和類。這對于不熟悉c++的開發人員比較吃力。下面的代碼示例了通過海康的SDK獲取熱成像畫面上某一點的具體溫度。hCNetSDK = CDLL('./libhcnetsdk.so')是海康sdk的目錄，可以是相對路徑也可以是絕對路徑。
1.dll動態庫python函數封裝
from ctypes import c_int32, c_char_p, c_void_p, c_float, c_size_t, c_ubyte, c_long, cdll, POINTER, CDLL, c_bool, c_long, c_short
from hk_class import *
import sys
if 'linux' == sys.platform:
hCNetSDK = CDLL('./libhcnetsdk.so')
else:
hCNetSDK = CDLL('./HCNetSDK.dll')
SERIALNO_LEN = 48 # 序列號長度
NAME_LEN = 32 # 用戶名長度
# //boolean NET_DVR_Init();
NET_DVR_Init = hCNetSDK.NET_DVR_Init
NET_DVR_Init.restype = c_bool
NET_DVR_Init.argtypes = ()
# boolean NET_DVR_Cleanup();
NET_DVR_Cleanup = hCNetSDK.NET_DVR_Cleanup
NET_DVR_Cleanup.restype = c_bool
NET_DVR_Cleanup.argtypes = ()
# NativeLong NET_DVR_Login_V30(String sDVRIP, short wDVRPort, String sUserName, String sPassword, NET_DVR_DEVICEINFO_V30 lpDeviceInfo);
NET_DVR_Login_V30 = hCNetSDK.NET_DVR_Login_V30
NET_DVR_Login_V30.restype = c_long
NET_DVR_Login_V30.argtypes = (c_char_p, c_short, c_char_p, c_char_p, POINTER(NET_DVR_DEVICEINFO_V30))
# boolean NET_DVR_Logout_V30(NativeLong lUserID);
NET_DVR_Logout_V30 = hCNetSDK.NET_DVR_Logout_V30
NET_DVR_Logout_V30.restype = c_bool
NET_DVR_Logout_V30.argtypes = (c_long,)
# boolean NET_DVR_SetSTDConfig(NativeLong lUserID, int dwCommand, NET_DVR_STD_CONFIG lpInConfigParam);
NET_DVR_SetSTDConfig = hCNetSDK.NET_DVR_SetSTDConfig
NET_DVR_SetSTDConfig.restype = c_bool
NET_DVR_SetSTDConfig.argtypes = (c_long, c_int32, NET_DVR_STD_CONFIG)
# boolean NET_DVR_GetSTDConfig(NativeLong lUserID, int dwCommand, NET_DVR_STD_CONFIG lpOutConfigParam);
NET_DVR_GetSTDConfig = hCNetSDK.NET_DVR_GetSTDConfig
NET_DVR_GetSTDConfig.restype = c_bool
NET_DVR_GetSTDConfig.argtypes = (c_long, c_int32, NET_DVR_STD_CONFIG)
# boolean NET_DVR_CaptureJPEGPicture_WithAppendData(NativeLong lUserID, int lChannel, NET_DVR_JPEGPICTURE_WITH_APPENDDATA lpJpegWithAppend);
NET_DVR_CaptureJPEGPicture_WithAppendData = hCNetSDK.NET_DVR_CaptureJPEGPicture_WithAppendData
NET_DVR_CaptureJPEGPicture_WithAppendData.restype = c_bool
NET_DVR_CaptureJPEGPicture_WithAppendData.argtypes = (c_long, c_int32, POINTER(NET_DVR_JPEGPICTURE_WITH_APPENDDATA))
# int NET_DVR_GetLastError();
NET_DVR_GetLastError = hCNetSDK.NET_DVR_GetLastError
NET_DVR_GetLastError.restype = c_int32
NET_DVR_GetLastError.argtypes = ()
# 啟用日志文件寫入接口
# boolean NET_DVR_SetLogToFile(int bLogEnable, String strLogDir, boolean bAutoDel);
NET_DVR_SetLogToFile = hCNetSDK.NET_DVR_SetLogToFile
NET_DVR_SetLogToFile.restype = c_bool
NET_DVR_SetLogToFile.argtypes = (c_int32, c_char_p, c_bool)
# 單幀數據捕獲并保存成JPEG存放在指定的內存空間中。
# BOOL NET_DVR_CaptureJPEGPicture_NEW(
# LONG lUserID,
# LONG lChannel,
# LPNET_DVR_JPEGPARA lpJpegPara,
# char *sJpegPicBuffer,
# DWORD dwPicSize,
# LPDWORD lpSizeReturned
# );
NET_DVR_CaptureJPEGPicture_new = hCNetSDK.NET_DVR_CaptureJPEGPicture_NEW
NET_DVR_CaptureJPEGPicture_new.restype = c_bool
NET_DVR_CaptureJPEGPicture_new.argtypes = (c_long, c_long, POINTER(NET_DVR_JPEGPARA), c_char_p, c_ulong, POINTER(c_ulong))
# BOOL NET_DVR_CaptureJPEGPicture_NEW(
# LONG lUserID,
# LONG lChannel,
# LPNET_DVR_JPEGPARA lpJpegPara,
# char *sJpegPicBuffer,
# DWORD dwPicSize,
# LPDWORD lpSizeReturned
# );
NET_DVR_CaptureJPEGPicture = hCNetSDK.NET_DVR_CaptureJPEGPicture
NET_DVR_CaptureJPEGPicture.restype = c_bool
NET_DVR_CaptureJPEGPicture.argtypes = (c_long, c_long, POINTER(NET_DVR_JPEGPARA), c_char_p)
2.結構體python類封裝
from ctypes import *
SERIALNO_LEN = 48 # 序列號長度
NAME_LEN = 32 # 用戶名長度py
class NET_DVR_DEVICEINFO_V30(Structure):
_fields_ = [('sSerialNumber', c_ubyte * SERIALNO_LEN), ('byAlarmInPortNum', c_byte), ('byAlarmOutPortNum', c_byte), ('byDiskNum', c_byte),
('byDVRType', c_byte), ('byChanNum', c_byte), ('byStartChan', c_byte), ('byAudioChanNum', c_byte), ('byIPChanNum', c_byte), ('byRes1', c_ubyte * 24)]
class NET_VCA_POINT(Structure):
_fields_ = [('fX', c_float), ('fY', c_float)]
class NET_VCA_POLYGON(Structure):
_fields_ = [('dwPointNum', c_ulong), ('struPos',NET_VCA_POINT * 10)]
class NET_DVR_THERMOMETRY_PRESETINFO_PARAM(Structure):
_fields_ = [('byEnabled', c_byte), ('byRuleID', c_short), ('wDistance', c_short), ('fEmissivity', c_float), ('byDistanceUnit', c_byte), ('byRes', c_ubyte * 2), ('byReflectiveEnabled', c_byte),
('fReflectiveTemperature', c_float), ('szRuleName', c_ubyte * NAME_LEN), ('byRes1', c_ubyte * 63), ('byRuleCalibType', c_byte), ('struPoint', NET_VCA_POINT), ('struRegion', NET_VCA_POLYGON)]
class NET_DVR_THERMOMETRY_PRESETINFO(Structure):
_fields_ = [('dwSize', c_ulong), ('wPresetNo', c_short), ('byRes', c_ubyte * 2),
('struPresetInfo', NET_DVR_THERMOMETRY_PRESETINFO_PARAM * 40)]
class NET_DVR_THERMOMETRY_COND(Structure):
_fields_ = [('dwSize', c_ulong), ('dwChannel', c_ulong),
('wPresetNo', c_short), ('byRes', c_ubyte * 62)]
class BYTE_ARRAY(Structure):
_fields_ = [('byValue', c_byte * 2097152)]
class NET_DVR_STD_CONFIG(Structure):
_fields_ = [('lpCondBuffer', POINTER(NET_DVR_THERMOMETRY_COND)), ('dwCondSize', c_ulong), ('lpInBuffer', POINTER(NET_DVR_THERMOMETRY_PRESETINFO)), ('dwInSize', c_ulong), ('lpOutBuffer', POINTER(NET_DVR_THERMOMETRY_PRESETINFO)), ('dwOutSize', c_ulong),
('lpStatusBuffer', POINTER(BYTE_ARRAY)), ('dwStatusSize', c_ulong), ('lpXmlBuffer', c_void_p), ('dwXmlSize', c_ulong), ('byDataType', c_bool), ('byRes', c_ubyte * 23)]
class NET_VCA_RECT(Structure):
_fields_ = [('fX', c_char),('fY', c_char),('fWidth', c_char),('fHeight', c_char)]
class NET_DVR_JPEGPICTURE_WITH_APPENDDATA(Structure):
_fields_ = [('dwSize', c_int32), ('dwChannel', c_int32), ('dwJpegPicLen', c_int32), ('pJpegPicBuff', POINTER(BYTE_ARRAY)), ('dwJpegPicWidth', c_int32),
('dwJpegPicHeight', c_int32), ('dwP2PDataLen', c_int32), ('pP2PDataBuff', POINTER(BYTE_ARRAY)), ('byIsFreezedata', c_byte), ('byRes', c_byte * 255)]
# JPEG圖像信息結構體。
# struct{
# WORD wPicSize;
# WORD wPicQuality;
# }NET_DVR_JPEGPARA,*LPNET_DVR_JPEGPARA;
class NET_DVR_JPEGPARA(Structure):
_fields_ = [('wPicSize', c_ulong),('wPicQuality', c_ulong)]
3.sdk自封裝
import hk_dll as hk_dll
import hk_class as hk_class
from ctypes import *
import struct
# from numba import njit
# 設備信息
m_strDeviceInfo = None
SERIALNO_LEN = 48 # 序列號長度
NAME_LEN = 32 # 用戶名長度py
point_bytes = (c_byte * 4)()
#用戶登錄信息
m_strDeviceInfo = None
#測溫信息
m_strJpegWithAppenData = None
# 初始化
def init():
return hk_dll.NET_DVR_Init()
# 登錄
def login(ip, port, username, password):
# 注冊
m_strDeviceInfo = hk_class.NET_DVR_DEVICEINFO_V30()
m_strDeviceInfo.sSerialNumber = (c_ubyte * SERIALNO_LEN)()
m_strDeviceInfo.byRes1 = (c_ubyte * 24)()
lUserID = hk_dll.NET_DVR_Login_V30(bytes(ip), port, bytes(username), bytes(password), byref(m_strDeviceInfo))
# 打開SDK寫日志的功能
hk_dll.NET_DVR_SetLogToFile(3, b'./sdklog', False)
return lUserID
# 退出登錄
def logout(lUserID):
hk_dll.NET_DVR_Logout_V30(lUserID)
# 釋放sdk
def cleanup():
hk_dll.NET_DVR_Cleanup()
# 獲取抓拍圖片最高溫度
def get_temperature_all(lUserID):
ret, m_strJpegWithAppenData = get_temperature0(lUserID)
max_temperature = -50
min_temperature = 120
byValue = m_strJpegWithAppenData.pP2PDataBuff.contents.byValue
if ret:
for x in range(m_strJpegWithAppenData.dwJpegPicWidth):
for y in range(m_strJpegWithAppenData.dwJpegPicHeight):
temperature = struct.unpack('<f', struct.pack('4b', *get_bytes(byValue , (m_strJpegWithAppenData.dwJpegPicWidth * y + x) * 4, 4)))[0]
max_temperature = temperature if temperature > max_temperature else max_temperature
min_temperature = temperature if temperature < min_temperature else min_temperature
return True, max_temperature, min_temperature
return False, max_temperature, min_temperature
# 獲取給定點列表最高溫度
def get_temperature_max(points, sourceWidth, sourceHeight, lUserID):
ret, m_strJpegWithAppenData = get_temperature0(lUserID)
if(len(points) < 2):
return False, -2
if ret:
x1, y1 = point2point(points[0][0], points[0][1], sourceWidth, sourceHeight,
m_strJpegWithAppenData.dwJpegPicWidth, m_strJpegWithAppenData.dwJpegPicHeight)
x2, y2 = point2point(points[1][0], points[1][1], sourceWidth, sourceHeight,
m_strJpegWithAppenData.dwJpegPicWidth, m_strJpegWithAppenData.dwJpegPicHeight)
if x1 > x2 or y1 > y2:
return False, -3
byValue = m_strJpegWithAppenData.pP2PDataBuff.contents.byValue
max_temperature = -50.0
for x in range(x1, x2 + 1):
for y in range(y1, y2 + 1):
# 160 * 120
temperature = struct.unpack('<f', struct.pack('4b', *get_bytes(byValue, (m_strJpegWithAppenData.dwJpegPicWidth * y + x) * 4, 4)))[0]
max_temperature = temperature if temperature > max_temperature else max_temperature
return True, max_temperature
return False, -1
# 獲取某點的溫度
def get_temperature(x, y, sourceWidth, sourceHeight, lUserID):
ret, m_strJpegWithAppenData = get_temperature0(lUserID)
if ret:
# m_strJpegWithAppenData.pP2PDataBuff.contents.byValue
x, y = point2point(x, y, sourceWidth, sourceHeight, m_strJpegWithAppenData.dwJpegPicWidth, m_strJpegWithAppenData.dwJpegPicHeight)
byValue = m_strJpegWithAppenData.pP2PDataBuff.contents.byValue
return True, struct.unpack('<f', struct.pack('4b', *get_bytes(byValue, (m_strJpegWithAppenData.dwJpegPicWidth * y + x) * 4, 4)))[0]
return False, 0.0
# 截取指定下標的和長度的返回數據
def get_bytes(src_bytes, offset, length):
global point_bytes
for i in range(length):
point_bytes[i] = src_bytes[offset + i]
# del src_bytes
return point_bytes
# 獲取溫度
def get_temperature0(lUserID):
bRet = False
global m_strJpegWithAppenData
if m_strJpegWithAppenData is None:
m_strJpegWithAppenData = hk_class.NET_DVR_JPEGPICTURE_WITH_APPENDDATA()
m_strJpegWithAppenData.byRes = (c_byte * 255)()
m_strJpegWithAppenData.dwChannel = 1
m_strJpegWithAppenData.pJpegPicBuff = pointer(
hk_class.BYTE_ARRAY((c_byte * 2097152)()))
m_strJpegWithAppenData.pP2PDataBuff = pointer(
hk_class.BYTE_ARRAY((c_byte * 2097152)()))
m_strJpegWithAppenData.dwSize = sizeof(m_strJpegWithAppenData)
bRet = hk_dll.NET_DVR_CaptureJPEGPicture_WithAppendData(lUserID, 2, byref(m_strJpegWithAppenData))
if bRet:
# 測溫數據
print(m_strJpegWithAppenData.dwP2PDataLen)
if m_strJpegWithAppenData.dwP2PDataLen > 0:
return True, m_strJpegWithAppenData
return False, None
# 坐標轉換
# @njit
def point2point(x, y, sourceWidth, sourceHeight, targetWidth, targetHeight):
x = x * targetWidth / sourceWidth
y = y * targetHeight / sourceHeight
x = x if x <= targetWidth else targetWidth
x = 0 if x < 0 else x
y = y if y <= targetHeight else targetHeight
y = 0 if y < 0 else y
return int(x), int(y)
#抓拍圖片
# lUserID 登錄用戶id
# lChannel 渠道號
# dir 文件保存路徑
def captureJPEGPicture(lUserID, lChannel, dir):
jpegpara= hk_class.NET_DVR_JPEGPARA()
jpegpara.wPicSize = 0xff
jpegpara.wPicQuality = 2
p = c_char_p()
s = byref(c_ulong())
# return hk_dll.NET_DVR_CaptureJPEGPicture(lUserID, lChannel, byref(jpegpara), p, 2048, s)
return hk_dll.NET_DVR_CaptureJPEGPicture(lUserID, lChannel, byref(jpegpara), dir)
def getLastError():
return hk_dll.NET_DVR_GetLastError()
4.測試調用
import hk_sdk as hk_sdk
import hk_class as hkclass
import time
def test():
result = hk_sdk.init()
if not result:
print('初始化失敗')
return False
print('初始化成功')
lUserIDs = []
lUserID = hk_sdk.login(b"192.168.8.16", 8000, b'admin', b'a1234567')
if lUserID != -1:
print('lUserID ', lUserID)
lUserIDs.append(lUserID)
time.sleep(5)
lUserID = hk_sdk.login(b"192.168.8.15", 8000, b'admin', b'a1234567')
if lUserID != -1:
print('lUserID ', lUserID)
lUserIDs.append(lUserID)
for i in range(10):
temperature_list = [[0, 0], [255, 255]]
# result, max_temperature, min_temperature = hk_sdk.get_temperature_all();
# if not result:
# print("獲取溫度失敗!")
# return False
# print("max溫度是" , max_temperature,"min溫度是", min_temperature)
userid = lUserIDs[int(i % 2)]
result, temperature = hk_sdk.get_temperature_max(temperature_list, 1280, 720, userid)
# result, temperature, m = hk_sdk.get_temperature_all();
if result:
print("獲取的溫度是", temperature, i)
else:
print("獲取溫度失敗 ", userid)
time.sleep(1)
# 退出登錄
hk_sdk.logout(lUserIDs[0])
# 釋放
hk_sdk.cleanup()
if __name__ == "__main__":
test()
好了代碼主要就這么多，喜歡的伙伴可以點贊留言加關注哦。
