Mercurial > hg > nsaunier > traffic-intelligence
view include/InputFrameListModule.h @ 595:17b02c8054d0
added tests and corrected one bug
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Sun, 07 Dec 2014 22:59:47 -0500 |
| parents | b829ebdc18e6 |
| children |
line wrap: on
line source
#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
