mirror of
https://github.com/thewtex/tmux-mem-cpu-load.git
synced 2026-09-10 07:16:16 -04:00
Corrections in CMakeLists.txt
This commit is contained in:
l0ner
Pawel "l0ner" Soltys
parent
94f455bedf
commit
d166539cd7
3
.gitignore
vendored
3
.gitignore
vendored
|
|
@ -16,6 +16,9 @@ cmake_install.cmake
|
|||
CMakeCache.txt
|
||||
install_manifest.txt
|
||||
version.h
|
||||
CTestTestfile.cmake
|
||||
tmux-mem-cpu-load
|
||||
DartConfiguration.tcl
|
||||
|
||||
# Packages #
|
||||
############
|
||||
|
|
|
|||
|
|
@ -4,6 +4,15 @@ if(COMMAND cmake_policy)
|
|||
endif(COMMAND cmake_policy)
|
||||
|
||||
project( tmux-mem-cpu-load )
|
||||
set(tmux-mem-cpu-load_VERSION_MAJOR 2)
|
||||
set(tmux-mem-cpu-load_VERSION_MINOR 3)
|
||||
set(tmux-mem-cpu-load_VERSION_PATCH 0)
|
||||
|
||||
# generate header file to handle version
|
||||
configure_file(
|
||||
"${PROJECT_SOURCE_DIR}/version.h.in"
|
||||
"${PROJECT_SOURCE_DIR}/version.h"
|
||||
)
|
||||
|
||||
if(NOT CMAKE_BUILD_TYPE)
|
||||
set(CMAKE_BUILD_TYPE MinSizeRel CACHE STRING
|
||||
|
|
@ -11,10 +20,43 @@ if(NOT CMAKE_BUILD_TYPE)
|
|||
FORCE)
|
||||
endif(NOT CMAKE_BUILD_TYPE)
|
||||
|
||||
add_executable(tmux-mem-cpu-load tmux-mem-cpu-load.cpp)
|
||||
install(TARGETS tmux-mem-cpu-load
|
||||
RUNTIME DESTINATION bin
|
||||
)
|
||||
# detect system type
|
||||
if(CMAKE_SYSTEM_NAME MATCHES "Linux")
|
||||
message( "Linux detected")
|
||||
SET( METER_SOURCES "linux/memory.cc" "linux/cpu.cc" "linux/load.cc" )
|
||||
ELSEIF(CMAKE_SYSTEM_NAME MATCHES "Darwin")
|
||||
message( "Darwin detected")
|
||||
SET( METER_SOURCES "osx/memory.cc" "osx/cpu.cc" "osx/load.cc" )
|
||||
# Mac OS X source setting will go here
|
||||
ELSEIF(CMAKE_SYSTEM_NAME MATCHES "FreeBSD")
|
||||
message( "FreeBSD detected")
|
||||
# FreeBSD STUFF HERE
|
||||
ELSEIF(CMAKE_SYSTEM_NAME MATCHES "OpenBSD")
|
||||
message( "OpenBSD detected")
|
||||
# OpenBSD Stuff Here
|
||||
ELSE(CMAKE_SYSTEM_NAME MATCHES "Linux")
|
||||
message( FATAL_ERROR "Cannot be compiled on this system" )
|
||||
endif()
|
||||
|
||||
# Search for boost
|
||||
# will be needed later
|
||||
set(Boost_USE_STATIC_LIBS OFF)
|
||||
set(Boost_USE_MULTITHREADED ON)
|
||||
set(Boost_SUE_STATIC_RUTIME OFF)
|
||||
find_package(Boost 1.52.0 REQUIRED COMPONENTS filesystem program_options system)
|
||||
|
||||
# compiler flags
|
||||
SET( GCC_COVERAGE_COMPILE_FLAGS "-std=c++11 " )
|
||||
SET( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${GCC_COVERAGE_COMPILE_FLAGS}" )
|
||||
|
||||
# add binary tree so we find version.h
|
||||
include_directories("${PROJECT_BINARY_DIR}" ${Boost_INCLUDE_DIRS})
|
||||
|
||||
add_executable(tmux-mem-cpu-load tmux-mem-cpu-load.cpp ${METER_SOURCES})
|
||||
target_link_libraries(tmux-mem-cpu-load ${Boost_LIBRARIES})
|
||||
install(TARGETS tmux-mem-cpu-load
|
||||
RUNTIME DESTINATION bin
|
||||
)
|
||||
|
||||
include( CTest )
|
||||
if( BUILD_TESTING )
|
||||
|
|
|
|||
|
|
@ -23,302 +23,73 @@
|
|||
#include <cmath> // for floorf
|
||||
#include <cstdlib> // EXIT_SUCCESS
|
||||
|
||||
// Apple specific.
|
||||
// Tmux color lookup tables for the different metrics.
|
||||
#include "luts.h"
|
||||
|
||||
#if defined(__APPLE__) && defined(__MACH__)
|
||||
// Mach kernel includes for getting memory and CPU statistics
|
||||
# include <mach/vm_statistics.h>
|
||||
# include <mach/processor_info.h>
|
||||
# include <mach/mach_types.h>
|
||||
# include <mach/mach_init.h>
|
||||
# include <mach/mach_host.h>
|
||||
# include <mach/host_info.h>
|
||||
# include <mach/mach_error.h>
|
||||
# include <mach/vm_map.h>
|
||||
# include <mach/mach.h>
|
||||
# include <sys/sysctl.h> // for sysctl
|
||||
# include <sys/types.h> // for integer types
|
||||
// Apple osx system
|
||||
#include "osx/cpu.h"
|
||||
#include "osx/memory.h"
|
||||
#include "osx/load.h"
|
||||
#elif defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__)
|
||||
// BSD system
|
||||
// TODO: Includes and *BSD support
|
||||
#define BSD_BASED 1
|
||||
// include _get_cpu_percentage (see osx/cpu.cc)
|
||||
// include cpu_percentage (see osx/cpu.cc)
|
||||
#else
|
||||
// assume linux system
|
||||
#include "linux/cpu.h"
|
||||
#include "linux/memory.h"
|
||||
#include "linux/load.h"
|
||||
#endif
|
||||
|
||||
// if we are on a BSD system
|
||||
#if defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__)
|
||||
// TODO: Includes and *BSD support
|
||||
# define BSD_BASED 1
|
||||
#endif
|
||||
|
||||
// Function declarations for global use.
|
||||
// Function declarations.
|
||||
// TODO: those should stay in separate headers
|
||||
// LINUX: DONE/partial
|
||||
// OSX: DONE/partial
|
||||
// BSD: TODO
|
||||
float cpu_percentage( unsigned int cpu_usage_delay );
|
||||
std::string mem_string();
|
||||
std::string load_string();
|
||||
|
||||
// This function is platform independent and therefore can be placed at the top
|
||||
std::string cpu_string( unsigned int cpu_usage_delay, unsigned int graph_lines )
|
||||
{
|
||||
std::string meter( graph_lines + 2, ' ' );
|
||||
meter[0] = '[';
|
||||
meter[meter.length() - 1] = ']';
|
||||
int meter_count = 0;
|
||||
float percentage;
|
||||
std::ostringstream oss;
|
||||
oss.precision( 1 );
|
||||
oss.setf( std::ios::fixed | std::ios::right );
|
||||
|
||||
percentage = cpu_percentage( cpu_usage_delay );
|
||||
float meter_step = 99.9 / graph_lines;
|
||||
meter_count = 1;
|
||||
while(meter_count*meter_step < percentage)
|
||||
{
|
||||
meter[meter_count] = '|';
|
||||
meter_count++;
|
||||
}
|
||||
|
||||
oss << meter;
|
||||
oss.width( 5 );
|
||||
oss << percentage;
|
||||
oss << "%";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
#if defined(BSD_BASED) || (defined(__APPLE__) && defined(__MACH__))
|
||||
// OSX or BSD based system, use BSD APIs instead
|
||||
|
||||
#if defined(__APPLE__) && defined(__MACH__)
|
||||
std::string mem_string()
|
||||
{
|
||||
// These values are in bytes
|
||||
int64_t total_mem;
|
||||
int64_t used_mem;
|
||||
int64_t unused_mem;
|
||||
// blah
|
||||
vm_size_t page_size;
|
||||
mach_port_t mach_port;
|
||||
mach_msg_type_number_t count;
|
||||
vm_statistics_data_t vm_stats;
|
||||
std::ostringstream oss;
|
||||
|
||||
// Get total physical memory
|
||||
int mib[2];
|
||||
mib[0] = CTL_HW;
|
||||
mib[1] = HW_MEMSIZE;
|
||||
size_t length = sizeof(int64_t);
|
||||
sysctl(mib, 2, &total_mem, &length, NULL, 0);
|
||||
|
||||
mach_port = mach_host_self();
|
||||
count = sizeof(vm_stats) / sizeof(natural_t);
|
||||
if (KERN_SUCCESS == host_page_size(mach_port, &page_size) &&
|
||||
KERN_SUCCESS == host_statistics(mach_port, HOST_VM_INFO, (host_info_t)&vm_stats, &count))
|
||||
{
|
||||
unused_mem = (int64_t)vm_stats.free_count * (int64_t)page_size;
|
||||
|
||||
used_mem = ((int64_t)vm_stats.active_count +
|
||||
(int64_t)vm_stats.inactive_count +
|
||||
(int64_t)vm_stats.wire_count) * (int64_t)page_size;
|
||||
}
|
||||
|
||||
oss << used_mem / 1024 / 1024 << '/' << total_mem / 1024 / 1024 << "MB";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
// See: http://www.opensource.apple.com/source/xnu/xnu-201/osfmk/mach/host_info.h
|
||||
// and: http://web.mit.edu/darwin/src/modules/xnu/osfmk/man/
|
||||
host_cpu_load_info_data_t _get_cpu_percentage()
|
||||
{
|
||||
kern_return_t error;
|
||||
mach_msg_type_number_t count;
|
||||
host_cpu_load_info_data_t r_load;
|
||||
mach_port_t mach_port;
|
||||
|
||||
count = HOST_CPU_LOAD_INFO_COUNT;
|
||||
mach_port = mach_host_self();
|
||||
error = host_statistics(mach_port, HOST_CPU_LOAD_INFO, (host_info_t)&r_load, &count);
|
||||
if (error != KERN_SUCCESS)
|
||||
return host_cpu_load_info_data_t();
|
||||
|
||||
return r_load;
|
||||
}
|
||||
|
||||
float cpu_percentage( unsigned int cpu_usage_delay )
|
||||
{
|
||||
// Get the load times from the XNU kernel
|
||||
host_cpu_load_info_data_t load1 = _get_cpu_percentage();
|
||||
usleep(cpu_usage_delay);
|
||||
host_cpu_load_info_data_t load2 = _get_cpu_percentage();
|
||||
|
||||
// Current load times
|
||||
unsigned long long current_user = load1.cpu_ticks[CPU_STATE_USER];
|
||||
unsigned long long current_system = load1.cpu_ticks[CPU_STATE_SYSTEM];
|
||||
unsigned long long current_nice = load1.cpu_ticks[CPU_STATE_NICE];
|
||||
unsigned long long current_idle = load1.cpu_ticks[CPU_STATE_IDLE];
|
||||
// Next load times
|
||||
unsigned long long next_user = load2.cpu_ticks[CPU_STATE_USER];
|
||||
unsigned long long next_system = load2.cpu_ticks[CPU_STATE_SYSTEM];
|
||||
unsigned long long next_nice = load2.cpu_ticks[CPU_STATE_NICE];
|
||||
unsigned long long next_idle = load2.cpu_ticks[CPU_STATE_IDLE];
|
||||
// Difference between the two
|
||||
unsigned long long diff_user = next_user - current_user;
|
||||
unsigned long long diff_system = next_system - current_system;
|
||||
unsigned long long diff_nice = next_nice - current_nice;
|
||||
unsigned long long 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;
|
||||
}
|
||||
|
||||
#else // APPLE
|
||||
|
||||
// BSD type systems (FreeBSD/NetBSD/OpenBSD)
|
||||
// Incomplete
|
||||
float cpu_percentage( unsigned int cpu_usage_delay )
|
||||
{
|
||||
return 0.0f;
|
||||
}
|
||||
|
||||
std::string mem_string()
|
||||
{
|
||||
return "0/0 MB";
|
||||
}
|
||||
|
||||
#endif // APPLE
|
||||
|
||||
// Both apple and BSD style systems have these api calls
|
||||
std::string load_string()
|
||||
{
|
||||
std::stringstream ss;
|
||||
// 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)
|
||||
ss << "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;
|
||||
ss << avg << " ";
|
||||
}
|
||||
}
|
||||
|
||||
return ss.str();
|
||||
}
|
||||
|
||||
#else // APPLE || BSD_BASED
|
||||
|
||||
// Linux based system, if it's windows, too much work, they can deal with their compile errors - CMD sucks anyway.
|
||||
std::string mem_string()
|
||||
{
|
||||
std::string cpu_string( unsigned int cpu_usage_delay,
|
||||
unsigned int graph_lines, bool use_colors = false ) {
|
||||
std::string meter( graph_lines + 2, ' ' );
|
||||
meter[0] = '[';
|
||||
meter[meter.length() - 1] = ']';
|
||||
int meter_count = 0;
|
||||
float percentage;
|
||||
std::ostringstream oss;
|
||||
oss.precision( 1 );
|
||||
oss.setf( std::ios::fixed | std::ios::right );
|
||||
|
||||
#if defined(BSD_BASED) || (defined(__APPLE__) && defined(__MACH__))
|
||||
// Both apple and BSD style systems have these api calls
|
||||
percentage = cpu_percentage( cpu_usage_delay );
|
||||
float meter_step = 99.9 / graph_lines;
|
||||
meter_count = 1;
|
||||
|
||||
// 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.
|
||||
while(meter_count*meter_step < percentage) {
|
||||
meter[meter_count] = '|';
|
||||
meter_count++;
|
||||
}
|
||||
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( "" );
|
||||
|
||||
#else // Linux
|
||||
std::ifstream loadavg_file( "/proc/loadavg" );
|
||||
std::string load_line;
|
||||
std::getline( loadavg_file, load_line );
|
||||
loadavg_file.close();
|
||||
if( use_colors )
|
||||
oss << cpu_percentage_lut[static_cast<unsigned int>( percentage )];
|
||||
|
||||
oss << meter;
|
||||
oss.width( 5 );
|
||||
oss << percentage;
|
||||
oss << "%";
|
||||
if( use_colors )
|
||||
oss << "#[fg=default,bg=default]";
|
||||
|
||||
#endif // platform
|
||||
|
||||
float cpu_percentage( unsigned int cpu_usage_delay )
|
||||
{
|
||||
std::string stat_line;
|
||||
size_t line_start_pos;
|
||||
size_t line_end_pos;
|
||||
unsigned long long current_user;
|
||||
unsigned long long current_system;
|
||||
unsigned long long current_nice;
|
||||
unsigned long long current_idle;
|
||||
unsigned long long next_user;
|
||||
unsigned long long next_system;
|
||||
unsigned long long next_nice;
|
||||
unsigned long long next_idle;
|
||||
unsigned long long diff_user;
|
||||
unsigned long long diff_system;
|
||||
unsigned long long diff_nice;
|
||||
unsigned long long diff_idle;
|
||||
std::istringstream iss;
|
||||
|
||||
std::ifstream stat_file("/proc/stat");
|
||||
std::getline(stat_file, stat_line);
|
||||
stat_file.close();
|
||||
|
||||
// skip "cpu"
|
||||
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_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 >> current_user >> current_nice >> current_system >> current_idle;
|
||||
iss.clear();
|
||||
|
||||
usleep( cpu_usage_delay );
|
||||
|
||||
stat_file.open("/proc/stat");
|
||||
std::getline(stat_file, stat_line);
|
||||
stat_file.close();
|
||||
|
||||
// skip "cpu"
|
||||
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_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 >> next_user >> next_nice >> next_system >> next_idle;
|
||||
iss.clear();
|
||||
|
||||
diff_user = next_user - current_user;
|
||||
diff_system = next_system - current_system;
|
||||
diff_nice = next_nice - current_nice;
|
||||
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 oss.str();
|
||||
}
|
||||
|
||||
std::string load_string()
|
||||
{
|
||||
std::ifstream loadavg_file( "/proc/loadavg" );
|
||||
std::string load_line;
|
||||
std::getline( loadavg_file, load_line );
|
||||
loadavg_file.close();
|
||||
|
||||
return load_line.substr( 0, 14 );
|
||||
}
|
||||
|
||||
#endif // BSD
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
int main(int argc, char** argv) {
|
||||
unsigned int cpu_usage_delay = 900000;
|
||||
int graph_lines = 10;
|
||||
bool use_colors = false;
|
||||
try
|
||||
{
|
||||
try {
|
||||
std::istringstream iss;
|
||||
iss.exceptions ( std::ifstream::failbit | std::ifstream::badbit );
|
||||
std::string current_arg;
|
||||
|
|
|
|||
Loading…
Reference in a new issue