Mercurial Hosting > traffic-intelligence
view 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 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