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