videoreader.h

#ifndef VIDEOREADER_H
#define VIDEOREADER_H
 
#include <QMutex>
#include <QMutexLocker>
#include <QObject>
#include <iostream>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/videoio.hpp>
 
#include "algorithmmanager.h"
#include "reader.h"
#include "readerfactory.h"
#include "readerparams.h"
#include "sequentialreaderimpl.h"
 
extern "C" {
#include <libavcodec/avcodec.h>
#include <libavformat/avformat.h>
#include <libavutil/rational.h>
#include <libswscale/swscale.h>
}
 
#define NOT_IMPLEMENTED assert(false && "Not implemtented yet.")
 
class VideoReader : public Reader
 
{
   public:
    explicit VideoReader(const QString& path,
                         std::shared_ptr<ReaderParams> readerParams);
    ~VideoReader();
    cv::Mat getPic(unsigned int index,
                   PictureProcessingFlags flags = APPLY_ALL) override;
    unsigned int getPicCount() override;
    QString getInputPath() override;
    double getFPS() override;
    double getVideoDuration() override;
    bool isDir() override;
    VideoReader *copy(std::shared_ptr<ReaderParams> params) override;
 
    std::vector<std::string> getFileVector() override;
 
    SequentialReader* createSequentialReader(
        std::vector<uint> indices,
        PictureProcessingFlags flags = APPLY_ALL) override;
 
    void addMetaData(MetaData* md) override;
    MetaData* getMetaData() override;
    bool isValid() override;
 
   private:
    // getter exposed members
    int m_currentIndex = -1;
    std::string m_path = "";
    size_t m_frameCount = -1;
    bool m_isValid = false;
    AVRational m_avgVideoFPS;
    MetaData* m_md = nullptr;
    //
 
    QMutex m_mutex;
    std::shared_ptr<ReaderParams> m_readerParams;
    int64_t m_startTimestamp = AV_NOPTS_VALUE;
    std::map<uint, AVFrame*> m_buffer;
    AVRational m_streamTimeBase = AV_TIME_BASE_Q;
    int m_streamId = -1;
    int m_lastFrameIdx = -1;
    // AV/SWS-Objects
    AVFormatContext* m_formatContext = nullptr;
    AVCodecContext* m_codecContext = nullptr;
    const AVCodec* m_codec = nullptr;
    struct SwsContext* m_swsContext = nullptr;
 
    // private helper functions
    cv::Mat avFrame2CvMat(const AVFrame* av_f);
    void openFormatContext();
    void selectVideoStream();
    void openCodec();
    void createSWS();
    int decodeNextPkg();
};
 
REGISTER_READER("VideoReader", VideoReader)
 
#endif  // VIDEOREADER_H