From a25e26917893aa2d11c19cfd872c37a14fa05c7e Mon Sep 17 00:00:00 2001
From: Luke Hubmayer-Werner <mongoose41@gmail.com>
Date: Sat, 3 Jun 2023 17:28:18 +0930
Subject: [PATCH] Program to generate the usage/temp strings without shell
 overhead

---
 cpu_monitor.rs | 12 +++++++++++
 gpu_monitor.rs | 57 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+)
 create mode 100644 cpu_monitor.rs
 create mode 100644 gpu_monitor.rs

diff --git a/cpu_monitor.rs b/cpu_monitor.rs
new file mode 100644
index 0000000..d424c72
--- /dev/null
+++ b/cpu_monitor.rs
@@ -0,0 +1,12 @@
+use std::fs;
+// use std::fmt;
+use std::path::Path;
+
+fn main() {
+	let hwmon_path = fs::read_dir(Path::new("/sys/devices/pci0000:00/0000:00:18.3/hwmon/"))?.next().unwrap()?.path();
+	// let temp_raw = fs::read_to_string(hwmon_path.join("temp1_input"))?;
+	let temperature = fs::read_to_string(hwmon_path.join("temp1_input"))?.trim_end().parse::<f32>()? / 1000.0;
+	let filename_usage = format!("{}/gpu_busy_percent", dir);
+	let usage = format!("{}", fs::read_to_string(filename_usage)?.trim_end());  // No need to convert from ASCII to int to ASCII
+	print!("{}%\r{}℃\n", usage, temperature);
+}
diff --git a/gpu_monitor.rs b/gpu_monitor.rs
new file mode 100644
index 0000000..2c3da7b
--- /dev/null
+++ b/gpu_monitor.rs
@@ -0,0 +1,57 @@
+use std::fs;
+// use std::fmt;
+use std::path::Path;
+
+fn read_gpu_stats(dir: String) -> Result<String,Box<dyn std::error::Error>> {
+	// print!("Opening {}\n", dir);
+	let filename_usage = format!("{}/gpu_busy_percent", dir);
+	let dirname = format!("{}/hwmon", dir);
+	let dir = Path::new(&dirname);
+
+	// let usage = fs::read_to_string(filename_usage)?.trim_end().parse::<i32>()?;
+	let usage = format!("{}", fs::read_to_string(filename_usage)?.trim_end());  // No need to convert from ASCII to int to ASCII
+	let hwmon_path = fs::read_dir(dir)?.next().unwrap()?.path();
+	let mut temp_str = "".to_owned();
+	for i in 1..=8 {
+		let path = hwmon_path.join(format!("temp{i}_input"));
+		let lbl = hwmon_path.join(format!("temp{i}_label"));
+		let symbol = match fs::read_to_string(lbl) {
+			Ok(s) => match s.trim_end() {
+				"junction" => "",
+				"mem" => "",
+				_ => "",  // Includes "edge"
+			},
+			_ => "",  // Default icon if no label found
+		};
+		match fs::read_to_string(path) {
+			Ok(s) => {
+				// let temperature_int = s.trim_end().parse::<i32>()? / 1000;
+				let t = s.get(0..s.len()-4).unwrap_or("");
+				if i > 1 {temp_str.push_str("  ")};
+				// temp_str.push_str(&format!("{symbol}{t:>3}℃"))
+				temp_str.push_str(&format!("{symbol}{t:>2}℃"))  // If the temp hits 100C we have bigger problems than a layout shift
+			},
+			_ => break,
+		};
+	}
+	// // let temperature = fs::read_to_string(path)?.trim_end().parse::<i32>()? / 1000;
+	// let temp_raw = fs::read_to_string(hwmon_path.join("temp1_input"))?;
+	// let temperature = format!("{}", temp_raw.get(0..temp_raw.len()-4).unwrap_or(""));
+	// return Ok(format!("{:>3}% {:>3}℃", usage, temperature));
+	return Ok(format!("{:>3}% {}", usage, temp_str));
+}
+
+fn main() {
+	let mut output_str = "       ".to_owned();
+	for i in 0..8 {
+		let dir = format!("/sys/class/drm/card{}/device", i);
+		match read_gpu_stats(dir) {
+			Ok(s) => {
+				if i > 0 {output_str.push_str("\r       ")};  // Gap between GPUs
+				output_str.push_str(&s)
+			},
+			_ => break,
+		};
+	}
+	print!("{}\n", output_str);
+}