tmux/tmux-plugins.tmux-cpu
1
0
Fork 0
mirror of https://github.com/tmux-plugins/tmux-cpu.git synced 2026-09-10 07:16:16 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds gram_total and gram_used values

Browse source
This commit is contained in:
Fortyseven 2024-09-30 21:02:37 -04:00
parent 98d787191b
commit 7d016adb73
3 changed files with 56 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cpu.tmux
View file

@ -44,6 +44,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)"

27
scripts/gram_total.sh Executable file
View file

@ -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 "$@"

27
scripts/gram_used.sh Executable file
View file

@ -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 "$@"