Add test for Predictor

[trackerpp.git] / src / Tracker.cpp

diff --git a/src/Tracker.cpp b/src/Tracker.cpp
index 27dd400..8ab2414 100644 (file)
--- a/src/Tracker.cpp
+++ b/src/Tracker.cpp
@@ -2,16 +2,16 @@
 
 using namespace suanzi;
 using namespace cv;
+using namespace std;
 
 static const int MaxLost = 5;
 
-Tracker::Tracker(int id) : id(id)
+Tracker::Tracker(const cv::Mat& image,int id) : id(id)
 {
     status = TrackerStatus::Fire;
     preStatus = TrackerStatus::Fire;
 
-    // TODO
-    // init KalmanFilter
+    // TODO: Kalman filter
     this->kf.transitionMatrix = (Mat_<float>(4, 4) << 
                                                 1, 0, 1, 0,
                                                 0, 1, 0, 1,
@@ -29,14 +29,7 @@ Tracker::Tracker(int id) : id(id)
 
 Tracker::~Tracker()
 {
-}
-
-void Tracker::addPatch(Patch* p)
-{
-    patches.push_back(p);
-    if (patches.size() > Metrics::MaxPatch){
-        patches.erase(patches.end());
-    }
+    patches.clear();
 }
 
 void Tracker::updateState(const Mat& image)
@@ -53,3 +46,8 @@ void Tracker::updateState(const Mat& image)
         status = TrackerStatus::Lost;
     }
 }
+
+void Tracker::addPatch(PatchPtr p)
+{
+    this->patches.push_back(p);
+}