qt ffmpeg 推流RTMP代码

以下是使用Qt和FFmpeg进行RTMP推流的示例代码：

extern "C" {
#include <libavutil/imgutils.h>
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
}

void pushRtmp()
{
    av_register_all();
    avformat_network_init();

    AVFormatContext *pFormatCtx = nullptr;
    AVOutputFormat *pOutputFmt = nullptr;
    AVStream *pStream = nullptr;
    AVCodecContext *pCodecCtx = nullptr;
    AVCodec *pCodec = nullptr;
    AVFrame *pFrame = nullptr;
    AVPacket pkt;

    const char *rtmpUrl = "rtmp://your_rtmp_server_url";

    // 创建输出上下文
    avformat_alloc_output_context2(&pFormatCtx, nullptr, "flv", rtmpUrl);
    if (!pFormatCtx) {
        qDebug() << "Failed to create output context";
        return;
    }

    pOutputFmt = pFormatCtx->oformat;

    // 添加视频流
    pStream = avformat_new_stream(pFormatCtx, nullptr);
    if (!pStream) {
        qDebug() << "Failed to create video stream";
        return;
    }

    pCodecCtx = pStream->codec;

    // 设置编码参数
    pCodecCtx->codec_id = pOutputFmt->video_codec;
    pCodecCtx->codec_type = AVMEDIA_TYPE_VIDEO;
    pCodecCtx->pix_fmt = AV_PIX_FMT_YUV420P;
    pCodecCtx->width = 640;
    pCodecCtx->height = 480;
    pCodecCtx->bit_rate = 400000;
    pCodecCtx->gop_size = 250;
    pCodecCtx->time_base.num = 1;
    pCodecCtx->time_base.den = 25;

    // 查找编码器
    pCodec = avcodec_find_encoder(pCodecCtx->codec_id);
    if (!pCodec) {
        qDebug() << "Failed to find encoder";
        return;
    }

    // 打开编码器
    if (avcodec_open2(pCodecCtx, pCodec, nullptr) < 0) {
        qDebug() << "Failed to open encoder";
        return;
    }

    // 创建帧并分配数据内存
    pFrame = av_frame_alloc();
    pFrame->width = pCodecCtx->width;
    pFrame->height = pCodecCtx->height;
    pFrame->format = pCodecCtx->pix_fmt;
    av_image_alloc(pFrame->data, pFrame->linesize,
                   pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, 32);

    // 打开输出URL
    if (avio_open(&pFormatCtx->pb, rtmpUrl, AVIO_FLAG_WRITE) < 0) {
        qDebug() << "Failed to open output URL";
        return;
    }

    // 写文件头
    avformat_write_header(pFormatCtx, nullptr);

    // 编码并推流
    for (;;) {
        // 从视频源中读取数据到AVFrame
        // ...

        // 编码
        avcodec_send_frame(pCodecCtx, pFrame);
        while (avcodec_receive_packet(pCodecCtx, &pkt) == 0) {
            // 时间基转换
            pkt.pts = av_rescale_q(pkt.pts, pCodecCtx->time_base, pStream->time_base);
            pkt.dts = av_rescale_q(pkt.dts, pCodecCtx->time_base, pStream->time_base);
            pkt.duration = av_rescale_q(pkt.duration, pCodecCtx->time_base, pStream->time_base);
            pkt.pos = -1;

            // 写入推流器
            av_interleaved_write_frame(pFormatCtx, &pkt);

            // 释放包资源
            av_packet_unref(&pkt);
        }
    }

    // 写文件尾
    av_write_trailer(pFormatCtx);

    // 释放资源
    avcodec_free_context(&pCodecCtx);
    av_frame_free(&pFrame);
    avformat_free_context(pFormatCtx);
    avio_close(pFormatCtx->pb);
    avformat_network_deinit();
}