VideoClient98/src/video/readstream.h

#ifndef READSTREAM_H
#define READSTREAM_H

#include <QDebug>
#include <QMutex>
#include <QObject>
#include <QQueue>
#include <QSize>
#include <QThread>
#include <QUdpSocket>

#include "avpacketqueuemanager.h"
#include "decodestream.h"
#include "ffmpeginclude.h"
#include "pushstream.h"
#include "savestream.h"

class ReadStream : public QObject {
    Q_OBJECT
public:
    explicit ReadStream(QObject *parent = nullptr);
    ~ReadStream();
    bool openFile(const QString &url);
    bool setStreamDecoder(DecodeStream *decodeStreamer);
    bool setUDPStreamDecoder(DecodeStream *decodeStreamer);
    bool setStreamSaver(SaveStream *streamSaver, bool isUDP = false);
    bool setStreamPusher(PushStream *streamPusher);  // 开启推流
    void close();

    void setUDPParms(QString ip, int port);
    bool initSocket(QString ip, int port);
public slots:
    void startPullStream();
    void startPullUDPStream();
    void closeUDPConnectionSLot();
signals:
    // void startDecodeSignal();
    void startPullStreamSignal();
    void startPullUDPStreamSignal();
    void sendErrorMessageSignal(QString message, int type);
    void closeUDPConnectionSignal();

private:
    void initFFmpeg();  // 初始化ffmpeg库（整个程序中只需加载一次）
    bool initObject();  // 初始化对象
    // void showError(int err);  // 显示ffmpeg执行错误时的错误信息
    // qreal rationalToDouble(AVRational* rational);  //
    // 将AVRational转换为double
    void clear();  // 清空读取缓冲
    void free();   // 释放

    /**
     * @brief 重连
     * @return 重连成功或失败
     */
    bool reconnect();

    bool isValidAVPacket(AVPacket *pkt);

    bool isMulticastAddress(const QString &ip);

    void udpDataReceivedSlot();

private:
    bool m_pushStreamFlag = false;  // 推流开始标识
    bool m_decodeStreamFlag = false;
    bool m_saveStreamFlag = false;
    QString m_pullURL;      // 拉流地址
    int MAXRECONNECT = 12;  // 最大重连次数
    qreal m_frameRate = 0;  // 视频帧率
    QSize m_size;           // 视频分辨率大小
    uchar *m_buffer = nullptr;
    bool m_start = true;
    bool m_end = false;
    int64_t m_startTime;

    AVFormatContext *m_formatContext = nullptr;  // 解封装上下文
    AVCodecContext *m_codecContext = nullptr;    // 解码器上下文
    // SwsContext* m_swsContext = nullptr;          // 图像转换上下文
    AVPacket *m_inputPacket = nullptr;  // 数据包
    int m_videoIndex = 0;               // 视频流索引
    int QUEUECAPACITY = 100;

    DecodeStream *m_streamDecoder = nullptr;
    SaveStream *m_streamSaver = nullptr;
    PushStream *m_streamPusher = nullptr;

    // QMutex m_mutex;
    AVPacketQueueManager m_queueManager;

private:
    QUdpSocket *udpSocket = nullptr;
    QString m_UDPIP;
    int m_UDPPort;
    AVCodecParserContext *parser = nullptr;
    AVCodecContext *codec_ctx = nullptr;
};

#endif  // READSTREAM_H