通过多线程同时获取H264和H265码流

一.RV1126 VI采集摄像头数据并同时编码H264、H265的大概流程编辑编辑

1.1初始化VI模块：

1.2H264、H265的VENC模块初始化：

1.3VI分别绑定H264的VENC层和H265的VENC层：

1.4开启H264线程采集H264的VENC数据：

1.5开启H265线程采集H265的VENC数据：

二.代码实战：

一.RV1126 VI采集摄像头数据并同时编码H264、H265的大概流程

RV1126利用多线程同时获取H264文件、H265文件的过程一般分为上图的7步骤，分别是：VI模块的初始化、H264的VENC模块初始化、H265的VENC模块初始化、VI绑定H264的VENC模块、VI绑定H265的VENC模块，开启H264线程获取H264码流并保存、开启H265线程获取H265码流并保存。

1.1初始化VI模块：

VI模块的初始化实际上就是对VI_CHN_ATTR_S的参数进行设置、然后调用RK_MPI_VI_SetChnAttr设置VI模块并使能RK_MPI_VI_EnableChn，伪代码如下：

VI_CHN_ATTR_S vi_chn_attr;

。。。。。。。。。。。。。。。(这里是设置VI的属性)

ret = RK_MPI_VI_SetChnAttr(CAMERA_ID, 0, &vi_chn_attr);

ret |= RK_MPI_VI_EnableChn(CAMERA_ID, 0);

1.2H264、H265的VENC模块初始化：

VENC_CHN_ATTR_S h264_venc_chn_attr;

..................................

RK_MPI_VENC_CreateChn(H264_VENC_CHN, &h264_venc_chn_attr);

VENC_CHN_ATTR_S h265_venc_chn_attr;

..................................

RK_MPI_VENC_CreateChn(H265_VENC_CHN, &h265_venc_chn_attr);

注意：这里需要创建两个编码器层，分别是H264编码器和H265编码器。

1.3VI分别绑定H264的VENC层和H265的VENC层：

VI节点分别绑定H264的VENC节点和H265节点，伪代码如下：

//VI模块节点的设置

MPP_CHN_S vi_chn_s;

vi_chn_s.enModId = RK_ID_VI;

vi_chn_s.s32ChnId = 0;

//H264的VENC模块节点设置

MPP_CHN_S h264_venc_chn_s;

h264_venc_chn_s.enModId = RK_ID_VENC;

h264_venc_chn_s.s32ChnId = H264_VENC_CHN;

ret = RK_MPI_SYS_Bind(&vi_chn_s, &h264_venc_chn_s);

//H265的VENC模块节点设置

MPP_CHN_S h265_venc_chn_s;

h265_venc_chn_s.enModId = RK_ID_VENC;

h265_venc_chn_s.s32ChnId = H265_VENC_CHN;

ret = RK_MPI_SYS_Bind(&vi_chn_s, &h265_venc_chn_s);

1.4开启H264线程采集H264的VENC数据：

开启一个线程去采集每一帧H264的VENC模块数据，使用的API是RK_MPI_SYS_GetMediaBuffer， 模块ID是RK_ID_VENC，通道号ID是H264 VENC创建的ID号。这个API伪代码如下：

while(1)

{

.........................

mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_VENC, H264_VENC_CHN, -1);

fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, h264_file);

.......................

}

1.5开启H265线程采集H265的VENC数据：

开启一个线程去采集每一帧H265的VENC模块数据，使用的API是RK_MPI_SYS_GetMediaBuffer， 模块ID是RK_ID_VENC，通道号ID是H265 VENC创建的ID号。这个API伪代码如下：

while(1)

{

.........................

mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_VENC, H265_VENC_CHN, -1);

fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, h265_file);

.......................

}

二.代码实战：

#include <assert.h>
#include <fcntl.h>
#include <getopt.h>
#include <pthread.h>
#include <signal.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <unistd.h>// #include "common/sample_common.h"
#include "rkmedia_api.h"#define CAMERA_PATH "rkispp_scale0"
#define CAMERA_ID 0
#define CAMERA_CHN 0
#define H264_VENC_CHN 0
#define H265_VENC_CHN 1//创建线程获取H264码流数据并保存
void * get_h264_stream_thread(void * args)
{pthread_detach(pthread_self());FILE * h264_file = fopen("test_camera.h264", "w+");MEDIA_BUFFER mb ;while (1){//获取每一帧H264编码码流mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_VENC, H264_VENC_CHN, -1);if(!mb){printf("Get H264_Venc Buffer break....\n");break;}fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, h264_file);RK_MPI_MB_ReleaseBuffer(mb);}return NULL;
}//创建线程获取H265码流数据并保存
void * get_h265_stream_thread(void * args)
{pthread_detach(pthread_self());FILE * h265_file = fopen("test_camera.h265", "w+");MEDIA_BUFFER mb ;while (1){//获取每一帧H265编码码流mb = RK_MPI_SYS_GetMediaBuffer(RK_ID_VENC, H265_VENC_CHN , -1);if(!mb){printf("Get H265_VENC Buffer break...\n");break;}fwrite(RK_MPI_MB_GetPtr(mb), RK_MPI_MB_GetSize(mb), 1, h265_file);RK_MPI_MB_ReleaseBuffer(mb);}return NULL;
}int main(int argc, char *argv[])
{int ret;VI_CHN_ATTR_S vi_chn_attr;vi_chn_attr.pcVideoNode = CAMERA_PATH; //设置视频设备节点路径vi_chn_attr.u32Width = 1920; //设置分辨率的宽度vi_chn_attr.u32Height = 1080; //设置分辨率的高度vi_chn_attr.enPixFmt = IMAGE_TYPE_NV12; //设置图像类型vi_chn_attr.enBufType = VI_CHN_BUF_TYPE_MMAP;//设置VI获取类型vi_chn_attr.u32BufCnt = 3; //设置缓冲数量vi_chn_attr.enWorkMode = VI_WORK_MODE_NORMAL; //设置VI工作类型ret = RK_MPI_VI_SetChnAttr(CAMERA_ID, CAMERA_CHN, &vi_chn_attr);if(ret){printf("Vi Set Attr Failed.....\n");return 0;}else{printf("Vi Set Attr Success.....\n");}ret = RK_MPI_VI_EnableChn(CAMERA_ID, CAMERA_CHN);if(ret){printf("Vi Enable Attr Failed.....\n");return 0;}else{printf("Vi Enable Attr Success.....\n");}VENC_CHN_ATTR_S h264_venc_chn_attr;//******   H264  设置VENC基础属性   ************************//h264_venc_chn_attr.stVencAttr.enType = RK_CODEC_TYPE_H264; //设置编码器类型h264_venc_chn_attr.stVencAttr.u32PicWidth = 1920;//设置编码分辨率宽度h264_venc_chn_attr.stVencAttr.u32PicHeight = 1080;//设置编码分辨率高度h264_venc_chn_attr.stVencAttr.u32VirWidth = 1920;//设置编码分辨率虚宽h264_venc_chn_attr.stVencAttr.u32VirHeight = 1080;//设置编码分辨率虚高h264_venc_chn_attr.stVencAttr.u32Profile = 66;//设置编码等级h264_venc_chn_attr.stVencAttr.imageType = IMAGE_TYPE_NV12;//设置编码图像类型h264_venc_chn_attr.stVencAttr.enRotation = VENC_ROTATION_0;//设置编码的旋转角度//********* H264 设置H264码率控制属性  *******************//h264_venc_chn_attr.stRcAttr.enRcMode = VENC_RC_MODE_H264CBR; //设置H264的CBR码率控制模式h264_venc_chn_attr.stRcAttr.stH264Cbr.u32Gop = 25;//设置GOP关键帧间隔//25/1 NUM/DEN == FrameRateh264_venc_chn_attr.stRcAttr.stH264Cbr.u32SrcFrameRateDen = 1; //设置源帧率分母h264_venc_chn_attr.stRcAttr.stH264Cbr.u32SrcFrameRateNum = 25;//设置源帧率分子h264_venc_chn_attr.stRcAttr.stH264Cbr.fr32DstFrameRateDen = 1;//设置目标帧率分母h264_venc_chn_attr.stRcAttr.stH264Cbr.fr32DstFrameRateNum = 25;//设置目标帧率分子h264_venc_chn_attr.stRcAttr.stH264Cbr.u32BitRate = 8388608; //设置码率大小ret = RK_MPI_VENC_CreateChn(H264_VENC_CHN, &h264_venc_chn_attr);if(ret){printf("Create H264 Venc Failed .....\n");return 0;}else{printf("Create H264 Venc Success .....\n");}VENC_CHN_ATTR_S h265_venc_chn_attr;//******   H265  设置VENC基础属性   ************************//h265_venc_chn_attr.stVencAttr.enType = RK_CODEC_TYPE_H265;//设置编码器类型h265_venc_chn_attr.stVencAttr.u32PicWidth = 1920;//设置编码分辨率宽度h265_venc_chn_attr.stVencAttr.u32PicHeight = 1080;//设置编码分辨率高度h265_venc_chn_attr.stVencAttr.u32VirWidth = 1920;//设置编码分辨率虚宽h265_venc_chn_attr.stVencAttr.u32VirHeight = 1080;//设置编码分辨率虚高h265_venc_chn_attr.stVencAttr.u32Profile = 77;//设置编码等级h265_venc_chn_attr.stVencAttr.imageType = IMAGE_TYPE_NV12;//设置编码图像类型h265_venc_chn_attr.stVencAttr.enRotation = VENC_ROTATION_0;//设置编码的旋转角度//********* H265 VENC RCMODE Set  *******************//h265_venc_chn_attr.stRcAttr.enRcMode = VENC_RC_MODE_H265CBR;//设置H265的CBR码率控制模式h265_venc_chn_attr.stRcAttr.stH264Cbr.u32Gop = 25;//设置GOP关键帧间隔//25/1 NUM/DEN == FrameRateh265_venc_chn_attr.stRcAttr.stH265Cbr.u32SrcFrameRateDen = 1; //设置源帧率分母h265_venc_chn_attr.stRcAttr.stH265Cbr.u32SrcFrameRateNum = 25;//设置源帧率分子h265_venc_chn_attr.stRcAttr.stH265Cbr.fr32DstFrameRateDen = 1;//设置目标帧率分母h265_venc_chn_attr.stRcAttr.stH265Cbr.fr32DstFrameRateNum = 25;//设置目标帧率分子h265_venc_chn_attr.stRcAttr.stH265Cbr.u32BitRate = 8388608; //设置码率大小ret = RK_MPI_VENC_CreateChn(H265_VENC_CHN, &h265_venc_chn_attr);if(ret){printf("Create H265 Venc Failed .....\n");return 0;}else{printf("Create H265 Venc Success .....\n");}//VI_CHNMPP_CHN_S vi_chn_s;vi_chn_s.enModId = RK_ID_VI;vi_chn_s.s32ChnId = CAMERA_CHN;//H264_VENC_CHNMPP_CHN_S h264_chn_s;h264_chn_s.enModId = RK_ID_VENC;h264_chn_s.s32ChnId = H264_VENC_CHN;//H265_VENC_CHNMPP_CHN_S h265_chn_s;h265_chn_s.enModId = RK_ID_VENC;h265_chn_s.s32ChnId = H265_VENC_CHN;//VI Bind H264_VENCret = RK_MPI_SYS_Bind(&vi_chn_s, &h264_chn_s);if(ret){printf("Vi Bind H264_Venc Failed .....\n");}else{printf("Vi Bind H264_Venc Success .....\n");}//VI Bind H265_VENCret = RK_MPI_SYS_Bind(&vi_chn_s, &h265_chn_s);if(ret){printf("Vi Bind H265_Venc Failed .....\n");}else{printf("Vi Bind H265_Venc Success .....\n");}pthread_t h264_pid, h265_pid;pthread_create(&h264_pid, NULL, get_h264_stream_thread, NULL);//创建线程获取H264码流数据并保存pthread_create(&h265_pid, NULL, get_h265_stream_thread, NULL);//创建线程获取H265码流数据并保存while (1){sleep(1);}RK_MPI_SYS_UnBind(&vi_chn_s, &h264_chn_s);RK_MPI_SYS_UnBind(&vi_chn_s, &h265_chn_s);RK_MPI_VI_DisableChn(0, 0);RK_MPI_VENC_DestroyChn(0);RK_MPI_VENC_DestroyChn(1);return 0;
}