From 4f25bb4a95122bd1938d2690c741595bd5de0258 Mon Sep 17 00:00:00 2001
From: Mederic Fourmy <mederic.fourmy@gmail.com>
Date: Wed, 13 Apr 2022 10:15:40 +0200
Subject: [PATCH] change type of color attribute from int to enum

---
 include/publisher_vision.h | 4 ++--
 src/publisher_vision.cpp   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/include/publisher_vision.h b/include/publisher_vision.h
index 5bcb933..6b00b2f 100644
--- a/include/publisher_vision.h
+++ b/include/publisher_vision.h
@@ -83,7 +83,7 @@ class PublisherVisionDebug : public Publisher
         int size_feature_kfs_pix_;
         double thickness_lmk_;
         double size_text_ID_;
-        int color_tracks_features_;
+        COLORFEATURES color_tracks_features_;
 
         image_transport::Publisher publisher_image_;
         image_transport::ImageTransport img_transport_;
@@ -99,7 +99,7 @@ class PublisherVisionDebug : public Publisher
             Function that return a unique RGB color vector depending on the lenght of the track given,
             so the longer is the track the darker it is
         */
-        std::vector<int> colorTrackAndFeatures(int _nb_feature_in_track, int _max_feature_in_tracks, int _min_feature_in_tracks, int _color_of_features);
+        std::vector<int> colorTrackAndFeatures(int _nb_feature_in_track, int _max_feature_in_tracks, int _min_feature_in_tracks, COLORFEATURES _color_of_features);
 
         /*
             Function that change an image to show tracks and features within it
diff --git a/src/publisher_vision.cpp b/src/publisher_vision.cpp
index d60ebe2..32a8e0b 100644
--- a/src/publisher_vision.cpp
+++ b/src/publisher_vision.cpp
@@ -183,7 +183,7 @@ std::pair<int, int> PublisherVisionDebug::minMaxFeatureInTrack(const std::map<Si
 std::vector<int> PublisherVisionDebug::colorTrackAndFeatures(int _nb_feature_in_track,
                                                              int _max_feature_in_tracks,
                                                              int _min_feature_in_tracks,
-                                                             int _color_of_features)
+                                                             COLORFEATURES _color_of_features)
 {
 
     // std::cout << "\n\ncolorTrackAndFeatures" << '\n';
-- 
GitLab