基于QT以及FFmpeg开发的录屏软件以及视频播放器代码

#include "videocodec.h" #include <QDebug> extern "C" { #include "libavcodec/avcodec.h" #include "libavformat/avformat.h" #include "libavutil/avutil.h" #include "libswscale/swscale.h" #include "libavutil/imgutils.h" #include "libavdevice/avdevice.h" } // 异常信息数组长度 #define ERROR_LEN 1024 #define PRINT_LOG 1 VideoCodec::VideoCodec() { } VideoCodec::~VideoCodec() { close(); } //初始化并打开视频编解码器，通过分配输出上下文、打开输出文件、查询编码器、设置编码器参数、打开编码器、向媒体文件添加新流、写入文件头等操作完成 bool VideoCodec::open(AVCodecContext *codecContext, QPoint point, const QString &fileName) { //检查输入参数codecContext和fileName是否为空 if(!codecContext || fileName.isEmpty()) { return false; } // 通过输出文件名为输出格式分配AVFormatContext //参数3编码器设置为空，由参数4文件名后缀推测合适的编码器 int ret = avformat_alloc_output_context2(&m_formatContext, nullptr, nullptr, fileName.toStdString().data()); if(ret < 0) { close(); showError(ret); return false; } // 创建并初始化AVIOContext以访问url所指示的资源。将打开的结果保存在m_formatContext->pb中 ret = avio_open(&m_formatContext->pb, fileName.toStdString().data(), AVIO_FLAG_WRITE); if(ret < 0) { close(); showError(ret); return false; } // 查询编码器 const AVCodec* codec = avcodec_find_encoder(m_formatContext->oformat->video_codec); //检查codec是否为空 if(!codec) { close(); showError(AVERROR(ENOMEM)); return false; } qDebug() << codec->id <<" " << codec->name; // 分配AVCodecContext并将其字段设置为默认值。 m_codecContext = avcodec_alloc_context3(codec); if(!m_codecContext) { close(); showError(AVERROR(ENOMEM)); return false; } // 设置编码器上下文参数 m_codecContext->width = codecContext->width; // 图片宽度/高度 m_codecContext->height = codecContext->height; m_codecContext->pix_fmt = codec->pix_fmts[0]; // 像素格式（这里通过编码器赋值，不需要自己指定） m_codecContext->time_base = {point.y(), point.x()}; //设置时间基，20为分母，1为分子，表示以1/20秒时间间隔播放一帧图像 m_codecContext->framerate = {point.x(), point.y()}; m_codecContext->bit_rate = bitrate; // 目标的码率，即采样的码率；显然，采样码率越大，视频大小越大，画质越高 m_codecContext->gop_size = 12; // I帧间隔(值越大，视频文件越小，编解码延时越长) m_codecContext->flags |= AV_CODEC_FLAG_GLOBAL_HEADER; //将AV_CODEC_FLAG_GLOBAL_HEADER标志设置到m_codecContext的flags字段中 // 打开编码器 ret = avcodec_open2(m_codecContext, nullptr, nullptr); if(ret < 0) { close(); showError(ret); return false; } // 向媒体文件添加新流 m_videoStream = avformat_new_stream(m_formatContext, nullptr); if(!m_videoStream) { close(); showError(AVERROR(ENOMEM)); return false; } //拷贝一些参数，给codecpar赋值 ret = avcodec_parameters_from_context(m_videoStream->codecpar,m_codecContext); if(ret < 0) { close(); showError(ret); return false; } //将m_codecContext中的参数拷贝给m_videoStream->codecpar，写入文件头 ret = avformat_write_header(m_formatContext, nullptr); if(ret < 0) { close(); showError(ret); return false; } m_writeHeader = true; // 分配一个AVPacket m_packet = av_packet_alloc(); if(!m_packet) { close(); showError(AVERROR(ENOMEM)); return false; } m_frame = av_frame_alloc(); if(!m_frame) { close(); showError(AVERROR(ENOMEM)); return false; } //设置m_frame的像素格式 m_frame->format = codec->pix_fmts[0]; qDebug() << "开始录制视频！"; return true; } //将视频帧进行编码并写入视频文件。如果需要，它会转换帧的格式，并将帧发送给编码器。然后，它循环接收和写入所有已编码的数据包到文件中 void VideoCodec::write(AVFrame *frame) { //获取互斥锁，以保证线程安全 QMutexLocker locker(&m_mutex); if(!m_packet) { return; } // 由于解码的图像格式和编码需要的图像格式不一定相同，所以需要转换一下格式 if(!swsFormat(frame)) { return; } //如果不为空，设置m_frame的pts为m_index（用于指定当前帧的时间戳），然后自增m_index if(m_frame) { // pts从0开始增加，保存的视频才会时间从0开始增加 m_frame->pts = m_index; m_index++; } // 将图像传入编码器 avcodec_send_frame(m_codecContext, m_frame); // 循环读取所有编码完的帧 while (true) { // 从编码器中读取图像帧 int ret = avcodec_receive_packet(m_codecContext, m_packet); if(ret < 0) { break; } // 将数据包中的有效时间字段（时间戳/持续时间）从一个时基转换为 输出流的时间 av_packet_rescale_ts(m_packet, m_codecContext->time_base, m_videoStream->time_base); av_write_frame(m_formatContext, m_packet); // 将数据包写入输出媒体文件 av_packet_unref(m_packet); } } //关闭视频编解码器，包括写入文件尾部、关闭AVIO上下文、释放分配的资源 void VideoCodec::close() { //传入空帧，读取所有编码数据 write(nullptr); // 如果不加锁可能在点击关闭时，write函数正在写入数据，导致崩溃 QMutexLocker locker(&m_mutex); //是否成功初始化了输出媒体文件上下文 if(m_formatContext) { // 写入文件尾 if(m_writeHeader) { //避免重复写入文件尾部 m_writeHeader = false; int ret = av_write_trailer(m_formatContext); if(ret < 0) { showError(ret); return; } } //关闭输出IO流 int ret = avio_close(m_formatContext->pb); if(ret < 0) { showError(ret); return; } //释放输出媒体文件上下文的资源 avformat_free_context(m_formatContext); m_formatContext = nullptr; m_videoStream = nullptr; } // 释放编解码器上下文并置空 if(m_codecContext) { avcodec_free_context(&m_codecContext); } if(m_packet) { av_packet_free(&m_packet); } // 释放上下文swsContext。 if(m_swsContext) { sws_freeContext(m_swsContext); m_swsContext = nullptr; // sws_freeContext不会把上下文置NULL } if(m_frame) { //释放帧占用的内存 av_frame_free(&m_frame); } m_index = 0; } //在函数执行过程中，记录和显示任何错误消息 void VideoCodec::showError(int err) { #if PRINT_LOG static char m_error[ERROR_LEN]; // 保存异常信息 memset(m_error, 0, ERROR_LEN); // 将数组置零 av_strerror(err, m_error, ERROR_LEN); //将错误码err转换为对�