diff include/InputVideoFileModule.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 2be846d36dec
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/include/InputVideoFileModule.h	Fri Dec 05 12:13:53 2014 -0500
@@ -0,0 +1,32 @@
+#ifndef INPUT_VIDEO_FILE_MODULE_H
+#define INPUT_VIDEO_FILE_MODULE_H
+
+#include "InputFrameProviderIface.h"
+#include <string>
+#include "opencv2/core/core.hpp"
+#include "opencv2/highgui/highgui.hpp"
+
+class InputVideoFileModule : public InputFrameProviderIface
+{
+public:
+  InputVideoFileModule(const std::string& videoPath);
+  ~InputVideoFileModule();
+
+  bool getNextFrame(cv::Mat&);
+
+  unsigned int getNbFrames(){		return mNumberOfFrame;}
+
+  bool isOpen() const { return mInit;}
+
+  void setFrameNumber(const unsigned int& frameNumber);
+
+  const cv::Size& getSize() const { return mSize;}
+	
+private:
+  cv::Size mSize;
+  cv::VideoCapture mVideoCapture;
+  bool mInit;
+  int mNumberOfFrame;
+};
+
+#endif