conky-config/module/Graphics.lua

329 lines
7.6 KiB
Lua

local Widget = require 'Widget'
local CriticalText = require 'CriticalText'
local Text = require 'Text'
local TextColumn = require 'TextColumn'
local Line = require 'Line'
local LabelPlot = require 'LabelPlot'
local util = require 'util'
local schema = require 'default_patterns'
local _TONUMBER = tonumber
--construction params
local MODULE_Y = 145
local SEPARATOR_SPACING = 20
local TEXT_SPACING = 20
local PLOT_SEC_BREAK = 20
local PLOT_HEIGHT = 56
local header = Widget.Header{
x = _G_INIT_DATA_.LEFT_X,
y = MODULE_Y,
width = _G_INIT_DATA_.SECTION_WIDTH,
header = 'NVIDIA GRAPHICS'
}
local RIGHT_X = _G_INIT_DATA_.LEFT_X + _G_INIT_DATA_.SECTION_WIDTH
local status = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = header.bottom_y,
text = 'Status'
},
value = Widget.Text{
x = RIGHT_X,
y = header.bottom_y,
x_align = 'right',
text_color = schema.blue,
text = '<status>'
}
}
local SEP_Y_1 = header.bottom_y + SEPARATOR_SPACING
local separator1 = Widget.Line{
p1 = {x = _G_INIT_DATA_.LEFT_X, y = SEP_Y_1},
p2 = {x = RIGHT_X, y = SEP_Y_1}
}
local INTERNAL_TEMP_Y = SEP_Y_1 + SEPARATOR_SPACING
local internal_temp = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = INTERNAL_TEMP_Y,
text = 'Internal Temperature'
},
value = Widget.CriticalText{
x = RIGHT_X,
y = INTERNAL_TEMP_Y,
x_align = 'right',
text_color = schema.blue,
text = '<gpu_temp>'
}
}
local PCI_UTIL_Y = INTERNAL_TEMP_Y + TEXT_SPACING
local pci_util = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = PCI_UTIL_Y,
text = 'PCI Utilization'
},
value = Widget.Text{
x = RIGHT_X,
y = PCI_UTIL_Y,
x_align = 'right',
text_color = schema.blue,
text = '<pci_util>'
}
}
local SEP_Y_2 = PCI_UTIL_Y + SEPARATOR_SPACING
local separator2 = Widget.Line{
p1 = {x = _G_INIT_DATA_.LEFT_X, y = SEP_Y_2},
p2 = {x = RIGHT_X, y = SEP_Y_2}
}
local CLOCK_SPEED_Y = SEP_Y_2 + SEPARATOR_SPACING
local clock_speed = {
labels = Widget.TextColumn{
x = _G_INIT_DATA_.LEFT_X,
y = CLOCK_SPEED_Y,
spacing = TEXT_SPACING,
'GPU Clock Speed',
'Memory Clock Speed'
},
values = Widget.TextColumn{
x = _G_INIT_DATA_.LEFT_X + _G_INIT_DATA_.SECTION_WIDTH,
y = CLOCK_SPEED_Y,
spacing = TEXT_SPACING,
x_align = 'right',
text_color = schema.blue,
num_rows = 2
}
}
local SEP_Y_3 = CLOCK_SPEED_Y + TEXT_SPACING * 2
local separator3 = Widget.Line{
p1 = {x = _G_INIT_DATA_.LEFT_X, y = SEP_Y_3},
p2 = {x = RIGHT_X, y = SEP_Y_3}
}
local GPU_UTIL_Y = SEP_Y_3 + SEPARATOR_SPACING
local GPU_UTIL_PLOT_Y = GPU_UTIL_Y + PLOT_SEC_BREAK
local gpu_util = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = GPU_UTIL_Y,
text = 'GPU Utilization'
},
value = Widget.Text{
x = RIGHT_X,
y = GPU_UTIL_Y,
x_align = 'right',
text_color = schema.blue,
text = '<gpu_util>'
},
plot = Widget.LabelPlot{
x = _G_INIT_DATA_.LEFT_X,
y = GPU_UTIL_PLOT_Y,
width = _G_INIT_DATA_.SECTION_WIDTH,
height = PLOT_HEIGHT
}
}
local MEM_UTIL_Y = GPU_UTIL_PLOT_Y + PLOT_HEIGHT + PLOT_SEC_BREAK
local MEM_UTIL_PLOT_Y = MEM_UTIL_Y + PLOT_SEC_BREAK
local mem_util = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = MEM_UTIL_Y,
text = 'Memory Utilization'
},
value = Widget.Text{
x = RIGHT_X,
y = MEM_UTIL_Y,
x_align = 'right',
text_color = schema.blue,
text = '<mem_util>'
},
plot = Widget.LabelPlot{
x = _G_INIT_DATA_.LEFT_X,
y = MEM_UTIL_PLOT_Y,
width = _G_INIT_DATA_.SECTION_WIDTH,
height = PLOT_HEIGHT
}
}
local VID_UTIL_Y = MEM_UTIL_PLOT_Y + PLOT_HEIGHT + PLOT_SEC_BREAK
local VID_UTIL_PLOT_Y = VID_UTIL_Y + PLOT_SEC_BREAK
local vid_util = {
label = Widget.Text{
x = _G_INIT_DATA_.LEFT_X,
y = VID_UTIL_Y,
text = 'Video Utilization'
},
value = Widget.Text{
x = RIGHT_X,
y = VID_UTIL_Y,
x_align = 'right',
text_color = schema.blue,
text = '<vid_util>'
},
plot = Widget.LabelPlot{
x = _G_INIT_DATA_.LEFT_X,
y = VID_UTIL_PLOT_Y,
width = _G_INIT_DATA_.SECTION_WIDTH,
height = PLOT_HEIGHT
}
}
--[[
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 = 'optirun nvidia-settings -c :8 -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 __nvidia_off = function(cr)
CriticalText.set(internal_temp.value, cr, 'N/A', 1)
Text.set(pci_util.value, cr, 'N/A')
TextColumn.set(clock_speed.values, cr, 1, 'N/A')
TextColumn.set(clock_speed.values, cr, 2, 'N/A')
Text.set(gpu_util.value, cr, 'N/A')
Text.set(mem_util.value, cr, 'N/A')
Text.set(vid_util.value, cr, 'N/A')
LabelPlot.update(gpu_util.plot, 0)
LabelPlot.update(mem_util.plot, 0)
LabelPlot.update(vid_util.plot, 0)
end
local __update = function(cr)
if util.read_file('/proc/acpi/bbswitch', '.+%s+(%u+)') == 'ON' then
if string.find(util.execute_cmd('ps -A -o comm'), 'optirun') == nil then
Text.set(status.value, cr, 'Mixed')
__nvidia_off(cr)
else
Text.set(status.value, cr, 'On')
local nvidia_settings_glob = util.execute_cmd(NV_QUERY)
local used_memory, total_memory, temp_reading, gpu_frequency,
memory_frequency, gpu_utilization, vid_utilization,
pci_utilization = string.match(nvidia_settings_glob, NV_REGEX)
local force = 1
if _TONUMBER(temp_reading) > 80 then force = 0 end
CriticalText.set(internal_temp.value, cr, temp_reading..'°C', force)
Text.set(pci_util.value, cr, pci_utilization..'%')
TextColumn.set(clock_speed.values, cr, 1, gpu_frequency..' Mhz')
TextColumn.set(clock_speed.values, cr, 2, memory_frequency..' Mhz')
local percent_used_memory = used_memory / total_memory
Text.set(gpu_util.value, cr, gpu_utilization..'%')
Text.set(mem_util.value, cr, util.round(percent_used_memory * 100)..'%')
Text.set(vid_util.value, cr, vid_utilization..'%')
LabelPlot.update(gpu_util.plot, gpu_utilization * 0.01)
LabelPlot.update(mem_util.plot, percent_used_memory)
LabelPlot.update(vid_util.plot, vid_utilization * 0.01)
end
else
Text.set(status.value, cr, 'Off')
__nvidia_off(cr)
end
end
Widget = nil
schema = nil
MODULE_Y = nil
SEPARATOR_SPACING = nil
TEXT_SPACING = nil
PLOT_SECTION_BREAK = nil
PLOT_HEIGHT = nil
RIGHT_X = nil
SEP_Y_1 = nil
SEP_Y_2 = nil
SEP_Y_3 = nil
INTERNAL_TEMP_Y = nil
PCI_UTIL_Y = nil
CLOCK_SPEED_Y = nil
GPU_UTIL_Y = nil
GPU_UTIL_PLOT_Y = nil
MEM_UTIL_Y = nil
MEM_UTIL_PLOT_Y = nil
VID_UTIL_Y = nil
VID_UTIL_PLOT_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)
Text.draw(status.label, cr)
Text.draw(status.value, cr)
Line.draw(separator1, cr)
Text.draw(internal_temp.label, cr)
Text.draw(internal_temp.value, cr)
Text.draw(pci_util.label, cr)
Text.draw(pci_util.value, cr)
Line.draw(separator2, cr)
TextColumn.draw(clock_speed.labels, cr)
TextColumn.draw(clock_speed.values, cr)
Line.draw(separator3, cr)
Text.draw(gpu_util.label, cr)
Text.draw(gpu_util.value, cr)
LabelPlot.draw(gpu_util.plot, cr)
Text.draw(mem_util.label, cr)
Text.draw(mem_util.value, cr)
LabelPlot.draw(mem_util.plot, cr)
Text.draw(vid_util.label, cr)
Text.draw(vid_util.value, cr)
LabelPlot.draw(vid_util.plot, cr)
end
end
return draw