FFmpeg开发-视频播放【软/硬解码 + OpenGL显示YUV/NV12】05 转载
1、概述
- 最近研究了一下FFmpeg开发,功能实在是太强大了,网上ffmpeg3、4的文章还是很多的,但是学习嘛,最新的还是不能放过,就选了一个最新的ffmpeg n5.1.2版本,和3、4版本api变化还是挺大的;
- 在这个Demo里主要使用Qt + FFmpeg开发一个简单的视频播放器,这里主要使用的是【硬解码】,软解码在之前的文章中有,同时也支持切换软解码;
- 同时为了尽可能的简单,这里没有进行音频解码和播放,只是单独的进行视频解码播放;
- 在之前的文章中使用了QPainter进行绘制显示,也讲了使用OpenGL显示RGB、YUV图像方式;
- 由于FFmpeg硬解码得到的像素格式为NV12,将NV12转换为RGB或者YUV都很麻烦,并且会消耗CPU资源,所以这里直接使用OpenGL显示NV12图像,(将NV12转RGB的步骤放到了GPU中进行);
- 由于去掉了YUV转RGB部分功能,所以这篇文章中的解码代码和之前文章中的有一点小的区别。
开发环境说明
2、实现效果
- 使用ffmpeg音视频库【软/硬解码】实现的视频播放器;
- 支持打开本地视频文件(如mp4、mov、avi等)、网络视频流(rtsp、rtmp、http等);
- 支持视频匀速播放;
- 采用【OpenGL显示YUV、NV12】图像,支持自适应窗口缩放,支持使用QOpenGLWidget、QOpenGLWindow显示;
- 将YUV/NV12转RGB的步骤由CPU转换改为使用GPU转换,降低CPU占用率;
- 使用av_hwframe_map替代av_hwframe_transfer_data,可将【耗时降低1/3】;
- 视频播放支持实时开始/关闭、暂停/继续播放;
- 视频解码、线程控制、显示各部分功能分离,低耦合度。
- 采用最新的【5.1.2版本】ffmpeg库进行开发,超详细注释信息,将所有踩过的坑、解决办法、注意事项都得很写清楚。
![动图封面](https://pic3.zhimg.com/v2-280be5419e72c03c89a2acc9964ca056_b.jpg)
- 使用GPU解码 + OpenGL绘制大大降低了CPU占用率
![](https://pic2.zhimg.com/80/v2-0bf678db7cbc4fc48a6681c5558797a5_1440w.webp)
3、FFmpeg硬解码流程
- 白色是软解码流程,蓝色为多出来的硬解码流程。
- 【更多音视频学习资料,点击下方链接免费领取↓↓,先码住不迷路~】
![](https://pic1.zhimg.com/80/v2-13450e772f6c44427986697a2fb9f078_1440w.webp)
4、优化av_hwframe_transfer_data()性能低问题
- FFmpeg使用硬解码时需要使用*av_hwframe_transfer_data()*函数将解码后的图像数据从GPU拷贝到CPU中,这一步性能非常低,导致使用硬解码和软解码区别不大;
- 但是可以通过使用av_hwframe_map()替代av_hwframe_transfer_data(),通过测试,使用av_hwframe_map耗时比av_hwframe_transfer_data,可将【降低1/3】;
- av_hwframe_map()在FFmpegV3.3以后版本有。
![](https://pic4.zhimg.com/80/v2-96506e36d50f4f2e2ba6906ea4c29d5f_1440w.webp)
5、主要代码
- 啥也不说了,直接上代码,一切有注释
5.1 解码代码
- videodecode.h文件
/******************************************************************************
* @文件名 videodecode.h
* @功能 视频解码类,在这个类中调用ffmpeg打开视频进行解码;
* 使用av_hwframe_map替代av_hwframe_transfer_data,可将【耗时降低1/3】;
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/09/15
* @备注
*****************************************************************************/
#ifndef VIDEODECODE_H
#define VIDEODECODE_H
#include <QString>
#include <QSize>
#include <qlist.h>
#include <qelapsedtimer.h>
struct AVFormatContext;
struct AVCodecContext;
struct AVRational;
struct AVPacket;
struct AVFrame;
struct AVCodec;
struct SwsContext;
struct AVBufferRef;
class QImage;
class VideoDecode
{
public:
VideoDecode();
~VideoDecode();
bool open(const QString& url = QString()); // 打开媒体文件,或者流媒体rtmp、strp、http
AVFrame* read(); // 读取视频图像
void close(); // 关闭
bool isEnd(); // 是否读取完成
const qint64& pts(); // 获取当前帧显示时间
void setHWDecoder(bool flag); // 是否使用硬件解码器
bool isHWDecoder();
private:
void initFFmpeg(); // 初始化ffmpeg库(整个程序中只需加载一次)
void initHWDecoder(const AVCodec* codec); // 初始化硬件解码器
bool initObject(); // 初始化对象
bool dataCopy(); // 硬件解码完成需要将数据从GPU复制到CPU
void showError(int err); // 显示ffmpeg执行错误时的错误信息
qreal rationalToDouble(AVRational* rational); // 将AVRational转换为double
void clear(); // 清空读取缓冲
void free(); // 释放
private:
AVFormatContext* m_formatContext = nullptr; // 解封装上下文
AVCodecContext* m_codecContext = nullptr; // 解码器上下文
SwsContext* m_swsContext = nullptr; // 图像转换上下文
AVPacket* m_packet = nullptr; // 数据包
AVFrame* m_frame = nullptr; // 解码后的视频帧
AVFrame* m_frameHW = nullptr; // 硬件解码后的视频帧
int m_videoIndex = 0; // 视频流索引
qint64 m_totalTime = 0; // 视频总时长
qint64 m_totalFrames = 0; // 视频总帧数
qint64 m_obtainFrames = 0; // 视频当前获取到的帧数
qint64 m_pts = 0; // 图像帧的显示时间
qreal m_frameRate = 0; // 视频帧率
QSize m_size; // 视频分辨率大小
char* m_error = nullptr; // 保存异常信息
bool m_end = false; // 视频读取完成
uchar* m_buffer = nullptr;
QList<int> m_HWDeviceTypes; // 保存当前环境支持的硬件解码器
AVBufferRef* hw_device_ctx = nullptr; // 对数据缓冲区的引用
bool m_HWDecoder = false; // 记录是否使用硬件解码
};
#endif // VIDEODECODE_H
- videodecode.cpp文件
#include "videodecode.h"
#include <QDebug>
#include <QImage>
#include <QMutex>
#include <qdatetime.h>
extern "C" { // 用C规则编译指定的代码
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/avutil.h"
#include "libswscale/swscale.h"
#include "libavutil/imgutils.h"
}
#define ERROR_LEN 1024 // 异常信息数组长度
#define PRINT_LOG 1
VideoDecode::VideoDecode()
{
// initFFmpeg(); // 5.1.2版本不需要调用了
m_error = new char[ERROR_LEN];
/*************************************** 获取当前环境支持的硬件解码器 *********************************************/
AVHWDeviceType type = AV_HWDEVICE_TYPE_NONE; // ffmpeg支持的硬件解码器
QStringList strTypes;
while ((type = av_hwdevice_iterate_types(type)) != AV_HWDEVICE_TYPE_NONE) // 遍历支持的设备类型。
{
m_HWDeviceTypes.append(type);
const char* ctype = av_hwdevice_get_type_name(type); // 获取AVHWDeviceType的字符串名称。
if(ctype)
{
strTypes.append(QString(ctype));
}
}
qDebug() << "支持的硬件解码器:" << strTypes;
/************************************************ END ******************************************************/
}
VideoDecode::~VideoDecode()
{
close();
}
/**
* @brief 初始化ffmpeg库(整个程序中只需加载一次)
* 旧版本的ffmpeg需要注册各种文件格式、解复用器、对网络库进行全局初始化。
* 在新版本的ffmpeg中纷纷弃用了,不需要注册了
*/
void VideoDecode::initFFmpeg()
{
static bool isFirst = true;
static QMutex mutex;
QMutexLocker locker(&mutex);
if(isFirst)
{
// av_register_all(); // 已经从源码中删除
/**
* 初始化网络库,用于打开网络流媒体,此函数仅用于解决旧GnuTLS或OpenSSL库的线程安全问题。
* 一旦删除对旧GnuTLS和OpenSSL库的支持,此函数将被弃用,并且此函数不再有任何用途。
*/
avformat_network_init();
isFirst = false;
}
}
/*********************************** FFmpeg获取GPU硬件解码帧格式的回调函数 *****************************************/
static enum AVPixelFormat g_pixelFormat;
/**
* @brief 回调函数,获取GPU硬件解码帧的格式
* @param s
* @param fmt
* @return
*/
AVPixelFormat get_hw_format(AVCodecContext* s, const enum AVPixelFormat* fmt)
{
Q_UNUSED(s)
const enum AVPixelFormat* p;
for (p = fmt; *p != -1; p++)
{
if(*p == g_pixelFormat)
{
return *p;
}
}
qDebug() << "无法获取硬件表面格式."; // 当同时打开太多路视频时,如果超过了GPU的能力,可能会返回找不到解码帧格式
return AV_PIX_FMT_NONE;
}
/************************************************ END ******************************************************/
/**************************************** FFmpeg初始化硬件解码器 **********************************************/
/**
* @brief 初始化硬件解码器
* @param codec
*/
void VideoDecode::initHWDecoder(const AVCodec *codec)
{
if(!codec) return;
for(int i = 0; ; i++)
{
const AVCodecHWConfig* config = avcodec_get_hw_config(codec, i); // 检索编解码器支持的硬件配置。
if(!config)
{
qDebug() << "打开硬件解码器失败!";
return; // 没有找到支持的硬件配置
}
if (config->methods & AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX) // 判断是否是设备类型
{
for(auto i : m_HWDeviceTypes)
{
if(config->device_type == AVHWDeviceType(i)) // 判断设备类型是否是支持的硬件解码器
{
g_pixelFormat = config->pix_fmt;
// 打开指定类型的设备,并为其创建AVHWDeviceContext。
int ret = av_hwdevice_ctx_create(&hw_device_ctx, config->device_type, nullptr, nullptr, 0);
if(ret < 0)
{
showError(ret);
free();
return ;
}
qDebug() << "打开硬件解码器:" << av_hwdevice_get_type_name(config->device_type);
m_codecContext->hw_device_ctx = av_buffer_ref(hw_device_ctx); // 创建一个对AVBuffer的新引用。
m_codecContext->get_format = get_hw_format; // 由一些解码器调用,以选择将用于输出帧的像素格式
return;
}
}
}
}
}
/************************************************ END ******************************************************/
/**
* @brief 打开媒体文件,或者流媒体,例如rtmp、strp、http
* @param url 视频地址
* @return true:成功 false:失败
*/
bool VideoDecode::open(const QString &url)
{
if(url.isNull()) return false;
AVDictionary* dict = nullptr;
av_dict_set(&dict, "rtsp_transport", "tcp", 0); // 设置rtsp流使用tcp打开,如果打开失败错误信息为【Error number -135 occurred】可以切换(UDP、tcp、udp_multicast、http),比如vlc推流就需要使用udp打开
av_dict_set(&dict, "max_delay", "3", 0); // 设置最大复用或解复用延迟(以微秒为单位)。当通过【UDP】 接收数据时,解复用器尝试重新排序接收到的数据包(因为它们可能无序到达,或者数据包可能完全丢失)。这可以通过将最大解复用延迟设置为零(通过max_delayAVFormatContext 字段)来禁用。
av_dict_set(&dict, "timeout", "1000000", 0); // 以微秒为单位设置套接字 TCP I/O 超时,如果等待时间过短,也可能会还没连接就返回了。
// 打开输入流并返回解封装上下文
int ret = avformat_open_input(&m_formatContext, // 返回解封装上下文
url.toStdString().data(), // 打开视频地址
nullptr, // 如果非null,此参数强制使用特定的输入格式。自动选择解封装器(文件格式)
&dict); // 参数设置
// 释放参数字典
if(dict)
{
av_dict_free(&dict);
}
// 打开视频失败
if(ret < 0)
{
showError(ret);
free();
return false;
}
// 读取媒体文件的数据包以获取流信息。
ret = avformat_find_stream_info(m_formatContext, nullptr);
if(ret < 0)
{
showError(ret);
free();
return false;
}
m_totalTime = m_formatContext->duration / (AV_TIME_BASE / 1000); // 计算视频总时长(毫秒)
#if PRINT_LOG
qDebug() << QString("视频总时长:%1 ms,[%2]").arg(m_totalTime).arg(QTime::fromMSecsSinceStartOfDay(int(m_totalTime)).toString("HH:mm:ss zzz"));
#endif
// 通过AVMediaType枚举查询视频流ID(也可以通过遍历查找),最后一个参数无用
m_videoIndex = av_find_best_stream(m_formatContext, AVMEDIA_TYPE_VIDEO, -1, -1, nullptr, 0);
if(m_videoIndex < 0)
{
showError(m_videoIndex);
free();
return false;
}
AVStream* videoStream = m_formatContext->streams[m_videoIndex]; // 通过查询到的索引获取视频流
// 获取视频图像分辨率(AVStream中的AVCodecContext在新版本中弃用,改为使用AVCodecParameters)
m_size.setWidth(videoStream->codecpar->width);
m_size.setHeight(videoStream->codecpar->height);
m_frameRate = rationalToDouble(&videoStream->avg_frame_rate); // 视频帧率
// 通过解码器ID获取视频解码器(新版本返回值必须使用const)
const AVCodec* codec = avcodec_find_decoder(videoStream->codecpar->codec_id);
m_totalFrames = videoStream->nb_frames;
#if PRINT_LOG
qDebug() << QString("分辨率:[w:%1,h:%2] 帧率:%3 总帧数:%4 解码器:%5")
.arg(m_size.width()).arg(m_size.height()).arg(m_frameRate).arg(m_totalFrames).arg(codec->name);
#endif
// 分配AVCodecContext并将其字段设置为默认值。
m_codecContext = avcodec_alloc_context3(codec);
if(!m_codecContext)
{
#if PRINT_LOG
qWarning() << "创建视频解码器上下文失败!";
#endif
free();
return false;
}
// 使用视频流的codecpar为解码器上下文赋值
ret = avcodec_parameters_to_context(m_codecContext, videoStream->codecpar);
if(ret < 0)
{
showError(ret);
free();
return false;
}
m_codecContext->flags2 |= AV_CODEC_FLAG2_FAST; // 允许不符合规范的加速技巧。
m_codecContext->thread_count = 8; // 使用8线程解码
if(m_HWDecoder)
{
initHWDecoder(codec); // 初始化硬件解码器(在avcodec_open2前调用)
}
// 初始化解码器上下文,如果之前avcodec_alloc_context3传入了解码器,这里设置NULL就可以
ret = avcodec_open2(m_codecContext, nullptr, nullptr);
if(ret < 0)
{
showError(ret);
free();
return false;
}
return initObject();
}
/**
* @brief 初始化需要用到的对象
* @return
*/
bool VideoDecode::initObject()
{
// 分配AVPacket并将其字段设置为默认值。
m_packet = av_packet_alloc();
if(!m_packet)
{
#if PRINT_LOG
qWarning() << "av_packet_alloc() Error!";
#endif
free();
return false;
}
// 分配AVFrame并将其字段设置为默认值。
m_frame = av_frame_alloc();
if(!m_frame)
{
#if PRINT_LOG
qWarning() << "av_frame_alloc() Error!";
#endif
free();
return false;
}
m_frameHW = av_frame_alloc();
if(!m_frameHW)
{
#if PRINT_LOG
qWarning() << "av_frame_alloc() Error!";
#endif
free();
return false;
}
// 由于传递时是浅拷贝,可能显示类还没处理完成,所以如果播放完成就释放可能会崩溃;
if(m_buffer)
{
delete [] m_buffer;
m_buffer = nullptr;
}
// 分配图像空间
int size = av_image_get_buffer_size(AV_PIX_FMT_RGBA, m_size.width(), m_size.height(), 4);
/**
* 【注意:】这里可以多分配一些,否则如果只是安装size分配,大部分视频图像数据拷贝没有问题,
* 但是少部分视频图像在使用sws_scale()拷贝时会超出数组长度,在使用使用msvc debug模式时delete[] m_buffer会报错(HEAP CORRUPTION DETECTED: after Normal block(#32215) at 0x000001AC442830370.CRT delected that the application wrote to memory after end of heap buffer)
* 特别是这个视频流http://vfx.mtime.cn/Video/2019/02/04/mp4/190204084208765161.mp4
*/
m_buffer = new uchar[size + 1000]; // 这里多分配1000个字节就基本不会出现拷贝超出的情况了,反正不缺这点内存
// m_image = new QImage(m_buffer, m_size.width(), m_size.height(), QImage::Format_RGBA8888); // 这种方式分配内存大部分情况下也可以,但是因为存在拷贝超出数组的情况,delete时也会报错
m_end = false;
return true;
}
/**
* @brief 读取并返回视频图像
* @return
*/
AVFrame* VideoDecode::read()
{
// 如果没有打开则返回
if(!m_formatContext)
{
return nullptr;
}
// 读取下一帧数据
int readRet = av_read_frame(m_formatContext, m_packet);
if(readRet < 0)
{
avcodec_send_packet(m_codecContext, m_packet); // 读取完成后向解码器中传如空AVPacket,否则无法读取出最后几帧
}
else
{
if(m_packet->stream_index == m_videoIndex) // 如果是图像数据则进行解码
{
// 计算当前帧时间(毫秒)
#if 1 // 方法一:适用于所有场景,但是存在一定误差
m_packet->pts = qRound64(m_packet->pts * (1000 * rationalToDouble(&m_formatContext->streams[m_videoIndex]->time_base)));
m_packet->dts = qRound64(m_packet->dts * (1000 * rationalToDouble(&m_formatContext->streams[m_videoIndex]->time_base)));
#else // 方法二:适用于播放本地视频文件,计算每一帧时间较准,但是由于网络视频流无法获取总帧数,所以无法适用
m_obtainFrames++;
m_packet->pts = qRound64(m_obtainFrames * (qreal(m_totalTime) / m_totalFrames));
#endif
// 将读取到的原始数据包传入解码器
int ret = avcodec_send_packet(m_codecContext, m_packet);
if(ret < 0)
{
showError(ret);
}
}
}
av_packet_unref(m_packet); // 释放数据包,引用计数-1,为0时释放空间
av_frame_unref(m_frame);
av_frame_unref(m_frameHW);
int ret = avcodec_receive_frame(m_codecContext, m_frame);
if(ret < 0)
{
av_frame_unref(m_frame);
if(readRet < 0)
{
m_end = true; // 当无法读取到AVPacket并且解码器中也没有数据时表示读取完成
}
return nullptr;
}
// 这样写是为了兼容软解码或者硬件解码打开失败情况
AVFrame* m_frameTemp = m_frame;
if(!m_frame->data[0]) // 如果是硬件解码就进入
{
m_frameTemp = m_frameHW;
// 将解码后的数据从GPU拷贝到CPU
if(!dataCopy())
{
return nullptr;
}
}
m_pts = m_frameTemp->pts;
return m_frameTemp;
}
/********************************* FFmpeg初始化硬件后将图像数据从GPU拷贝到CPU *************************************/
/**
* @brief 硬件解码完成需要将数据从GPU复制到CPU
* @return
*/
bool VideoDecode::dataCopy()
{
if(m_frame->format != g_pixelFormat)
{
av_frame_unref(m_frame);
return false;
}
#if 1 // av_hwframe_map在ffmpeg3.3以后才有,经过测试av_hwframe_transfer_data的耗时大概是av_hwframe_map的【1.5倍】
int ret = av_hwframe_map(m_frameHW, m_frame, 0); // 映射硬件数据帧
if(ret < 0)
{
showError(ret);
av_frame_unref(m_frame);
return false;
}
m_frameHW->width = m_frame->width;
m_frameHW->height = m_frame->height;
#else
int ret = av_hwframe_transfer_data(m_frameHW, m_frame, 0); // 将解码后的数据从GPU复制到CPU(m_frameHW) 这一步比较耗时,在这一步之前硬解码速度比软解码快很多
if(ret < 0)
{
showError(ret);
av_frame_unref(m_frame);
return false;
}
av_frame_copy_props(m_frameHW, m_frame); // 仅将“metadata”字段从src复制到dst。
#endif
return true;
}
/************************************************ END ******************************************************/
/**
* @brief 关闭视频播放并释放内存
*/
void VideoDecode::close()
{
clear();
free();
m_totalTime = 0;
m_videoIndex = 0;
m_totalFrames = 0;
m_obtainFrames = 0;
m_pts = 0;
m_frameRate = 0;
m_size = QSize(0, 0);
}
/**
* @brief 视频是否读取完成
* @return
*/
bool VideoDecode::isEnd()
{
return m_end;
}
/**
* @brief 返回当前帧图像播放时间
* @return
*/
const qint64 &VideoDecode::pts()
{
return m_pts;
}
/**
* @brief 设置是否使用硬件解码
* @param flag true:使用 false:不使用
*/
void VideoDecode::setHWDecoder(bool flag)
{
m_HWDecoder = flag;
}
/**
* @brief 返回当前是否使用硬件解码
* @return
*/
bool VideoDecode::isHWDecoder()
{
return m_HWDecoder;
}
/**
* @brief 显示ffmpeg函数调用异常信息
* @param err
*/
void VideoDecode::showError(int err)
{
#if PRINT_LOG
memset(m_error, 0, ERROR_LEN); // 将数组置零
av_strerror(err, m_error, ERROR_LEN);
qWarning() << "DecodeVideo Error:" << m_error;
#else
Q_UNUSED(err)
#endif
}
/**
* @brief 将AVRational转换为double,用于计算帧率
* @param rational
* @return
*/
qreal VideoDecode::rationalToDouble(AVRational* rational)
{
qreal frameRate = (rational->den == 0) ? 0 : (qreal(rational->num) / rational->den);
return frameRate;
}
/**
* @brief 清空读取缓冲
*/
void VideoDecode::clear()
{
// 因为avformat_flush不会刷新AVIOContext (s->pb)。如果有必要,在调用此函数之前调用avio_flush(s->pb)。
if(m_formatContext && m_formatContext->pb)
{
avio_flush(m_formatContext->pb);
}
if(m_formatContext)
{
avformat_flush(m_formatContext); // 清理读取缓冲
}
}
void VideoDecode::free()
{
// 释放上下文swsContext。
if(m_swsContext)
{
sws_freeContext(m_swsContext);
m_swsContext = nullptr; // sws_freeContext不会把上下文置NULL
}
// 释放编解码器上下文和与之相关的所有内容,并将NULL写入提供的指针
if(m_codecContext)
{
avcodec_free_context(&m_codecContext);
}
// 关闭并失败m_formatContext,并将指针置为null
if(m_formatContext)
{
avformat_close_input(&m_formatContext);
}
if(hw_device_ctx)
{
av_buffer_unref(&hw_device_ctx);
}
if(m_packet)
{
av_packet_free(&m_packet);
}
if(m_frame)
{
av_frame_free(&m_frame);
}
if(m_frameHW)
{
av_frame_free(&m_frameHW);
}
}
5.2 OpenGL显示RGB图像代码
- 鼠标右键->Add New…
![](https://pic3.zhimg.com/80/v2-866af0f372433478ead44398ca07fb4a_1440w.webp)
- 创建两个GLSL着色器文件
![](https://pic3.zhimg.com/80/v2-a462e9a5db49e99947aedec04c075b8e_1440w.webp)
- 创建一个资源文件,将刚创建的两个GLSL文件添加进资源文件
![](https://pic1.zhimg.com/80/v2-a8fcadf5a37d8b9b0ab92b34e741dc00_1440w.webp)
- 结果如下图所示
![](https://pic1.zhimg.com/80/v2-54b11b42782a0b88968ce13a175800e4_1440w.webp)
- 顶点着色器 vertex.vsh
#version 330 core
layout (location = 0) in vec3 aPos;
layout (location = 1) in vec2 aTexCord;
out vec2 TexCord; // 纹理坐标
void main()
{
gl_Position = vec4(aPos.x, -aPos.y, aPos.z, 1.0); // 图像坐标和OpenGL坐标Y轴相反,
TexCord = aTexCord;
}
- 片段着色器fragment.fsh,由于YUV420P转RGB和NV12转RGB方式不一样,所以片段着色器中需要加上判断。
#version 330 core
in vec2 TexCord; // 纹理坐标
uniform int format = -1; // 像素格式
uniform sampler2D tex_y;
uniform sampler2D tex_u;
uniform sampler2D tex_v;
uniform sampler2D tex_uv;
void main()
{
vec3 yuv;
vec3 rgb;
if(format == 0) // YUV420P转RGB
{
yuv.x = texture2D(tex_y, TexCord).r;
yuv.y = texture2D(tex_u, TexCord).r-0.5;
yuv.z = texture2D(tex_v, TexCord).r-0.5;
}
else if(format == 23) // NV12转RGB
{
yuv.x = texture2D(tex_y, TexCord.st).r;
yuv.y = texture2D(tex_uv, TexCord.st).r - 0.5;
yuv.z = texture2D(tex_uv, TexCord.st).g - 0.5;
}
else
{
}
rgb = mat3(1.0, 1.0, 1.0,
0.0, -0.39465, 2.03211,
1.13983, -0.58060, 0.0) * yuv;
gl_FragColor = vec4(rgb, 1.0);
}
- OpenGL显示YUV420P/NV12图像这里可以采用QOpenGLWidget或者QOpenGLWIndow进行显示,直接将解码后的AVFrame传入,由于是两种不同的格式,所以将纹理创建、销毁、更新功能分为两部分。
- playimage.h
/******************************************************************************
* @文件名 playimage.h
* @功能 使用OpenGL实现YUV图像的绘制,可通过USE_WINDOW宏切换使用QOpenGLWindow还是QOpenGLWidget
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/10/14
* @备注
*****************************************************************************/
#ifndef PLAYIMAGE_H
#define PLAYIMAGE_H
#include <QWidget>
#include <QOpenGLFunctions_3_3_Core>
#include <qopenglshaderprogram.h>
#include <QOpenGLTexture>
#include <qopenglpixeltransferoptions.h>
struct AVFrame;
#define USE_WINDOW 0 // 1:使用QOpenGLWindow显示, 0:使用QOpenGLWidget显示
#if USE_WINDOW
#include <QOpenGLWindow>
class PlayImage : public QOpenGLWindow, public QOpenGLFunctions_3_3_Core
#else
#include <QOpenGLWidget>
class PlayImage : public QOpenGLWidget, public QOpenGLFunctions_3_3_Core
#endif
{
Q_OBJECT
public:
#if USE_WINDOW
explicit PlayImage(UpdateBehavior updateBehavior = NoPartialUpdate, QWindow *parent = nullptr);
#else
explicit PlayImage(QWidget* parent = nullptr, Qt::WindowFlags f = Qt::WindowFlags());
#endif
~PlayImage() override;
void repaint(AVFrame* frame); // 设置需要绘制的图像帧
protected:
void initializeGL() override; // 初始化gl
void resizeGL(int w, int h) override; // 窗口尺寸变化
void paintGL() override; // 刷新显示
private:
// YUV420图像数据更新
void repaintTexYUV420P(AVFrame* frame);
void initTexYUV420P(AVFrame* frame);
void freeTexYUV420P();
// NV12图像数据更新
void repaintTexNV12(AVFrame* frame);
void initTexNV12(AVFrame* frame);
void freeTexNV12();
private:
QOpenGLShaderProgram* m_program = nullptr;
QOpenGLTexture* m_texY = nullptr;
QOpenGLTexture* m_texU = nullptr;
QOpenGLTexture* m_texV = nullptr;
QOpenGLTexture* m_texUV = nullptr;
QOpenGLPixelTransferOptions m_options;
GLuint VBO = 0; // 顶点缓冲对象,负责将数据从内存放到缓存,一个VBO可以用于多个VAO
GLuint VAO = 0; // 顶点数组对象,任何随后的顶点属性调用都会储存在这个VAO中,一个VAO可以有多个VBO
GLuint EBO = 0; // 元素缓冲对象,它存储 OpenGL 用来决定要绘制哪些顶点的索引
QSize m_size;
QSizeF m_zoomSize;
QPointF m_pos;
int m_format; // 像素格式
};
#endif // PLAYIMAGE_H
- playimage.cpp
#include "playimage.h"
extern "C" { // 用C规则编译指定的代码
#include "libavcodec/avcodec.h"
}
#if USE_WINDOW
PlayImage::PlayImage(QOpenGLWindow::UpdateBehavior updateBehavior, QWindow *parent):QOpenGLWindow(updateBehavior, parent)
{
// 初始化视图大小,由于Shader里面有YUV转RGB的代码,会初始化显示为绿色,这里通过将视图大小设置为0避免显示绿色背景
m_pos = QPointF(0, 0);
m_zoomSize = QSize(0, 0);
}
#else
PlayImage::PlayImage(QWidget *parent, Qt::WindowFlags f): QOpenGLWidget(parent, f)
{
// 初始化视图大小,由于Shader里面有YUV转RGB的代码,会初始化显示为绿色,这里通过将视图大小设置为0避免显示绿色背景
m_pos = QPointF(0, 0);
m_zoomSize = QSize(0, 0);
}
#endif
PlayImage::~PlayImage()
{
if(!isValid()) return; // 如果控件和OpenGL资源(如上下文)已成功初始化,则返回true。
this->makeCurrent(); // 通过将相应的上下文设置为当前上下文并在该上下文中绑定帧缓冲区对象,为呈现此小部件的OpenGL内容做准备。
freeTexYUV420P();
freeTexNV12();
this->doneCurrent(); // 释放上下文
// 释放
glDeleteBuffers(1, &VBO);
glDeleteBuffers(1, &EBO);
glDeleteVertexArrays(1, &VAO);
}
void PlayImage::repaint(AVFrame *frame)
{
if(!frame) return;
m_format = frame->format;
switch (m_format)
{
case AV_PIX_FMT_YUV420P: // ffmpeg软解码的像素格式为YUV420P
{
repaintTexYUV420P(frame);
break;
}
case AV_PIX_FMT_NV12: // 由于ffmpeg硬件解码的像素格式为NV12,不是YUV,所以需要单独处理
{
repaintTexNV12(frame);
break;
}
default: break;
}
av_frame_unref(frame); // 取消引用帧引用的所有缓冲区并重置帧字段。
this->update();
}
/**
* @brief 更新YUV420P图像数据纹理
* @param frame
*/
void PlayImage::repaintTexYUV420P(AVFrame *frame)
{
// 当切换显示的视频后,如果分辨率不同则需要重新创建纹理,否则会崩溃
if(frame->width != m_size.width() || frame->height != m_size.height())
{
freeTexYUV420P();
}
initTexYUV420P(frame);
m_options.setImageHeight(frame->height);
m_options.setRowLength(frame->linesize[0]);
m_texY->setData(QOpenGLTexture::Red, QOpenGLTexture::UInt8, static_cast<const void *>(frame->data[0]), &m_options); // 设置图像数据 Y
m_options.setRowLength(frame->linesize[1]);
m_texU->setData(QOpenGLTexture::Red, QOpenGLTexture::UInt8, static_cast<const void *>(frame->data[1]), &m_options); // 设置图像数据 U
m_options.setRowLength(frame->linesize[2]);
m_texV->setData(QOpenGLTexture::Red, QOpenGLTexture::UInt8, static_cast<const void *>(frame->data[2]), &m_options); // 设置图像数据 V
}
/**
* @brief 初始化YUV420P图像纹理
* @param frame
*/
void PlayImage::initTexYUV420P(AVFrame *frame)
{
if(!m_texY) // 初始化纹理
{
// 创建2D纹理
m_texY = new QOpenGLTexture(QOpenGLTexture::Target2D);
// 设置纹理大小
m_texY->setSize(frame->width, frame->height);
// 设置放大、缩小过滤器
m_texY->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
// 设置图像格式
m_texY->setFormat(QOpenGLTexture::R8_UNorm);
// 分配内存
m_texY->allocateStorage();
// 记录图像分辨率
m_size.setWidth(frame->width);
m_size.setHeight(frame->height);
resizeGL(this->width(), this->height());
}
if(!m_texU)
{
m_texU = new QOpenGLTexture(QOpenGLTexture::Target2D);
m_texU->setSize(frame->width / 2, frame->height / 2);
m_texU->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
m_texU->setFormat(QOpenGLTexture::R8_UNorm);
m_texU->allocateStorage();
}
if(!m_texV) // 初始化纹理
{
m_texV = new QOpenGLTexture(QOpenGLTexture::Target2D);
m_texV->setSize(frame->width / 2, frame->height / 2);
m_texV->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
m_texV->setFormat(QOpenGLTexture::R8_UNorm);
m_texV->allocateStorage();
}
}
/**
* @brief 释放YUV420P图像纹理
*/
void PlayImage::freeTexYUV420P()
{
// 释放纹理
if(m_texY)
{
m_texY->destroy();
delete m_texY;
m_texY = nullptr;
}
if(m_texU)
{
m_texU->destroy();
delete m_texU;
m_texU = nullptr;
}
if(m_texV)
{
m_texV->destroy();
delete m_texV;
m_texV = nullptr;
}
}
/**
* @brief 更新NV12图像数据纹理
* @param frame
*/
void PlayImage::repaintTexNV12(AVFrame *frame)
{
// 当切换显示的视频后,如果分辨率不同则需要重新创建纹理,否则会崩溃
if(frame->width != m_size.width() || frame->height != m_size.height())
{
freeTexNV12();
}
initTexNV12(frame);
m_options.setImageHeight(frame->height);
m_options.setRowLength(frame->linesize[0]);
m_texY->setData(QOpenGLTexture::Red, QOpenGLTexture::UInt8, static_cast<const void *>(frame->data[0]), &m_options); // 设置图像数据 Y
m_options.setImageHeight(frame->height / 2);
m_options.setRowLength(frame->linesize[1] / 2);
m_texUV->setData(QOpenGLTexture::RG, QOpenGLTexture::UInt8, static_cast<const void *>(frame->data[1]), &m_options); // 设置图像数据 UV
}
/**
* @brief 初始化NV12图像纹理
* @param frame
*/
void PlayImage::initTexNV12(AVFrame *frame)
{
if(!m_texY) // 初始化纹理
{
// 创建2D纹理
m_texY = new QOpenGLTexture(QOpenGLTexture::Target2D);
// 设置纹理大小
m_texY->setSize(frame->width, frame->height);
// 设置放大、缩小过滤器
m_texY->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
// 设置图像格式
m_texY->setFormat(QOpenGLTexture::R8_UNorm);
// 分配内存
m_texY->allocateStorage();
// 记录图像分辨率
m_size.setWidth(frame->width);
m_size.setHeight(frame->height);
resizeGL(this->width(), this->height());
}
if(!m_texUV)
{
m_texUV = new QOpenGLTexture(QOpenGLTexture::Target2D);
m_texUV->setSize(frame->width / 2, frame->height / 2);
m_texUV->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
m_texUV->setFormat(QOpenGLTexture::RG8_UNorm);
m_texUV->allocateStorage();
}
}
/**
* @brief 释放NV12图像纹理
*/
void PlayImage::freeTexNV12()
{
// 释放纹理
if(m_texY)
{
m_texY->destroy();
delete m_texY;
m_texY = nullptr;
}
if(m_texUV)
{
m_texUV->destroy();
delete m_texUV;
m_texUV = nullptr;
}
}
// 三个顶点坐标XYZ,VAO、VBO数据播放,范围时[-1 ~ 1]直接
static GLfloat vertices[] = { // 前三列点坐标,后两列为纹理坐标
1.0f, 1.0f, 0.0f, 1.0f, 1.0f, // 右上角
1.0f, -1.0f, 0.0f, 1.0f, 0.0f, // 右下
-1.0f, -1.0f, 0.0f, 0.0f, 0.0f, // 左下
-1.0f, 1.0f, 0.0f, 0.0f, 1.0f // 左上
};
static GLuint indices[] = {
0, 1, 3,
1, 2, 3
};
void PlayImage::initializeGL()
{
initializeOpenGLFunctions();
// 加载shader脚本程序
m_program = new QOpenGLShaderProgram(this);
m_program->addShaderFromSourceFile(QOpenGLShader::Vertex, ":/vertex.vsh");
m_program->addShaderFromSourceFile(QOpenGLShader::Fragment, ":/fragment.fsh");
m_program->link();
// 绑定YUV 变量值
m_program->bind();
m_program->setUniformValue("tex_y", 0);
m_program->setUniformValue("tex_u", 1);
m_program->setUniformValue("tex_v", 2);
m_program->setUniformValue("tex_uv", 3);
// 返回属性名称在此着色器程序的参数列表中的位置。如果名称不是此着色器程序的有效属性,则返回-1。
GLuint posAttr = GLuint(m_program->attributeLocation("aPos"));
GLuint texCord = GLuint(m_program->attributeLocation("aTexCord"));
glGenVertexArrays(1, &VAO);
glBindVertexArray(VAO);
glGenBuffers(1, &VBO);
glBindBuffer(GL_ARRAY_BUFFER, VBO);
glGenBuffers(1, &EBO); // 创建一个EBO
glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, EBO);
// 为当前绑定到的缓冲区对象创建一个新的数据存储target。任何预先存在的数据存储都将被删除。
glBufferData(GL_ARRAY_BUFFER, // 为VBO缓冲绑定顶点数据
sizeof (vertices), // 数组字节大小
vertices, // 需要绑定的数组
GL_STATIC_DRAW); // 指定数据存储的预期使用模式,GL_STATIC_DRAW: 数据几乎不会改变
glBufferData(GL_ELEMENT_ARRAY_BUFFER, sizeof(indices), indices, GL_STATIC_DRAW); // 将顶点索引数组传入EBO缓存
// 设置顶点坐标数据
glVertexAttribPointer(posAttr, // 指定要修改的通用顶点属性的索引
3, // 指定每个通用顶点属性的组件数(如vec3:3,vec4:4)
GL_FLOAT, // 指定数组中每个组件的数据类型(数组中一行有几个数)
GL_FALSE, // 指定在访问定点数据值时是否应规范化 ( GL_TRUE) 或直接转换为定点值 ( GL_FALSE),如果vertices里面单个数超过-1或者1可以选择GL_TRUE
5 * sizeof(GLfloat), // 指定连续通用顶点属性之间的字节偏移量。
nullptr); // 指定当前绑定到目标的缓冲区的数据存储中数组中第一个通用顶点属性的第一个组件的偏移量。初始值为0 (一个数组从第几个字节开始读)
// 启用通用顶点属性数组
glEnableVertexAttribArray(posAttr); // 属性索引是从调用glGetAttribLocation接收的,或者传递给glBindAttribLocation。
// 设置纹理坐标数据
glVertexAttribPointer(texCord, 2, GL_FLOAT, GL_FALSE, 5 * sizeof(GLfloat), reinterpret_cast<const GLvoid *>(3 * sizeof (GLfloat))); // 指定当前绑定到目标的缓冲区的数据存储中数组中第一个通用顶点属性的第一个组件的偏移量。初始值为0 (一个数组从第几个字节开始读)
// 启用通用顶点属性数组
glEnableVertexAttribArray(texCord); // 属性索引是从调用glGetAttribLocation接收的,或者传递给glBindAttribLocation。
// 释放
glBindBuffer(GL_ARRAY_BUFFER, 0);
glBindVertexArray(0); // 设置为零以破坏现有的顶点数组对象绑定
glClearColor(0.0f, 0.0f, 0.0f, 1.0f); // 指定颜色缓冲区的清除值(背景色)
}
void PlayImage::resizeGL(int w, int h)
{
if(m_size.width() < 0 || m_size.height() < 0) return;
// 计算需要显示图片的窗口大小,用于实现长宽等比自适应显示
if((double(w) / h) < (double(m_size.width()) / m_size.height()))
{
m_zoomSize.setWidth(w);
m_zoomSize.setHeight(((double(w) / m_size.width()) * m_size.height())); // 这里不使用QRect,使用QRect第一次设置时有误差bug
}
else
{
m_zoomSize.setHeight(h);
m_zoomSize.setWidth((double(h) / m_size.height()) * m_size.width());
}
m_pos.setX(double(w - m_zoomSize.width()) / 2);
m_pos.setY(double(h - m_zoomSize.height()) / 2);
this->update(QRect(0, 0, w, h));
}
void PlayImage::paintGL()
{
glClear(GL_COLOR_BUFFER_BIT); // 将窗口的位平面区域(背景)设置为先前由glClearColor、glClearDepth和选择的值
glViewport(m_pos.x(), m_pos.y(), m_zoomSize.width(), m_zoomSize.height()); // 设置视图大小实现图片自适应
m_program->bind(); // 绑定着色器
m_program->setUniformValue("format", m_format);
// 绑定纹理
switch (m_format)
{
case AV_PIX_FMT_YUV420P:
{
if(m_texY && m_texU && m_texV)
{
m_texY->bind(0);
m_texU->bind(1);
m_texV->bind(2);
}
break;
}
case AV_PIX_FMT_NV12:
{
if(m_texY && m_texUV)
{
m_texY->bind(0);
m_texUV->bind(3);
}
break;
}
default: break;
}
glBindVertexArray(VAO); // 绑定VAO
glDrawElements(GL_TRIANGLES, // 绘制的图元类型
6, // 指定要渲染的元素数(点数)
GL_UNSIGNED_INT, // 指定索引中值的类型(indices)
nullptr); // 指定当前绑定到GL_ELEMENT_array_buffer目标的缓冲区的数据存储中数组中第一个索引的偏移量。
glBindVertexArray(0);
// 释放纹理
switch (m_format)
{
case AV_PIX_FMT_YUV420P:
{
if(m_texY && m_texU && m_texV)
{
m_texY->release();
m_texU->release();
m_texV->release();
}
break;
}
case AV_PIX_FMT_NV12:
{
if(m_texY && m_texUV)
{
m_texY->release();
m_texUV->release();
}
break;
}
default: break;
}
m_program->release();
}
6、完整源代码
https://gitee.com/mahuifa/QMDemo
https://zhuanlan.zhihu.com/p/595036632