前言
测试环境:
- ffmpeg的4.3.2自行编译版本
- windows环境
- qt5.12
完整代码:
H264DncodeThread.h
#ifndef H264DNCODETHREAD_H
#define H264DNCODETHREAD_H
#include <QObject>
#include <QThread>
extern "C" {
#include <libavutil/avutil.h>
}
typedef struct {
const char *filename;
int width;
int height;
AVPixelFormat pixFmt;
int fps;
} VideoDecodeSpec;
class H264DncodeThread : public QThread
{
Q_OBJECT
public:
explicit H264DncodeThread(QObject *parent = nullptr);
~H264DncodeThread();
static void h264Decode(const char *inFilename,
VideoDecodeSpec &out);
signals:
// QThread interface
protected:
virtual void run() override;
};
#endif // H264DNCODETHREAD_H
H264DncodeThread.cpp
#include "h264dncodethread.h"
#include <QDebug>
#include <QFile>
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>
#include <libavutil/imgutils.h>
}
#define ERROR_BUF(ret) \
char errbuf[1024]; \
av_strerror(ret, errbuf, sizeof (errbuf));
// 输入缓冲区的大小
#define IN_DATA_SIZE 4096
static int frameIdx = 0;
static int decode(AVCodecContext *ctx,
AVPacket *pkt,
AVFrame *frame,
QFile &outFile) {
// 发送压缩数据到解码器
int ret = avcodec_send_packet(ctx, pkt);
if (ret < 0) {
ERROR_BUF(ret);
qDebug() << "avcodec_send_packet error" << errbuf;
return ret;
}
while (true) {
// 获取解码后的数据
ret = avcodec_receive_frame(ctx, frame);
if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
return 0;
} else if (ret < 0) {
ERROR_BUF(ret);
qDebug() << "avcodec_receive_frame error" << errbuf;
return ret;
}
qDebug() << "解码出第" << ++frameIdx << "帧";
// 将解码后的数据写入文件
// 写入Y平面
outFile.write((char *) frame->data[0],
frame->linesize[0] * ctx->height);
// 写入U平面
outFile.write((char *) frame->data[1],
frame->linesize[1] * ctx->height >> 1);
// 写入V平面
outFile.write((char *) frame->data[2],
frame->linesize[2] * ctx->height >> 1);
// qDebug() << frame->data[0] << frame->data[1] << frame->data[2];
/*
* frame->data[0] 0xd08c400 0x8c400
* frame->data[1] 0xd0d79c0 0xd79c0
* frame->data[2] 0xd0ea780 0xea780
*
* frame->data[1] - frame->data[0] = 308672 = y平面的大小
* frame->data[2] - frame->data[1] = 77248 = u平面的大小
*
* y平面的大小 640x480*1 = 307200
* u平面的大小 640x480*0.25 = 76800
* v平面的大小 640x480*0.25
*/
// // 将解码后的数据写入文件(460800)
// int imgSize = av_image_get_buffer_size(ctx->pix_fmt, ctx->width, ctx->height, 1);
// // outFile.write((char *) frame->data[0], frame->linesize[0]);
// outFile.write((char *) frame->data[0], imgSize);
}
}
H264DncodeThread::H264DncodeThread(QObject *parent) : QThread(parent)
{
// 当监听到线程结束时(finished),就调用deleteLater回收内存
connect(this,&H264DncodeThread::finished,this,[=](){
this->deleteLater();
qDebug()<<"H264DncodeThread线程结束,线程指针被dlete";
});
}
H264DncodeThread::~H264DncodeThread()
{
// 断开所有的连接
disconnect();
//强制关闭窗口时,线程也能安全关闭
requestInterruption();
wait();
qDebug()<<"H264DncodeThread析构函数";
}
void H264DncodeThread::h264Decode(const char *inFilename, VideoDecodeSpec &out)
{
// 返回结果
int ret = 0;
// 用来存放读取的输入文件数据(h264)
// 加上AV_INPUT_BUFFER_PADDING_SIZE是为了防止某些优化过的reader一次性读取过多导致越界
char inDataArray[IN_DATA_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
char *inData = inDataArray;
// 每次从输入文件中读取的长度(h264)
// 输入缓冲区中,剩下的等待进行解码的有效数据长度
int inLen;
// 是否已经读取到了输入文件的尾部
int inEnd = 0;
// 文件
QFile inFile(inFilename);
QFile outFile(out.filename);
// 解码器
AVCodec *codec = nullptr;
// 上下文
AVCodecContext *ctx = nullptr;
// 解析器上下文
AVCodecParserContext *parserCtx = nullptr;
// 存放解码前的数据(h264)
AVPacket *pkt = nullptr;
// 存放解码后的数据(yuv)
AVFrame *frame = nullptr;
// 获取解码器
// codec = avcodec_find_decoder_by_name("h264");
codec = avcodec_find_decoder(AV_CODEC_ID_H264);
if (!codec) {
qDebug() << "decoder not found";
return;
}
// 初始化解析器上下文
parserCtx = av_parser_init(codec->id);
if (!parserCtx) {
qDebug() << "av_parser_init error";
return;
}
// 创建上下文
ctx = avcodec_alloc_context3(codec);
if (!ctx) {
qDebug() << "avcodec_alloc_context3 error";
goto end;
}
// 创建AVPacket
pkt = av_packet_alloc();
if (!pkt) {
qDebug() << "av_packet_alloc error";
goto end;
}
// 创建AVFrame
frame = av_frame_alloc();
if (!frame) {
qDebug() << "av_frame_alloc error";
goto end;
}
// 打开解码器
ret = avcodec_open2(ctx, codec, nullptr);
if (ret < 0) {
ERROR_BUF(ret);
qDebug() << "avcodec_open2 error" << errbuf;
goto end;
}
// 打开文件
if (!inFile.open(QFile::ReadOnly)) {
qDebug() << "file open error:" << inFilename;
goto end;
}
if (!outFile.open(QFile::WriteOnly)) {
qDebug() << "file open error:" << out.filename;
goto end;
}
// 读取文件数据
do {
inLen = inFile.read(inDataArray, IN_DATA_SIZE);
// 设置是否到了文件尾部
inEnd = !inLen;
// 让inData指向数组的首元素
inData = inDataArray;
// 只要输入缓冲区中还有等待进行解码的数据
while (inLen > 0 || inEnd) {
// 到了文件尾部(虽然没有读取任何数据,但也要调用av_parser_parse2,修复bug)
// 经过解析器解析
ret = av_parser_parse2(parserCtx, ctx,
&pkt->data, &pkt->size,
(uint8_t *) inData, inLen,
AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
if (ret < 0) {
ERROR_BUF(ret);
qDebug() << "av_parser_parse2 error" << errbuf;
goto end;
}
// 跳过已经解析过的数据
inData += ret;
// 减去已经解析过的数据大小
inLen -= ret;
qDebug() << inEnd << pkt->size << ret;
// 解码
if (pkt->size > 0 && decode(ctx, pkt, frame, outFile) < 0) {
goto end;
}
// 如果到了文件尾部
if (inEnd) break;
}
} while (!inEnd);
// 刷新缓冲区
// pkt->data = nullptr;
// pkt->size = 0;
// decode(ctx, pkt, frame, outFile);
decode(ctx, nullptr, frame, outFile);
// 赋值输出参数
out.width = ctx->width;
out.height = ctx->height;
out.pixFmt = ctx->pix_fmt;
// 用framerate.num获取帧率,并不是time_base.den
out.fps = ctx->framerate.num;
end:
inFile.close();
outFile.close();
av_packet_free(&pkt);
av_frame_free(&frame);
av_parser_close(parserCtx);
avcodec_free_context(&ctx);
// bug fix
// https://patchwork.ffmpeg.org/project/ffmpeg/patch/tencent_609A2E9F73AB634ED670392DD89A63400008@qq.com/
//
// while ((inLen = inFile.read(inDataArray, IN_DATA_SIZE)) > 0)
// while (inLen > 0) {
// // 经过解析器解析
// ret = av_parser_parse2(parserCtx, ctx,
// &pkt->data, &pkt->size,
// (uint8_t *) inData, inLen,
// AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
// if (ret < 0) {
// ERROR_BUF(ret);
// qDebug() << "av_parser_parse2 error" << errbuf;
// goto end;
// }
// // 跳过已经解析过的数据
// inData += ret;
// // 减去已经解析过的数据大小
// inLen -= ret;
// // 解码
// if (pkt->size > 0 && decode(ctx, pkt, frame, outFile) < 0) {
// goto end;
// }
// }
// }
}
void H264DncodeThread::run()
{
VideoDecodeSpec out;
out.filename = "E:/media/out-yuv420p-decode.yuv";
h264Decode("E:/media/out-yuv420p.h264", out);
qDebug() << out.width << out.height
<< out.fps << av_get_pix_fmt_name(out.pixFmt);
}
线程调用:
void MainWindow::on_pushButton_h264_decode_clicked()
{
m_pH264DncodeThread=new H264DncodeThread(this);
m_pH264DncodeThread->start();
}
注意:.h文件中提前声明了以下全局变量
H264DncodeThread *m_pH264DncodeThread=nullptr;
注意:本文为个人记录,新手照搬可能会出现各种问题,请谨慎使用
码字不易,如果这篇博客对你有帮助,麻烦点赞收藏,非常感谢!有不对的地方