Mercurial > hg > nsaunier > traffic-intelligence
comparison include/InputFrameListModule.h @ 399:c389fae9689a
Added a class to read list of image instead of video. This is controlled by the use of the database-filename and folder-data parameters in the config file.
| author | Jean-Philippe Jodoin <jpjodoin@gmail.com> |
|---|---|
| date | Mon, 29 Jul 2013 17:12:45 -0400 |
| parents | |
| children | 7ef1071e3cc3 |
comparison
equal
deleted
inserted
replaced
| 398:3399bd48cb40 | 399:c389fae9689a |
|---|---|
| 1 #ifndef INPUT_FRAME_LIST_MODULE_H | |
| 2 #define INPUT_FRAME_LIST_MODULE_H | |
| 3 | |
| 4 #include "InputFrameProviderIface.h" | |
| 5 #include <string> | |
| 6 #include <vector> | |
| 7 | |
| 8 class InputFrameListModule : public InputFrameProviderIface | |
| 9 { | |
| 10 public: | |
| 11 InputFrameListModule(const std::string& basePath,const std::string& pictureList); | |
| 12 ~InputFrameListModule(); | |
| 13 | |
| 14 | |
| 15 | |
| 16 bool getNextFrame(cv::Mat&); | |
| 17 unsigned int getNbFrames(); | |
| 18 bool isOpen() const { return mInit;} | |
| 19 | |
| 20 | |
| 21 | |
| 22 | |
| 23 | |
| 24 | |
| 25 | |
| 26 | |
| 27 | |
| 28 | |
| 29 virtual const cv::Size& getSize() const { return mSize;} | |
| 30 private: | |
| 31 void loadFileList(const std::string& path); | |
| 32 std::vector<std::string> mFileList; | |
| 33 int mCurrentIdx; | |
| 34 bool mInit; | |
| 35 std::string mBasePath; | |
| 36 cv::Mat mCurrentFrame; | |
| 37 cv::Size mSize; | |
| 38 | |
| 39 }; | |
| 40 | |
| 41 #endif |