SENSVideoStream.h

Go to the documentation of this file.

#ifndef SENS_VIDEOSTREAM_H
#define SENS_VIDEOSTREAM_H
 
#include <opencv2/opencv.hpp>
#include "SENSFrame.h"
#include "SENSCalibration.h"
 
class SENSVideoStream
{
public:
    SENSVideoStream(const std::string& videoFileName, bool videoLoops, bool mirrorH, bool mirrorV, float targetFps = 0);
    SENSFramePtr grabNextFrame();
    SENSFramePtr grabNextResampledFrame();
    SENSFramePtr grabPreviousResampledFrame();
    SENSFramePtr grabPreviousFrame();
 
    cv::Size getFrameSize() const { return _videoFrameSize; }
 
    const std::string& videoFilename() const { return _videoFileName; }
    int                nextFrameIndex() const { return (int)_cap.get(cv::CAP_PROP_POS_FRAMES); }
    int                frameCount() const { return _frameCount; }
    float              fps() const { return _fps; }
 
    bool isOpened() const { return _cap.isOpened(); }
 
    const SENSCalibration* const calibration() const
    {
        return _calibration.get();
    }
 
    void setCalibration(SENSCalibration calibration, bool buildUndistortionMaps);
    void guessAndSetCalibration(float fovGuess);
 
private:
    void moveCapturePosition(int n);
 
    cv::VideoCapture _cap;
    cv::Size2i       _videoFrameSize;
    int              _frameCount = 0;
    std::string      _videoFileName;
 
    bool  _videoLoops = false;
    float _fps        = 0.f;
    float _targetFps;
 
    bool _mirrorH = false;
    bool _mirrorV = false;
 
    std::unique_ptr<SENSCalibration> _calibration;
};
 
#endif //SENS_VIDEOSTREAM_H