From d60af79363bc81bc80a30a5157cf500dff37557c Mon Sep 17 00:00:00 2001
From: Abhishek Keshri <iam2kabhishek@gmail.com>
Date: Tue, 9 Aug 2022 06:28:35 +0530
Subject: [PATCH] Add gpu_usage.sh

---
 scripts/gpu_usage.sh | 48 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100755 scripts/gpu_usage.sh

diff --git a/scripts/gpu_usage.sh b/scripts/gpu_usage.sh
new file mode 100755
index 0000000..4babdc0
--- /dev/null
+++ b/scripts/gpu_usage.sh
@@ -0,0 +1,48 @@
+#!/usr/bin/env bash
+# setting the locale, some users have issues with different locales, this forces the correct one
+export LC_ALL=en_US.UTF-8
+
+current_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source $current_dir/utils.sh
+
+get_platform()
+{
+  case $(uname -s) in
+    Linux)
+      gpu=$(lspci -v | grep VGA | head -n 1 | awk '{print $5}')
+      echo $gpu
+      ;;
+
+    Darwin)
+      # TODO - Darwin/Mac compatability
+      ;;
+
+    CYGWIN*|MINGW32*|MSYS*|MINGW*)
+      # TODO - windows compatability
+      ;;
+  esac
+}
+
+get_gpu()
+{
+  gpu=$(get_platform)
+  if [[ "$gpu" == NVIDIA ]]; then
+    usage=$(nvidia-smi | grep '%' | awk '{ sum += $13 } END { printf("%d%%\n", sum / NR) }')
+  else
+    usage='unknown'
+  fi
+  normalize_percent_len $usage
+}
+
+main()
+{
+  # storing the refresh rate in the variable RATE, default is 5
+  RATE=$(get_tmux_option "@tmux2k-refresh-rate" 5)
+  gpu_label=$(get_tmux_option "@tmux2k-gpu-usage-label" "GPU")
+  gpu_usage=$(get_gpu)
+  echo "$gpu_label $gpu_usage"
+  sleep $RATE
+}
+
+# run the main driver
+main