海康攝像頭opencv實時回調

由於項目的需求，需要對海康攝像頭進行取流操作。

#include <iostream>
#include <opencv2\opencv.hpp>
#include <HCNetSDK.h>
#include <plaympeg4.h>
using namespace std;
using namespace cv;

LONG nPort = -1;
HWND hWnd = NULL;
Mat image;

//解碼回調 視頻爲YUV數據(YV12)，音頻爲PCM數據
void CALLBACK DecCBFun(long nPort, char * pBuf, long nSize, FRAME_INFO * pFrameInfo, long nReserved1, long nReserved2)
{
	if (pFrameInfo->nType == T_YV12)
	{
		if (image.empty())
		{
			image.create(pFrameInfo->nHeight, pFrameInfo->nWidth, CV_8UC3);
		}
		Mat YUVImage(pFrameInfo->nHeight + pFrameInfo->nHeight / 2, pFrameInfo->nWidth, CV_8UC1, (unsigned char*)pBuf);

		cvtColor(YUVImage, image, COLOR_YUV2BGR_YV12);
		resize(image, image, Size(1000, 600));
		imshow("view", image);
		waitKey(15);
		YUVImage.~Mat();
	}
}

///實時流回調
void CALLBACK fRealDataCallBack_V30(LONG lRealHandle, DWORD dwDataType, BYTE *pBuffer, DWORD dwBufSize, void *pUser)
{
	switch (dwDataType)
	{
	case NET_DVR_SYSHEAD:    //系統頭
		if (!PlayM4_GetPort(&nPort)) //獲取播放庫未使用的通道號
		{
			nPort = -1;
			break;
		}
		if (dwBufSize > 0)
		{
			//實時流播放模式
			if (!PlayM4_SetStreamOpenMode(nPort, STREAME_REALTIME))
				break;
			//打開流接口
			if (!PlayM4_OpenStream(nPort, pBuffer, dwBufSize, 1024 * 1024 * 8))
				break;
			//播放開始
			if (!PlayM4_Play(nPort, NULL))
				break;
			//設置解碼回調函數 僅僅解碼不顯示
			if (!PlayM4_SetDecCallBack(nPort, DecCBFun))
				break;
		}
		break;
	case NET_DVR_STREAMDATA:   //碼流數據
		if (dwBufSize > 0 && nPort != -1)
		{
			if (!PlayM4_InputData(nPort, pBuffer, dwBufSize))
				break;
		}
		break;
	}
}


int main() 
{
	/*設備SDK初始化*/
	NET_DVR_Init();
	NET_DVR_SetConnectTime(2000, 1);
	NET_DVR_SetReconnect(2000, true);
	NET_DVR_SetRecvTimeOut(5000);

	//---------------------------------------
	// 註冊設備
	LONG lUserID;
	NET_DVR_DEVICEINFO_V30 struDeviceInfo;
	lUserID = NET_DVR_Login_V30("192.168.1.64", 8000, "admin", "123456", &struDeviceInfo);
	if (lUserID < 0)
	{
		printf("Login error, %d\n", NET_DVR_GetLastError());
		NET_DVR_Cleanup();
		return -1;
	}
	printf("Login success!\n");

	//--------------------------------------
	long lRealPlayHandle;
	//HWND hWnd = GetConsoleWindow(); //獲取窗口句柄
	NET_DVR_PREVIEWINFO struPlayInfo = { 0 };
	struPlayInfo.hPlayWnd = NULL; //需要 SDK 解碼時句柄設爲有效值，僅取流不解碼時可爲空(設置回調函數)
	struPlayInfo.lChannel = 1; //預覽通道號
	struPlayInfo.dwStreamType = 0; //0-主碼流，1-子碼流，2-碼流 3，3-碼流 4，以此類推
	struPlayInfo.dwLinkMode = 0; //0- TCP 方式，1- UDP 方式，2- 多播方式，3- RTP 方式，4-RTP/RTSP，5-RSTP/HTTP
	struPlayInfo.bBlocked = 0; //0- 非阻塞取流，1- 阻塞取流
	lRealPlayHandle = NET_DVR_RealPlay_V40(lUserID, &struPlayInfo,fRealDataCallBack_V30, NULL);
	if (lRealPlayHandle < 0) //預覽失敗
	{
		printf("NET_DVR_RealPlay_V40 error, %d\n", NET_DVR_GetLastError());
		NET_DVR_Logout(lUserID);
		NET_DVR_Cleanup();
		return-1;
	}
	Sleep(5000);
	//關閉預覽
	NET_DVR_StopRealPlay(lRealPlayHandle);
	//註銷用戶
	NET_DVR_Logout(lUserID);
	//釋放 SDK 資源
	NET_DVR_Cleanup();



	return 0;
}