mirror of
https://github.com/rad4day/Waybar.git
synced 2023-12-21 10:22:59 +01:00
89 lines
2.5 KiB
C++
89 lines
2.5 KiB
C++
#include "modules/cpu.hpp"
|
|
|
|
waybar::modules::Cpu::Cpu(const std::string& id, const Json::Value& config)
|
|
: ALabel(config, "{usage}%", 10)
|
|
{
|
|
label_.set_name("cpu");
|
|
if (!id.empty()) {
|
|
label_.get_style_context()->add_class(id);
|
|
}
|
|
thread_ = [this] {
|
|
dp.emit();
|
|
thread_.sleep_for(interval_);
|
|
};
|
|
}
|
|
|
|
auto waybar::modules::Cpu::update() -> void
|
|
{
|
|
// TODO: as creating dynamic fmt::arg arrays is buggy we have to calc both
|
|
auto cpu_load = getCpuLoad();
|
|
auto [cpu_usage, tooltip] = getCpuUsage();
|
|
label_.set_tooltip_text(tooltip);
|
|
label_.set_markup(fmt::format(format_,
|
|
fmt::arg("load", cpu_load), fmt::arg("usage", cpu_usage)));
|
|
}
|
|
|
|
uint16_t waybar::modules::Cpu::getCpuLoad()
|
|
{
|
|
struct sysinfo info = {0};
|
|
if (sysinfo(&info) == 0) {
|
|
float f_load = 1.f / (1u << SI_LOAD_SHIFT);
|
|
uint16_t load = info.loads[0] * f_load * 100 / get_nprocs();
|
|
return load;
|
|
}
|
|
throw std::runtime_error("Can't get Cpu load");
|
|
}
|
|
|
|
std::tuple<uint16_t, std::string> waybar::modules::Cpu::getCpuUsage()
|
|
{
|
|
if (prev_times_.empty()) {
|
|
prev_times_ = parseCpuinfo();
|
|
std::this_thread::sleep_for(chrono::milliseconds(100));
|
|
}
|
|
std::vector<std::tuple<size_t, size_t>> curr_times = parseCpuinfo();
|
|
std::string tooltip;
|
|
uint16_t usage = 0;
|
|
for (size_t i = 0; i < curr_times.size(); ++i) {
|
|
auto [curr_idle, curr_total] = curr_times[i];
|
|
auto [prev_idle, prev_total] = prev_times_[i];
|
|
const float delta_idle = curr_idle - prev_idle;
|
|
const float delta_total = curr_total - prev_total;
|
|
uint16_t tmp = 100 * (1 - delta_idle / delta_total);
|
|
if (i == 0) {
|
|
usage = tmp;
|
|
tooltip = fmt::format("Total: {}%", tmp);
|
|
} else {
|
|
tooltip = tooltip + fmt::format("\nCore{}: {}%", i - 1, tmp);
|
|
}
|
|
}
|
|
prev_times_ = curr_times;
|
|
return {usage, tooltip};
|
|
}
|
|
|
|
std::vector<std::tuple<size_t, size_t>> waybar::modules::Cpu::parseCpuinfo()
|
|
{
|
|
std::ifstream info(data_dir_);
|
|
if (!info.is_open()) {
|
|
throw std::runtime_error("Can't open " + data_dir_);
|
|
}
|
|
std::vector< std::tuple<size_t, size_t> > cpuinfo;
|
|
std::string line;
|
|
while (getline(info, line)) {
|
|
if (line.substr(0,3).compare("cpu") != 0) {
|
|
break;
|
|
}
|
|
std::stringstream sline(line.substr(5));
|
|
std::vector<size_t> times;
|
|
for (size_t time; sline >> time; times.push_back(time));
|
|
|
|
size_t idle_time = 0;
|
|
size_t total_time = 0;
|
|
if (times.size() >= 4) {
|
|
idle_time = times[3];
|
|
total_time = std::accumulate(times.begin(), times.end(), 0);
|
|
}
|
|
cpuinfo.push_back({idle_time, total_time});
|
|
}
|
|
return cpuinfo;
|
|
}
|