当前位置： 首页 > article >正文

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

article 2025/3/4 12:25:59

目录

一.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 == FrameRate
    h264_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 == FrameRate
    h265_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_CHN
    MPP_CHN_S vi_chn_s;
    vi_chn_s.enModId = RK_ID_VI;
    vi_chn_s.s32ChnId = CAMERA_CHN;

    //H264_VENC_CHN
    MPP_CHN_S h264_chn_s;
    h264_chn_s.enModId = RK_ID_VENC;
    h264_chn_s.s32ChnId = H264_VENC_CHN;

    //H265_VENC_CHN
    MPP_CHN_S h265_chn_s;
    h265_chn_s.enModId = RK_ID_VENC;
    h265_chn_s.s32ChnId = H265_VENC_CHN;


    //VI Bind H264_VENC
    ret = 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_VENC
    ret = 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;
}


http://www.kler.cn/a/570123.html

相关文章：

  • scala基础学习-匹配模式
  • Kettle 最常见的性能优化面试题
  • 计算机网络——因特网和互联网
  • 【AI论文】MedVLM-R1：通过强化学习激励视觉语言模型（VLMs）的医疗推理能力
  • 鸿蒙5.0实战案例：基于hvigor插件定制构建
  • 深入探索像ChatGPT这样的大语言模型-01-Base Model-Pretraining阶段
  • ArcGIS Pro应用指南：如何为栅格图精确添加坐标信息
  • 拓扑排序_走多远
  • SQL Server下载和安装细节
  • 4.Linux操作系统命令
  • Microk8s Ingress实现七层负载均衡
  • 玩转ChatGPT：Claude 3.7 Sonnet进行数据分析（分类）
  • 如何评估大语言模型（LLMs）
  • LLM - Attention Is All You Need 的理解
  • C++ 二叉树代码
  • SQL 中为什么参数多了not in 比 in 慢多了，怎么优化
  • Linux常见操作命令
  • AI 赋能 RPA：一键生成热点话题文章的奥秘
  • c++ accumulate、find、count、fill、fill_n、copy、sort、unique 泛型算法
  • 【实战篇】【深度解析DeepSeek：从机器学习到深度学习的全场景落地指南】