C++ ffmpeg 编码转换示例

在C++中使用FFmpeg进行编码转换，通常会涉及到头文件的包含、初始化FFmpeg上下文、设置编码参数和调用编码函数等步骤。下面是一个简单的H264到AV1编码转换的示例，假设已经包含了必要的FFmpeg库：

#include <iostream>
#include <string>
#include "libavutil/imgutils.h"
#include "libavformat/avformat.h"
#include "libavcodec/avcodec.h"

//... （省略FFmpeg库初始化代码）

std::string input_file = "input.h264";
std::string output_file = "output.av1";

AVFormatContext* input_ctx = nullptr;
AVFormatContext* output_ctx = nullptr;

int main() {
    // 读取H264源文件
    if (avformat_open_input(&input_ctx, input_file.c_str(), NULL, NULL) != 0) {
        std::cerr << "Error opening file: " << av_err2str(AVERROR_UNKNOWN) << std::endl;
        return 1;
    }

    // 解析文件
    if (avformat_find_stream_info(input_ctx, NULL) < 0) {
        std::cerr << "Error finding stream info" << std::endl;
        return 1;
    }

    // 获取视频流索引
    int video_idx = -1;
    for (int i = 0; i < input_ctx->nb_streams; ++i) {
        AVStream* stream = input_ctx->streams[i];
        if (stream->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
            video_idx = i;
            break;
        }
    }

    if (video_idx == -1) {
        std::cerr << "No video stream found" << std::endl;
        return 1;
    }

    // 创建AVCodecContext并选择编码器
    AVCodecContext* codec_ctx = input_ctx->streams[video_idx]->codec;
    AVCodec* encoder = avcodec_find_encoder(AV_CODEC_ID_AV1);
    if (!encoder) {
        std::cerr << "Encoder not found for AV1" << std::endl;
        return 1;
    }
    if (avcodec_open2(codec_ctx, encoder, NULL) < 0) {
        std::cerr << "Error opening codec" << std::endl;
        return 1;
    }

    // 创建输出AVFormatContext
    av_register_all();
    if (!(output_ctx = avformat_alloc_output_context2(NULL, NULL, AV ?");
    //（此处应填入“AV_PIX_FMT_YUV420P”或其他适合的AV_PIX_FMT_...值），NULL))) {
    //...（继续创建输出格式上下文）
    if (!output_ctx) {
        std::cerr << "Error creating output context" << std::endl;
        return 1;
    }

    // 将H264编码器输出的图像转换为AV1所需的格式
    // 这里省略了具体的图像转换过程，因为FFmpeg本身并不直接支持这种跨格式的转换，通常需要先解码再编码

    // 开始编码
    while (1) {
        int got_frame = 0;
        AVPacket pkt;
        if (av_read_frame(input_ctx, &pkt) >= 0) {
            if (pkt.stream_index == video_idx) {
                AVFrame* frame = av_frame_alloc();
                if (!frame) {
                    std::cerr << "Memory allocation failed" << std::endl;
                    return 1;
                }

                if (avcodec_decode_video2(codec_ctx, frame, &got_frame, &pkt) == 0) {
                    // 将解码后的帧转换为AV1格式并编码
                    // ...（此处编写实际的编码过程，可能涉及av_image_fill_arrays等函数）

                    av_packet_unref(&pkt); // 清理旧的AVPacket
                    if (av_interleaved_write_frame(output_ctx, frame) < 0) {
                        std::cerr << "Error writing frame to output" << std::endl;
                        return 1;
                    }
                    av_frame_free(&frame);
                } else {
                    std::cerr << "Error decoding frame" << std::endl;
                    return 1;
                }
            }
            av_packet_unref(&pkt);
        } else {
            if (avio.EOF == pkt.pos)
                break; // End of stream
            std::cerr << "Error reading frame" << std::endl;
            return 1;
        }
    }

    // 写入文件和关闭文件
    if (avio_close(output_ctx->pb) < 0) {
        std::cerr << "Error closing output file" << std::endl;
        return 1;
    }
    avformat_write_header(output_ctx, NULL);

    if (av_write_trailer(output_ctx) < 0) {
        std::cerr << "Error writing trailer" << std::endl;
        return 1;
    }

    // 关闭编码器和解码器
    avcodec_close(codec_ctx);
    avformat_free_context(input_ctx);
    avformat_free_context(output_ctx);

    return 0;
}

//...（可能需要清理其他FFmpeg资源的代码）

注意这个示例只是一个基本框架，实际编码过程中你需要处理更多细节，如错误处理、帧处理等。同时，FFmpeg的官方文档是学习和使用的重要参考。