Mercurial > hg > nsaunier > traffic-intelligence
comparison include/InputVideoFileModule.h @ 400:7ef1071e3cc3
clean up of input classes for list of images and video files
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Mon, 29 Jul 2013 18:06:55 -0400 |
| parents | c389fae9689a |
| children | b829ebdc18e6 |
comparison
equal
deleted
inserted
replaced
| 399:c389fae9689a | 400:7ef1071e3cc3 |
|---|---|
| 7 #include "opencv2/highgui/highgui.hpp" | 7 #include "opencv2/highgui/highgui.hpp" |
| 8 | 8 |
| 9 class InputVideoFileModule : public InputFrameProviderIface | 9 class InputVideoFileModule : public InputFrameProviderIface |
| 10 { | 10 { |
| 11 public: | 11 public: |
| 12 InputVideoFileModule(const std::string& videoPath); | 12 InputVideoFileModule(const std::string& videoPath); |
| 13 ~InputVideoFileModule(); | 13 ~InputVideoFileModule(); |
| 14 | 14 |
| 15 bool getNextFrame(cv::Mat&); | |
| 15 | 16 |
| 16 bool isOpen() const { return mInit;} | 17 unsigned int getNbFrames(){ return mNumberOfFrame;} |
| 17 const cv::Size& getSize() const { return mSize;} | |
| 18 | 18 |
| 19 bool isOpen() const { return mInit;} | |
| 19 | 20 |
| 21 void setFrameNumber(const int& frameNumber); | |
| 20 | 22 |
| 21 | 23 const cv::Size& getSize() const { return mSize;} |
| 22 | |
| 23 bool getNextFrame(cv::Mat&); | |
| 24 | |
| 25 unsigned int getNbFrames(){ return mNumberOfFrame;} | |
| 26 | |
| 27 | |
| 28 | |
| 29 | 24 |
| 30 private: | 25 private: |
| 31 cv::Size mSize; | 26 cv::Size mSize; |
| 32 cv::VideoCapture mVideoCapture; | 27 cv::VideoCapture mVideoCapture; |
| 33 bool mInit; | 28 bool mInit; |
| 34 int mNumberOfFrame; | 29 int mNumberOfFrame; |
| 35 }; | 30 }; |
| 36 | 31 |
| 37 #endif | 32 #endif |
