From 0e44f5a0e0205902f144daef160c70a101f6e165 Mon Sep 17 00:00:00 2001
From: Eduardo Almeida <enmsa@outlook.pt>
Date: Thu, 24 Nov 2022 18:25:59 +0000
Subject: [PATCH] examples: Improve wifi-ofdmt-eht-validation.cc

---
 examples/wireless/wifi-ofdm-eht-validation.cc | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/examples/wireless/wifi-ofdm-eht-validation.cc b/examples/wireless/wifi-ofdm-eht-validation.cc
index 2989da050..266ba1d11 100644
--- a/examples/wireless/wifi-ofdm-eht-validation.cc
+++ b/examples/wireless/wifi-ofdm-eht-validation.cc
@@ -124,8 +124,8 @@ main(int argc, char* argv[])
     std::stringstream plotExtra;
     plotExtra << "set xrange [-5:55]\n\
 set yrange [0:1]\n";
-    uint8_t lineNumber = 1;
-    const std::string colors[14] = {
+
+    const std::vector<std::string> colors{
         "green",
         "blue",
         "red",
@@ -141,10 +141,13 @@ set yrange [0:1]\n";
         "olive",
         "beige",
     };
-    for (uint32_t i = 0; i < modes.size(); i++)
+
+    NS_ASSERT_MSG(colors.size() == modes.size(), "Colors and modes vectors have different sizes");
+
+    for (std::size_t i = 0; i < modes.size(); i++)
     {
-        plotExtra << "set style line " << +lineNumber++ << " linewidth 5 linecolor rgb \""
-                  << colors[i] << "\" \n";
+        plotExtra << "set style line " << (i + 1) << " linewidth 5 linecolor rgb \"" << colors[i]
+                  << "\" \n";
     }
     plotExtra << "set style increment user";