241 lines
7.2 KiB
Lua
241 lines
7.2 KiB
Lua
local pure = require 'pure'
|
|
local i_o = require 'i_o'
|
|
local common = require 'common'
|
|
local geometry = require 'geometry'
|
|
|
|
return function(update_freq, point)
|
|
local SEPARATOR_SPACING = 20
|
|
local TEXT_SPACING = 20
|
|
local PLOT_SEC_BREAK = 20
|
|
local PLOT_HEIGHT = 56
|
|
local NA = 'N/A'
|
|
local __string_match = string.match
|
|
local __tonumber = tonumber
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- helper functions
|
|
|
|
local _from_state = function(def, f, set)
|
|
return function(s)
|
|
if s.error == false then
|
|
set(f(s))
|
|
else
|
|
set(def)
|
|
end
|
|
end
|
|
end
|
|
|
|
local _mk_plot = function(label, getter, y)
|
|
local obj = common.make_tagged_maybe_percent_timeseries(
|
|
point.x,
|
|
y,
|
|
geometry.SECTION_WIDTH,
|
|
PLOT_HEIGHT,
|
|
PLOT_SEC_BREAK,
|
|
label,
|
|
update_freq
|
|
)
|
|
local update = _from_state(
|
|
false,
|
|
getter,
|
|
pure.partial(common.tagged_maybe_percent_timeseries_set, obj)
|
|
)
|
|
local static = pure.partial(common.tagged_percent_timeseries_draw_static, obj)
|
|
local dynamic = pure.partial(common.tagged_percent_timeseries_draw_dynamic, obj)
|
|
return common.mk_acc(PLOT_HEIGHT + PLOT_SEC_BREAK, update, static, dynamic)
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- header
|
|
|
|
local mk_header = pure.partial(
|
|
common.mk_header,
|
|
'NVIDIA GRAPHICS',
|
|
geometry.SECTION_WIDTH,
|
|
point.x
|
|
)
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu status
|
|
|
|
local mk_status = function(y)
|
|
local obj = common.make_text_row(
|
|
point.x,
|
|
y,
|
|
geometry.SECTION_WIDTH,
|
|
'Status'
|
|
)
|
|
local update = function(s)
|
|
if s.error == false then
|
|
common.text_row_set(obj, 'On')
|
|
else
|
|
common.text_row_set(obj, s.error)
|
|
end
|
|
end
|
|
local static = pure.partial(common.text_row_draw_static, obj)
|
|
local dynamic = pure.partial(common.text_row_draw_dynamic, obj)
|
|
return common.mk_acc(0, update, static, dynamic)
|
|
end
|
|
|
|
local mk_sep = pure.partial(
|
|
common.mk_seperator,
|
|
geometry.SECTION_WIDTH,
|
|
point.x
|
|
)
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu temperature
|
|
|
|
local mk_temp = function(y)
|
|
local obj = common.make_threshold_text_row(
|
|
point.x,
|
|
y,
|
|
geometry.SECTION_WIDTH,
|
|
'Internal Temperature',
|
|
function(s)
|
|
if s == -1 then return NA else return string.format('%s°C', s) end
|
|
end,
|
|
80
|
|
)
|
|
local update = _from_state(
|
|
-1,
|
|
function(s) return __tonumber(s.temp_reading) end,
|
|
pure.partial(common.threshold_text_row_set, obj)
|
|
)
|
|
local static = pure.partial(common.threshold_text_row_draw_static, obj)
|
|
local dynamic = pure.partial(common.threshold_text_row_draw_dynamic, obj)
|
|
return common.mk_acc(0, update, static, dynamic)
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu clock speeds
|
|
|
|
local mk_clock = function(y)
|
|
local obj = common.make_text_rows(
|
|
point.x,
|
|
y,
|
|
geometry.SECTION_WIDTH,
|
|
TEXT_SPACING,
|
|
{'GPU Clock Speed', 'memory Clock Speed'}
|
|
)
|
|
local update = function(s)
|
|
if s.error == false then
|
|
common.text_rows_set(obj, 1, s.gpu_frequency..' Mhz')
|
|
common.text_rows_set(obj, 2, s.memory_frequency..' Mhz')
|
|
else
|
|
common.text_rows_set(obj, 1, NA)
|
|
common.text_rows_set(obj, 2, NA)
|
|
end
|
|
end
|
|
local static = pure.partial(common.text_rows_draw_static, obj)
|
|
local dynamic = pure.partial(common.text_rows_draw_dynamic, obj)
|
|
return common.mk_acc(TEXT_SPACING, update, static, dynamic)
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu utilization plot
|
|
|
|
local mk_gpu_util = pure.partial(
|
|
_mk_plot,
|
|
'GPU utilization',
|
|
function(s) return s.gpu_utilization end
|
|
)
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu memory consumption plot
|
|
|
|
local mk_mem_util = pure.partial(
|
|
_mk_plot,
|
|
'Memory utilization',
|
|
function(s) return s.used_memory / s.total_memory * 100 end
|
|
)
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- gpu video utilization plot
|
|
|
|
local mk_vid_util = pure.partial(
|
|
_mk_plot,
|
|
'Video utilization',
|
|
function(s) return s.vid_utilization end
|
|
)
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- nvidia state
|
|
|
|
-- vars to process the nv settings glob
|
|
--
|
|
-- glob will be of the form:
|
|
-- <used_mem>
|
|
-- <total_mem>
|
|
-- <temp>
|
|
-- <gpu_freq>,<mem_freq>
|
|
-- graphics=<gpu_util>, memory=<mem_util>, video=<vid_util>, PCIe=<pci_util>
|
|
local NV_QUERY = 'nvidia-settings -t'..
|
|
' -q UsedDedicatedGPUmemory'..
|
|
' -q TotalDedicatedGPUmemory'..
|
|
' -q ThermalSensorReading'..
|
|
' -q [gpu:0]/GPUCurrentClockFreqs'..
|
|
' -q [gpu:0]/GPUutilization'
|
|
|
|
local NV_REGEX = '(%d+)\n'..
|
|
'(%d+)\n'..
|
|
'(%d+)\n'..
|
|
'(%d+),(%d+)\n'..
|
|
'graphics=(%d+), memory=%d+, video=(%d+), PCIe=%d+\n'
|
|
|
|
local GPU_BUS_CTRL = '/sys/bus/pci/devices/0000:01:00.0/power/control'
|
|
|
|
local state = {
|
|
error = false,
|
|
used_memory = 0,
|
|
total_memory = 0,
|
|
temp_reading = 0,
|
|
gpu_frequency = 0,
|
|
memory_frequency = 0,
|
|
gpu_utilization = 0,
|
|
vid_utilization = 0
|
|
}
|
|
|
|
local update_state = function()
|
|
if i_o.read_file(GPU_BUS_CTRL, nil, '*l') == 'on' then
|
|
local nvidia_settings_glob = i_o.execute_cmd(NV_QUERY)
|
|
if nvidia_settings_glob == '' then
|
|
state.error = 'Error'
|
|
else
|
|
state.used_memory,
|
|
state.total_memory,
|
|
state.temp_reading,
|
|
state.gpu_frequency,
|
|
state.memory_frequency,
|
|
state.gpu_utilization,
|
|
state.vid_utilization
|
|
= __string_match(nvidia_settings_glob, NV_REGEX)
|
|
state.error = false
|
|
end
|
|
else
|
|
state.error = 'Off'
|
|
end
|
|
return state
|
|
end
|
|
|
|
-----------------------------------------------------------------------------
|
|
-- main drawing functions
|
|
|
|
local rbs = common.reduce_blocks_(
|
|
point.y,
|
|
{
|
|
common.mk_block(mk_header, true, 0),
|
|
common.mk_block(mk_status, true, 0),
|
|
common.mk_block(mk_sep, true, SEPARATOR_SPACING),
|
|
common.mk_block(mk_temp, true, TEXT_SPACING),
|
|
common.mk_block(mk_sep, true, SEPARATOR_SPACING),
|
|
common.mk_block(mk_clock, true, SEPARATOR_SPACING),
|
|
common.mk_block(mk_sep, true, SEPARATOR_SPACING),
|
|
common.mk_block(mk_gpu_util, true, PLOT_SEC_BREAK),
|
|
common.mk_block(mk_mem_util, true, PLOT_SEC_BREAK),
|
|
common.mk_block(mk_vid_util, true, PLOT_SEC_BREAK)
|
|
}
|
|
)
|
|
return pure.map_at("update", function(f) return function() f(update_state()) end end, rbs)
|
|
end
|