62 lines
1.7 KiB
C++
62 lines
1.7 KiB
C++
|
#include <string>
|
||
|
|
#include <sstream>
|
||
|
|
#include <fstream>
|
||
|
|
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <stdlib.h> // getloadavg()
|
||
|
|
|
||
|
|
std::string load_string( bool use_colors = false ) {
|
||
|
|
std::ostringstream oss;
|
||
|
|
|
||
|
|
// Both apple and BSD style systems have these api calls
|
||
|
|
// Only get 3 load averages
|
||
|
|
int nelem = 3;
|
||
|
|
double averages[3];
|
||
|
|
// based on:
|
||
|
|
// http://www.opensource.apple.com/source/Libc/Libc-262/gen/getloadavg.c
|
||
|
|
if( getloadavg(averages, nelem) < 0 )
|
||
|
|
oss << "0.00 0.00 0.00"; // couldn't get averages.
|
||
|
|
else
|
||
|
|
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<float>(averages[i]) * 100 + 0.5) / 100;
|
||
|
|
oss << avg << " ";
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string load_line( oss.str() );
|
||
|
|
oss.str( "" );
|
||
|
|
|
||
|
|
if( use_colors ) {
|
||
|
|
std::ifstream stat_file( "/proc/stat" );
|
||
|
|
std::string stat_line;
|
||
|
|
std::getline( stat_file, stat_line );
|
||
|
|
|
||
|
|
// Likely does not work on BSD, but not tested
|
||
|
|
unsigned number_of_cpus = sysconf( _SC_NPROCESSORS_ONLN );
|
||
|
|
|
||
|
|
std::istringstream iss( load_line.substr( 0, 4 ) );
|
||
|
|
float recent_load;
|
||
|
|
iss >> recent_load;
|
||
|
|
// colors range from zero to twice the number of
|
||
|
|
// cpu's for the most recent load metric
|
||
|
|
|
||
|
|
unsigned load_percent = static_cast< unsigned int >(
|
||
|
|
recent_load / number_of_cpus * 0.5f * 100.0f );
|
||
|
|
|
||
|
|
if( load_percent > 100 )
|
||
|
|
load_percent = 100;
|
||
|
|
|
||
|
|
oss << load_lut[load_percent];
|
||
|
|
}
|
||
|
|
|
||
|
|
oss << load_line.substr( 0, 14 );
|
||
|
|
|
||
|
|
if( use_colors )
|
||
|
|
oss << "#[fg=default,bg=default]";
|
||
|
|
|
||
|
|
return oss.str();
|
||
|
|
}
|
||
|
|
|