Add worker thread

[trackerpp.git] / include / MultiTracker.h

diff --git a/include/MultiTracker.h b/include/MultiTracker.h
index 58ef922..66ea04e 100644 (file)
--- a/include/MultiTracker.h
+++ b/include/MultiTracker.h
@@ -7,17 +7,19 @@
 #include "PredictorWrapper.h"
 #include <opencv2/opencv.hpp>
 #include <utility>
+#include "Engine.h"
 
 namespace suanzi {
 
     TK_DECLARE_PTR(Patch);
     TK_DECLARE_PTR(MultiTracker);
     TK_DECLARE_PTR(Tracker);
+    TK_DECLARE_PTR(Engine);
 
     class MultiTracker 
     {
     public:
-        MultiTracker();
+        MultiTracker(EngineWPtr e);
         virtual ~MultiTracker();
         void update(unsigned int total, const Detection* d, const cv::Mat& image);
 
@@ -28,6 +30,7 @@ namespace suanzi {
         double distance(TrackerPtr t, const cv::Mat& image, const Detection& d);
         PredictorWrapperPtr predictor;
         cv::HOGDescriptor descriptor;
+        EngineWPtr engine;
     };
 
     class Patch