improvement

[trackerpp.git] / src / Engine.h

diff --git a/src/Engine.h b/src/Engine.h
index 66b1341..a0da903 100644 (file)
--- a/src/Engine.h
+++ b/src/Engine.h
@@ -6,6 +6,7 @@
 #include<set>
 #include "Tracker.h"
 #include "Detector.h"
+#include "VideoReader.h"
 
 namespace suanzi{
 
@@ -18,15 +19,17 @@ public:
     void destroy();
     virtual void start();
     void addObserver(EngineObserver* o);
-    void setVideoSrc(const std::string& url);
+    void setVideoSrc(VideoSrcType type, const std::string& url);
 
 private:
     Engine();
     virtual ~Engine();
+    void run();
     Tracker* tracker;
     Detector* detector;
     std::set<EngineObserver *> observer_list;
     std::string videoSrc;
+    VideoReader* reader;
 };
 
 struct Person