Mercurial Hosting > traffic-intelligence
diff include/InputFrameListModule.h @ 401:b829ebdc18e6
simplified input of directories of video frames (simply use the video filename parameter to point at the directory)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 29 Jul 2013 18:58:05 -0400 |
parents | 7ef1071e3cc3 |
children |
line wrap: on
line diff
--- a/include/InputFrameListModule.h Mon Jul 29 18:06:55 2013 -0400 +++ b/include/InputFrameListModule.h Mon Jul 29 18:58:05 2013 -0400 @@ -2,32 +2,29 @@ #define INPUT_FRAME_LIST_MODULE_H #include "InputFrameProviderIface.h" + #include <string> #include <vector> class InputFrameListModule : public InputFrameProviderIface { public: - InputFrameListModule(const std::string& basePath,const std::string& pictureList); + InputFrameListModule(const std::string& _dirname); ~InputFrameListModule(); - - bool getNextFrame(cv::Mat&); unsigned int getNbFrames(); bool isOpen() const { return mInit;} - void setFrameNumber(const int& frameNumber) {} // does nothing for now + void setFrameNumber(const unsigned int& frameNumber); virtual const cv::Size& getSize() const { return mSize;} private: - void loadFileList(const std::string& path); - std::vector<std::string> mFileList; + void loadImageList(void); + std::vector<std::string> filenames; unsigned int mCurrentIdx; bool mInit; - std::string mBasePath; - cv::Mat mCurrentFrame; + std::string dirname; cv::Size mSize; - }; #endif