From 05b1a8da26148fc9084e06a2817575e20bfd331c Mon Sep 17 00:00:00 2001
From: Ben Kyd <benjaminkyd@gmail.com>
Date: Tue, 23 May 2023 21:11:16 +0100
Subject: [PATCH] add nvidia smi to polybar

---
 common/.config/polybar/main-bar.ini   |  2 +-
 common/.config/polybar/modules.ini    | 13 +++++++++++++
 common/.config/polybar/scripts/gpu.sh |  1 +
 3 files changed, 15 insertions(+), 1 deletion(-)
 create mode 100755 common/.config/polybar/scripts/gpu.sh

diff --git a/common/.config/polybar/main-bar.ini b/common/.config/polybar/main-bar.ini
index d1e5a2c..d5769cf 100644
--- a/common/.config/polybar/main-bar.ini
+++ b/common/.config/polybar/main-bar.ini
@@ -22,7 +22,7 @@ border-bottom-color = ${colors.background}
 
 modules-left = i3 empty-space title
 modules-center = time
-modules-right = empty-space spotify empty-space backlight empty-space audio empty-space cpu empty-space temperature empty-space memory empty-space battery empty-space arrow
+modules-right = empty-space spotify empty-space backlight empty-space audio empty-space gpu cpu empty-space temperature empty-space memory empty-space battery empty-space arrow
 
 cursor-click = pointer
 cursor-scroll = ns-resize
diff --git a/common/.config/polybar/modules.ini b/common/.config/polybar/modules.ini
index 2a3be00..33113ec 100644
--- a/common/.config/polybar/modules.ini
+++ b/common/.config/polybar/modules.ini
@@ -107,6 +107,19 @@ format-foreground = ${colors.sapphire}
 
 # =======================================
 
+[module/gpu]
+type = custom/script
+exec = $HOME/.config/polybar/scripts/gpu.sh
+interval = 0.5
+tail = true
+format = <label>
+label = "%output%%  "
+format-prefix = "  "
+format-prefix-foreground = ${colors.green}
+format-foreground = ${colors.green}
+
+# =======================================
+
 [module/temperature]
 type = custom/script
 exec = sensors | grep 'Package id 0:\|Tdie' | grep ':[ ]*+[0-9]*.[0-9]*°C' -o | grep '[0-9]*.[0-9]*°C' -o
diff --git a/common/.config/polybar/scripts/gpu.sh b/common/.config/polybar/scripts/gpu.sh
new file mode 100755
index 0000000..bcf91de
--- /dev/null
+++ b/common/.config/polybar/scripts/gpu.sh
@@ -0,0 +1 @@
+nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits | awk '{printf "%02d\n", $1}'