ffmpeg 编码yuv pcm 数据保存文件

将YUV和PCM数据编码成视频和音频文件需要使用FFmpeg库，具体的实现步骤如下：

1. 初始化FFmpeg库

在使用FFmpeg库之前，需要先进行初始化。使用av_register_all函数可以注册FFmpeg库中的所有编解码器、格式器和协议等。

av_register_all();

2. 打开输出文件

使用avformat_alloc_output_context2和avio_open2函数打开输出文件，创建AVFormatContext结构体并分配内存，将输出文件与该结构体关联。

AVFormatContext *out_ctx = NULL;
int ret = avformat_alloc_output_context2(&out_ctx, NULL, NULL, output_file);
if (ret < 0) {
    // 创建AVFormatContext失败
    return;
}
if (!(out_ctx->oformat->flags & AVFMT_NOFILE)) {
    ret = avio_open2(&out_ctx->pb, output_file, AVIO_FLAG_WRITE, NULL, NULL);
    if (ret < 0) {
        // 打开输出文件失败
        return;
    }
}

3. 创建音视频流

使用avformat_new_stream函数创建音视频流，并设置音视频流的相关参数，如编码器、帧率、码率、采样率等。

AVStream *video_stream = avformat_new_stream(out_ctx, NULL);
if (video_stream == NULL) {
    // 创建视频流失败
    return;
}
AVCodecParameters *codecpar = video_stream->codecpar;
codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
codecpar->width = width;
codecpar->height = height;
codecpar->format = AV_PIX_FMT_YUV420P;
codecpar->codec_id = AV_CODEC_ID_H264;
codecpar->bit_rate = bit_rate;
codecpar->framerate = {fps, 1};

AVStream *audio_stream = avformat_new_stream(out_ctx, NULL);
if (audio_stream == NULL) {
    // 创建音频流失败
    return;
}
codecpar = audio_stream->codecpar;
codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
codecpar->sample_rate = sample_rate;
codecpar->format = AV_SAMPLE_FMT_S16;
codecpar->channels = channels;
codecpar->channel_layout = av_get_default_channel_layout(channels);
codecpar->codec_id = AV_CODEC_ID_AAC;
codecpar->bit_rate = bit_rate;

4. 打开视频和音频编码器

使用avcodec_find_encoder函数查找视频和音频编码器，并使用avcodec_open2打开编码器。

AVCodec *video_codec = avcodec_find_encoder(video_stream->codecpar->codec_id);
if (video_codec == NULL) {
    // 查找视频编码器失败
    return;
}
AVCodecContext *video_cctx = avcodec_alloc_context3(video_codec);
if (video_cctx == NULL) {
    // 创建视频编码器上下文失败
    return;
}
ret = avcodec_open2(video_cctx, video_codec, NULL);
if (ret < 0) {
    // 打开视频编码器失败
    return;
}

AVCodec *audio_codec = avcodec_find_encoder(audio_stream->codecpar->codec_id);
if (audio_codec == NULL) {
    // 查找音频编码器失败
    return;
}
AVCodecContext *audio_cctx = avcodec_alloc_context3(audio_codec);
if (audio_cctx == NULL) {
    // 创建音频编码器上下文失败
    return;
}
ret = avcodec_open2(audio_cctx, audio_codec, NULL);
if (ret < 0) {
    // 打开音频编码器失败
    return;
}

5. 写入视频和音频数据

使用av_frame_alloc函数创建AVFrame结构体，填充YUV和PCM数据，并使用avcodec_send_frame和avcodec_receive_packet函数将数据编码成视频和音频包，最后使用av_write_frame函数将包写入输出文件。

AVFrame *video_frame = av_frame_alloc();
// 填充YUV数据到video_frame中

AVPacket *video_packet = av_packet_alloc();
ret = avcodec_send_frame(video_cctx, video_frame);
if (ret < 0) {
    // 向视频编码器发送数据失败
    return;
}
while (ret >= 0) {
    ret = avcodec_receive_packet(video_cctx, video_packet);
    if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
        break;
    }
    if (ret < 0) {
        // 从视频编码器接收数据失败
        return;
    }
    av_packet_rescale_ts(video_packet, video_cctx->time_base, video_stream->time_base);
    video_packet->stream_index = video_stream->index;
    ret = av_write_frame(out_ctx, video_packet);
    if (ret < 0) {
        // 写入视频数据失败
        return;
    }
}

AVFrame *audio_frame = av_frame_alloc();
// 填充PCM数据到audio_frame中

AVPacket *audio_packet = av_packet_alloc();
ret = avcodec_send_frame(audio_cctx, audio_frame);
if (ret < 0) {
    // 向音频编码器发送数据失败
    return;
}
while (ret >= 0) {
    ret = avcodec_receive_packet(audio_cctx, audio_packet);
    if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
        break;
    }
    if (ret < 0) {
        // 从音频编码器接收数据失败
        return;
    }
    av_packet_rescale_ts(audio_packet, audio_cctx->time_base, audio_stream->time_base);
    audio_packet->stream_index = audio_stream->index;
    ret = av_write_frame(out_ctx, audio_packet);
    if (ret < 0) {
        // 写入音频数据失败
        return;
    }
}

6. 关闭编码器和输出文件

使用av_write_trailer、avcodec_free_context和avformat_free_context函数释放资源并关闭编码器和输出文件。

av_write_trailer(out_ctx);
avcodec_free_context(&video_cctx);
avcodec_free_context(&audio_cctx);
avformat_free_context(out_ctx);

以上是将YUV和PCM数据编码成视频和音频文件的基本流程，需要注意的是各项参数的设置和数据的填充。如果需要进行更详细的配置和处理，可以参考FFmpeg库的官方文档。