From 564de038cf752039a63e90b2e1c0071d2ad79165 Mon Sep 17 00:00:00 2001
From: Fortyseven <fortyseven@network47.org>
Date: Mon, 30 Sep 2024 21:02:37 -0400
Subject: [PATCH] Adds `gram_total` and `gram_used` values

Fix: adds missing `gram_used` and `gram_total` interpolations

Removes errant 'i'
---
 cpu.tmux              |  4 ++++
 scripts/gram_total.sh | 27 +++++++++++++++++++++++++++
 scripts/gram_used.sh  | 27 +++++++++++++++++++++++++++
 3 files changed, 58 insertions(+)
 create mode 100755 scripts/gram_total.sh
 create mode 100755 scripts/gram_used.sh

diff --git a/cpu.tmux b/cpu.tmux
index 40d8124..c9a2810 100755
--- a/cpu.tmux
+++ b/cpu.tmux
@@ -18,6 +18,8 @@ cpu_interpolation=(
   "\#{ram_bg_color}"
   "\#{ram_fg_color}"
   "\#{gram_percentage}"
+  "\#{gram_total}"
+  "\#{gram_used}"
   "\#{gram_icon}"
   "\#{gram_bg_color}"
   "\#{gram_fg_color}"
@@ -44,6 +46,8 @@ cpu_commands=(
   "#($CURRENT_DIR/scripts/ram_bg_color.sh)"
   "#($CURRENT_DIR/scripts/ram_fg_color.sh)"
   "#($CURRENT_DIR/scripts/gram_percentage.sh)"
+  "#($CURRENT_DIR/scripts/gram_total.sh)"
+  "#($CURRENT_DIR/scripts/gram_used.sh)"
   "#($CURRENT_DIR/scripts/gram_icon.sh)"
   "#($CURRENT_DIR/scripts/gram_bg_color.sh)"
   "#($CURRENT_DIR/scripts/gram_fg_color.sh)"
diff --git a/scripts/gram_total.sh b/scripts/gram_total.sh
new file mode 100755
index 0000000..18ae7f9
--- /dev/null
+++ b/scripts/gram_total.sh
@@ -0,0 +1,27 @@
+#!/usr/bin/env bash
+
+CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# shellcheck source=scripts/helpers.sh
+source "$CURRENT_DIR/helpers.sh"
+
+gram_total_format="%dk"
+
+print_gram_total() {
+  gram_total_format=$(get_tmux_option "@gram_total_format" "$gram_total_format")
+
+  if command_exists "nvidia-smi"; then
+    loads=$(cached_eval nvidia-smi | sed -nr 's/.*\s([0-9]+)MiB\s*\/\s*([0-9]+)MiB.*/\1 \2/p')
+  elif command_exists "cuda-smi"; then
+    loads=$(cached_eval cuda-smi | sed -nr 's/.*\s([0-9.]+) of ([0-9.]+) MB.*/\1 \2/p' | awk '{print $2-$1" "$2}')
+  else
+    echo "No GPU"
+    return
+  fi
+  echo "$loads" | awk -v format="$gram_total_format" '{total+=$2} END {printf format, total}'
+}
+
+main() {
+  print_gram_total
+}
+main "$@"
diff --git a/scripts/gram_used.sh b/scripts/gram_used.sh
new file mode 100755
index 0000000..721068c
--- /dev/null
+++ b/scripts/gram_used.sh
@@ -0,0 +1,27 @@
+#!/usr/bin/env bash
+
+CURRENT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+# shellcheck source=scripts/helpers.sh
+source "$CURRENT_DIR/helpers.sh"
+
+gram_used_format="%dk"
+
+print_gram_usage() {
+  gram_used_format=$(get_tmux_option "@gram_used_format" "$gram_used_format")
+
+  if command_exists "nvidia-smi"; then
+    loads=$(cached_eval nvidia-smi | sed -nr 's/.*\s([0-9]+)MiB\s*\/\s*([0-9]+)MiB.*/\1 \2/p')
+  elif command_exists "cuda-smi"; then
+    loads=$(cached_eval cuda-smi | sed -nr 's/.*\s([0-9.]+) of ([0-9.]+) MB.*/\1 \2/p' | awk '{print $2-$1" "$2}')
+  else
+    echo "No GPU"
+    return
+  fi
+  echo "$loads" | awk -v format="$gram_used_format" '{used+=$1} END {printf format, used}'
+}
+
+main() {
+  print_gram_usage
+}
+main "$@"