view 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 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