提交 c6644e7d 编写于 作者: mahuifa's avatar mahuifa

Merge branch 'Dev'

......@@ -6,10 +6,12 @@
## 1、说明
| 类名 | 功能 |
| ----------- | ---------------------------------------------- |
| VideoPlay | 使用ffmpeg音视频库【软解码】实现的视频播放器; |
| VideoPlayHW | 使用ffmpeg音视频库【硬解码】实现的视频播放器; |
| 类名 | 功能 |
| ------------ | ------------------------------------------------------------ |
| VideoPlay | 使用ffmpeg音视频库【软解码】实现的视频播放器; |
| VideoPlayGL1 | 使用ffmpeg音视频库【软解码 + OpenGL显示RGB图像】实现的视频播放器; |
| VideoPlayGL2 | 使用ffmpeg音视频库【软解码 + OpenGL显示YUV图像】实现的视频播放器; |
| VideoPlayHW | 使用ffmpeg音视频库【硬解码】实现的视频播放器; |
......@@ -34,7 +36,44 @@
![VideoPlay-tuya](FFmpegDemo.assets/VideoPlay-tuya.gif)
### 1.2 VideoPlayHW
### 1.2 VideoPlayGL1
> 1. 使用ffmpeg音视频库【软解码】实现的视频播放器;
> 2. 支持打开本地视频文件(如mp4、mov、avi等)、网络视频流(rtsp、rtmp、http等);
> 3. 支持视频匀速播放;
> 4. 采用【OpenGL显示RGB】图像,支持自适应窗口缩放,支持使用QOpenGLWidget、QOpenGLWindow显
> 5. 视频播放支持实时开始/关闭、暂停/继续播放;
> 6. 视频解码、线程控制、显示各部分功能分离,低耦合度。
> 7. 采用最新的5.1.2版本ffmpeg库进行开发,超详细注释信息,将所有踩过的坑、解决办法、注意事项都得很写清楚。
>
* 下图中使用OpenGL显示RGB图像CPU占用率是使用QPainter显示的<mark>一半</mark>,由于我使用的是非常老的笔记本的集显测试,所以GPU占用率比较高。
![image-20221015204308041](FFmpegDemo.assets/image-20221015204308041.png)
### 1.3 VideoPlayGL2
> 1. 使用ffmpeg音视频库【软解码】实现的视频播放器;
> 2. 支持打开本地视频文件(如mp4、mov、avi等)、网络视频流(rtsp、rtmp、http等);
> 3. 支持视频匀速播放;
> 4. 采用【OpenGL显示YUV】图像,支持自适应窗口缩放,支持使用QOpenGLWidget、QOpenGLWindow显示;
> 5. 将YUV转RGB的步骤由CPU转换改为使用GPU转换,降低CPU占用率;
> 6. 视频播放支持实时开始/关闭、暂停/继续播放;
> 7. 视频解码、线程控制、显示各部分功能分离,低耦合度。
> 8. 采用最新的5.1.2版本ffmpeg库进行开发,超详细注释信息,将所有踩过的坑、解决办法、注意事项都得很写清楚。
>
* 下图中使用OpenGL显示YUV图像CPU占用率是使用QPainter显示的<mark>1/3左右</mark>,由于我使用的是非常老的笔记本的集显测试,所以GPU占用率比较高。
![image-20221017232820037](FFmpegDemo.assets/image-20221017232820037.png)
### 1.4 VideoPlayHW
> 1. 使用ffmpeg音视频库【硬解码】实现的视频播放器,采用GPU解码, 大幅降低对CPU的暂用率;
> 2. 支持打开本地视频文件(如mp4、mov、avi等)、网络视频流(rtsp、rtmp、http等);
......
......@@ -11,4 +11,6 @@
TEMPLATE = subdirs
SUBDIRS += VideoPlay # 使用软解码实现的视频播放器
SUBDIRS += VideoPlayGL1 # 使用软解码实现的视频播放器 使用OpenGL显示RGB图像
SUBDIRS += VideoPlayGL2 # 使用软解码实现的视频播放器 使用OpenGL显示YUV图像
SUBDIRS += VideoPlayHW # 使用硬件解码实现的视频播放器
......@@ -87,6 +87,6 @@ void Widget::on_playState(ReadThread::PlayState state)
{
ui->but_open->setText("开始播放");
ui->but_pause->setText("暂停");
this->setWindowTitle(QString("Qt+ffmpeg视频播放Demo V%1").arg(APP_VERSION));
this->setWindowTitle(QString("Qt+ffmpeg视频播放(软解码)Demo V%1").arg(APP_VERSION));
}
}
#---------------------------------------------------------------------------------------
# @功能: ffmpeg读取视频图像模块
# @编译器: Desktop Qt 5.12.5 MSVC2017 64bit(也支持其它编译器)
# @Qt IDE: D:/Qt/Qt5.12.5/Tools/QtCreator/share/qtcreator
#
# @开发者 mhf
# @邮箱 1603291350@qq.com
# @时间 2022-09-15 14:27:25
# @备注
#---------------------------------------------------------------------------------------
# 加载库,ffmpeg n5.1.2版本
win32{
LIBS += -LE:/lib/ffmpeg5-1-2/lib/ -lavcodec -lavfilter -lavformat -lswscale -lavutil -lswresample -lavdevice
INCLUDEPATH += E:/lib/ffmpeg5-1-2/include
DEPENDPATH += E:/lib/ffmpeg5-1-2/include
}
unix:!macx{
LIBS += -L/home/mhf/lib/ffmpeg/ffmpeg-5-1-2/lib -lavcodec -lavfilter -lavformat -lswscale -lavutil -lswresample -lavdevice
INCLUDEPATH += /home/mhf/lib/ffmpeg/ffmpeg-5-1-2/include
DEPENDPATH += /home/mhf/lib/ffmpeg/ffmpeg-5-1-2/include
}
HEADERS += \
$$PWD/readthread.h \
$$PWD/videodecode.h
SOURCES += \
$$PWD/readthread.cpp \
$$PWD/videodecode.cpp
#include "readthread.h"
#include "videodecode.h"
#include <QEventLoop>
#include <QTimer>
#include <QDebug>
#include <qimage.h>
#include <playimage.h>
ReadThread::ReadThread(QObject *parent) : QThread(parent)
{
m_videoDecode = new VideoDecode();
qRegisterMetaType<PlayState>("PlayState"); // 注册自定义枚举类型,否则信号槽无法发送
}
ReadThread::~ReadThread()
{
if(m_videoDecode)
{
delete m_videoDecode;
}
}
/**
* @brief 传入播放的视频地址并开启线程
* @param url
*/
void ReadThread::open(const QString &url)
{
if(!this->isRunning())
{
m_url = url;
emit this->start();
}
}
/**
* @brief 控制暂停、继续
* @param flag true:暂停 fasle:继续
*/
void ReadThread::pause(bool flag)
{
m_pause = flag;
}
/**
* @brief 关闭播放
*/
void ReadThread::close()
{
m_play = false;
m_pause = false;
}
/**
* @brief 返回当前播放的地址
* @return
*/
const QString &ReadThread::url()
{
return m_url;
}
/**
* @brief 非阻塞延时
* @param msec 延时毫秒
*/
void sleepMsec(int msec)
{
if(msec <= 0) return;
QEventLoop loop; //定义一个新的事件循环
QTimer::singleShot(msec, &loop, SLOT(quit()));//创建单次定时器,槽函数为事件循环的退出函数
loop.exec(); //事件循环开始执行,程序会卡在这里,直到定时时间到,本循环被退出
}
void ReadThread::run()
{
bool ret = m_videoDecode->open(m_url); // 打开网络流时会比较慢,如果放到Ui线程会卡
if(ret)
{
m_play = true;
m_etime1.start();
m_etime2.start();
emit playState(play);
}
else
{
qWarning() << "打开失败!";
}
// 循环读取视频图像
while (m_play)
{
// 暂停
while (m_pause)
{
sleepMsec(200);
}
QImage image = m_videoDecode->read(); // 读取视频图像
if(!image.isNull())
{
// 1倍速播放
#if 0
sleepMsec(int(m_decodeVideo->pts() - m_etime1.elapsed())); // 不支持后退
#else
sleepMsec(int(m_videoDecode->pts() - m_etime2.elapsed())); // 支持后退
#endif
emit updateImage(image);
}
else
{
// 当前读取到无效图像时判断是否读取完成
if(m_videoDecode->isEnd())
{
break;
}
sleepMsec(1); // 这里不能使用QThread::msleep()延时,否则会很不稳定
}
}
qDebug() << "播放结束!";
m_videoDecode->close();
emit playState(end);
}
/******************************************************************************
* @文件名 readthread.h
* @功能 读取视频图像数据线程,在线程中解码视频
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/09/15
* @备注
*****************************************************************************/
#ifndef READTHREAD_H
#define READTHREAD_H
#include <QElapsedTimer>
#include <QThread>
#include <QTime>
class VideoDecode;
class PlayImage;
class ReadThread : public QThread
{
Q_OBJECT
public:
enum PlayState // 视频播放状态
{
play,
end
};
public:
explicit ReadThread(QObject *parent = nullptr);
~ReadThread() override;
void open(const QString& url = QString()); // 打开视频
void pause(bool flag); // 暂停视频
void close(); // 关闭视频
const QString& url(); // 获取打开的视频地址
protected:
void run() override;
signals:
void updateImage(const QImage& image); // 将读取到的视频图像发送出去
void playState(PlayState state); // 视频播放状态发送改变时触发
private:
VideoDecode* m_videoDecode = nullptr; // 视频解码类
QString m_url; // 打开的视频地址
bool m_play = false; // 播放控制
bool m_pause = false; // 暂停控制
QElapsedTimer m_etime1; // 控制视频播放速度(更精确,但不支持视频后退)
QTime m_etime2; // 控制视频播放速度(支持视频后退)
};
#endif // READTHREAD_H
#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];
}
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;
}
}
/**
* @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线程解码
// 初始化解码器上下文,如果之前avcodec_alloc_context3传入了解码器,这里设置NULL就可以
ret = avcodec_open2(m_codecContext, nullptr, nullptr);
if(ret < 0)
{
showError(ret);
free();
return false;
}
// 分配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;
}
// 分配图像空间
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
*/
QImage VideoDecode::read()
{
// 如果没有打开则返回
if(!m_formatContext)
{
return QImage();
}
// 读取下一帧数据
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时释放空间
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 QImage();
}
m_pts = m_frame->pts;
// 为什么图像转换上下文要放在这里初始化呢,是因为m_frame->format,如果使用硬件解码,解码出来的图像格式和m_codecContext->pix_fmt的图像格式不一样,就会导致无法转换为QImage
if(!m_swsContext)
{
// 获取缓存的图像转换上下文。首先校验参数是否一致,如果校验不通过就释放资源;然后判断上下文是否存在,如果存在直接复用,如不存在进行分配、初始化操作
m_swsContext = sws_getCachedContext(m_swsContext,
m_frame->width, // 输入图像的宽度
m_frame->height, // 输入图像的高度
(AVPixelFormat)m_frame->format, // 输入图像的像素格式
m_size.width(), // 输出图像的宽度
m_size.height(), // 输出图像的高度
AV_PIX_FMT_RGBA, // 输出图像的像素格式
SWS_BILINEAR, // 选择缩放算法(只有当输入输出图像大小不同时有效),一般选择SWS_FAST_BILINEAR
nullptr, // 输入图像的滤波器信息, 若不需要传NULL
nullptr, // 输出图像的滤波器信息, 若不需要传NULL
nullptr); // 特定缩放算法需要的参数(?),默认为NULL
if(!m_swsContext)
{
#if PRINT_LOG
qWarning() << "sws_getCachedContext() Error!";
#endif
free();
return QImage();
}
}
// AVFrame转QImage
uchar* data[] = {m_buffer};
int lines[4];
av_image_fill_linesizes(lines, AV_PIX_FMT_RGBA, m_frame->width); // 使用像素格式pix_fmt和宽度填充图像的平面线条大小。
ret = sws_scale(m_swsContext, // 缩放上下文
m_frame->data, // 原图像数组
m_frame->linesize, // 包含源图像每个平面步幅的数组
0, // 开始位置
m_frame->height, // 行数
data, // 目标图像数组
lines); // 包含目标图像每个平面的步幅的数组
QImage image(m_buffer, m_frame->width, m_frame->height, QImage::Format_RGBA8888);
av_frame_unref(m_frame);
return image;
}
/**
* @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 显示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(m_packet)
{
av_packet_free(&m_packet);
}
if(m_frame)
{
av_frame_free(&m_frame);
}
if(m_buffer)
{
delete [] m_buffer;
m_buffer = nullptr;
}
}
/******************************************************************************
* @文件名 videodecode.h
* @功能 视频解码类,在这个类中调用ffmpeg打开视频进行解码
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/09/15
* @备注
*****************************************************************************/
#ifndef VIDEODECODE_H
#define VIDEODECODE_H
#include <QString>
#include <QSize>
struct AVFormatContext;
struct AVCodecContext;
struct AVRational;
struct AVPacket;
struct AVFrame;
struct SwsContext;
struct AVBufferRef;
class QImage;
class VideoDecode
{
public:
VideoDecode();
~VideoDecode();
bool open(const QString& url = QString()); // 打开媒体文件,或者流媒体rtmp、strp、http
QImage read(); // 读取视频图像
void close(); // 关闭
bool isEnd(); // 是否读取完成
const qint64& pts(); // 获取当前帧显示时间
private:
void initFFmpeg(); // 初始化ffmpeg库(整个程序中只需加载一次)
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; // 解码后的视频帧
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; // YUV图像需要转换位RGBA图像,这里保存转换后的图形数据
};
#endif // VIDEODECODE_H
#---------------------------------------------------------------------------------------
# @功能: 使用ffmpeg音视频库【软解码】实现的视频播放器;
# @编译器: Desktop Qt 5.12.5 MSVC2017 64bit 32bit(也支持其它编译器)
# @Qt IDE D:/Qt/Qt5.12.5/Tools/QtCreator/share/qtcreator
#
# @开发者 mhf
# @邮箱 1603291350@qq.com
# @时间 2022-09-14 14:13:52
# @备注 1、使用ffmpeg音视频库【软解码】实现的视频播放器;
# 2、支持打开本地视频文件(如mp4movavi等)、网络视频流(rtsprtmphttp等);
# 3、支持视频匀速播放;
# 4、采用【OpenGL显示RGB】图像,支持自适应窗口缩放,支持使用QOpenGLWidgetQOpenGLWindow显示;
# 5、视频播放支持实时开始/关闭、暂停/继续播放;
# 6、视频解码、线程控制、显示各部分功能分离,低耦合度。
# 7、采用最新的5.1.2版本ffmpeg库进行开发,超详细注释信息,将所有踩过的坑、解决办法、注意事项都得很写清楚。
#---------------------------------------------------------------------------------------
QT += core gui
greaterThan(QT_MAJOR_VERSION, 4): QT += widgets
CONFIG += c++11
DEFINES += QT_DEPRECATED_WARNINGS
SOURCES += \
playimage.cpp \
main.cpp \
widget.cpp
HEADERS += \
playimage.h \
widget.h
FORMS += \
widget.ui
# Default rules for deployment.
qnx: target.path = /tmp/$${TARGET}/bin
else: unix:!android: target.path = /opt/$${TARGET}/bin
!isEmpty(target.path): INSTALLS += target
# ffmpeg读取视频图像模块
include(./VideoPlay/VideoPlay.pri)
INCLUDEPATH += ./VideoPlay
# 定义程序版本号
VERSION = 1.0.0
DEFINES += APP_VERSION=\\\"$$VERSION\\\"
TARGET = VideoPlayGL1
contains(QT_ARCH, i386){ # 使用32位编译器
DESTDIR = $$PWD/../bin # 程序输出路径
}else{
DESTDIR = $$PWD/../bin64 # 使用64位编译器
}
# msvc 编译器使用utf-8编码
msvc {
QMAKE_CFLAGS += /utf-8
QMAKE_CXXFLAGS += /utf-8
}
DISTFILES += \
fragment.fsh \
vertex.vsh
RESOURCES += \
rc.qrc
#version 330 core
out vec4 FragColor;
in vec2 TexCord; // 纹理坐标
uniform sampler2D texture; // 纹理采样器
void main()
{
FragColor = texture2D(texture, TexCord); // 采样纹理函数
}
#include "widget.h"
#include <QApplication>
int main(int argc, char *argv[])
{
QApplication a(argc, argv);
Widget w;
w.show();
return a.exec();
}
#include "playimage.h"
#if USE_WINDOW
PlayImage::PlayImage(QOpenGLWindow::UpdateBehavior updateBehavior, QWindow *parent):QOpenGLWindow(updateBehavior, parent)
{
}
#else
PlayImage::PlayImage(QWidget *parent, Qt::WindowFlags f): QOpenGLWidget(parent, f)
{
}
#endif
PlayImage::~PlayImage()
{
if(!isValid()) return; // 如果控件和OpenGL资源(如上下文)已成功初始化,则返回true。
this->makeCurrent(); // 通过将相应的上下文设置为当前上下文并在该上下文中绑定帧缓冲区对象,为呈现此小部件的OpenGL内容做准备。
// 释放纹理
if(m_texture)
{
m_texture->destroy();
delete m_texture;
}
this->doneCurrent(); // 释放上下文
// 释放
glDeleteBuffers(1, &VBO);
glDeleteBuffers(1, &EBO);
glDeleteVertexArrays(1, &VAO);
}
/**
* @brief 传入Qimage图片显示
* @param image
*/
void PlayImage::updateImage(const QImage& image)
{
if(image.isNull()) return;
m_size = image.size();
if(!m_texture)
{
m_texture = new QOpenGLTexture(image.mirrored());
resizeGL(this->width(), this->height());
}
else
{
m_texture->destroy();
m_texture->setData(image.mirrored());
}
this->update();
}
// 三个顶点坐标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();
m_program = new QOpenGLShaderProgram(this);
m_program->addShaderFromSourceFile(QOpenGLShader::Vertex, ":/vertex.vsh");
m_program->addShaderFromSourceFile(QOpenGLShader::Fragment, ":/fragment.fsh");
m_program->link();
// 返回属性名称在此着色器程序的参数列表中的位置。如果名称不是此着色器程序的有效属性,则返回-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(); // 绑定着色器
if(m_texture)
{
m_texture->bind();
}
glBindVertexArray(VAO); // 绑定VAO
glDrawElements(GL_TRIANGLES, // 绘制的图元类型
6, // 指定要渲染的元素数(点数)
GL_UNSIGNED_INT, // 指定索引中值的类型(indices)
nullptr); // 指定当前绑定到GL_ELEMENT_array_buffer目标的缓冲区的数据存储中数组中第一个索引的偏移量。
glBindVertexArray(0);
if(m_texture)
{
m_texture->release();
}
m_program->release();
}
/******************************************************************************
* @文件名 playimage.h
* @功能 使用OpenGL实现RGB图像的绘制,可通过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>
#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 updateImage(const QImage& image);
protected:
void initializeGL() override; // 初始化gl
void resizeGL(int w, int h) override; // 窗口尺寸变化
void paintGL() override; // 刷新显示
private:
QOpenGLShaderProgram* m_program = nullptr;
QOpenGLTexture* m_texture = nullptr;
GLuint VBO = 0; // 顶点缓冲对象,负责将数据从内存放到缓存,一个VBO可以用于多个VAO
GLuint VAO = 0; // 顶点数组对象,任何随后的顶点属性调用都会储存在这个VAO中,一个VAO可以有多个VBO
GLuint EBO = 0; // 元素缓冲对象,它存储 OpenGL 用来决定要绘制哪些顶点的索引
QSize m_size;
QSizeF m_zoomSize;
QPointF m_pos;
};
#endif // PLAYIMAGE_H
<RCC>
<qresource prefix="/">
<file>fragment.fsh</file>
<file>vertex.vsh</file>
</qresource>
</RCC>
#version 330 core
layout (location = 0) in vec3 aPos;
layout (location = 1) in vec2 aTexCord;
out vec2 TexCord; // 纹理坐标
void main()
{
gl_Position = vec4(aPos, 1.0);
TexCord = aTexCord;
}
#include "widget.h"
#include "ui_widget.h"
#include <QFileDialog>
Widget::Widget(QWidget *parent)
: QWidget(parent)
, ui(new Ui::Widget)
{
ui->setupUi(this);
this->setWindowTitle(QString("Qt+ffmpeg视频播放(软解码 + OpenGL显示RGB)Demo V%1").arg(APP_VERSION));
// 使用QOpenGLWindow绘制
playImage = new PlayImage;
#if USE_WINDOW
ui->verticalLayout->addWidget(QWidget::createWindowContainer(playImage)); // 这一步加载速度要比OpenGLWidget慢一点
#else
ui->verticalLayout->addWidget(playImage);
#endif
m_readThread = new ReadThread();
connect(m_readThread, &ReadThread::updateImage, playImage, &PlayImage::updateImage);
connect(m_readThread, &ReadThread::playState, this, &Widget::on_playState);
}
Widget::~Widget()
{
// 释放视频读取线程
if(m_readThread)
{
m_readThread->close();
m_readThread->wait();
delete m_readThread;
}
delete ui;
}
/**
* @brief 获取本地视频路径
*/
void Widget::on_but_file_clicked()
{
QString strName = QFileDialog::getOpenFileName(this, "选择播放视频~!", "/", "视频 (*.mp4 *.m4v *.mov *.avi *.flv);; 其它(*)");
if(strName.isEmpty())
{
return;
}
ui->com_url->setCurrentText(strName);
}
/**
* @brief 视频播放/停止
*/
void Widget::on_but_open_clicked()
{
if(ui->but_open->text() == "开始播放")
{
m_readThread->open(ui->com_url->currentText());
}
else
{
m_readThread->close();
}
}
/**
* @brief 视频暂停/继续
*/
void Widget::on_but_pause_clicked()
{
if(ui->but_pause->text() == "暂停")
{
m_readThread->pause(true);
ui->but_pause->setText("继续");
}
else
{
m_readThread->pause(false);
ui->but_pause->setText("暂停");
}
}
/**
* @brief 根据视频播放状态切换界面设置
* @param state
*/
void Widget::on_playState(ReadThread::PlayState state)
{
if(state == ReadThread::play)
{
this->setWindowTitle(QString("正在播放:%1").arg(m_readThread->url()));
ui->but_open->setText("停止播放");
}
else
{
ui->but_open->setText("开始播放");
ui->but_pause->setText("暂停");
this->setWindowTitle(QString("Qt+ffmpeg视频播放(软解码 + OpenGL显示RGB)Demo V%1").arg(APP_VERSION));
}
}
#ifndef WIDGET_H
#define WIDGET_H
#include <QWidget>
#include "readthread.h"
#include "playimage.h"
QT_BEGIN_NAMESPACE
namespace Ui { class Widget; }
QT_END_NAMESPACE
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = nullptr);
~Widget();
private slots:
void on_but_file_clicked();
void on_but_open_clicked();
void on_but_pause_clicked();
void on_playState(ReadThread::PlayState state);
private:
Ui::Widget *ui;
PlayImage* playImage = nullptr;
ReadThread* m_readThread = nullptr;
};
#endif // WIDGET_H
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>Widget</class>
<widget class="QWidget" name="Widget">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>800</width>
<height>600</height>
</rect>
</property>
<property name="windowTitle">
<string>Widget</string>
</property>
<layout class="QVBoxLayout" name="verticalLayout">
<property name="leftMargin">
<number>0</number>
</property>
<property name="topMargin">
<number>0</number>
</property>
<property name="rightMargin">
<number>0</number>
</property>
<property name="bottomMargin">
<number>0</number>
</property>
<item>
<layout class="QHBoxLayout" name="horizontalLayout">
<item>
<widget class="QComboBox" name="com_url">
<property name="editable">
<bool>true</bool>
</property>
<property name="insertPolicy">
<enum>QComboBox::InsertAtBottom</enum>
</property>
<property name="sizeAdjustPolicy">
<enum>QComboBox::AdjustToMinimumContentsLengthWithIcon</enum>
</property>
<property name="duplicatesEnabled">
<bool>false</bool>
</property>
<item>
<property name="text">
<string>rtsp://wowzaec2demo.streamlock.net/vod/mp4:BigBuckBunny_115k.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>rtmp://ns8.indexforce.com/home/mystream</string>
</property>
</item>
<item>
<property name="text">
<string>rtmp://58.200.131.2:1935/livetv/cctv1</string>
</property>
</item>
<item>
<property name="text">
<string>http://vfx.mtime.cn/Video/2019/02/04/mp4/190204084208765161.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>http://playertest.longtailvideo.com/adaptive/bipbop/gear4/prog_index.m3u8</string>
</property>
</item>
<item>
<property name="text">
<string>http://clips.vorwaerts-gmbh.de/big_buck_bunny.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>http://vjs.zencdn.net/v/oceans.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>https://media.w3.org/2010/05/sintel/trailer.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>https://sf1-hscdn-tos.pstatp.com/obj/media-fe/xgplayer_doc_video/flv/xgplayer-demo-360p.flv</string>
</property>
</item>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_file">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>选择</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_open">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>开始播放</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_pause">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>暂停</string>
</property>
</widget>
</item>
</layout>
</item>
</layout>
</widget>
<resources/>
<connections/>
</ui>
#---------------------------------------------------------------------------------------
# @功能: ffmpeg读取视频图像模块
# @编译器: Desktop Qt 5.12.5 MSVC2017 64bit(也支持其它编译器)
# @Qt IDE: D:/Qt/Qt5.12.5/Tools/QtCreator/share/qtcreator
#
# @开发者 mhf
# @邮箱 1603291350@qq.com
# @时间 2022-09-15 14:27:25
# @备注
#---------------------------------------------------------------------------------------
# 加载库,ffmpeg n5.1.2版本
win32{
LIBS += -LE:/lib/ffmpeg5-1-2/lib/ -lavcodec -lavfilter -lavformat -lswscale -lavutil -lswresample -lavdevice
INCLUDEPATH += E:/lib/ffmpeg5-1-2/include
DEPENDPATH += E:/lib/ffmpeg5-1-2/include
}
unix:!macx{
LIBS += -L/home/mhf/lib/ffmpeg/ffmpeg-5-1-2/lib -lavcodec -lavfilter -lavformat -lswscale -lavutil -lswresample -lavdevice
INCLUDEPATH += /home/mhf/lib/ffmpeg/ffmpeg-5-1-2/include
DEPENDPATH += /home/mhf/lib/ffmpeg/ffmpeg-5-1-2/include
}
HEADERS += \
$$PWD/readthread.h \
$$PWD/videodecode.h
SOURCES += \
$$PWD/readthread.cpp \
$$PWD/videodecode.cpp
#include "readthread.h"
#include "videodecode.h"
#include<QVariant>
#include <QEventLoop>
#include <QTimer>
#include <QDebug>
#include <qimage.h>
#include <playimage.h>
extern "C" { // 用C规则编译指定的代码
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/avutil.h"
#include "libswscale/swscale.h"
#include "libavutil/imgutils.h"
}
ReadThread::ReadThread(QObject *parent) : QThread(parent)
{
m_videoDecode = new VideoDecode();
// 注册自定义枚举类型,否则信号槽无法发送
qRegisterMetaType<PlayState>("PlayState");
}
ReadThread::~ReadThread()
{
if(m_videoDecode)
{
delete m_videoDecode;
}
}
/**
* @brief 传入播放的视频地址并开启线程
* @param url
*/
void ReadThread::open(const QString &url)
{
if(!this->isRunning())
{
m_url = url;
emit this->start();
}
}
/**
* @brief 控制暂停、继续
* @param flag true:暂停 fasle:继续
*/
void ReadThread::pause(bool flag)
{
m_pause = flag;
}
/**
* @brief 关闭播放
*/
void ReadThread::close()
{
m_play = false;
m_pause = false;
}
/**
* @brief 返回当前播放的地址
* @return
*/
const QString &ReadThread::url()
{
return m_url;
}
/**
* @brief 非阻塞延时
* @param msec 延时毫秒
*/
void sleepMsec(int msec)
{
if(msec <= 0) return;
QEventLoop loop; //定义一个新的事件循环
QTimer::singleShot(msec, &loop, SLOT(quit()));//创建单次定时器,槽函数为事件循环的退出函数
loop.exec(); //事件循环开始执行,程序会卡在这里,直到定时时间到,本循环被退出
}
void ReadThread::run()
{
bool ret = m_videoDecode->open(m_url); // 打开网络流时会比较慢,如果放到Ui线程会卡
if(ret)
{
m_play = true;
m_etime1.start();
m_etime2.start();
emit playState(play);
}
else
{
qWarning() << "打开失败!";
}
// 循环读取视频图像
while (m_play)
{
// 暂停
while (m_pause)
{
sleepMsec(200);
}
AVFrame* frame = m_videoDecode->read(); // 读取视频图像
if(frame)
{
// 1倍速播放
#if 0
sleepMsec(int(m_decodeVideo->pts() - m_etime1.elapsed())); // 不支持后退
#else
sleepMsec(int(m_videoDecode->pts() - m_etime2.elapsed())); // 支持后退
#endif
emit repaint(frame);
}
else
{
// 当前读取到无效图像时判断是否读取完成
if(m_videoDecode->isEnd())
{
break;
}
sleepMsec(1); // 这里不能使用QThread::msleep()延时,否则会很不稳定
}
}
qDebug() << "播放结束!";
m_videoDecode->close();
emit playState(end);
}
/******************************************************************************
* @文件名 readthread.h
* @功能 读取视频图像数据线程,在线程中解码视频
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/09/15
* @备注
*****************************************************************************/
#ifndef READTHREAD_H
#define READTHREAD_H
#include <QElapsedTimer>
#include <QThread>
#include <QTime>
class VideoDecode;
class PlayImage;
struct AVFrame;
class ReadThread : public QThread
{
Q_OBJECT
public:
enum PlayState // 视频播放状态
{
play,
end
};
public:
explicit ReadThread(QObject *parent = nullptr);
~ReadThread() override;
void open(const QString& url = QString()); // 打开视频
void pause(bool flag); // 暂停视频
void close(); // 关闭视频
const QString& url(); // 获取打开的视频地址
protected:
void run() override;
signals:
void repaint(AVFrame* frame); // 重绘
void playState(PlayState state); // 视频播放状态发送改变时触发
private:
VideoDecode* m_videoDecode = nullptr; // 视频解码类
QString m_url; // 打开的视频地址
bool m_play = false; // 播放控制
bool m_pause = false; // 暂停控制
QElapsedTimer m_etime1; // 控制视频播放速度(更精确,但不支持视频后退)
QTime m_etime2; // 控制视频播放速度(支持视频后退)
};
#endif // READTHREAD_H
#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];
}
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;
}
}
/**
* @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线程解码
// 初始化解码器上下文,如果之前avcodec_alloc_context3传入了解码器,这里设置NULL就可以
ret = avcodec_open2(m_codecContext, nullptr, nullptr);
if(ret < 0)
{
showError(ret);
free();
return false;
}
// 分配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;
}
// 分配图像空间
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);
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;
}
m_pts = m_frame->pts;
return m_frame;
}
/**
* @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 显示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(m_packet)
{
av_packet_free(&m_packet);
}
if(m_frame)
{
av_frame_free(&m_frame);
}
if(m_buffer)
{
delete [] m_buffer;
m_buffer = nullptr;
}
}
/******************************************************************************
* @文件名 videodecode.h
* @功能 视频解码类,在这个类中调用ffmpeg打开视频进行解码
*
* @开发者 mhf
* @邮箱 1603291350@qq.com
* @时间 2022/09/15
* @备注
*****************************************************************************/
#ifndef VIDEODECODE_H
#define VIDEODECODE_H
#include <QString>
#include <QSize>
struct AVFormatContext;
struct AVCodecContext;
struct AVRational;
struct AVPacket;
struct AVFrame;
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(); // 获取当前帧显示时间
private:
void initFFmpeg(); // 初始化ffmpeg库(整个程序中只需加载一次)
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; // 解码后的视频帧
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; // YUV图像需要转换位RGBA图像,这里保存转换后的图形数据
};
#endif // VIDEODECODE_H
#---------------------------------------------------------------------------------------
# @功能: 使用ffmpeg音视频库【软解码】实现的视频播放器;
# @编译器: Desktop Qt 5.12.5 MSVC2017 64bit 32bit(也支持其它编译器)
# @Qt IDE D:/Qt/Qt5.12.5/Tools/QtCreator/share/qtcreator
#
# @开发者 mhf
# @邮箱 1603291350@qq.com
# @时间 2022-09-14 14:13:52
# @备注 1、使用ffmpeg音视频库【软解码】实现的视频播放器;
# 2、支持打开本地视频文件(如mp4movavi等)、网络视频流(rtsprtmphttp等);
# 3、支持视频匀速播放;
# 4、采用【OpenGL显示YUV】图像,支持自适应窗口缩放,支持使用QOpenGLWidgetQOpenGLWindow显示;
# 5、将YUVRGB的步骤由CPU转换改为使用GPU转换,降低CPU占用率;
# 6、视频播放支持实时开始/关闭、暂停/继续播放;
# 7、视频解码、线程控制、显示各部分功能分离,低耦合度。
# 8、采用最新的5.1.2版本ffmpeg库进行开发,超详细注释信息,将所有踩过的坑、解决办法、注意事项都得很写清楚。
#---------------------------------------------------------------------------------------
QT += core gui
greaterThan(QT_MAJOR_VERSION, 4): QT += widgets
CONFIG += c++11
DEFINES += QT_DEPRECATED_WARNINGS
SOURCES += \
playimage.cpp \
main.cpp \
widget.cpp
HEADERS += \
playimage.h \
widget.h
FORMS += \
widget.ui
# Default rules for deployment.
qnx: target.path = /tmp/$${TARGET}/bin
else: unix:!android: target.path = /opt/$${TARGET}/bin
!isEmpty(target.path): INSTALLS += target
# ffmpeg读取视频图像模块
include(./VideoPlay/VideoPlay.pri)
INCLUDEPATH += ./VideoPlay
# 定义程序版本号
VERSION = 1.0.0
DEFINES += APP_VERSION=\\\"$$VERSION\\\"
TARGET = VideoPlayGL2
contains(QT_ARCH, i386){ # 使用32位编译器
DESTDIR = $$PWD/../bin # 程序输出路径
}else{
DESTDIR = $$PWD/../bin64 # 使用64位编译器
}
# msvc 编译器使用utf-8编码
msvc {
QMAKE_CFLAGS += /utf-8
QMAKE_CXXFLAGS += /utf-8
}
DISTFILES += \
fragment.fsh \
vertex.vsh
RESOURCES += \
rc.qrc
#version 330 core
in vec2 TexCord; // 纹理坐标
uniform sampler2D texture; // 纹理采样器
uniform sampler2D tex_y;
uniform sampler2D tex_u;
uniform sampler2D tex_v;
void main()
{
vec3 yuv;
vec3 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;
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);
}
#include "widget.h"
#include <QApplication>
int main(int argc, char *argv[])
{
QApplication a(argc, argv);
Widget w;
w.show();
return a.exec();
}
#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内容做准备。
// 释放纹理
if(m_texY)
{
m_texY->destroy();
delete m_texY;
}
if(m_texU)
{
m_texU->destroy();
delete m_texU;
}
if(m_texV)
{
m_texV->destroy();
delete m_texV;
}
this->doneCurrent(); // 释放上下文
// 释放
glDeleteBuffers(1, &VBO);
glDeleteBuffers(1, &EBO);
glDeleteVertexArrays(1, &VAO);
}
void PlayImage::repaint(AVFrame *frame)
{
if(!frame) return;
// 当切换显示的视频后,如果分辨率不同则需要重新创建纹理,否则会崩溃
if(frame->width != m_size.width() || frame->height != m_size.height())
{
if(m_texY && m_texU && m_texV)
{
m_texY->destroy();
m_texU->destroy();
m_texV->destroy();
delete m_texY;
delete m_texU;
delete m_texV;
m_texY = nullptr;
m_texU = nullptr;
m_texV = nullptr;
}
}
if(!m_texY) // 初始化纹理
{
// 创建2D纹理
m_texY = new QOpenGLTexture(QOpenGLTexture::Target2D);
m_texU = new QOpenGLTexture(QOpenGLTexture::Target2D);
m_texV = new QOpenGLTexture(QOpenGLTexture::Target2D);
// 设置纹理大小
m_texY->setSize(frame->width, frame->height);
m_texU->setSize(frame->width / 2, frame->height / 2);
m_texV->setSize(frame->width / 2, frame->height / 2);
// 设置放大、缩小过滤器
m_texY->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
m_texU->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
m_texV->setMinMagFilters(QOpenGLTexture::LinearMipMapLinear,QOpenGLTexture::Linear);
// 设置图像格式
m_texY->setFormat(QOpenGLTexture::R8_UNorm);
m_texU->setFormat(QOpenGLTexture::R8_UNorm);
m_texV->setFormat(QOpenGLTexture::R8_UNorm);
// 分配内存
m_texY->allocateStorage();
m_texU->allocateStorage();
m_texV->allocateStorage();
// 记录图像分辨率
m_size.setWidth(frame->width);
m_size.setHeight(frame->height);
resizeGL(this->width(), this->height());
}
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
av_frame_unref(frame); // 取消引用帧引用的所有缓冲区并重置帧字段。
this->update();
}
// 三个顶点坐标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);
// 返回属性名称在此着色器程序的参数列表中的位置。如果名称不是此着色器程序的有效属性,则返回-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(); // 绑定着色器
// 绑定纹理
if(m_texY && m_texU && m_texV)
{
m_texY->bind(0);
m_texU->bind(1);
m_texV->bind(2);
}
glBindVertexArray(VAO); // 绑定VAO
glDrawElements(GL_TRIANGLES, // 绘制的图元类型
6, // 指定要渲染的元素数(点数)
GL_UNSIGNED_INT, // 指定索引中值的类型(indices)
nullptr); // 指定当前绑定到GL_ELEMENT_array_buffer目标的缓冲区的数据存储中数组中第一个索引的偏移量。
glBindVertexArray(0);
if(m_texY && m_texU && m_texV)
{
m_texY->release();
m_texU->release();
m_texV->release();
}
m_program->release();
}
/******************************************************************************
* @文件名 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:
QOpenGLShaderProgram* m_program = nullptr;
QOpenGLTexture* m_texY = nullptr;
QOpenGLTexture* m_texU = nullptr;
QOpenGLTexture* m_texV = 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;
};
#endif // PLAYIMAGE_H
<RCC>
<qresource prefix="/">
<file>fragment.fsh</file>
<file>vertex.vsh</file>
</qresource>
</RCC>
#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;
}
#include "widget.h"
#include "ui_widget.h"
#include <QFileDialog>
extern "C" { // 用C规则编译指定的代码
#include "libavcodec/avcodec.h"
}
Q_DECLARE_METATYPE(AVFrame) //注册结构体,否则无法通过信号传递AVFrame
Widget::Widget(QWidget *parent)
: QWidget(parent)
, ui(new Ui::Widget)
{
ui->setupUi(this);
this->setWindowTitle(QString("Qt+ffmpeg视频播放(软解码 + OpenGL显示YUV)Demo V%1").arg(APP_VERSION));
// 使用QOpenGLWindow绘制
playImage = new PlayImage;
#if USE_WINDOW
ui->verticalLayout->addWidget(QWidget::createWindowContainer(playImage)); // 这一步加载速度要比OpenGLWidget慢一点
#else
ui->verticalLayout->addWidget(playImage);
#endif
m_readThread = new ReadThread();
connect(m_readThread, &ReadThread::repaint, playImage, &PlayImage::repaint, Qt::BlockingQueuedConnection);
connect(m_readThread, &ReadThread::playState, this, &Widget::on_playState);
}
Widget::~Widget()
{
// 释放视频读取线程
if(m_readThread)
{
m_readThread->close();
m_readThread->wait();
delete m_readThread;
}
delete ui;
}
/**
* @brief 获取本地视频路径
*/
void Widget::on_but_file_clicked()
{
QString strName = QFileDialog::getOpenFileName(this, "选择播放视频~!", "/", "视频 (*.mp4 *.m4v *.mov *.avi *.flv);; 其它(*)");
if(strName.isEmpty())
{
return;
}
ui->com_url->setCurrentText(strName);
}
/**
* @brief 视频播放/停止
*/
void Widget::on_but_open_clicked()
{
if(ui->but_open->text() == "开始播放")
{
m_readThread->open(ui->com_url->currentText());
}
else
{
m_readThread->close();
}
}
/**
* @brief 视频暂停/继续
*/
void Widget::on_but_pause_clicked()
{
if(ui->but_pause->text() == "暂停")
{
m_readThread->pause(true);
ui->but_pause->setText("继续");
}
else
{
m_readThread->pause(false);
ui->but_pause->setText("暂停");
}
}
/**
* @brief 根据视频播放状态切换界面设置
* @param state
*/
void Widget::on_playState(ReadThread::PlayState state)
{
if(state == ReadThread::play)
{
this->setWindowTitle(QString("正在播放:%1").arg(m_readThread->url()));
ui->but_open->setText("停止播放");
}
else
{
ui->but_open->setText("开始播放");
ui->but_pause->setText("暂停");
this->setWindowTitle(QString("Qt+ffmpeg视频播放(软解码 + OpenGL显示YUV)Demo V%1").arg(APP_VERSION));
}
}
#ifndef WIDGET_H
#define WIDGET_H
#include <QWidget>
#include "readthread.h"
#include "playimage.h"
QT_BEGIN_NAMESPACE
namespace Ui { class Widget; }
QT_END_NAMESPACE
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = nullptr);
~Widget();
private slots:
void on_but_file_clicked();
void on_but_open_clicked();
void on_but_pause_clicked();
void on_playState(ReadThread::PlayState state);
private:
Ui::Widget *ui;
PlayImage* playImage = nullptr;
ReadThread* m_readThread = nullptr;
};
#endif // WIDGET_H
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>Widget</class>
<widget class="QWidget" name="Widget">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>800</width>
<height>600</height>
</rect>
</property>
<property name="windowTitle">
<string>Widget</string>
</property>
<layout class="QVBoxLayout" name="verticalLayout">
<property name="leftMargin">
<number>0</number>
</property>
<property name="topMargin">
<number>0</number>
</property>
<property name="rightMargin">
<number>0</number>
</property>
<property name="bottomMargin">
<number>0</number>
</property>
<item>
<layout class="QHBoxLayout" name="horizontalLayout">
<item>
<widget class="QComboBox" name="com_url">
<property name="editable">
<bool>true</bool>
</property>
<property name="insertPolicy">
<enum>QComboBox::InsertAtBottom</enum>
</property>
<property name="sizeAdjustPolicy">
<enum>QComboBox::AdjustToMinimumContentsLengthWithIcon</enum>
</property>
<property name="duplicatesEnabled">
<bool>false</bool>
</property>
<item>
<property name="text">
<string>rtsp://wowzaec2demo.streamlock.net/vod/mp4:BigBuckBunny_115k.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>rtmp://ns8.indexforce.com/home/mystream</string>
</property>
</item>
<item>
<property name="text">
<string>rtmp://58.200.131.2:1935/livetv/cctv1</string>
</property>
</item>
<item>
<property name="text">
<string>http://vfx.mtime.cn/Video/2019/02/04/mp4/190204084208765161.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>http://playertest.longtailvideo.com/adaptive/bipbop/gear4/prog_index.m3u8</string>
</property>
</item>
<item>
<property name="text">
<string>http://clips.vorwaerts-gmbh.de/big_buck_bunny.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>http://vjs.zencdn.net/v/oceans.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>https://media.w3.org/2010/05/sintel/trailer.mp4</string>
</property>
</item>
<item>
<property name="text">
<string>https://sf1-hscdn-tos.pstatp.com/obj/media-fe/xgplayer_doc_video/flv/xgplayer-demo-360p.flv</string>
</property>
</item>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_file">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>选择</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_open">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>开始播放</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_pause">
<property name="sizePolicy">
<sizepolicy hsizetype="Fixed" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>暂停</string>
</property>
</widget>
</item>
</layout>
</item>
</layout>
</widget>
<resources/>
<connections/>
</ui>
......@@ -116,6 +116,6 @@ void Widget::on_playState(ReadThread::PlayState state)
{
ui->but_open->setText("开始播放");
ui->but_pause->setText("暂停");
this->setWindowTitle(QString("Qt+ffmpeg视频播放Demo V%1").arg(APP_VERSION));
this->setWindowTitle(QString("Qt+ffmpeg视频播放(硬解码)Demo V%1").arg(APP_VERSION));
}
}
......@@ -9,9 +9,11 @@
| 工程 | 功能 |
| -------------------- | ------------------------------------------------------------ |
| TableModel | 展示了使用QSqlTableModel来显示数据库信息 |
| TableModel2 | 在TableModel的基础上实现创建空白数据行、自增key |
| RelationalTableModel | 展示了使用QSqlRelationalTableModel来使用数据库[外键] |
| SqlWidgetMapper | 将数据库内容通过QDataWidgetMapper将模型数据映射到小部件 |
| CachedTable | 通过QSqlTableModel显示和修改数据,通过按键保存修改内容和恢复修改内容 |
| SignIn | 使用QSqlite数据库实现用户登录、后台管理用户功能 |
......@@ -21,9 +23,21 @@
![image-20220518112255014](QSql.assets/image-20220518112255014.png)
### 1.2 TableModel2
> 这个Demo中展示了如何使用具有表视图的专用 SQL 表模型(QSqlTableModel)来编辑数据库中的信息
> 1、通过按键新建 空白数据行;
> 2、使用自增Key;
> 3、通过按键更新数据;
> 4、判断表是否存在,不存在则创建。
![TableModel2-tuya](QSql.assets/TableModel2-tuya.gif)
### 1.2 RelationalTableModel
![RelationalTableModel](C:/Users/mhf/Desktop/RelationalTableModel.gif)
![RelationalTableModel](QSql.assets/RelationalTableModel.gif)
### 1.3 SqlWidgetMapper
......@@ -31,4 +45,22 @@
### 1.4 CachedTable
![CachedTable](C:/Users/mhf/Desktop/CachedTable.gif)
![CachedTable](QSql.assets/CachedTable.gif)
### 1.5 SignIn
> 使用QSqlite数据库实现用户登录、后台管理用户功能
> 1、通过按键新建 空白数据行;
> 2、使用自增Key;
> 3、通过按键更新数据;
> 4、判断表是否存在,不存在则创建。
> 5、用户登录功能,默认创建超级管理员账号root
> 6、支持用户后台管理,通过后台创建、修改、删除用户
> 7、用户分为超级管理员、普通管理员、普通用户三个等级;
> 8、超级管理员有所有权限,可创建、修改、删除普通管理员、普通用户;
> 9、普通管理员可创建、修改、删除普通用户,可新建、修改、查询数据;
> 10、普通用户不可修改用户信息,不可新建、修改数据库信息,只可查询。
![SignIn-tuya](QSql.assets/SignIn-tuya.gif)
......@@ -5,4 +5,5 @@ SUBDIRS += \
RelationalTableModel \ # 本示例展示了如何使用QSqlRelationalTableModel(关系表模型)来可视化数据库中[外键]的使用。
SqlWidgetMapper \ # 通过QDataWidgetMapper将数据库数据映射到小部件
TableModel \ # 这个Demo中展示了如何使用具有表视图的专用 SQL 表模型(QSqlTableModel)来编辑数据库中的信息。
TableModel2 # 这个Demo中展示了如何使用具有表视图的专用 SQL 表模型(QSqlTableModel)来编辑数据库中的信息,实现创建空白数据行、自增key
TableModel2\ # 这个Demo中展示了如何使用具有表视图的专用 SQL 表模型(QSqlTableModel)来编辑数据库中的信息,实现创建空白数据行、自增key
SignIn # 使用QSqlite数据库实现用户登录、后台管理用户功能
#---------------------------------------------------------------------------------------
# @功能: 使用QSqlite数据库实现用户登录、后台管理用户功能
# 1、通过按键新建 空白数据行;
# 2、使用自增Key
# 3、通过按键更新数据;
# 4、判断表是否存在,不存在则创建。
# 5、用户登录功能,默认创建超级管理员账号root
# 6、支持用户后台管理,通过后台创建、修改、删除用户
# 7、用户分为超级管理员、普通管理员、普通用户三个等级;
# 8、超级管理员有所有权限,可创建、修改、删除普通管理员、普通用户;
# 9、普通管理员可创建、修改、删除普通用户,可新建、修改、查询数据;
# 10、普通用户不可修改用户信息,不可新建、修改数据库信息,只可查询。
# @编译器: Desktop Qt 5.12.5 MSVC2017 64bit(也支持其它编译器)
# @Qt IDE D:/Qt/Qt5.12.5/Tools/QtCreator/share/qtcreator
#
# @开发者 mhf
# @邮箱 1603291350@qq.com
# @时间 2022-10-07 11:29:40
# @备注
#---------------------------------------------------------------------------------------
QT += core gui sql
greaterThan(QT_MAJOR_VERSION, 4): QT += widgets
CONFIG += c++11
DEFINES += QT_DEPRECATED_WARNINGS
SOURCES += \
dialog.cpp \
main.cpp \
userbackstage.cpp \
widget.cpp
HEADERS += \
dialog.h \
userbackstage.h \
widget.h
FORMS += \
dialog.ui \
userbackstage.ui \
widget.ui
# Default rules for deployment.
qnx: target.path = /tmp/$${TARGET}/bin
else: unix:!android: target.path = /opt/$${TARGET}/bin
!isEmpty(target.path): INSTALLS += target
# 定义程序版本号
VERSION = 1.0.0
DEFINES += APP_VERSION=\\\"$$VERSION\\\"
contains(QT_ARCH, i386){ # 使用32位编译器
DESTDIR = $$PWD/../bin # 程序输出路径
}else{
DESTDIR = $$PWD/../bin64 # 使用64位编译器
}
# msvc 编译器使用utf-8编码
msvc {
QMAKE_CFLAGS += /utf-8
QMAKE_CXXFLAGS += /utf-8
}
#include "dialog.h"
#include "ui_dialog.h"
#include <QSqlQuery>
#include <qmessagebox.h>
#include <QDebug>
bool Dialog::m_loggedIn = false; // 登录状态
QString Dialog::m_userName; // 登录的用户名
UserBackstage::UserType Dialog::m_userType; // 用户类型
Dialog::Dialog(QWidget *parent) :
QDialog(parent),
ui(new Ui::Dialog)
{
ui->setupUi(this);
this->setWindowTitle("登录");
m_db = QSqlDatabase::addDatabase("QSQLITE"); // 使用数据库驱动(Qsqlite)和默认连接名称(qt_sql_default_connection)添加一个数据库
m_db.setDatabaseName("SignIn.db"); // 使用文件数据库(可生成数据库文件,数据一直有效)
if(!m_db.open()) // 打开数据库
{
QMessageBox::critical(nullptr, "Error", "打开数据库失败!");
return ;
}
// 创建一个用于保存用户信息的表
if(!isTableExists("User"))
{
QSqlQuery query;
// 创建一个表person,包含id、firstname、lastname三个字段
bool ret = query.exec("create table User ("
"id integer primary key," // 索引(自增key)
"userName varchar(20)," // 用户名
"password varchar(20)," // 密码
"type int)"); // 用户类型
if(!ret)
{
qDebug() << "创建表失败:";
}
else
{
QSqlQuery query;
query.prepare("insert into User(userName, password, type)"
"values (:userName, :password, :type)");
query.bindValue(":userName", "root");
query.bindValue(":password", "123456");
query.bindValue(":type", int(UserBackstage::Root));
query.exec();
}
}
}
Dialog::~Dialog()
{
if(m_db.isOpen())
{
m_db.close(); // 关闭数据库
}
delete ui;
}
/**
* @brief 返回登录状态
* @return true登录 false未登录
*/
bool Dialog::loggedIn()
{
return m_loggedIn;
}
/**
* @brief 返回登录的用户名
* @return 如果没有则为空
*/
QString Dialog::userName()
{
return m_userName;
}
/**
* @brief 返回登录的用户类型
* @return
*/
UserBackstage::UserType Dialog::type()
{
return m_userType;
}
/**
* @brief 判断表是否存在
* @param table
* @return
*/
bool Dialog::isTableExists(const QString &table)
{
QSqlQuery query;
QString sql = QString("select * from sqlite_master where name = '%1';").arg(table); // 查询sqlite_master表中是否存在表名
if(query.exec(sql))
{
return query.next();
}
return false;
}
/**
* @brief 登录
*/
void Dialog::on_but_signIn_clicked()
{
QString userName = ui->line_user->text().trimmed();
QString password = ui->line_password->text().trimmed();
if(userName.isEmpty())
{
QMessageBox::about(this, "注意!", "用户名不能为空");
return;
}
if(password.isEmpty())
{
QMessageBox::about(this, "注意!", "用户密码不能为空");
return;
}
// 从数据库中查询用户和密码,完成登录功能
QSqlQuery query;
QString sql = QString("select * from User where userName = '%1';").arg(userName); // 查询用户
if(query.exec(sql))
{
if(query.next()) // true则用户存在
{
if(password == query.value("password").toString()) // 密码相等
{
m_userName = userName;
m_userType = UserBackstage::UserType(query.value("type").toInt());
m_loggedIn = true;
this->close();
}
else
{
QMessageBox::about(this, "注意", "输入密码错误!");
}
}
else
{
QMessageBox::about(this, "注意", "用户不存在!");
}
}
else
{
QMessageBox::about(this, "注意", "sql指令执行失败!");
}
}
#ifndef DIALOG_H
#define DIALOG_H
#include <QDialog>
#include <qsqldatabase.h>
#include "userbackstage.h"
namespace Ui {
class Dialog;
}
class Dialog : public QDialog
{
Q_OBJECT
public:
explicit Dialog(QWidget *parent = nullptr);
~Dialog();
static bool loggedIn();
static QString userName();
static UserBackstage::UserType type();
private slots:
bool isTableExists(const QString& table);
void on_but_signIn_clicked();
private:
Ui::Dialog *ui;
QSqlDatabase m_db;
static bool m_loggedIn; // 登录状态
static QString m_userName; // 登录的用户名
static UserBackstage::UserType m_userType; // 用户类型
};
#endif // DIALOG_H
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>Dialog</class>
<widget class="QDialog" name="Dialog">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>400</width>
<height>300</height>
</rect>
</property>
<property name="windowTitle">
<string>Dialog</string>
</property>
<layout class="QGridLayout" name="gridLayout_3">
<item row="0" column="1">
<spacer name="verticalSpacer">
<property name="orientation">
<enum>Qt::Vertical</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
<width>20</width>
<height>74</height>
</size>
</property>
</spacer>
</item>
<item row="1" column="0">
<spacer name="horizontalSpacer">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
<width>50</width>
<height>20</height>
</size>
</property>
</spacer>
</item>
<item row="1" column="1">
<layout class="QGridLayout" name="gridLayout_2">
<property name="verticalSpacing">
<number>20</number>
</property>
<item row="0" column="0" colspan="2">
<layout class="QGridLayout" name="gridLayout">
<item row="1" column="1">
<widget class="QLineEdit" name="line_password">
<property name="minimumSize">
<size>
<width>200</width>
<height>30</height>
</size>
</property>
<property name="text">
<string/>
</property>
<property name="echoMode">
<enum>QLineEdit::Password</enum>
</property>
<property name="dragEnabled">
<bool>false</bool>
</property>
</widget>
</item>
<item row="0" column="1">
<widget class="QLineEdit" name="line_user">
<property name="minimumSize">
<size>
<width>200</width>
<height>30</height>
</size>
</property>
</widget>
</item>
<item row="0" column="0">
<widget class="QLabel" name="label">
<property name="text">
<string>用户:</string>
</property>
</widget>
</item>
<item row="1" column="0">
<widget class="QLabel" name="label_2">
<property name="text">
<string>密码:</string>
</property>
</widget>
</item>
</layout>
</item>
<item row="1" column="0" colspan="2">
<widget class="QPushButton" name="but_signIn">
<property name="minimumSize">
<size>
<width>0</width>
<height>35</height>
</size>
</property>
<property name="text">
<string>登录</string>
</property>
</widget>
</item>
</layout>
</item>
<item row="1" column="2">
<spacer name="horizontalSpacer_2">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
<width>50</width>
<height>20</height>
</size>
</property>
</spacer>
</item>
<item row="2" column="1">
<spacer name="verticalSpacer_2">
<property name="orientation">
<enum>Qt::Vertical</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
<width>20</width>
<height>65</height>
</size>
</property>
</spacer>
</item>
</layout>
</widget>
<resources/>
<connections/>
</ui>
#include "widget.h"
#include "dialog.h"
#include <QApplication>
int main(int argc, char *argv[])
{
QApplication a(argc, argv);
// 打开登录界面
Dialog signIn;
signIn.exec();
if(!signIn.loggedIn()) // 登录失败直接退出
{
return 0;
}
// 登录成功打开主界面
Widget w;
w.show();
return a.exec();
}
#include "userbackstage.h"
#include "ui_userbackstage.h"
#include <QDebug>
#include <qmessagebox.h>
#include <qsqlquery.h>
UserBackstage::UserBackstage(QWidget *parent) :
QDialog(parent),
ui(new Ui::UserBackstage)
{
ui->setupUi(this);
m_model = new QSqlTableModel(this);
m_model->setTable("User"); // 设置需要显示的数据库表
ui->tableView->setEditTriggers(QAbstractItemView::NoEditTriggers);
m_model->setHeaderData(0, Qt::Horizontal, "ID");
m_model->setHeaderData(1, Qt::Horizontal, "用户名");
m_model->setHeaderData(2, Qt::Horizontal, "密码");
m_model->setHeaderData(3, Qt::Horizontal, "用户类型");
ui->tableView->setModel(m_model);
m_model->select(); // 获取数据库中的数据
ui->tableView->resizeColumnsToContents(); // 根据表格中的内容自动调整列宽
}
UserBackstage::~UserBackstage()
{
delete ui;
}
/**
* @brief 设置登录的用户类型
* @param type
*/
void UserBackstage::setUserType(UserBackstage::UserType type)
{
m_userType = type;
// 添加用户类型
ui->com_type->clear();
switch (type)
{
case User:
{
break;
}
case Admin:
{
ui->com_type->addItem("普通用户", User);
m_model->setFilter(QString("type = '%1'").arg(int(User))); // 只显示普通用户的信息
break;
}
case Root:
{
ui->com_type->addItem("普通用户", User);
ui->com_type->addItem("管理员", Admin);
break;
}
}
}
/**
* @brief 设置用户信息
*/
void UserBackstage::on_but_set_clicked()
{
QString userName = ui->line_user->text().trimmed();
QString password = ui->line_password->text().trimmed();
UserType userType = UserType(ui->com_type->currentIndex());
if(userName.isEmpty())
{
QMessageBox::about(this, "注意!", "用户名不能为空");
return;
}
if(password.isEmpty())
{
QMessageBox::about(this, "注意!", "用户密码不能为空");
return;
}
QSqlQuery query;
QString sql = QString("select * from User where userName = '%1';").arg(userName); // 查询用户
if(query.exec(sql))
{
if(query.next()) // true则用户存在则更新数据
{
int type = query.value("type").toInt();
if(type >= m_userType) // 如果修改的用户等级超过登录用户等级,则修改失败
{
return;
}
sql = QString("update User set password='%1', type=%2 where userName = '%3';")
.arg(password)
.arg(int(userType))
.arg(userName);
query.exec(sql);
}
else // 用户不存在则插入数据
{
query.prepare("insert into User(userName, password, type)"
"values (:userName, :password, :type)");
query.bindValue(":userName", userName);
query.bindValue(":password", password);
query.bindValue(":type", int(userType));
query.exec();
}
m_model->select(); // 获取数据库中的数据
}
else
{
qDebug() << "指令执行失败";
}
}
/**
* @brief 删除用户信息
*/
void UserBackstage::on_but_delete_clicked()
{
QString userName = ui->line_user->text().trimmed();
if(userName.isEmpty())
{
QMessageBox::about(this, "注意!", "用户名不能为空");
return;
}
QSqlQuery query;
QString sql = QString("delete from User where userName = '%1';").arg(userName); // 查询用户
if(query.exec(sql))
{
m_model->select(); // 获取数据库中的数据
}
else
{
qDebug() << "指令执行失败";
}
}
#ifndef USERBACKSTAGE_H
#define USERBACKSTAGE_H
#include <QDialog>
#include <qsqltablemodel.h>
namespace Ui {
class UserBackstage;
}
class UserBackstage : public QDialog
{
Q_OBJECT
public:
enum UserType {
User, // 普通用户
Admin, // 普通管理员
Root // 超级管理员(唯一)
};
public:
explicit UserBackstage(QWidget *parent = nullptr);
~UserBackstage();
void setUserType(UserType type);
private slots:
void on_but_set_clicked();
void on_but_delete_clicked();
private:
Ui::UserBackstage *ui;
QSqlTableModel* m_model = nullptr; // 创建一个 单个数据库表的可编辑数据模型
UserType m_userType;
};
#endif // USERBACKSTAGE_H
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>UserBackstage</class>
<widget class="QDialog" name="UserBackstage">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>582</width>
<height>404</height>
</rect>
</property>
<property name="windowTitle">
<string>Dialog</string>
</property>
<layout class="QGridLayout" name="gridLayout">
<item row="0" column="0" rowspan="2">
<widget class="QTableView" name="tableView"/>
</item>
<item row="0" column="1">
<layout class="QVBoxLayout" name="verticalLayout_3">
<property name="spacing">
<number>6</number>
</property>
<item>
<layout class="QVBoxLayout" name="verticalLayout_2">
<item>
<widget class="QLabel" name="label_4">
<property name="text">
<string>用户名:</string>
</property>
</widget>
</item>
<item>
<widget class="QLineEdit" name="line_user">
<property name="sizePolicy">
<sizepolicy hsizetype="Minimum" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
</widget>
</item>
<item>
<widget class="QLabel" name="label_5">
<property name="text">
<string>密码:</string>
</property>
</widget>
</item>
<item>
<widget class="QLineEdit" name="line_password">
<property name="sizePolicy">
<sizepolicy hsizetype="Minimum" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
</widget>
</item>
<item>
<widget class="QLabel" name="label_6">
<property name="text">
<string>类型:</string>
</property>
</widget>
</item>
<item>
<widget class="QComboBox" name="com_type">
<property name="sizePolicy">
<sizepolicy hsizetype="Minimum" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
</widget>
</item>
</layout>
</item>
<item>
<widget class="QPushButton" name="but_set">
<property name="sizePolicy">
<sizepolicy hsizetype="Minimum" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>设置用户</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="but_delete">
<property name="sizePolicy">
<sizepolicy hsizetype="Minimum" vsizetype="Fixed">
<horstretch>0</horstretch>
<verstretch>0</verstretch>
</sizepolicy>
</property>
<property name="text">
<string>删除用户</string>
</property>
</widget>
</item>
</layout>
</item>
<item row="1" column="1">
<spacer name="verticalSpacer">
<property name="orientation">
<enum>Qt::Vertical</enum>
</property>
<property name="sizeHint" stdset="0">
<size>
<width>20</width>
<height>189</height>
</size>
</property>
</spacer>
</item>
</layout>
</widget>
<resources/>
<connections/>
</ui>
#include "widget.h"
#include "ui_widget.h"
#include "dialog.h"
#include "userbackstage.h"
#include <QMessageBox>
#include <QSqlDatabase>
#include <QDebug>
Widget::Widget(QWidget *parent)
: QWidget(parent)
, ui(new Ui::Widget)
{
ui->setupUi(this);
QString user;
switch (Dialog::type())
{
case UserBackstage::User:
{
ui->but_user->setVisible(false); // 隐藏用户管理功能
ui->but_add->setVisible(false); // 隐藏数据新建功能
ui->tableView->setEditTriggers(QAbstractItemView::NoEditTriggers); // 普通用户禁用修改数据功能
user = QString("普通用户:%1").arg(Dialog::userName());
break;
}
case UserBackstage::Admin:
{
user = QString("普通管理员:%1").arg(Dialog::userName());
break;
}
case UserBackstage::Root:
{
user = QString("超级管理员:%1").arg(Dialog::userName());
break;
}
}
this->setWindowTitle(QString("QSql-用户登录Demo - V%1 %2").arg(APP_VERSION).arg(user));
}
Widget::~Widget()
{
delete ui;
}
void Widget::on_but_connect_clicked()
{
if(ui->but_connect->text() == "关闭数据库")
{
ui->but_connect->setText("连接数据库");
m_db.close();
}
else
{
m_db = QSqlDatabase::addDatabase("QSQLITE"); // 使用数据库驱动(Qsqlite)和默认连接名称(qt_sql_default_connection)添加一个数据库
qDebug() << QSqlDatabase::defaultConnection; // 打印默认数据库连接名称
m_db.setDatabaseName("SignIn.db"); // 使用文件数据库(可生成数据库文件,数据一直有效)
if(!m_db.open()) // 打开数据库
{
QMessageBox::critical(nullptr, "Error", "打开数据库失败!");
return ;
}
// 如果表不存在则创建表
if(!isTableExists("person"))
{
QSqlQuery query;
// 创建一个表person,包含id、firstname、lastname三个字段
bool ret = query.exec("create table person ("
"id integer primary key," // 索引(自增key),使用integer默认为自增, int不能设置主键自增
"firstname varchar(20)," // 名
"lastname varchar(20))"); // 姓
if(!ret)
{
qDebug() << "创建表失败:";
}
}
initModel();
ui->but_connect->setText("关闭数据库");
}
}
/**
* @brief 判断表是否存在
* @param table 表名称
* @return true存在 false不存在
*/
bool Widget::isTableExists(const QString &table)
{
QSqlQuery query;
QString sql = QString("select * from sqlite_master where name = '%1';").arg(table); // 查询sqlite_master表中是否存在表名
if(query.exec(sql))
{
return query.next();
}
return false;
}
/**
* @brief SQL 表模型(QSqlTableModel)来编辑数据库中的信息
*/
void Widget::initModel()
{
if(m_model)
{
m_model->clear();
delete m_model;
m_model = nullptr;
}
m_model = new QSqlTableModel(this, m_db);
m_model->setTable("person"); // 设置需要显示的数据库表
#if 1
m_model->setEditStrategy(QSqlTableModel::OnFieldChange); // 在界面上修改后数据立刻保存到数据库
#else
m_model->setEditStrategy(QSqlTableModel::OnManualSubmit); // 将将编辑数据库中值的策略设置为[在调用 submitAll() 或 revertAll() 之前,所有更改都将缓存在模型中(即在界面上修改数据后不会立刻存入数据库)]
#endif
m_model->setHeaderData(0, Qt::Horizontal, "ID");
m_model->setHeaderData(1, Qt::Horizontal, "名称");
m_model->setHeaderData(2, Qt::Horizontal, "姓氏");
ui->tableView->setModel(m_model);
}
/**
* @brief 添加空白数据行
*/
void Widget::on_but_add_clicked()
{
QSqlQuery query;
query.prepare("insert into person(firstname, lastname)" // 写入数据时不需写入id字段,实现自增
"values (:firstname, :lastname)");
query.bindValue(":firstname", "");
query.bindValue(":lastname", "");
query.exec();
m_model->select(); // 获取数据库中的数据
}
/**
* @brief 查询数据库更新界面
*/
void Widget::on_but_read_clicked()
{
if(!m_model) return;
m_model->select(); // 获取数据库中的数据
ui->tableView->resizeColumnsToContents(); // 根据表格中的内容自动调整列宽
}
void Widget::on_but_user_clicked()
{
UserBackstage user;
user.setUserType(Dialog::type());
user.exec();
}
#ifndef WIDGET_H
#define WIDGET_H
#include <QWidget>
#include <QSqlTableModel>
#include <qsqlquery.h>
QT_BEGIN_NAMESPACE
namespace Ui { class Widget; }
QT_END_NAMESPACE
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = nullptr);
~Widget();
private slots:
void initModel();
bool isTableExists(const QString& table);
void on_but_connect_clicked();
void on_but_add_clicked();
void on_but_read_clicked();
void on_but_user_clicked();
private:
Ui::Widget *ui;
QSqlTableModel* m_model = nullptr; // 创建一个 单个数据库表的可编辑数据模型
QSqlDatabase m_db;
};
#endif // WIDGET_H
<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
<class>Widget</class>
<widget class="QWidget" name="Widget">
<property name="geometry">
<rect>
<x>0</x>
<y>0</y>
<width>522</width>
<height>402</height>
</rect>
</property>
<property name="windowTitle">
<string>Widget</string>
</property>
<layout class="QGridLayout" name="gridLayout">
<item row="1" column="1">
<widget class="QPushButton" name="but_add">
<property name="text">
<string>新增数据行</string>
</property>
</widget>
</item>
<item row="1" column="0">
<widget class="QPushButton" name="but_connect">
<property name="text">
<string>连接数据库</string>
</property>
</widget>
</item>
<item row="1" column="2">
<widget class="QPushButton" name="but_read">
<property name="text">
<string>读取数据</string>
</property>
</widget>
</item>
<item row="1" column="3">
<widget class="QPushButton" name="but_user">
<property name="text">
<string>用户管理</string>
</property>
</widget>
</item>
<item row="0" column="0" colspan="4">
<widget class="QTableView" name="tableView"/>
</item>
</layout>
</widget>
<resources/>
<connections/>
</ui>
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册