view include/InputVideoFileModule.h @ 589:5800a87f11ae

corrected one bug and changed attribute names
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 05 Dec 2014 15:14:44 -0500
parents 2be846d36dec
children
line wrap: on
line source

#ifndef INPUT_VIDEO_FILE_MODULE_H
#define INPUT_VIDEO_FILE_MODULE_H

#include "InputFrameProviderIface.h"
#include <string>
#include "opencv2/core/core.hpp"
#include "opencv2/highgui/highgui.hpp"

class InputVideoFileModule : public InputFrameProviderIface
{
public:
  InputVideoFileModule(const std::string& videoPath);
  ~InputVideoFileModule();

  bool getNextFrame(cv::Mat&);

  unsigned int getNbFrames(){		return mNumberOfFrame;}

  bool isOpen() const { return mInit;}

  void setFrameNumber(const unsigned int& frameNumber);

  const cv::Size& getSize() const { return mSize;}
	
private:
  cv::Size mSize;
  cv::VideoCapture mVideoCapture;
  bool mInit;
  int mNumberOfFrame;
};

#endif