From 6178ba9aa304e8a588cea2d15b2e8d71025c497b Mon Sep 17 00:00:00 2001
From: "henry.potgieter" <henry.potgieter@viscousbits.ca>
Date: Sat, 25 Mar 2023 11:07:55 -0400
Subject: [PATCH] Add option to display CPU graph as vertical bar chart

---
 README.rst      |  2 ++
 common/graph.cc | 20 +++++++++++++++++++-
 common/graph.h  |  1 +
 common/main.cc  | 21 +++++++++++++++++----
 4 files changed, 39 insertions(+), 5 deletions(-)

diff --git a/README.rst b/README.rst
index 2e8df39..566594d 100644
--- a/README.rst
+++ b/README.rst
@@ -144,6 +144,8 @@ The full usage::
           Use tmux colors in output
   -p, --powerline-left
 	  Use powerline left symbols throughout the output, enables --colors
+  -n, --nerd-graph
+	  Use NerdFont characters to render CPU graph as vertical bar chart
   -q, --powerline-right
 	  Use powerline right symbols throughout the output, enables --colors
   -i <value>, --interval <value>
diff --git a/common/graph.cc b/common/graph.cc
index a8694e5..1e37d54 100644
--- a/common/graph.cc
+++ b/common/graph.cc
@@ -18,10 +18,11 @@
 
 #include <string>
 #include <cstring>
+#include <map>
 
 #include "graph.h"
 
-std::string get_graph_by_percentage( unsigned value, unsigned len ) 
+std::string get_graph_by_percentage( unsigned value, unsigned len )
 {
   unsigned step = 0;
   std::string bars;
@@ -60,3 +61,20 @@ std::string get_graph_by_value( unsigned value, unsigned max, unsigned len )
   return bars;
 }
 
+std::string get_graph_nerd( unsigned value )
+{
+  static const std::map<unsigned, std::string> graph_chars = {
+    { 0, " " }, { 10, "▁" }, { 20, "▂" }, { 30, "▃" }, { 40, "▄" },
+    { 50, "▅" }, { 60, "▆" }, { 70, "▇" }, { 80, "█" }, { 90, "▲" }
+  };
+
+  for( auto it = graph_chars.rbegin(); it != graph_chars.rend(); ++it )
+  {
+    if( value >= it->first )
+    {
+      return it->second;
+    }
+  }
+
+  return " "; // default return in case value doesn't match map options
+}
diff --git a/common/graph.h b/common/graph.h
index 4034161..ed97eb7 100644
--- a/common/graph.h
+++ b/common/graph.h
@@ -23,5 +23,6 @@
 
 std::string get_graph_by_percentage( unsigned, unsigned len = 10 );
 std::string get_graph_by_value( unsigned, unsigned, unsigned len = 10 );
+std::string get_graph_nerd( unsigned );
 
 #endif
diff --git a/common/main.cc b/common/main.cc
index bab7597..b3eb6e3 100644
--- a/common/main.cc
+++ b/common/main.cc
@@ -37,7 +37,7 @@
 
 std::string cpu_string( CPU_MODE cpu_mode, unsigned int cpu_usage_delay, unsigned int graph_lines,
     bool use_colors = false,
-    bool use_powerline_left = false, bool use_powerline_right = false )
+    bool use_powerline_left = false, bool use_powerline_right = false, bool nerd_graph = false)
 {
 
   float percentage;
@@ -80,7 +80,13 @@ std::string cpu_string( CPU_MODE cpu_mode, unsigned int cpu_usage_delay, unsigne
     }
   }
 
-  if( graph_lines > 0)
+  if( nerd_graph )
+  {
+    oss << "▕";
+    oss << get_graph_nerd( unsigned( percentage ) );
+    oss << "▏";
+  }
+  else if( graph_lines > 0)
   {
     oss << " [";
     oss << get_graph_by_percentage( unsigned( percentage ), graph_lines );
@@ -123,6 +129,8 @@ void print_help()
     << "\tUse powerline left symbols throughout the output, enables --colors\n"
     << "-q, --powerline-right\n"
     << "\tUse powerline right symbols throughout the output, enables --colors\n"
+    << "-n, --nerd-graph\n"
+    << "\tUse NerdFont symbols to render CPU graph as vertical bar chart\n"
     << "-i <value>, --interval <value>\n"
     << "\tSet tmux status refresh interval in seconds. Default: 1 second\n"
     << "-g <value>, --graph-lines <value>\n"
@@ -146,6 +154,7 @@ int main( int argc, char** argv )
   bool use_powerline_right = false;
   MEMORY_MODE mem_mode = MEMORY_MODE_DEFAULT;
   CPU_MODE cpu_mode = CPU_MODE_DEFAULT;
+  bool nerd_graph = false;
 
   static struct option long_options[] =
   {
@@ -157,6 +166,7 @@ int main( int argc, char** argv )
     { "colors", no_argument, NULL, 'c' },
     { "powerline-left", no_argument, NULL, 'p' },
     { "powerline-right", no_argument, NULL, 'q' },
+    { "nerd-graph", no_argument, NULL, 'n' },
     { "interval", required_argument, NULL, 'i' },
     { "graph-lines", required_argument, NULL, 'g' },
     { "mem-mode", required_argument, NULL, 'm' },
@@ -167,7 +177,7 @@ int main( int argc, char** argv )
 
   int c;
   // while c != -1
-  while( (c = getopt_long( argc, argv, "hi:cpqg:m:a:t:", long_options, NULL) ) != -1 )
+  while( (c = getopt_long( argc, argv, "hi:cpqng:m:a:t:", long_options, NULL) ) != -1 )
   {
     switch( c )
     {
@@ -186,6 +196,9 @@ int main( int argc, char** argv )
         use_colors = true;
         use_powerline_right = true;
         break;
+      case 'n': // --nerd-graph
+        nerd_graph = true;
+        break;
       case 'i': // --interval, -i
         if( atoi( optarg ) < 1 )
           {
@@ -247,7 +260,7 @@ int main( int argc, char** argv )
   MemoryStatus memory_status;
   mem_status( memory_status );
   std::cout << mem_string( memory_status, mem_mode, use_colors, use_powerline_left, use_powerline_right )
-    << cpu_string( cpu_mode, cpu_usage_delay, graph_lines, use_colors, use_powerline_left, use_powerline_right )
+    << cpu_string( cpu_mode, cpu_usage_delay, graph_lines, use_colors, use_powerline_left, use_powerline_right, nerd_graph )
     << load_string( use_colors, use_powerline_left, use_powerline_right, averages_count );
 
   std::cout << std::endl;