devbox
盐析白兔
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

C++ 使用海康威视SDK将视频推流到rtmp服务器_相机sdk回调函数yuv推流rtmp

作者：盐析白兔 | 2024-03-14 11:04:40

赞

踩

相机sdk回调函数yuv推流rtmp

研究FFmpeg有两三年了，一直没写过这方面的文章，今天记一下。

由于工作关系，需要将化工企业内部的视频发布到一个部署在公网的视频服务器，然后由相关人员浏览。由于是化工企业，企业严禁外部的机器直接访问视频网络，最多提供一个跳板机。因此，两年多前，针对这种情况，基于FFmpeg研发了一个推流系统。

随着接入视频数量的增加，发现不能单纯的使用RTSP协议获取硬盘录像机视频数据了，海康威视的硬盘录像机，最多允许5个用户同时访问，如果使用RTSP的话，每路RTSP都相当于一个访问用户，因此，需要使用海康卫视的SDK将视频流转为FFmpeg帧。

其实海康威视在开发文档中给出了相应示例，这里我贴一下我写的这部分代码：

 

初始化程序，连接硬盘录像机：

  1. void HKVideo::initInput() {
  2.     inputSuccess = false;
  3.     NET_DVR_USER_LOGIN_INFO struLoginInfo = { 0 };
  4.     NET_DVR_DEVICEINFO_V40 struDeviceInfoV40 = { 0 };
  5.     while (true)
  6.     {
  7.         //---------------------------------------
  8.     // 初始化
  9.         NET_DVR_Init();
  10.         //设置连接时间与重连时间
  11.         NET_DVR_SetConnectTime(2000, 1);
  12.         NET_DVR_SetReconnect(2000, true);
  13.  
  14.         //---------------------------------------
  15.         //设置异常消息回调函数
  16.         NET_DVR_SetExceptionCallBack_V30(0, NULL, g_ExceptionCallBack, NULL);
  17.  
  18.         //---------------------------------------
  19.         // 注册设备
  20.  
  21.  
  22.         //登录参数，包括设备地址、登录用户、密码等
  23.         
  24.         struLoginInfo.bUseAsynLogin = 0; //同步登录方式
  25.         strcpy(struLoginInfo.sDeviceAddress, ip.c_str()); //设备IP地址
  26.         struLoginInfo.wPort = port; //设备服务端口
  27.         strcpy(struLoginInfo.sUserName, user.c_str()); //设备登录用户名
  28.         strcpy(struLoginInfo.sPassword, psd.c_str()); //设备登录密码
  29.  
  30.         //设备信息, 输出参数
  31.         
  32.  
  33.         lUserID = NET_DVR_Login_V40(&struLoginInfo, &struDeviceInfoV40);
  34.         printf("Login  code: %d\n", NET_DVR_GetLastError());
  35.         if (lUserID < 0)
  36.         {
  37.             
  38.             NET_DVR_Cleanup();
  39.             Sleep(3000);
  40.             continue;
  41.         }
  42.         else {
  43.             printf("lUserID:%d\n", lUserID);
  44.             break;
  45.         }
  46.         
  47.     }
  48.     int size = rtsp_json["channel"].size();
  49.     for (size_t i = 0; i < size; i++)
  50.     {
  51.         int lChannel = rtsp_json["channel"][i].asInt();
  52.         HKVideoPush* push = new HKVideoPush;
  53.         stringstream stream;
  54.         stream << rtmp;
  55.         stream << "_";
  56.         stream << lChannel;
  57.         string rtmp0;
  58.         stream >> rtmp0;
  59.         push->init(rtmp0);
  60.         printf("Channel:%s\n",push->rtmp);
  61.         //---------------------------------------
  62.         //启动预览并设置回调数据流
  63.         //HWND  h = (HWND)cvGetWindowHandle("Mywindow");
  64.         push->struPlayInfo.hPlayWnd = NULL;         //需要SDK解码时句柄设为有效值，仅取流不解码时可设为空
  65.         push->struPlayInfo.lChannel = lChannel;       //预览通道号
  66.         push->struPlayInfo.dwStreamType = 0;       //0-主码流，1-子码流，2-码流3，3-码流4，以此类推
  67.         push->struPlayInfo.dwLinkMode = 0;       //0- TCP方式，1- UDP方式，2- 多播方式，3- RTP方式，4-RTP/RTSP，5-RSTP/HTTP
  68.         push->struPlayInfo.bBlocked = 1;       //0- 非阻塞取流，1- 阻塞取流
  69.  
  70.         push->lRealPlayHandle = NET_DVR_RealPlay_V40(lUserID, &push->struPlayInfo, g_RealDataCallBack_V30, push);
  71.         if (push->lRealPlayHandle < 0)
  72.         {
  73.             printf("NET_DVR_RealPlay_V40 error, %d\n", NET_DVR_GetLastError());
  74.             NET_DVR_Logout(lUserID);
  75.             NET_DVR_Cleanup();
  76.         }
  77.         HKVideos[lChannel] = push;
  78.  
  79.         HANDLE h = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)StartHKPushThread, (PVOID)push, 1, 0); //创建子线程
  80.         ResumeThread(h);  //启动子线程
  81.     }
  82.     inputSuccess = true;
  83.     
  84. };

 

 回调函数：

  1. void HKVideoPush::AVFrame2AVPacket(unsigned char* pYUV, int srcWidth, int srcHeight) {
  2. //关键函数，将海康威视视频帧转为FFmpeg帧
  3.     AVFrame* avframe_tmp = av_frame_alloc();//申请一个新的帧
  4.     AVPacket* pkt = av_packet_alloc();//申请一个新的视频包
  5.     avframe_tmp->format = AV_PIX_FMT_YUV420P;
  6.     avframe_tmp->width = srcWidth;
  7.     avframe_tmp->height = srcHeight;
  8.     
  9.     avpicture_fill((AVPicture*)avframe_tmp, pYUV, AV_PIX_FMT_YUV420P, srcWidth, srcHeight);//将海康威视帧数据填充到FFmpeg视频帧中
  10.     //下面这一步很重要，如果没有下面三行代码，rtmp看到的视频颜色会失常
  11.     uint8_t* ptmp = avframe_tmp->data[1];
  12.     avframe_tmp->data[1] = avframe_tmp->data[2];
  13.     avframe_tmp->data[2] = ptmp;
  14.  
  15.     this->vpts += 1;
  16.     this->lock();//锁定，
  17.     avframe_tmp->pts = this->vpts;
  18.     int ret = avcodec_send_frame(this->vc, avframe_tmp);
  19.     
  20.     if (ret < 0) {
  21.         printf("Error avcodec_send_frame:%d\n", ret);
  22.         av_frame_free(&avframe_tmp);
  23.         this->unlock();
  24.         return;
  25.     }
  26.     av_frame_free(&avframe_tmp);
  27.     ret = avcodec_receive_packet(this->vc, pkt);
  28.     if (ret < 0)
  29.     {
  30.         printf("Error avcodec_receive_packet:%d\n", ret);
  31.         this->unlock();
  32.         return;
  33.     }
  34.     //this->lock();
  35.     this->Pkts.push_back(pkt);//将最终的到的视频包，推入一个vector中，其他线程会从vector中取出包，推流给rtmp
  36.     this->unlock();//解锁
  37. };
  38.  
  39. void CALLBACK DecCBFun(long nPort, char* pBuf, long nSize, FRAME_INFO* pFrameInfo, long nReserved1, long nReserved2)
  40. {
  41.     long lFrameType = pFrameInfo->nType;
  42.     if (lFrameType == T_YV12)
  43.     {
  44.         //将海康威视包转为FFmpeg帧
  45.         AVFrame2AVPacket((unsigned char*)pBuf, pFrameInfo->nWidth, pFrameInfo->nHeight);
  46.  
  47.     }
  48. }
  49.  
  50. void CALLBACK g_RealDataCallBack_V30(LONG lRealHandle, DWORD dwDataType, BYTE* pBuffer, DWORD dwBufSize, void* dwUser)
  51. {
  52.     HKVideoPush* push = (HKVideoPush*)dwUser;
  53.     int dRet = 0;
  54.     switch (dwDataType)
  55.     {
  56.     case NET_DVR_SYSHEAD: //系统头
  57.         //if (push->lPort >= 0)
  58.         //{
  59.         //    break;  //该通道取流之前已经获取到句柄，后续接口不需要再调用
  60.         //}
  61.  
  62.         if (push->lPort >= 0)
  63.         {
  64.             break;  //该通道取流之前已经获取到句柄，后续接口不需要再调用
  65.         }
  66.  
  67.         if (!PlayM4_GetPort(&push->lPort))  //获取播放库未使用的通道号
  68.         {
  69.             break;
  70.         }
  71.         //m_iPort = lPort; //第一次回调的是系统头，将获取的播放库port号赋值给全局port，下次回调数据时即使用此port号播放
  72.         if (dwBufSize > 0)
  73.         {
  74.  
  75.             if (!PlayM4_OpenStream(push->lPort, pBuffer, dwBufSize, 1600 * 1600))
  76.             {
  77.                 dRet = PlayM4_GetLastError(push->lPort);
  78.                 break;
  79.             }
  80.             //设置解码回调函数 仅仅解码不显示
  81.             if (!PlayM4_SetDecCallBack(push->lPort, DecCBFun))
  82.             {
  83.                 dRet = PlayM4_GetLastError(push->lPort);
  84.                 break;
  85.             }
  86.  
  87.             //设置解码回调函数 解码且显示
  88.             //if (!PlayM4_SetDecCallBackEx(nPort,DecCBFun,NULL,NULL))
  89.             //{
  90.             //  dRet=PlayM4_GetLastError(nPort);
  91.             //  break;
  92.             //}
  93.  
  94.             //打开视频解码
  95.             if (!PlayM4_Play(push->lPort, push->hWnd))
  96.             {
  97.                 dRet = PlayM4_GetLastError(push->lPort);
  98.                 break;
  99.             }
  100.  
  101.  
  102.         }
  103.         break;
  104.     case NET_DVR_STREAMDATA:   //码流数据
  105.         if (dwBufSize > 0 && push->lPort != -1)
  106.         {
  107.             BOOL inData = PlayM4_InputData(push->lPort, pBuffer, dwBufSize);
  108.             while (!inData)
  109.             {
  110.                 inData = PlayM4_InputData(push->lPort, pBuffer, dwBufSize);
  111.                 OutputDebugString("PlayM4_InputData failed \n");
  112.  
  113.             }
  114.         }
  115.         break;
  116.     default: //其他数据
  117.         if (dwBufSize > 0)
  118.         {
  119.  
  120.         }
  121.         break;
  122.     }
  123. }
  124. void CALLBACK g_ExceptionCallBack(DWORD dwType, LONG lUserID, LONG lHandle, void* pUser)
  125. {
  126.     printf("异常回调%X\n", dwType);
  127.     switch (dwType)
  128.     {
  129.     case EXCEPTION_RECONNECT:    //预览时重连
  130.         break;
  131.     default:
  132.         break;
  133.     }
  134.  
  135. }

 

 总结：

缺点：这种方式，会造成大量计算，每一帧都会经过计算，我在实际项目使用的时候，使用了一个CPU为J1900的工控机，读取一个硬盘录像机的5路摄像头，CPU占用率达到90% 以上。主要耗费CPU的就是AVFrame2AVPacket函数，它需要大量计算

优点：节省带宽。。。因为海康卫视的SDK解出来的视频包很小，子码流时每路最多200Kb/s（要知道我用RTSP时，子码流每路800Kb）

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/233744
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号