Mercurial Hosting > traffic-intelligence
diff include/InputFrameListModule.h @ 614:5e09583275a4
Merged Nicolas/trafficintelligence into default
author | Mohamed Gomaa <eng.m.gom3a@gmail.com> |
---|---|
date | Fri, 05 Dec 2014 12:13:53 -0500 |
parents | b829ebdc18e6 |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/include/InputFrameListModule.h Fri Dec 05 12:13:53 2014 -0500 @@ -0,0 +1,30 @@ +#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& _dirname); + ~InputFrameListModule(); + + bool getNextFrame(cv::Mat&); + unsigned int getNbFrames(); + bool isOpen() const { return mInit;} + void setFrameNumber(const unsigned int& frameNumber); + + virtual const cv::Size& getSize() const { return mSize;} + private: + void loadImageList(void); + std::vector<std::string> filenames; + unsigned int mCurrentIdx; + bool mInit; + std::string dirname; + cv::Size mSize; +}; + +#endif