FFmpeg RGB转MP4

Posted 2021-03-18 linuxandmcu

下面是一个读取本地 RGB 文件，转换并输出 MP4 视频文件的一个例子，具体步骤如下：

1、创建编码器

2、创建输出视频上下文

3、添加视频流

4、rgb转yuv

5、写视频文件头

6、写视频文件，循环内部进行H264编码

完整代码如下：

#include <iostream>

extern "C"
{
    #include <libavformat/avformat.h>
    #include <libswscale/swscale.h>
}

#pragma comment(lib,"avformat.lib")
#pragma comment(lib,"avcodec.lib")
#pragma comment(lib,"avutil.lib")
#pragma comment(lib,"swscale.lib")

using namespace std;

int main()
{
    char infile[] = "out.rgb";
    char outfile[] = "rgb.mp4";

    // 注册所有和编解码器有关的组件
    av_register_all();

    // 打开RGB文件
    FILE *fp = fopen(infile, "rb");
    if (!fp)
    {
        cout << infile << " open failed!" << endl;
        getchar();
        return -1;
    }

    // 源图像参数
    int width = 848;
    int height = 480;
    int fps = 25;

    ///1 创建编码器
    // 查找编码器
    AVCodec *codec = avcodec_find_encoder(AV_CODEC_ID_H264);
    if (!codec)
    {
        cout << " avcodec_find_encoder AV_CODEC_ID_H264 failed!" << endl;
        getchar();
        return -1;
    }

    // 给编码器分配内存，返回对应编码器上下文
    AVCodecContext *c = avcodec_alloc_context3(codec);
    if (!c)
    {
        cout << " avcodec_alloc_context3  failed!" << endl;
        getchar();
        return -1;
    }
    // 配置编码器上下文的成员
    c->bit_rate = 400000000; // 比特率(码率)，越高视频质量越好
    c->width = width; // 设置编码视频宽度 
    c->height = height; // 设置编码视频高度
    c->time_base.num = 1;
    c->time_base.den = 25; // 设置帧率，num为分子，den为分母，如果是1/25则表示25帧/s

    c->gop_size = 50; // 画面组大小，关键帧
    c->max_b_frames = 0; // 设置B帧最大数,该值表示在两个非B帧之间，所允许插入的B帧的最大帧数

    c->pix_fmt = AV_PIX_FMT_YUV420P; // 设置输出像素格式
    c->codec_id = AV_CODEC_ID_H264; // 设置编码格式
    c->thread_count = 8; // 线程数量
    c->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; // 全局的编码信息

    // 打开编码器
    int ret = avcodec_open2(c, codec, NULL);
    if (ret < 0)
    {
        cout << " avcodec_open2  failed!" << endl;
        getchar();
        return -1;
    }
    cout << "avcodec_open2 success!" << endl;

    ///2 创建输出视频上下文
    AVFormatContext *oc = NULL;
    avformat_alloc_output_context2(&oc, 0, 0, outfile);

    ///3 添加视频流
    AVStream *st = avformat_new_stream(oc, NULL);
    st->id = 0;
    st->codecpar->codec_tag = 0;
    // 将AVCodecContext信息拷贝到AVCodecParameterst结构体中
    avcodec_parameters_from_context(st->codecpar, c);

    cout << "===============================================" << endl;
    // 打印AVFormatContext的内容
    av_dump_format(oc, 0, outfile, 1);
    cout << "===============================================" << endl;

    ///4 rgb转yuv
    SwsContext *ctx = NULL;
    ctx = sws_getCachedContext(ctx,
        width, height, AV_PIX_FMT_BGRA,
        width, height, AV_PIX_FMT_YUV420P,
        SWS_BICUBIC,
        NULL, NULL, NULL
        );
    // 分配输入空间
    unsigned char *rgb = new unsigned char[width*height * 4];

    // 分配输出空间
    AVFrame *yuv = av_frame_alloc();
    yuv->format = AV_PIX_FMT_YUV420P;
    yuv->width = width;
    yuv->height = height;
    ret = av_frame_get_buffer(yuv, 32);
    if (ret < 0)
    {
        cout << " av_frame_get_buffer  failed!" << endl;
        getchar();
        return -1;
    }

    ///5 写视频文件头
    // 创建并初始化AVIOContext以访问outfile指示的资源
    ret = avio_open(&oc->pb, outfile, AVIO_FLAG_WRITE);
    if (ret < 0)
    {
        cout << " avio_open  failed!" << endl;
        getchar();
        return -1;
    }
    // 写视频文件头
    ret = avformat_write_header(oc, NULL);
    if (ret < 0)
    {
        cout << " avformat_write_header  failed!" << endl;
        getchar();
        return -1;
    }

    // 循环写视频文件
    int pts = 0;
    for (;;)
    {
        // 读取RGB文件的像素数据长度
        int len = fread(rgb, 1, width*height * 4, fp);
        if (len <= 0)
        {
            break;
        }

        // 像素格式转换：RGB转YUV
        uint8_t *indata[AV_NUM_DATA_POINTERS] = { 0 };
        indata[0] = rgb;
        int inlinesize[AV_NUM_DATA_POINTERS] = { 0 };
        inlinesize[0] = width * 4;
        int h = sws_scale(ctx, indata, inlinesize, 0, height,
            yuv->data, yuv->linesize
            );
        if (h <= 0)
            break;

        ///6 H264编码
        // 将未压缩的AVFrame数据(yuv)给编码器
        yuv->pts = pts;
        //yuv->pict_type = AV_PICTURE_TYPE_I;
        pts = pts + 3600;
        ret = avcodec_send_frame(c, yuv);
        if (ret != 0)
        {
            continue;
        }
        // 将编码数据保存在AVPacket
        AVPacket pkt;
        av_init_packet(&pkt);
        ret = avcodec_receive_packet(c, &pkt);
        if (ret != 0)
            continue;

        // 将AVPacket写入输出媒体文件
        //av_write_frame(oc, &pkt);
        //av_packet_unref(&pkt);
        av_interleaved_write_frame(oc, &pkt);

        cout << "<"<<pkt.size<<">";
    }
    
    //写入视频索引
    av_write_trailer(oc);

    //关闭视频输出io
    avio_close(oc->pb);

    //清理封装输出上下文
    avformat_free_context(oc);

    //关闭编码器
    avcodec_close(c);

    //清理编码器上下文
    avcodec_free_context(&c);

    //清理视频重采样上下文
    sws_freeContext(ctx);

    cout << "======================end=========================" << endl;

    delete rgb;
    getchar();
    return 0;
}