| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784 |
- #include "av_decoder.h"
- #include <QDebug>
- #include <QThread>
- #include "threadpool.h"
- using AVTool::Decoder;
- Decoder::Decoder()
- : m_fmtCtx(nullptr)
- , m_maxFrameQueueSize(6)
- , m_maxPacketQueueSize(12)
- , m_audioIndex(-1)
- , m_videoIndex(-1)
- , m_exit(0)
- , m_duration(0)
- {
- if (!init())
- qDebug() << "Decoder init failed!\n";
- setInitVal();
- }
- Decoder::~Decoder()
- {
- exit();
- }
- AVTool::MediaInfo* Decoder::detectMediaInfo(const QString& url)
- {
- int ret = 0;
- int duration = 0;
- //解封装初始化
- AVFormatContext* fmtCtx = avformat_alloc_context();
- //用于获取流时长
- AVDictionary* formatOpts = nullptr;
- av_dict_set(&formatOpts, "probesize", "32", 0);
- av_dict_set(&formatOpts, "analyzeduration", "0", 0);
- av_dict_set(&formatOpts, "rtsp_transport", "tcp", 0);
- av_dict_set(&formatOpts, "fflags", "nobuffer", 0);
- av_dict_set(&formatOpts, "max_delay", "0", 0);
- ret = avformat_open_input(&fmtCtx, url.toUtf8().constData(), nullptr, &formatOpts);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avformat_open_input error:" << m_errBuf;
- av_dict_free(&formatOpts);
- //打开失败释放分配的AVFormatContext内存
- avformat_free_context(fmtCtx);
- return Q_NULLPTR;
- }
- // 低延迟:关闭内部缓冲
- if (fmtCtx) {
- fmtCtx->flags |= AVFMT_FLAG_NOBUFFER;
- fmtCtx->max_delay = 0;
- }
- ret = avformat_find_stream_info(fmtCtx, nullptr);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avformat_find_stream_info error:" << m_errBuf;
- av_dict_free(&formatOpts);
- return Q_NULLPTR;
- }
- //记录流时长
- AVRational q = {1, AV_TIME_BASE};
- duration = (uint32_t) (fmtCtx->duration * av_q2d(q));
- av_dict_free(&formatOpts);
- int videoIndex = av_find_best_stream(fmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, nullptr, 0);
- if (videoIndex < 0) {
- // 仅音频:返回时长和占位图
- AVTool::MediaInfo* info = new AVTool::MediaInfo;
- info->duration = duration;
- info->tipImg = QImage(300, 300, QImage::Format_RGB888);
- info->tipImg.fill(Qt::black);
- avformat_close_input(&fmtCtx);
- return info;
- }
- //视频解码初始化
- AVCodecParameters* videoCodecPar = fmtCtx->streams[videoIndex]->codecpar;
- if (!videoCodecPar) {
- qDebug() << "videocodecpar is nullptr!";
- return Q_NULLPTR;
- }
- AVCodecContext* codecCtx = avcodec_alloc_context3(nullptr);
- ret = avcodec_parameters_to_context(codecCtx, videoCodecPar);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
- return Q_NULLPTR;
- }
- const AVCodec* videoCodec = avcodec_find_decoder(codecCtx->codec_id);
- if (!videoCodec) {
- qDebug() << "avcodec_find_decoder failed!";
- return Q_NULLPTR;
- }
- codecCtx->codec_id = videoCodec->id;
- ret = avcodec_open2(codecCtx, videoCodec, nullptr);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_open2:" << m_errBuf;
- return Q_NULLPTR;
- }
- AVPacket* pkt = av_packet_alloc();
- AVFrame* frame = av_frame_alloc();
- bool flag = false;
- while (1) {
- ret = av_read_frame(fmtCtx, pkt);
- if (ret != 0) {
- return Q_NULLPTR;
- }
- if (pkt->stream_index == videoIndex) {
- ret = avcodec_send_packet(codecCtx, pkt);
- av_packet_unref(pkt);
- if (ret < 0 || ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avcodec_send_packet error:" << m_errBuf;
- continue;
- }
- while (1) {
- ret = avcodec_receive_frame(codecCtx, frame);
- if (ret == 0) {
- flag = true;
- break;
- } else if (ret == AVERROR(EAGAIN)) {
- break;
- } else {
- return Q_NULLPTR;
- }
- }
- if (flag)
- break;
- } else {
- av_packet_unref(pkt);
- }
- }
- int imageWidth = videoCodecPar->width;
- int imageHeight = videoCodecPar->height;
- enum AVPixelFormat dstPixFmt = AV_PIX_FMT_RGB24;
- int swsFlags = SWS_BICUBIC;
- uint8_t* pixels[4];
- int pitch[4];
- //分配存储转换后帧数据的buffer内存
- int bufSize = av_image_get_buffer_size(dstPixFmt, imageWidth, imageHeight, 1);
- uint8_t* buffer = (uint8_t*) av_malloc(bufSize * sizeof(uint8_t));
- av_image_fill_arrays(pixels, pitch, buffer, dstPixFmt, imageWidth, imageHeight, 1);
- SwsContext* swsCtx = sws_getCachedContext(nullptr,
- frame->width,
- frame->height,
- (enum AVPixelFormat) frame->format,
- imageWidth,
- imageHeight,
- dstPixFmt,
- swsFlags,
- nullptr,
- nullptr,
- nullptr);
- if (swsCtx)
- sws_scale(swsCtx, frame->data, frame->linesize, 0, frame->height, pixels, pitch);
- av_frame_unref(frame);
- AVTool::MediaInfo* info = new AVTool::MediaInfo;
- info->duration = duration;
- info->tipImg = QImage(pixels[0], imageWidth, imageHeight, pitch[0], QImage::Format_RGB888);
- av_packet_free(&pkt);
- av_frame_free(&frame);
- avformat_close_input(&fmtCtx);
- avcodec_free_context(&codecCtx);
- return info;
- }
- bool Decoder::init()
- {
- if (!ThreadPool::init()) {
- qDebug() << "threadpool init failed!\n";
- return false;
- }
- m_audioPacketQueue.pktVec.resize(m_maxPacketQueueSize);
- m_videoPacketQueue.pktVec.resize(m_maxPacketQueueSize);
- m_audioFrameQueue.frameVec.resize(m_maxFrameQueueSize);
- m_videoFrameQueue.frameVec.resize(m_maxFrameQueueSize);
- m_audioPktDecoder.codecCtx = nullptr;
- m_videoPktDecoder.codecCtx = nullptr;
- return true;
- }
- void Decoder::setInitVal()
- {
- m_audioPacketQueue.size = 0;
- m_audioPacketQueue.pushIndex = 0;
- m_audioPacketQueue.readIndex = 0;
- m_audioPacketQueue.serial = 0;
- m_videoPacketQueue.size = 0;
- m_videoPacketQueue.pushIndex = 0;
- m_videoPacketQueue.readIndex = 0;
- m_videoPacketQueue.serial = 0;
- m_audioFrameQueue.size = 0;
- m_audioFrameQueue.readIndex = 0;
- m_audioFrameQueue.pushIndex = 0;
- m_audioFrameQueue.shown = 0;
- m_videoFrameQueue.size = 0;
- m_videoFrameQueue.readIndex = 0;
- m_videoFrameQueue.pushIndex = 0;
- m_videoFrameQueue.shown = 0;
- m_exit = 0;
- m_isSeek = 0;
- m_audSeek = 0;
- m_vidSeek = 0;
- m_audioPktDecoder.serial = 0;
- m_videoPktDecoder.serial = 0;
- }
- int Decoder::decode(const QString& url)
- {
- int ret = 0;
- //解封装初始化
- m_fmtCtx = avformat_alloc_context();
- //用于获取流时长
- AVDictionary* formatOpts = nullptr;
- av_dict_set(&formatOpts, "probesize", "32", 0);
- av_dict_set(&formatOpts, "analyzeduration", "0", 0);
- av_dict_set(&formatOpts, "rtsp_transport", "tcp", 0);
- av_dict_set(&formatOpts, "fflags", "nobuffer", 0);
- av_dict_set(&formatOpts, "max_delay", "0", 0);
- ret = avformat_open_input(&m_fmtCtx, url.toUtf8().constData(), nullptr, &formatOpts);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avformat_open_input error:" << m_errBuf;
- av_dict_free(&formatOpts);
- //打开失败释放分配的AVFormatContext内存
- avformat_free_context(m_fmtCtx);
- return 0;
- }
- // 低延迟:关闭内部缓冲
- if (m_fmtCtx) {
- m_fmtCtx->flags |= AVFMT_FLAG_NOBUFFER;
- m_fmtCtx->max_delay = 0;
- }
- ret = avformat_find_stream_info(m_fmtCtx, nullptr);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avformat_find_stream_info error:" << m_errBuf;
- av_dict_free(&formatOpts);
- return 0;
- }
- //记录流时长
- AVRational q = {1, AV_TIME_BASE};
- m_duration = (uint32_t) (m_fmtCtx->duration * av_q2d(q));
- av_dict_free(&formatOpts);
- // 查找音频/视频流,允许缺失
- m_audioIndex = av_find_best_stream(m_fmtCtx, AVMEDIA_TYPE_AUDIO, -1, -1, nullptr, 0);
- if (m_audioIndex < 0) {
- qDebug() << "no audio stream!";
- }
- m_videoIndex = av_find_best_stream(m_fmtCtx, AVMEDIA_TYPE_VIDEO, -1, -1, nullptr, 0);
- if (m_videoIndex < 0) {
- qDebug() << "no video stream!";
- }
- if (m_audioIndex < 0 && m_videoIndex < 0) {
- qDebug() << "no audio and video stream!";
- return 0;
- }
- // 音频解码初始化(可选)
- if (m_audioIndex >= 0) {
- AVCodecParameters* audioCodecPar = m_fmtCtx->streams[m_audioIndex]->codecpar;
- if (!audioCodecPar) {
- qDebug() << "audio par is nullptr!";
- return 0;
- }
- m_audioPktDecoder.codecCtx = avcodec_alloc_context3(nullptr);
- ret = avcodec_parameters_to_context(m_audioPktDecoder.codecCtx, audioCodecPar);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
- return 0;
- }
- const AVCodec* audioCodec = avcodec_find_decoder(m_audioPktDecoder.codecCtx->codec_id);
- if (!audioCodec) {
- qDebug() << "avcodec_find_decoder failed!";
- return 0;
- }
- m_audioPktDecoder.codecCtx->codec_id = audioCodec->id;
- // 低延迟解码设置(音频)
- m_audioPktDecoder.codecCtx->flags |= AV_CODEC_FLAG_LOW_DELAY;
- m_audioPktDecoder.codecCtx->flags2 |= AV_CODEC_FLAG2_FAST;
- m_audioPktDecoder.codecCtx->thread_type = FF_THREAD_SLICE;
- m_audioPktDecoder.codecCtx->thread_count = 1;
- ret = avcodec_open2(m_audioPktDecoder.codecCtx, audioCodec, nullptr);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_open2:" << m_errBuf;
- return 0;
- }
- }
- // 视频解码初始化(可选)
- if (m_videoIndex >= 0) {
- AVCodecParameters* videoCodecPar = m_fmtCtx->streams[m_videoIndex]->codecpar;
- if (!videoCodecPar) {
- qDebug() << "videocodecpar is nullptr!";
- return 0;
- }
- m_videoPktDecoder.codecCtx = avcodec_alloc_context3(nullptr);
- ret = avcodec_parameters_to_context(m_videoPktDecoder.codecCtx, videoCodecPar);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_parameters_to_context:" << m_errBuf;
- return 0;
- }
- const AVCodec* videoCodec = avcodec_find_decoder(m_videoPktDecoder.codecCtx->codec_id);
- if (!videoCodec) {
- qDebug() << "avcodec_find_decoder failed!";
- return 0;
- }
- m_videoPktDecoder.codecCtx->codec_id = videoCodec->id;
- // 低延迟解码设置(视频)
- m_videoPktDecoder.codecCtx->flags |= AV_CODEC_FLAG_LOW_DELAY;
- m_videoPktDecoder.codecCtx->flags2 |= AV_CODEC_FLAG2_FAST;
- m_videoPktDecoder.codecCtx->thread_type = FF_THREAD_SLICE;
- m_videoPktDecoder.codecCtx->thread_count = 1;
- ret = avcodec_open2(m_videoPktDecoder.codecCtx, videoCodec, nullptr);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "error info_avcodec_open2:" << m_errBuf;
- return 0;
- }
- //记录视频帧率
- m_vidFrameRate = av_guess_frame_rate(m_fmtCtx, m_fmtCtx->streams[m_videoIndex], nullptr);
- }
- // 启动解复用/解码线程
- setInitVal();
- ThreadPool::addTask(std::bind(&Decoder::demux, this, std::placeholders::_1),
- std::make_shared<int>(1));
- if (m_audioIndex >= 0) {
- ThreadPool::addTask(std::bind(&Decoder::audioDecode, this, std::placeholders::_1),
- std::make_shared<int>(2));
- }
- if (m_videoIndex >= 0) {
- ThreadPool::addTask(std::bind(&Decoder::videoDecode, this, std::placeholders::_1),
- std::make_shared<int>(3));
- }
- return 1;
- }
- void Decoder::exit()
- {
- m_exit = 1;
- QThread::msleep(200);
- clearQueueCache();
- if (m_fmtCtx) {
- avformat_close_input(&m_fmtCtx);
- m_fmtCtx = nullptr;
- }
- if (m_audioPktDecoder.codecCtx) {
- avcodec_free_context(&m_audioPktDecoder.codecCtx);
- m_audioPktDecoder.codecCtx = nullptr;
- }
- if (m_videoPktDecoder.codecCtx) {
- avcodec_free_context(&m_videoPktDecoder.codecCtx);
- m_videoPktDecoder.codecCtx = nullptr;
- }
- }
- void Decoder::clearQueueCache()
- {
- std::lock_guard<std::mutex> lockAP(m_audioPacketQueue.mutex);
- std::lock_guard<std::mutex> lockVP(m_videoPacketQueue.mutex);
- while (m_audioPacketQueue.size) {
- av_packet_unref(&m_audioPacketQueue.pktVec[m_audioPacketQueue.readIndex].pkt);
- m_audioPacketQueue.readIndex = (m_audioPacketQueue.readIndex + 1) % m_maxPacketQueueSize;
- m_audioPacketQueue.size--;
- }
- while (m_videoPacketQueue.size) {
- av_packet_unref(&m_videoPacketQueue.pktVec[m_videoPacketQueue.readIndex].pkt);
- m_videoPacketQueue.readIndex = (m_videoPacketQueue.readIndex + 1) % m_maxPacketQueueSize;
- m_videoPacketQueue.size--;
- }
- std::lock_guard<std::mutex> lockAF(m_audioFrameQueue.mutex);
- std::lock_guard<std::mutex> lockVF(m_videoFrameQueue.mutex);
- while (m_audioFrameQueue.size) {
- av_frame_unref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
- m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
- m_audioFrameQueue.size--;
- }
- while (m_videoFrameQueue.size) {
- av_frame_unref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex].frame);
- m_videoFrameQueue.readIndex = (m_videoFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
- m_videoFrameQueue.size--;
- }
- }
- void Decoder::packetQueueFlush(PacketQueue* queue)
- {
- std::lock_guard<std::mutex> lockAP(queue->mutex);
- while (queue->size) {
- av_packet_unref(&queue->pktVec[queue->readIndex].pkt);
- queue->readIndex = (queue->readIndex + 1) % m_maxPacketQueueSize;
- queue->size--;
- }
- queue->serial++;
- }
- void Decoder::seekTo(int32_t target, int32_t seekRel)
- {
- //上次跳转未完成不处理跳转请求
- if (m_isSeek == 1)
- return;
- if (target < 0)
- target = 0;
- m_seekTarget = target;
- m_seekRel = seekRel;
- m_isSeek = 1;
- }
- void Decoder::demux(std::shared_ptr<void> par)
- {
- int ret = -1;
- AVPacket* pkt = av_packet_alloc();
- while (1) {
- if (m_exit) {
- break;
- }
- if (m_audioPacketQueue.size >= m_maxPacketQueueSize
- || m_videoPacketQueue.size >= m_maxPacketQueueSize) {
- std::this_thread::sleep_for(std::chrono::milliseconds(50));
- continue;
- }
- if (m_isSeek) {
- //AVRational sq={1,AV_TIME_BASE};
- //int64_t seekMin = m_seekRel > 0 ? m_seekTarget-m_seekRel+2 : INT64_MIN;
- //int64_t seekMax = m_seekRel < 0 ? m_seekTarget-m_seekRel-2 : INT64_MAX;
- //qDebug()<<"seekMin:"<<seekMin<<" seekMax:"<<seekMax<<" seekTarget:"<<m_seekTarget<<endl;
- //ret=avformat_seek_file(m_fmtCtx,m_audioIndex,seekMin,m_seekTarget,seekMax,AVSEEK_FLAG_BACKWARD);
- int64_t seekTarget = m_seekTarget * AV_TIME_BASE;
- ret = av_seek_frame(m_fmtCtx, -1, seekTarget, AVSEEK_FLAG_BACKWARD);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avformat_seek_file error:" << m_errBuf;
- } else {
- packetQueueFlush(&m_audioPacketQueue);
- packetQueueFlush(&m_videoPacketQueue);
- m_audSeek = 1;
- m_vidSeek = 1;
- }
- m_isSeek = 0;
- }
- ret = av_read_frame(m_fmtCtx, pkt);
- if (ret != 0) {
- // 仅清理内部引用,避免与循环末尾的 av_packet_free 重复释放
- av_packet_unref(pkt);
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "av_read_frame error:" << m_errBuf;
- break;
- }
- if (pkt->stream_index == m_audioIndex) {
- //插入音频包队列
- //qDebug()<<pkt->pts*av_q2d(m_fmtCtx->streams[m_audioIndex]->time_base)<<endl;
- pushPacket(&m_audioPacketQueue, pkt);
- } else if (pkt->stream_index == m_videoIndex) {
- //插入视频包队列
- pushPacket(&m_videoPacketQueue, pkt);
- //av_packet_unref(pkt);
- } else {
- av_packet_unref(pkt);
- }
- }
- av_packet_free(&pkt);
- //是读到文件末尾退出的才清空,强制退出不重复此操作
- if (!m_exit) {
- // 根据实际存在的流等待对应帧队列清空
- if (m_audioIndex >= 0) {
- while (m_audioFrameQueue.size)
- QThread::msleep(50);
- }
- if (m_videoIndex >= 0) {
- while (m_videoFrameQueue.size)
- QThread::msleep(50);
- }
- exit();
- }
- //qDebug() << "demuxthread exit";
- }
- void Decoder::audioDecode(std::shared_ptr<void> par)
- {
- AVPacket* pkt = av_packet_alloc();
- AVFrame* frame = av_frame_alloc();
- while (1) {
- if (m_exit) {
- break;
- }
- //音频帧队列长度控制
- if (m_audioFrameQueue.size >= m_maxFrameQueueSize) {
- std::this_thread::sleep_for(std::chrono::milliseconds(10));
- continue;
- }
- //从音频包队列取音频包
- int ret = getPacket(&m_audioPacketQueue, pkt, &m_audioPktDecoder);
- if (ret) {
- ret = avcodec_send_packet(m_audioPktDecoder.codecCtx, pkt);
- av_packet_unref(pkt);
- if (ret < 0) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avcodec_send_packet error:" << m_errBuf;
- continue;
- }
- while (1) {
- ret = avcodec_receive_frame(m_audioPktDecoder.codecCtx, frame);
- if (ret == 0) {
- if (m_audSeek) {
- int pts = (int) frame->pts
- * av_q2d(m_fmtCtx->streams[m_audioIndex]->time_base);
- //qDebug()<<"audFrame pts:"<<pts<<endl;
- if (pts < m_seekTarget) {
- av_frame_unref(frame);
- continue;
- } else {
- m_audSeek = 0;
- }
- }
- //添加到待播放音频帧队列
- pushAFrame(frame);
- } else {
- break;
- }
- }
- } else {
- //qDebug() << "audio packetQueue is empty for decoding!";
- std::this_thread::sleep_for(std::chrono::milliseconds(20));
- }
- }
- av_packet_free(&pkt);
- av_frame_free(&frame);
- //qDebug() << "audioDecode exit";
- }
- void Decoder::videoDecode(std::shared_ptr<void> par)
- {
- AVPacket* pkt = av_packet_alloc();
- AVFrame* frame = av_frame_alloc();
- while (1) {
- if (m_exit) {
- break;
- }
- if (m_videoFrameQueue.size >= m_maxFrameQueueSize) { //视频帧队列长度控制
- std::this_thread::sleep_for(std::chrono::milliseconds(10));
- continue;
- }
- int ret = getPacket(&m_videoPacketQueue, pkt, &m_videoPktDecoder);
- if (ret) {
- ret = avcodec_send_packet(m_videoPktDecoder.codecCtx, pkt);
- av_packet_unref(pkt);
- if (ret < 0 || ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) {
- av_strerror(ret, m_errBuf, sizeof(m_errBuf));
- qDebug() << "avcodec_send_packet error:" << m_errBuf;
- continue;
- }
- while (1) {
- ret = avcodec_receive_frame(m_videoPktDecoder.codecCtx, frame);
- if (ret == 0) {
- if (m_vidSeek) {
- int pts = (int) frame->pts
- * av_q2d(m_fmtCtx->streams[m_videoIndex]->time_base);
- if (pts < m_seekTarget) {
- av_frame_unref(frame);
- continue;
- } else {
- m_vidSeek = 0;
- }
- }
- //AVRational vidRational=av_guess_frame_rate(m_fmtCtx,
- // m_fmtCtx->streams[m_videoIndex],frame);
- //添加到待播放视频帧队列
- pushVFrame(frame);
- } else {
- break;
- }
- }
- } else {
- //qDebug() << "video packetQueue is empty for decoding!";
- std::this_thread::sleep_for(std::chrono::milliseconds(20));
- }
- }
- av_packet_free(&pkt);
- av_frame_free(&frame);
- //qDebug() << "videoDecode exit";
- }
- int Decoder::getPacket(PacketQueue* queue, AVPacket* pkt, PktDecoder* decoder)
- {
- std::unique_lock<std::mutex> lock(queue->mutex);
- while (!queue->size) {
- bool ret = queue->cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
- return queue->size && !m_exit;
- });
- if (!ret)
- return 0;
- }
- if (queue->serial != decoder->serial) {
- //序列号不连续的帧证明发生了跳转操作则直接丢弃
- //并清空解码器缓存
- avcodec_flush_buffers(decoder->codecCtx);
- decoder->serial = queue->pktVec[queue->readIndex].serial;
- return 0;
- }
- av_packet_move_ref(pkt, &queue->pktVec[queue->readIndex].pkt);
- decoder->serial = queue->pktVec[queue->readIndex].serial;
- queue->readIndex = (queue->readIndex + 1) % m_maxPacketQueueSize;
- queue->size--;
- return 1;
- }
- void Decoder::pushPacket(PacketQueue* queue, AVPacket* pkt)
- {
- std::lock_guard<std::mutex> lock(queue->mutex);
- av_packet_move_ref(&queue->pktVec[queue->pushIndex].pkt, pkt);
- queue->pktVec[queue->pushIndex].serial = queue->serial;
- if (queue->serial == 1)
- qDebug() << "";
- queue->pushIndex = (queue->pushIndex + 1) % m_maxPacketQueueSize;
- queue->size++;
- }
- void Decoder::pushAFrame(AVFrame* frame)
- {
- std::lock_guard<std::mutex> lock(m_audioFrameQueue.mutex);
- av_frame_move_ref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.pushIndex].frame, frame);
- m_audioFrameQueue.frameVec[m_audioFrameQueue.pushIndex].serial = m_audioPktDecoder.serial;
- m_audioFrameQueue.pushIndex = (m_audioFrameQueue.pushIndex + 1) % m_maxFrameQueueSize;
- m_audioFrameQueue.size++;
- }
- int Decoder::getAFrame(AVFrame* frame)
- {
- if (!frame)
- return 0;
- std::unique_lock<std::mutex> lock(m_audioFrameQueue.mutex);
- while (!m_audioFrameQueue.size) {
- bool ret = m_audioFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
- return !m_exit && m_audioFrameQueue.size;
- });
- if (!ret)
- return 0;
- }
- if (m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].serial
- != m_audioPacketQueue.serial) {
- av_frame_unref(&m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
- m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
- m_audioFrameQueue.size--;
- return 0;
- }
- av_frame_move_ref(frame, &m_audioFrameQueue.frameVec[m_audioFrameQueue.readIndex].frame);
- m_audioFrameQueue.readIndex = (m_audioFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
- m_audioFrameQueue.size--;
- return 1;
- }
- void Decoder::pushVFrame(AVFrame* frame)
- {
- std::lock_guard<std::mutex> lock(m_videoFrameQueue.mutex);
- m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].serial = m_videoPktDecoder.serial;
- m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].duration
- = (m_vidFrameRate.den && m_vidFrameRate.den)
- ? av_q2d(AVRational{m_vidFrameRate.den, m_vidFrameRate.num})
- : 0.00;
- m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].pts
- = frame->pts * av_q2d(m_fmtCtx->streams[m_videoIndex]->time_base);
- av_frame_move_ref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.pushIndex].frame, frame);
- m_videoFrameQueue.pushIndex = (m_videoFrameQueue.pushIndex + 1) % m_maxFrameQueueSize;
- m_videoFrameQueue.size++;
- //qDebug()<<"RemainingVFrame:"<<m_videoFrameQueue.size-m_videoFrameQueue.shown;
- }
- Decoder::MyFrame* Decoder::peekLastVFrame()
- {
- Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex];
- return frame;
- }
- Decoder::MyFrame* Decoder::peekVFrame()
- {
- while (!m_videoFrameQueue.size) {
- std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
- bool ret = m_videoFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
- return !m_exit && m_videoFrameQueue.size;
- });
- if (!ret)
- return nullptr;
- }
- int index = (m_videoFrameQueue.readIndex + m_videoFrameQueue.shown) % m_maxFrameQueueSize;
- Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[index];
- return frame;
- }
- Decoder::MyFrame* Decoder::peekNextVFrame()
- {
- while (m_videoFrameQueue.size < 2) {
- std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
- bool ret = m_videoFrameQueue.cond.wait_for(lock, std::chrono::milliseconds(100), [&]() {
- return !m_exit && m_videoFrameQueue.size;
- });
- if (!ret)
- return nullptr;
- }
- int index = (m_videoFrameQueue.readIndex + m_videoFrameQueue.shown + 1) % m_maxFrameQueueSize;
- Decoder::MyFrame* frame = &m_videoFrameQueue.frameVec[index];
- return frame;
- }
- void Decoder::setNextVFrame()
- {
- std::unique_lock<std::mutex> lock(m_videoFrameQueue.mutex);
- if (!m_videoFrameQueue.size)
- return;
- if (!m_videoFrameQueue.shown) {
- m_videoFrameQueue.shown = 1;
- return;
- }
- av_frame_unref(&m_videoFrameQueue.frameVec[m_videoFrameQueue.readIndex].frame);
- m_videoFrameQueue.readIndex = (m_videoFrameQueue.readIndex + 1) % m_maxFrameQueueSize;
- m_videoFrameQueue.size--;
- }
- int Decoder::getRemainingVFrame()
- {
- if (!m_videoFrameQueue.size)
- return 0;
- return m_videoFrameQueue.size - m_videoFrameQueue.shown;
- }
|