生成一个qt下使用opencv 进行 rtmp推流例程

以下是一个使用Qt和OpenCV进行RTMP推流的例程：

首先，需要确保已经安装了OpenCV和FFmpeg，并且Qt项目中已经添加了OpenCV和FFmpeg的库文件。

然后，需要添加以下头文件：

#include <opencv2/opencv.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/videoio/videoio.hpp>
#include <opencv2/videoio/legacy/constants_c.h>

#include <QDebug>
#include <QThread>
#include <QString>
#include <QMutex>
#include <QDateTime>
#include <QStringList>
#include <QCoreApplication>

接下来，需要声明一些变量：

// RTMP 推流地址
const char *rtmpUrl = "rtmp://localhost/live/stream";

// 视频编码器
const char *codecName = "libx264";

// 视频分辨率
const int width = 640;
const int height = 480;

// 视频帧率
const int fps = 25;

// 视频比特率
const int bitrate = 1000000;

// 帧计数器
int frameCount = 0;

// RTMP 上下文
RTMP *rtmp = NULL;

// 视频编码器上下文
AVCodecContext *codecCtx = NULL;

// 视频帧
AVFrame *frame = NULL;

// 视频帧缓存区
uint8_t *frameBuffer = NULL;

// 视频帧缓存区大小
int frameBufferSize = 0;

// 视频编码器
AVCodec *codec = NULL;

// 编码器参数
AVDictionary *codecParams = NULL;

// 编码器帧计数器
int64_t codecFrameCount = 0;

// 编码器 PTS 计数器
int64_t codecPtsCount = 0;

// 编码器上一帧 PTS
int64_t codecLastPts = 0;

// 编码器上一帧 DTS
int64_t codecLastDts = 0;

// 编码器视频流
AVStream *codecStream = NULL;

// 编码器视频帧
AVFrame *codecFrame = NULL;

// 编码器视频帧缓存区
uint8_t *codecFrameBuffer = NULL;

// 编码器视频帧缓存区大小
int codecFrameBufferSize = 0;

// 编码器视频帧计数器
int codecFrameCount = 0;

// 编码器视频帧时间戳
int64_t codecFramePts = 0;

// 编码器视频帧数据大小
int codecFrameDataSize = 0;

// 编码器视频帧数据
uint8_t *codecFrameData = NULL;

// 编码器视频帧数据指针
uint8_t *codecFrameDataPtr = NULL;

// 编码器视频帧数据长度
int codecFrameDataLen = 0;

// 编码器视频帧类型
int codecFrameType = 0;

// 编码器视频帧是否为关键帧
int codecFrameKeyframe = 0;

// 编码器视频帧时间基数
AVRational codecTimeBase = {1, 1000};

// 编码器视频帧时间戳增量
AVRational codecTimeIncrement = {1, fps};

// 编码器视频帧时间戳
AVRational codecPtsTimeBase = {1, AV_TIME_BASE};

接着，需要定义一个函数用于初始化编码器：

void initEncoder()
{
    // 初始化编码器
    avcodec_register_all();

    // 获取编码器
    codec = avcodec_find_encoder_by_name(codecName);

    // 创建编码器上下文
    codecCtx = avcodec_alloc_context3(codec);
    codecCtx->width = width;
    codecCtx->height = height;
    codecCtx->bit_rate = bitrate;
    codecCtx->time_base = codecTimeBase;
    codecCtx->framerate = codecTimeIncrement;
    codecCtx->gop_size = fps;
    codecCtx->pix_fmt = AV_PIX_FMT_YUV420P;
    codecCtx->flags |= AV_CODEC_FLAG_GLOBAL_HEADER;

    // 设置编码器参数
    av_dict_set(&codecParams, "preset", "ultrafast", 0);
    av_dict_set(&codecParams, "tune", "zerolatency", 0);

    // 打开编码器
    avcodec_open2(codecCtx, codec, &codecParams);

    // 创建视频帧
    codecFrame = av_frame_alloc();
    codecFrame->format = codecCtx->pix_fmt;
    codecFrame->width = codecCtx->width;
    codecFrame->height = codecCtx->height;
    codecFrame->pts = 0;
    av_frame_get_buffer(codecFrame, 0);

    // 创建视频帧缓存区
    codecFrameBufferSize = av_image_get_buffer_size(codecCtx->pix_fmt, codecCtx->width, codecCtx->height, 1);
    codecFrameBuffer = (uint8_t *)av_malloc(codecFrameBufferSize);
    av_image_fill_arrays(codecFrame->data, codecFrame->linesize, codecFrameBuffer, codecCtx->pix_fmt, codecCtx->width, codecCtx->height, 1);
}

然后，需要定义一个函数用于连接RTMP服务器：

bool connectRTMP()
{
    // 初始化 RTMP
    RTMP_Init(rtmp);

    // 设置 RTMP URL
    if (!RTMP_SetupURL(rtmp, (char *)rtmpUrl))
    {
        qDebug() << "RTMP_SetupURL error:" << rtmpUrl;
        return false;
    }

    // 设置连接超时时间
    rtmp->Link.timeout = 5;

    // 开始连接 RTMP 服务器
    if (!RTMP_Connect(rtmp, NULL))
    {
        qDebug() << "RTMP_Connect error:" << rtmpUrl;
        return false;
    }

    // 开始连接 RTMP 流
    if (!RTMP_ConnectStream(rtmp, 0))
    {
        qDebug() << "RTMP_ConnectStream error:" << rtmpUrl;
        return false;
    }

    return true;
}

接下来，需要定义一个函数用于初始化视频帧：

void initFrame()
{
    // 创建视频帧
    frame = cvCreateMat(height, width, CV_8UC3);

    // 创建视频帧缓存区
    frameBufferSize = av_image_get_buffer_size(AV_PIX_FMT_YUV420P, width, height, 1);
    frameBuffer = (uint8_t *)av_malloc(frameBufferSize);
}

然后，需要定义一个函数用于发送视频帧：

bool sendFrame()
{
    // 将视频帧转换为 YUV 格式
    cvCvtColor(frame, frame, CV_BGR2YUV_I420);

    // 将视频帧数据写入编码器缓存区
    codecFrame->pts = codecPtsCount;
    codecFrameDataPtr = codecFrameData;
    codecFrameDataLen = avcodec_encode_video2(codecCtx, codecFrameDataPtr, codecFrameBufferSize, codecFrame);
    codecFrameDataPtr += codecFrameDataLen;

    // 如果编码器缓存区已满，则发送视频帧
    if (codecFrameDataLen > 0)
    {
        // 计算视频帧时间戳
        codecFramePts = av_rescale_q(codecFrameCount, codecTimeIncrement, codecTimeBase);

        // 发送视频帧
        AVPacket pkt;
        av_init_packet(&pkt);
        pkt.data = codecFrameData;
        pkt.size = codecFrameDataLen;
        pkt.pts = codecFramePts;
        pkt.dts = codecFramePts;
        pkt.duration = av_rescale_q(1, codecTimeIncrement, codecTimeBase);
        pkt.stream_index = codecStream->index;
        av_interleaved_write_frame(rtmp->opaque, &pkt);
        av_packet_unref(&pkt);

        // 更新帧计数器
        codecFrameCount++;
        codecFrameData = NULL;
        codecFrameDataLen = 0;

        return true;
    }

    return false;
}

最后，需要定义一个函数用于发送视频流：

void sendStream()
{
    while (true)
    {
        // 读取视频帧
        cv::Mat image;
        capture >> image;

        // 如果视频帧为空，则退出
        if (image.empty())
        {
            break;
        }

        // 将视频帧数据写入视频帧缓存区
        memcpy(frameBuffer, image.data, frameBufferSize);

        // 将视频帧数据写入视频帧
        cvSetData(frame, frameBuffer, width * 3);

        // 发送视频帧
        if (sendFrame())
        {
            qDebug() << "sendFrame:" << frameCount << ":" << codecFrameCount << ":" << codecPtsCount;
        }

        // 更新帧计数器
        frameCount++;
        codecPtsCount += av_rescale_q(1, codecTimeIncrement, codecTimeBase);
    }
}

最后，在主函数中调用以上函数即可：

int main(int argc, char *argv[])
{
    // 初始化 RTMP
    rtmp = RTMP_Alloc();
    if (!rtmp)
    {
        qDebug() << "RTMP_Alloc error";
        return -1;
    }

    // 初始化编码器
    initEncoder();

    // 初始化视频帧
    initFrame();

    // 连接 RTMP 服务器
    if (!connectRTMP())
    {
        return -1;
    }

    // 发送视频流
    sendStream();

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

    // 关闭 RTMP 流
    RTMP_Close(rtmp);

    // 释放 RTMP
    RTMP_Free(rtmp);

    // 释放视频帧缓存区
    av_free(frameBuffer);

    // 释放视频帧
    cvReleaseMat(&frame);

    return 0;
}

以上就是一个使用Qt和OpenCV进行RTMP推流的例程。需要注意的是，由于涉及到视频编码和网络传输等技术，代码实现较为复杂，需要仔细理解和调试。