Adding support for four points model.

Can't get it to work correctly though.

1 files changed, 2 insertions, 2 deletions

diff --git a/tracker-easy/point-extractor.cpp b/tracker-easy/point-extractor.cpp
index 61175a57..0a5ca81d 100644
--- a/tracker-easy/point-extractor.cpp
+++ b/tracker-easy/point-extractor.cpp
@@ -35,7 +35,7 @@ namespace EasyTracker
     }
 
 
-    void PointExtractor::ExtractPoints(const cv::Mat& aFrame, cv::Mat* aPreview, std::vector<cv::Point>& aPoints)
+    void PointExtractor::ExtractPoints(const cv::Mat& aFrame, cv::Mat* aPreview, int aNeededPointCount, std::vector<cv::Point>& aPoints)
     {
         //TODO: Assert if channel size is neither one nor two
         // Make sure our frame channel is 8 bit
@@ -133,7 +133,7 @@ namespace EasyTracker
         // Typically noise comming from zippers and metal parts on your clothing.
         // With a cap tracker it also successfully discards noise from glasses.
         // However it may not work as good with a clip user wearing glasses.
-        while (aPoints.size() > KPointCount) // Until we have no more than three points
+        while (aPoints.size() > aNeededPointCount) // Until we have no more than three points
         {
             int maxY = 0;
             size_t index = -1;