video-pc/libs/AVPlayer/av_decoder.h

#ifndef AV_DECODER_H
#define AV_DECODER_H

#include <QImage>
#include <QVector>
#include <condition_variable>
#include <mutex>
#include <atomic>

extern "C" {
}
#include "ffmpeg_compat.h"

QT_BEGIN_NAMESPACE
namespace AVTool {
typedef struct MediaInfo
{
    int duration;
    QImage tipImg;
} MediaInfo;

class Decoder
{
public:
    typedef struct MyFrame
    {
        AVFrame frame;
        int serial;
        double duration;
        double pts;
    } MyFrame;

private:
    typedef struct MyPacket
    {
        AVPacket pkt;
        int serial;
    } MyPacket;

    typedef struct PacketQueue
    {
        QVector<MyPacket> pktVec;
        int readIndex;
        int pushIndex;
        int size;
        int serial;
        std::mutex mutex;
        std::condition_variable cond;
    } PacketQueue;

    typedef struct FrameQueue
    {
        QVector<MyFrame> frameVec;
        int readIndex;
        int pushIndex;
        int shown;
        int size;
        std::mutex mutex;
        std::condition_variable cond;
    } FrameQueue;

    typedef struct PktDecoder
    {
        AVCodecContext* codecCtx;
        int serial;
    } PktDecoder;

public:
    Decoder();
    ~Decoder();

    //成功探测到媒体信息返回类型实例，否则返回nullptr
    AVTool::MediaInfo* detectMediaInfo(const QString& url);

    void seekTo(int32_t target, int32_t seekRel);

    int getAFrame(AVFrame* frame);

    int getRemainingVFrame();
    //查看上一帧（即当前显示的画面帧）
    Decoder::MyFrame* peekLastVFrame();
    //查看将要显示的帧
    Decoder::MyFrame* peekVFrame();
    //查看将要显示帧再下一帧
    Decoder::MyFrame* peekNextVFrame();
    //将读索引后移一位
    void setNextVFrame();

    inline int vidPktSerial() const { return m_videoPacketQueue.serial; }

    inline int audioIndex() const { return m_audioIndex; }

    inline int videoIndex() const { return m_videoIndex; }

    inline AVFormatContext* formatContext() const { return m_fmtCtx; }

    inline AVCodecParameters* audioCodecPar() const
    {
        return m_fmtCtx->streams[m_audioIndex]->codecpar;
    }

    inline AVCodecParameters* videoCodecPar() const
    {
        return m_fmtCtx->streams[m_videoIndex]->codecpar;
    }

    inline uint32_t avDuration() { return m_duration; }

    inline int isExit() { return m_exit; }

    int decode(const QString& url);

    void exit();

private:
    PacketQueue m_audioPacketQueue;
    PacketQueue m_videoPacketQueue;

    FrameQueue m_audioFrameQueue;
    FrameQueue m_videoFrameQueue;

    PktDecoder m_audioPktDecoder;
    PktDecoder m_videoPktDecoder;

    AVFormatContext* m_fmtCtx;

    const int m_maxFrameQueueSize;
    const int m_maxPacketQueueSize;

    AVRational m_vidFrameRate;

    int m_audioIndex;
    int m_videoIndex;

    std::atomic<int> m_exit;

    //是否执行跳转
    int m_isSeek;

    //跳转后等待目标帧标志
    int m_vidSeek;
    int m_audSeek;

    //跳转相对时间
    int64_t m_seekRel;

    //跳转绝对时间
    int64_t m_seekTarget;

    //流总时长/S
    uint32_t m_duration;

    char m_errBuf[100];

private:
    bool init();

    void setInitVal();

    void packetQueueFlush(PacketQueue* queue);

    void clearQueueCache();

    void demux(std::shared_ptr<void> par);

    void audioDecode(std::shared_ptr<void> par);
    void videoDecode(std::shared_ptr<void> par);

    int getPacket(PacketQueue* queue, AVPacket* pkt, PktDecoder* decoder);
    void pushPacket(PacketQueue* queue, AVPacket* pkt);

    void pushAFrame(AVFrame* frame);

    void pushVFrame(AVFrame* frame);
};
} // namespace AVTool
QT_END_NAMESPACE

#endif // AV_DECODER_H