conky-config/module/Processor.lua

256 lines
6.0 KiB
Lua

local Widget = require 'Widget'
local Arc = require 'Arc'
local CompoundDial = require 'CompoundDial'
local CriticalText = require 'CriticalText'
local Text = require 'Text'
local Line = require 'Line'
local LabelPlot = require 'LabelPlot'
local Table = require 'Table'
local util = require 'util'
local schema = require 'default_patterns'
local CORETEMP_PATH = '/sys/devices/platform/coretemp.0/hwmon/hwmon%i/%s'
local MODULE_Y = 636
local NUM_PHYSICAL_CORES = 4
local NUM_PHYSICAL_CORE_THREADS = 2
local NUM_ROWS = 5
local TABLE_CONKY = {{}, {}, {}}
for r = 1, NUM_ROWS do
TABLE_CONKY[1][r] = '${top name '..r..'}'
TABLE_CONKY[2][r] = '${top pid '..r..'}'
TABLE_CONKY[3][r] = '${top cpu '..r..'}'
end
--construction params
local DIAL_INNER_RADIUS = 30
local DIAL_OUTER_RADIUS = 42
local DIAL_SPACING = 1
local TEXT_Y_OFFSET = 15
local SEPARATOR_SPACING = 20
local PLOT_SECTION_BREAK = 23
local PLOT_HEIGHT = 56
local TABLE_SECTION_BREAK = 20
local TABLE_HEIGHT = 114
local CREATE_CORE = function(cores, id, x, y)
local conky_threads = {}
for c = 0, NUM_PHYSICAL_CORES * NUM_PHYSICAL_CORE_THREADS - 1 do
if util.read_file('/sys/devices/system/cpu/cpu'..c..'/topology/core_id', nil, '*n') == id then
table.insert(conky_threads, '${cpu cpu'..c..'}')
end
end
local hwmon_index = -1
while util.read_file(string.format(CORETEMP_PATH, hwmon_index, 'name'), nil, '*l') ~= 'coretemp' do
hwmon_index = hwmon_index + 1
end
cores[id +1] = {
dials = Widget.CompoundDial{
x = x,
y = y,
inner_radius = DIAL_INNER_RADIUS,
outer_radius = DIAL_OUTER_RADIUS,
spacing = DIAL_SPACING,
num_dials = NUM_PHYSICAL_CORE_THREADS,
critical_limit = '>0.8'
},
inner_ring = Widget.Arc{
x = x,
y = y,
radius = DIAL_INNER_RADIUS - 2,
theta0 = 0,
theta1 = 360
},
coretemp_text = Widget.CriticalText{
x = x,
y = y,
x_align = 'center',
y_align = 'center',
append_end = '°C',
critical_limit = '>90'
},
coretemp_path = string.format(CORETEMP_PATH, hwmon_index, 'temp'..(id + 2)..'_input'),
conky_threads = conky_threads
}
end
local header = Widget.Header{
x = G_DIMENSIONS_.LEFT_X,
y = MODULE_Y,
width = G_DIMENSIONS_.SECTION_WIDTH,
header = "PROCESSOR"
}
local HEADER_BOTTOM_Y = header.bottom_y
--we assume that this cpu has 4 physical cores with 2 logical each
local cores = {}
for c = 0, NUM_PHYSICAL_CORES - 1 do
local dial_x = G_DIMENSIONS_.LEFT_X + DIAL_OUTER_RADIUS + (G_DIMENSIONS_.SECTION_WIDTH - 2 * DIAL_OUTER_RADIUS) * c / 3
CREATE_CORE(cores, c, dial_x, HEADER_BOTTOM_Y + DIAL_OUTER_RADIUS)
end
local RIGHT_X = G_DIMENSIONS_.LEFT_X + G_DIMENSIONS_.SECTION_WIDTH
local PROCESS_Y = HEADER_BOTTOM_Y + DIAL_OUTER_RADIUS * 2 + PLOT_SECTION_BREAK
local process = {
labels = Widget.Text{
x = G_DIMENSIONS_.LEFT_X,
y = PROCESS_Y,
text = 'R | S | D | T | Z'
},
values = Widget.Text{
x = RIGHT_X,
y = PROCESS_Y,
x_align = 'right',
text_color = schema.blue,
text = '<R.S.D.T.Z>'
}
}
local SEP_Y = PROCESS_Y + SEPARATOR_SPACING
local separator = Widget.Line{
p1 = {x = G_DIMENSIONS_.LEFT_X, y = SEP_Y},
p2 = {x = RIGHT_X, y = SEP_Y}
}
local LOAD_Y = SEP_Y + SEPARATOR_SPACING
local total_load = {
label = Widget.Text{
x = G_DIMENSIONS_.LEFT_X,
y = LOAD_Y,
text = 'Total Load'
},
value = Widget.CriticalText{
x = RIGHT_X,
y = LOAD_Y,
x_align = 'right',
append_end = '%',
critical_limit = '>80'
}
}
local PLOT_Y = LOAD_Y + PLOT_SECTION_BREAK
local plot = Widget.LabelPlot{
x = G_DIMENSIONS_.LEFT_X,
y = PLOT_Y,
width = G_DIMENSIONS_.SECTION_WIDTH,
height = PLOT_HEIGHT
}
local TABLE_Y = PLOT_Y + PLOT_HEIGHT + TABLE_SECTION_BREAK
local tbl = Widget.Table{
x = G_DIMENSIONS_.LEFT_X,
y = TABLE_Y,
width = G_DIMENSIONS_.SECTION_WIDTH,
height = TABLE_HEIGHT,
num_rows = NUM_ROWS,
'Name',
'PID',
'CPU (%)'
}
local __update = function(cr)
local conky = util.conky
local char_count = util.char_count
local sum = 0
for c = 1, NUM_PHYSICAL_CORES do
local core = cores[c]
local conky_threads = core.conky_threads
for t = 1, NUM_PHYSICAL_CORE_THREADS do
local percent = util.conky_numeric(conky_threads[t]) * 0.01
CompoundDial.set(core.dials, t, percent)
sum = sum + percent
end
CriticalText.set(core.coretemp_text, cr, util.round(0.001 * util.read_file(core.coretemp_path, nil, '*n')))
end
local process_glob = util.execute_cmd('ps -A -o s')
--subtract one from running b/c ps will always be "running"
Text.set(process.values, cr, (char_count(process_glob, 'R') - 1)..' | '..
char_count(process_glob, 'S')..' | '..
char_count(process_glob, 'D')..' | '..
char_count(process_glob, 'T')..' | '..
char_count(process_glob, 'Z'))
local load_percent = util.round(sum / NUM_PHYSICAL_CORES / NUM_PHYSICAL_CORE_THREADS, 2)
CriticalText.set(total_load.value, cr, load_percent * 100)
LabelPlot.update(plot, load_percent)
for c = 1, 3 do
local column = TABLE_CONKY[c]
for r = 1, NUM_ROWS do
Table.set(tbl, cr, c, r, conky(column[r], '(%S+)'))
end
end
end
Widget = nil
schema = nil
MODULE_Y = nil
DIAL_INNER_RADIUS = nil
DIAL_OUTER_RADIUS = nil
DIAL_SPACING = nil
TEXT_Y_OFFSET = nil
SEPARATOR_SPACING = nil
PLOT_SECTION_BREAK = nil
PLOT_HEIGHT = nil
TABLE_SECTION_BREAK = nil
TABLE_HEIGHT = nil
CREATE_CORE = nil
HEADER_BOTTOM_Y = nil
LOAD_Y = nil
RIGHT_X = nil
SEP_Y = nil
PROCESS_Y = nil
PLOT_Y = nil
TABLE_Y = nil
local draw = function(cr, current_interface)
__update(cr)
if current_interface == 0 then
Text.draw(header.text, cr)
Line.draw(header.underline, cr)
for c = 1, NUM_PHYSICAL_CORES do
local core = cores[c]
CompoundDial.draw(core.dials, cr)
Arc.draw(core.inner_ring, cr)
CriticalText.draw(core.coretemp_text, cr)
end
Text.draw(process.labels, cr)
Text.draw(process.values, cr)
Line.draw(separator, cr)
Text.draw(total_load.label, cr)
CriticalText.draw(total_load.value, cr)
LabelPlot.draw(plot, cr)
Table.draw(tbl, cr)
end
end
return draw