Mercurial Hosting > traffic-intelligence
diff 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 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/include/InputFrameListModule.h Mon Jul 29 17:12:45 2013 -0400 @@ -0,0 +1,41 @@ +#ifndef INPUT_FRAME_LIST_MODULE_H +#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(); + + + + bool getNextFrame(cv::Mat&); + unsigned int getNbFrames(); + bool isOpen() const { return mInit;} + + + + + + + + + + + virtual const cv::Size& getSize() const { return mSize;} +private: + void loadFileList(const std::string& path); + std::vector<std::string> mFileList; + int mCurrentIdx; + bool mInit; + std::string mBasePath; + cv::Mat mCurrentFrame; + cv::Size mSize; + +}; + +#endif \ No newline at end of file