2 files changed, 6 insertions, 1 deletions
diff --git a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp
index 97830c9a..e216d319 100644
--- a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp
+++ b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.cpp
@@ -214,6 +214,8 @@ void Tracker::run()
             grayscale = channel[2];
         } else
             cv::cvtColor(color, grayscale, cv::COLOR_BGR2GRAY);
+        
+        gain.tick(camera, grayscale);
 
         const int scale = frame.cols > 480 ? 2 : 1;
         detector.setThresholdParams(scale > 1 ? 11 : 7, 4);
diff --git a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h
index 81332a26..5416bb52 100644
--- a/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h
+++ b/ftnoir_tracker_aruco/ftnoir_tracker_aruco.h
@@ -21,6 +21,9 @@
 #include "facetracknoir/options.h"
 #include "ftnoir_tracker_aruco/trans_calib.h"
 #include "facetracknoir/plugin-api.hpp"
+
+#include "facetracknoir/gain-control.hpp"
+
 using namespace options;
 
 struct settings {
@@ -65,9 +68,9 @@ private:
     cv::VideoCapture camera;
     cv::Matx33f r;
     cv::Vec3f t;
+    Gain gain;
 };
 
-// Widget that has controls for FTNoIR protocol client-settings.
 class TrackerControls : public QWidget, public ITrackerDialog
 {
     Q_OBJECT