the-btfash-foundation/tmux-mem-cpu-load
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix the formatting to use tabs instead of spaces, easier to read this way.

Browse Source
This commit is contained in:
Justin Crawford 2013-04-15 11:35:38 -07:00
parent 7261b34ace
commit dd3cb1e27d
1 changed files with 129 additions and 128 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

257
tmux-mem-cpu-load.cpp
View File

@ -22,169 +22,170 @@
float cpu_percentage( unsigned int cpu_usage_delay ) float cpu_percentage( unsigned int cpu_usage_delay )
{ {
std::string stat_line; std::string stat_line;
size_t line_start_pos; size_t line_start_pos;
size_t line_end_pos; size_t line_end_pos;
unsigned long long current_user; unsigned long long current_user;
unsigned long long current_system; unsigned long long current_system;
unsigned long long current_nice; unsigned long long current_nice;
unsigned long long current_idle; unsigned long long current_idle;
unsigned long long next_user; unsigned long long next_user;
unsigned long long next_system; unsigned long long next_system;
unsigned long long next_nice; unsigned long long next_nice;
unsigned long long next_idle; unsigned long long next_idle;
unsigned long long diff_user; unsigned long long diff_user;
unsigned long long diff_system; unsigned long long diff_system;
unsigned long long diff_nice; unsigned long long diff_nice;
unsigned long long diff_idle; unsigned long long diff_idle;
std::istringstream iss; std::istringstream iss;
std::ifstream stat_file("/proc/stat"); std::ifstream stat_file("/proc/stat");
std::getline(stat_file, stat_line); std::getline(stat_file, stat_line);
stat_file.close(); stat_file.close();
// skip "cpu" // skip "cpu"
line_start_pos = stat_line.find_first_not_of(" ", 3); line_start_pos = stat_line.find_first_not_of(" ", 3);
line_end_pos = stat_line.find_first_of(' ', line_start_pos); line_end_pos = stat_line.find_first_of(' ', line_start_pos);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
iss.str( stat_line.substr( line_start_pos, line_end_pos - line_start_pos ) ); iss.str( stat_line.substr( line_start_pos, line_end_pos - line_start_pos ) );
iss >> current_user >> current_nice >> current_system >> current_idle; iss >> current_user >> current_nice >> current_system >> current_idle;
iss.clear(); iss.clear();
usleep( cpu_usage_delay ); usleep( cpu_usage_delay );
stat_file.open("/proc/stat"); stat_file.open("/proc/stat");
std::getline(stat_file, stat_line); std::getline(stat_file, stat_line);
stat_file.close(); stat_file.close();
// skip "cpu" // skip "cpu"
line_start_pos = stat_line.find_first_not_of(" ", 3); line_start_pos = stat_line.find_first_not_of(" ", 3);
line_end_pos = stat_line.find_first_of(' ', line_start_pos); line_end_pos = stat_line.find_first_of(' ', line_start_pos);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1); line_end_pos = stat_line.find_first_of(' ', line_end_pos + 1);
iss.str( stat_line.substr( line_start_pos, line_end_pos - line_start_pos ) ); iss.str( stat_line.substr( line_start_pos, line_end_pos - line_start_pos ) );
iss >> next_user >> next_nice >> next_system >> next_idle; iss >> next_user >> next_nice >> next_system >> next_idle;
iss.clear(); iss.clear();
diff_user = next_user - current_user; diff_user = next_user - current_user;
diff_system = next_system - current_system; diff_system = next_system - current_system;
diff_nice = next_nice - current_nice; diff_nice = next_nice - current_nice;
diff_idle = next_idle - current_idle; diff_idle = next_idle - current_idle;
return static_cast<float>(diff_user + diff_system + diff_nice)/static_cast<float>(diff_user + diff_system + diff_nice + diff_idle)*100.0; return static_cast<float>(diff_user + diff_system + diff_nice)/static_cast<float>(diff_user + diff_system + diff_nice + diff_idle)*100.0;
} }
std::string cpu_string( unsigned int cpu_usage_delay, unsigned int graph_lines ) std::string cpu_string( unsigned int cpu_usage_delay, unsigned int graph_lines )
{ {
std::string meter( graph_lines + 2, ' ' ); std::string meter( graph_lines + 2, ' ' );
meter[0] = '['; meter[0] = '[';
meter[meter.length() - 1] = ']'; meter[meter.length() - 1] = ']';
int meter_count = 0; int meter_count = 0;
float percentage; float percentage;
std::ostringstream oss; std::ostringstream oss;
oss.precision( 1 ); oss.precision( 1 );
oss.setf( std::ios::fixed | std::ios::right ); oss.setf( std::ios::fixed | std::ios::right );
percentage = cpu_percentage( cpu_usage_delay ); percentage = cpu_percentage( cpu_usage_delay );
float meter_step = 99.9 / graph_lines; float meter_step = 99.9 / graph_lines;
meter_count = 1; meter_count = 1;
while(meter_count*meter_step < percentage) while(meter_count*meter_step < percentage)
{ {
meter[meter_count] = '|'; meter[meter_count] = '|';
meter_count++; meter_count++;
} }
oss << meter; oss << meter;
oss.width( 5 ); oss.width( 5 );
oss << percentage; oss << percentage;
oss << "%"; oss << "%";
return oss.str(); return oss.str();
} }
std::string mem_string() std::string mem_string()
{ {
unsigned int total_mem; unsigned int total_mem;
unsigned int used_mem; unsigned int used_mem;
unsigned int unused_mem; unsigned int unused_mem;
size_t line_start_pos; size_t line_start_pos;
size_t line_end_pos; size_t line_end_pos;
std::istringstream iss; std::istringstream iss;
std::ostringstream oss; std::ostringstream oss;
std::string mem_line; std::string mem_line;
std::ifstream meminfo_file( "/proc/meminfo" ); std::ifstream meminfo_file( "/proc/meminfo" );
std::getline( meminfo_file, mem_line ); std::getline( meminfo_file, mem_line );
line_start_pos = mem_line.find_first_of( ':' ); line_start_pos = mem_line.find_first_of( ':' );
line_start_pos++; line_start_pos++;
line_end_pos = mem_line.find_first_of( 'k' ); line_end_pos = mem_line.find_first_of( 'k' );
iss.str( mem_line.substr( line_start_pos, line_end_pos - line_start_pos ) ); iss.str( mem_line.substr( line_start_pos, line_end_pos - line_start_pos ) );
iss >> total_mem; iss >> total_mem;
used_mem = total_mem; used_mem = total_mem;
for( unsigned int i = 0; i < 3; i++ ) for( unsigned int i = 0; i < 3; i++ )
{ {
std::getline( meminfo_file, mem_line ); std::getline( meminfo_file, mem_line );
line_start_pos = mem_line.find_first_of( ':' ); line_start_pos = mem_line.find_first_of( ':' );
line_start_pos++; line_start_pos++;
line_end_pos = mem_line.find_first_of( 'k' ); line_end_pos = mem_line.find_first_of( 'k' );
iss.str( mem_line.substr( line_start_pos, line_end_pos - line_start_pos ) ); iss.str( mem_line.substr( line_start_pos, line_end_pos - line_start_pos ) );
iss >> unused_mem; iss >> unused_mem;
used_mem -= unused_mem; used_mem -= unused_mem;
} }
meminfo_file.close(); meminfo_file.close();
oss << used_mem / 1024 << '/' << total_mem / 1024 << "MB"; oss << used_mem / 1024 << '/' << total_mem / 1024 << "MB";
return oss.str(); return oss.str();
} }
std::string load_string() std::string load_string()
{ {
std::ifstream loadavg_file( "/proc/loadavg" ); std::ifstream loadavg_file( "/proc/loadavg" );
std::string load_line; std::string load_line;
std::getline( loadavg_file, load_line ); std::getline( loadavg_file, load_line );
loadavg_file.close(); loadavg_file.close();
return load_line.substr( 0, 14 ); return load_line.substr( 0, 14 );
} }
int main(int argc, char** argv) int main(int argc, char** argv)
{ {
unsigned int cpu_usage_delay = 900000; unsigned int cpu_usage_delay = 900000;
unsigned int graph_lines = 10; unsigned int graph_lines = 10;
try try
{ {
std::istringstream iss; std::istringstream iss;
iss.exceptions ( std::ifstream::failbit | std::ifstream::badbit ); iss.exceptions ( std::ifstream::failbit | std::ifstream::badbit );
if( argc > 1 ) if( argc > 1 )
{ {
iss.str( argv[1] ); iss.str( argv[1] );
unsigned int status_interval; unsigned int status_interval;
iss >> status_interval; iss >> status_interval;
cpu_usage_delay = status_interval * 1000000 - 100000; cpu_usage_delay = status_interval * 1000000 - 100000;
} }
if( argc > 2 )
{
iss.str( argv[2] );
iss.clear();
iss >> graph_lines;
}
}
catch(const std::exception &e)
{
std::cerr << "Usage: " << argv[0] << " [tmux_status-interval(seconds)] [graph lines]" << std::endl;
return 1;
}
std::cout << mem_string() << ' ' << cpu_string( cpu_usage_delay, graph_lines ) << ' ' << load_string(); if( argc > 2 )
{
iss.str( argv[2] );
iss.clear();
iss >> graph_lines;
}
}
catch(const std::exception &e)
{
std::cerr << "Usage: " << argv[0] << " [tmux_status-interval(seconds)] [graph lines]" << std::endl;
return 1;
}
return 0; std::cout << mem_string() << ' ' << cpu_string( cpu_usage_delay, graph_lines ) << ' ' << load_string();
return 0;
} }