add color helper code

Author: rdementi

src/pcm.cpp

@@ -89,6 +89,7 @@ void print_help(const string & prog_name)
     cout << "  -yc   | --yescores  | /yc          => enable specific cores to output\n";
     cout << "  -ns   | --nosockets | /ns          => hide socket related output\n";
     cout << "  -nsys | --nosystem  | /nsys        => hide system related output\n";
+    cout << "  --color                            => use ASCII colors\n";
     cout << "  -csv[=file.csv] | /csv[=file.csv]  => output compact CSV format to screen or\n"
         << "                                        to a file, in case filename is provided\n"
         << "                                        the format used is documented here: https://www.intel.com/content/www/us/en/developer/articles/technical/intel-pcm-column-names-decoder-ring.html\n";
@@ -169,10 +170,43 @@ void print_output(PCM * m,
     const bool show_partial_core_output,
     const bool show_socket_output,
     const bool show_system_output,
-    const int metricVersion
+    const int metricVersion,
+    const bool color
     )
 {
     cout << "\n";
+    auto setColor = [&color](const char * colorStr)
+    {
+        return color ? colorStr : "";
+    };
+    std::vector<const char *> colorTable = {
+            ASCII_GREEN,
+            ASCII_YELLOW,
+            ASCII_MAGENTA,
+            ASCII_CYAN,
+            ASCII_BRIGHT_GREEN,
+            ASCII_BRIGHT_YELLOW,
+            ASCII_BRIGHT_BLUE,
+            ASCII_BRIGHT_MAGENTA,
+            ASCII_BRIGHT_CYAN,
+            ASCII_BRIGHT_WHITE
+    };
+    size_t currentColor = 0;
+    auto setNextColor = [&setColor,&currentColor,colorTable]()
+    {
+        const auto result = setColor(colorTable[currentColor++]);
+        if (currentColor == colorTable.size())
+        {
+            currentColor = 0;
+        }
+        return result;
+    };
+    auto resetColor = [&setColor, &currentColor]()
+    {
+        currentColor = 0;
+        return setColor(ASCII_RESET_COLOR);
+    };
+
     switch (metricVersion)
     {
         case 2:
@@ -1292,6 +1326,7 @@ int mainThrows(int argc, char * argv[])
     bool disable_JKT_workaround = false; // as per http://software.intel.com/en-us/articles/performance-impact-when-sampling-certain-llc-events-on-snb-ep-with-vtune
     bool enforceFlush = false;
     int metricVersion = 2;
+    bool color = false;
 
     parsePID(argc, argv, pid);
 
@@ -1371,6 +1406,11 @@ int mainThrows(int argc, char * argv[])
             show_system_output = false;
             continue;
         }
+        else if (check_argument_equals(*argv, {"--color"}))
+        {
+            color = true;
+            continue;
+        }
         else if (check_argument_equals(*argv, {"-csv", "/csv"}))
         {
             csv_output = true;
@@ -1524,7 +1564,8 @@ int mainThrows(int argc, char * argv[])
             cpu_model, show_core_output, show_partial_core_output, show_socket_output, show_system_output);
         else
             print_output(m, cstates1, cstates2, sktstate1, sktstate2, ycores, sstate1, sstate2,
-            cpu_model, show_core_output, show_partial_core_output, show_socket_output, show_system_output, metricVersion);
+                cpu_model, show_core_output, show_partial_core_output, show_socket_output, show_system_output,
+                metricVersion, color);
 
         std::swap(sstate1, sstate2);
         std::swap(sktstate1, sktstate2);

src/utils.h

@@ -145,6 +145,24 @@ struct null_stream : public std::streambuf
 #pragma clang diagnostic pop
 #endif
 
+constexpr const char* ASCII_BLACK = "\033[0;30m";
+constexpr const char* ASCII_RED = "\033[0;31m";
+constexpr const char* ASCII_GREEN = "\033[0;32m";
+constexpr const char* ASCII_YELLOW = "\033[0;33m";
+constexpr const char* ASCII_BLUE = "\033[0;34m";
+constexpr const char* ASCII_MAGENTA = "\033[0;35m";
+constexpr const char* ASCII_CYAN = "\033[0;36m";
+constexpr const char* ASCII_WHITE = "\033[0;37m";
+constexpr const char* ASCII_BRIGHT_BLACK = "\033[1;30m";
+constexpr const char* ASCII_BRIGHT_RED = "\033[1;31m";
+constexpr const char* ASCII_BRIGHT_GREEN = "\033[1;32m";
+constexpr const char* ASCII_BRIGHT_YELLOW = "\033[1;33m";
+constexpr const char* ASCII_BRIGHT_BLUE = "\033[1;34m";
+constexpr const char* ASCII_BRIGHT_MAGENTA = "\033[1;35m";
+constexpr const char* ASCII_BRIGHT_CYAN = "\033[1;36m";
+constexpr const char* ASCII_BRIGHT_WHITE = "\033[1;37m";
+constexpr const char* ASCII_RESET_COLOR = "\033[0m";
+
 template <class IntType>
 inline std::string unit_format(IntType n)
 {