How to convert RGB from YUV420p for ffmpeg encoder?

在ffmpeg中将RGB格式的图像或视频转换为YUV420p格式，并进行编码，可以通过以下几个步骤完成：

步骤 1: 准备视频源

首先，你需要一个RGB格式的视频源。这可以是一个RGB格式的视频文件，或者是实时生成的RGB图像数据。

步骤 2: 初始化ffmpeg

在使用ffmpeg之前，你需要确保已经安装了ffmpeg库，并且你的开发环境配置正确。你可以在命令行工具中输入 ffmpeg 来检查是否已经安装。

步骤 3: 设置转换参数

使用ffmpeg的命令行工具或API来设置输入文件（如果是文件的话）和输出文件的参数。如果是使用命令行，基本的命令格式如下：

bash
ffmpeg -i input.rgb -pix_fmt yuv420p output.mp4

这里的关键是 -pix_fmt yuv420p，它指定输出使用YUV420p像素格式。input.rgb 是RGB格式的输入文件，output.mp4 是要创建的输出文件。

步骤 4: 处理并编码

执行上述ffmpeg命令后，ffmpeg会读取RGB数据，将其转换为YUV420p格式，然后编码为指定的视频格式（在这个例子中是MP4）。

示例 - 实际应用场景

假设我们有一个应用程序，它实时生成RGB格式的图像数据，并需要将这些数据转换为YUV420p格式，然后编码为H.264视频流。我们可以使用ffmpeg的API来实现这一功能。以下是用C语言编写的简化代码片段：

c
#include <libavutil/imgutils.h>
#include <libavcodec/avcodec.h>
#include <libswscale/swscale.h>

void encode_rgb_to_yuv420p(uint8_t *rgb_buffer, int width, int height) {
    AVCodec *codec;
    AVCodecContext *c= NULL;
    int ret;
    AVFrame *frame;
    AVPacket *pkt;
    struct SwsContext *sws_ctx;

    codec = avcodec_find_encoder(AV_CODEC_ID_H264);
    if (!codec) {
        fprintf(stderr, "Codec not found\n");
        exit(1);
    }

    c = avcodec_alloc_context3(codec);
    c->bit_rate = 400000;
    c->width = width;
    c->height = height;
    c->time_base = (AVRational){1, 25};
    c->framerate = (AVRational){25, 1};
    c->gop_size = 10;
    c->max_b_frames = 1;
    c->pix_fmt = AV_PIX_FMT_YUV420P;

    if (avcodec_open2(c, codec, NULL) < 0) {
        fprintf(stderr, "Could not open codec\n");
        exit(1);
    }

    frame = av_frame_alloc();
    frame->format = c->pix_fmt;
    frame->width  = c->width;
    frame->height = c->height;

    sws_ctx = sws_getContext(c->width, c->height, AV_PIX_FMT_RGB24,
                             c->width, c->height, AV_PIX_FMT_YUV420P,
                             SWS_BILINEAR, NULL, NULL, NULL);

    // 这里是将RGB转换为YUV420P
    sws_scale(sws_ctx, (const uint8_t * const *)&rgb_buffer, linesizes, 
              0, c->height, frame->data, frame->linesize);
    
    // 编码处理
    // ...
}

这段代码展示了如何初始化一个编码器，设置为YUV420p格式，然后使用sws_scale函数来转换RGB到YUV420p。之后，你可以将帧发送到编码器进行编码。