Add --color or -c option for customizable UI colors

Author: metaspartan

README.md

@@ -21,6 +21,8 @@
 - Disk Activity Read/Write
 - Easy-to-read terminal UI
 - Two layouts: default and alternative
+- Customizable UI color (green, red, blue, cyan, magenta, yellow, and white)
+- Customizable update interval (default is 1000ms)
 - Support for all Apple Silicon models.
 
 ## Install via Homebrew
@@ -83,6 +85,7 @@ sudo ./mactop
 ## mactop Flags
 
 - `--interval` or `-i`: Set the powermetrics update interval in milliseconds. Default is 1000. (For low-end M chips, you may want to increase this value)
+- `--color` or `-c`: Set the UI color. Default is white. Options are 'green', 'red', 'blue', 'cyan', 'magenta', 'yellow', and 'white'. (-c green)
 - `--version` or `-v`: Print the version of mactop.
 - `--help` or `-h`: Show a help message about these flags and how to run mactop.
 
@@ -92,6 +95,10 @@ Use the following keys to interact with the application while its running:
 - `r`: Refresh the UI data manually.
 - `l`: Toggle the current layout.
 
+## Example Theme (Green) Screenshot (sudo mactop -c green)
+
+![mactop theme](screenshot3.png)
+
 ## Confirmed tested working M series chips
 
 - M1

main.go

@@ -1,12 +1,18 @@
 // Copyright (c) 2024 Carsen Klock under MIT License
 // mactop is a simple terminal based Apple Silicon power monitor written in Go Lang!
+//                _
+//  _ __  __ _ __| |_ ___ _ __
+// | '  \/ _` / _|  _/ _ \ '_ \
+// |_|_|_\__,_\__|\__\___/ .__/
+//  					 |_|
 
 package main
 
 import (
 	"bufio"
 	"fmt"
 	"log"
+	"math"
 	"os"
 	"os/exec"
 	"os/signal"
@@ -15,6 +21,7 @@ import (
 	"strconv"
 	"strings"
 	"syscall"
+	"time"
 
 	ui "github.com/gizak/termui/v3"
 	w "github.com/gizak/termui/v3/widgets"
@@ -78,11 +85,13 @@ type MemoryMetrics struct {
 
 var (
 	cpu1Gauge, cpu2Gauge, gpuGauge, aneGauge        *w.Gauge
-	TotalPowerChart                                 *w.Plot
+	TotalPowerChart                                 *w.BarChart
 	memoryGauge                                     *w.Gauge
 	modelText, PowerChart, NetworkInfo, ProcessInfo *w.Paragraph
 	grid                                            *ui.Grid
 
+	powerValues       []float64
+	lastUpdateTime    time.Time
 	stderrLogger      = log.New(os.Stderr, "", 0)
 	currentGridLayout = "default"
 	updateInterval    = 1000
@@ -154,12 +163,16 @@ func setupUI() {
 	ProcessInfo = w.NewParagraph()
 	ProcessInfo.Title = "Process Info"
 
-	TotalPowerChart = w.NewPlot()
-	TotalPowerChart.Title = "Total Power Usage (W)"
-	TotalPowerChart.Data = make([][]float64, 1)
-	TotalPowerChart.Data[0] = []float64{1, 2, 3, 4, 5}
-	TotalPowerChart.AxesColor = ui.ColorGreen
-	TotalPowerChart.LineColors = []ui.Color{ui.ColorCyan}
+	TotalPowerChart = w.NewBarChart()
+	TotalPowerChart.Title = "~ W Total Power"
+	TotalPowerChart.SetRect(50, 0, 75, 10)
+	TotalPowerChart.BarWidth = 5 // Adjust the bar width to fill the available space
+	TotalPowerChart.BarGap = 1   // Remove the gap between the bars
+	TotalPowerChart.PaddingBottom = 0
+	TotalPowerChart.PaddingTop = 1
+	TotalPowerChart.NumFormatter = func(num float64) string {
+		return ""
+	}
 
 	memoryGauge = w.NewGauge()
 	memoryGauge.Title = "Memory Usage"
@@ -251,6 +264,7 @@ func main() {
 		fmt.Println("--help: Show this help message")
 		fmt.Println("--version: Show the version of mactop")
 		fmt.Println("--interval: Set the powermetrics update interval in milliseconds. Default is 1000.")
+		fmt.Println("--color: Set the UI color. Default is white. Options are 'green', 'red', 'blue', 'cyan', 'magenta', 'yellow', and 'white'. (-c green)")
 		fmt.Println("You must use sudo to run mactop, as powermetrics requires root privileges.")
 		fmt.Println("For more information, see https://github.com/context-labs/mactop")
 		os.Exit(0)
@@ -274,6 +288,35 @@ func main() {
 		os.Exit(1)
 	}
 
+	if len(os.Args) > 2 && os.Args[1] == "--color" || len(os.Args) > 2 && os.Args[1] == "-c" {
+		colorName := strings.ToLower(os.Args[2])
+		var color ui.Color
+		switch colorName {
+		case "green":
+			color = ui.ColorGreen
+		case "red":
+			color = ui.ColorRed
+		case "blue":
+			color = ui.ColorBlue
+		case "cyan":
+			color = ui.ColorCyan
+		case "magenta":
+			color = ui.ColorMagenta
+		case "yellow":
+			color = ui.ColorYellow
+		case "white":
+			color = ui.ColorWhite
+		default:
+			stderrLogger.Printf("Unsupported color: %s. Using default color.\n", colorName)
+			color = ui.ColorWhite
+		}
+		ui.Theme.Block.Title.Fg = color
+		ui.Theme.Block.Border.Fg = color
+		ui.Theme.Paragraph.Text.Fg = color
+		ui.Theme.BarChart.Bars = []ui.Color{color}
+		ui.Theme.Gauge.Label.Fg = color
+	}
+
 	if len(os.Args) > 1 && os.Args[1] == "--interval" || len(os.Args) > 1 && os.Args[1] == "-i" {
 		interval, err := strconv.Atoi(os.Args[2])
 		if err != nil {
@@ -312,7 +355,7 @@ func main() {
 	signal.Notify(quit, os.Interrupt, syscall.SIGTERM)
 
 	go collectMetrics(done, cpuMetricsChan, gpuMetricsChan, netdiskMetricsChan, processMetricsChan)
-
+	lastUpdateTime = time.Now()
 	go func() {
 		for {
 			select {
@@ -375,21 +418,15 @@ func main() {
 }
 
 func setupLogfile() (*os.File, error) {
-	// create the log directory
 	if err := os.MkdirAll("logs", 0755); err != nil {
 		return nil, fmt.Errorf("failed to make the log directory: %v", err)
 	}
-	// open the log file
 	logfile, err := os.OpenFile("logs/mactop.log", os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0660)
 	if err != nil {
 		return nil, fmt.Errorf("failed to open log file: %v", err)
 	}
-
-	// log time, filename, and line number
 	log.SetFlags(log.Ltime | log.Lshortfile)
-	// log to file
 	log.SetOutput(logfile)
-
 	return logfile, nil
 }
 
@@ -445,21 +482,22 @@ func collectMetrics(done chan struct{}, cpumetricsChan chan CPUMetrics, gpumetri
 }
 
 func updateTotalPowerChart(newPowerValue float64) {
-	// stderrLogger.Printf("Rendering TotalPowerChart with data: %v\n", TotalPowerChart.Data)
-	if len(TotalPowerChart.Data[0]) == 0 {
-		TotalPowerChart.Data[0] = append(TotalPowerChart.Data[0], 0) // Ensure there's at least one data point
-	}
-
-	TotalPowerChart.Data[0] = append(TotalPowerChart.Data[0], newPowerValue)
-
-	if len(TotalPowerChart.Data[0]) > 250 {
-		TotalPowerChart.Data[0] = TotalPowerChart.Data[0][1:]
-	}
-
-	if len(TotalPowerChart.Data[0]) > 0 {
+	currentTime := time.Now()
+	powerValues = append(powerValues, newPowerValue)
+	if currentTime.Sub(lastUpdateTime) >= 2*time.Second {
+		var sum float64
+		for _, value := range powerValues {
+			sum += value
+		}
+		averagePower := sum / float64(len(powerValues))
+		averagePower = math.Round(averagePower)
+		TotalPowerChart.Data = append([]float64{averagePower}, TotalPowerChart.Data...)
+		if len(TotalPowerChart.Data) > 25 {
+			TotalPowerChart.Data = TotalPowerChart.Data[:25]
+		}
+		powerValues = nil
+		lastUpdateTime = currentTime
 		ui.Render(TotalPowerChart)
-	} else {
-		log.Println("No data to render for TotalPowerChart")
 	}
 }
 
@@ -477,7 +515,6 @@ func updateCPUUI(cpuMetrics CPUMetrics) {
 	PowerChart.Text = fmt.Sprintf("CPU Power: %.1f W\nGPU Power: %.1f W\nANE Power: %.1f W\nTotal Power: %.1f W", cpuMetrics.CPUW, cpuMetrics.GPUW, cpuMetrics.ANEW, cpuMetrics.PackageW)
 
 	memoryMetrics := getMemoryMetrics()
-
 	memoryGauge.Title = fmt.Sprintf("Memory Usage: %.2f GB / %.2f GB (Swap: %.2f/%.2f GB)", float64(memoryMetrics.Used)/1024/1024/1024, float64(memoryMetrics.Total)/1024/1024/1024, float64(memoryMetrics.SwapUsed)/1024/1024/1024, float64(memoryMetrics.SwapTotal)/1024/1024/1024)
 	memoryGauge.Percent = int((float64(memoryMetrics.Used) / float64(memoryMetrics.Total)) * 100)
 
@@ -536,7 +573,6 @@ func parseProcessMetrics(powermetricsOutput string, processMetrics []ProcessMetr
 	sort.Slice(processMetrics, func(i, j int) bool {
 		return processMetrics[i].CPUUsage > processMetrics[j].CPUUsage
 	})
-
 	return processMetrics
 }
 
@@ -689,7 +725,6 @@ func parseCPUMetrics(powermetricsOutput string, cpuMetrics CPUMetrics) CPUMetric
 		cpuMetrics.EClusterActive = (cpuMetrics.E0ClusterActive + cpuMetrics.E1ClusterActive) / 2
 		cpuMetrics.EClusterFreqMHz = max(cpuMetrics.E0ClusterFreqMHz, cpuMetrics.E1ClusterFreqMHz)
 	}
-
 	if cpuMetrics.P3ClusterActive != 0 {
 		// M1 Ultra
 		cpuMetrics.PClusterActive = (cpuMetrics.P0ClusterActive + cpuMetrics.P1ClusterActive + cpuMetrics.P2ClusterActive + cpuMetrics.P3ClusterActive) / 4
@@ -780,7 +815,6 @@ func getSOCInfo() map[string]interface{} {
 		"gpu_core_count": getGPUCores(),
 	}
 
-	// TDP (power)
 	switch socInfo["name"] {
 	case "Apple M1 Max":
 		socInfo["cpu_max_power"] = 30
@@ -802,7 +836,6 @@ func getSOCInfo() map[string]interface{} {
 		socInfo["gpu_max_power"] = 20
 	}
 
-	// Bandwidth
 	switch socInfo["name"] {
 	case "Apple M1 Max":
 		socInfo["cpu_max_bw"] = 250