diff --git a/common/load.cc b/common/load.cc index 926ada8..cbc1cf0 100644 --- a/common/load.cc +++ b/common/load.cc @@ -3,6 +3,7 @@ * Copyright 2012 Matthew McCormick * Copyright 2013 Justin Crawford * Copyright 2015 Pawel 'l0ner' Soltys + * Copyright 2016 Compilenix * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -33,59 +34,64 @@ #include "powerline.h" // Load Averages -std::string load_string( bool use_colors, bool use_powerline ) +std::string load_string( bool use_colors, bool use_powerline, short num_averages ) { std::ostringstream ss; - // Get only 3 load averages - const int nelem = 3; - double averages[nelem]; + double averages[num_averages]; // based on: opensource.apple.com/source/Libc/Libc-262/gen/getloadavg.c - if( getloadavg( averages, nelem ) < 0 ) + if( num_averages > 0) { - ss << " 0.00 0.00 0.00"; // couldn't get averages. + if( getloadavg( averages, num_averages ) < 0 ) + { + ss << " 0.00 0.00 0.00"; // couldn't get averages. + } + else + { + if( use_colors ) + { + unsigned load_percent = static_cast( averages[0] / + get_cpu_count() * 0.5f * 100.0f ); + + if( load_percent > 100 ) + { + load_percent = 100; + } + powerline(ss, load_lut[load_percent], use_powerline); + } + + ss << ' '; + for( int i = 0; i < num_averages; ++i ) + { + // Round to nearest, make sure this is only a 0.00 value not a 0.0000 + float avg = floorf( static_cast( averages[i] ) * 100 + 0.5 ) / 100; + // Don't print trailing whitespace for last element + if ( i == num_averages-1 ) + { + ss << avg; + } + else + { + ss << avg << " "; + } + } + + if( use_colors ) + { + if( use_powerline ) + { + ss << ' '; + } + else + { + ss << "#[fg=default,bg=default]"; + } + } + } } else { - if( use_colors ) - { - unsigned load_percent = static_cast( averages[0] / - get_cpu_count() * 0.5f * 100.0f ); - - if( load_percent > 100 ) - { - load_percent = 100; - } - powerline(ss, load_lut[load_percent], use_powerline); - } - - ss << ' '; - for( int i = 0; i < nelem; ++i ) - { - // Round to nearest, make sure this is only a 0.00 value not a 0.0000 - float avg = floorf( static_cast( averages[i] ) * 100 + 0.5 ) / 100; - // Don't print trailing whitespace for last element - if ( i == nelem-1 ) - { - ss << avg; - } - else - { - ss << avg << " "; - } - } - - if( use_colors ) - { - if( use_powerline ) - { - ss << ' '; - } - else - { - ss << "#[fg=default,bg=default]"; - } - } + ss << (char) 0; } return ss.str(); diff --git a/common/load.h b/common/load.h index b8d50f9..17df7fa 100644 --- a/common/load.h +++ b/common/load.h @@ -2,6 +2,7 @@ * * Copyright 2012 Matthew McCormick * Copyright 2015 Pawel 'l0ner' Soltys + * Copyright 2016 Compilenix * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -21,6 +22,6 @@ #include -std::string load_string( bool use_colors = false, bool use_powerline = false ); +std::string load_string( bool use_colors = false, bool use_powerline = false, short num_averages = 3 ); #endif diff --git a/common/main.cc b/common/main.cc index 7d3f265..5d1e910 100644 --- a/common/main.cc +++ b/common/main.cc @@ -100,12 +100,15 @@ void print_help() << "\tSet how many lines should be drawn in a graph. Default: 10\n" << "-m , --mem-mode \n" << "\tSet memory display mode. 0: Default, 1: Free memory, 2: Usage percent.\n" + << "-a , --averages-count \n" + << "\tSet how many load-averages should be drawn. Default: 3\n" << endl; } int main( int argc, char** argv ) { unsigned cpu_usage_delay = 990000; + short averages_count = 3; short graph_lines = 10; // max 32767 should be enough bool use_colors = false; bool use_powerline = false; @@ -123,12 +126,13 @@ int main( int argc, char** argv ) { "interval", required_argument, NULL, 'i' }, { "graph-lines", required_argument, NULL, 'g' }, { "mem-mode", required_argument, NULL, 'm' }, + { "averages-count", required_argument, NULL, 'a' }, { 0, 0, 0, 0 } // used to handle unknown long options }; int c; // while c != -1 - while( (c = getopt_long( argc, argv, "hi:g:m:", long_options, NULL) ) != -1 ) + while( (c = getopt_long( argc, argv, "hi:g:m:a:", long_options, NULL) ) != -1 ) { switch( c ) { @@ -167,6 +171,14 @@ int main( int argc, char** argv ) } mem_mode = static_cast< MEMORY_MODE >( atoi( optarg ) ); break; + case 'a': // --averages-count, -a + if( atoi( optarg ) < 0 || atoi( optarg ) > 3 ) + { + std::cerr << "Valid averages-count arguments are: 0, 1, 2, 3\n"; + return EXIT_FAILURE; + } + averages_count = atoi( optarg ); + break; case '?': // getopt_long prints error message automatically return EXIT_FAILURE; @@ -189,7 +201,7 @@ int main( int argc, char** argv ) mem_status( memory_status ); std::cout << mem_string( memory_status, mem_mode, use_colors, use_powerline ) << cpu_string( cpu_usage_delay, graph_lines, use_colors, use_powerline ) - << load_string( use_colors, use_powerline ); + << load_string( use_colors, use_powerline, averages_count ); return EXIT_SUCCESS; }