Add config for progress bars

x

Author: nik-rev

Cargo.lock

@@ -71,7 +71,7 @@ default = ["deploy"]
 
 # Should not be included in builds.
 logging = ["fern", "log", "time"]
-generate_schema = ["schemars", "serde_json", "strum"]
+generate_schema = ["schemars", "serde_json", "strum", "mitsein/schemars"]
 
 [dependencies]
 anyhow = "1.0.100"
@@ -84,6 +84,7 @@ dirs = "6.0.0"
 humantime = "2.3.0"
 indexmap = "2.11.4"
 indoc = "2.0.7"
+mitsein = { version = "0.8", features = ["serde"] }
 itertools = "0.14.0"
 nohash = "0.2.0"
 nvml-wrapper = { version = "0.11.0", optional = true, features = ["legacy-functions"] }

Cargo.toml

@@ -166,3 +166,4 @@ These can be set under `[styles.widgets]`:
 | `selected_text`         | Text styling for text when representing something that is selected                           | `selected_text = { color = "black", bg_color = "blue", bold = true }` |
 | `disabled_text`         | Text styling for text when representing something that is disabled                           | `disabled_text = { color = "black", bg_color = "blue", bold = true }` |
 | `thread_text`           | Text styling for text when representing process threads. Only usable on Linux at the moment. | `thread_text = { color = "green", bg_color = "blue", bold = true }`   |
+| `progress_bar_chars`    | Characters to use for progress bars                                                          | `progress_bar_chars = ["▏", "▎", "▍", "▌", "▋", "▊", "▉", "█"]`       |

docs/content/configuration/config-file/styling.md

@@ -1,3 +1,4 @@
+use mitsein::NonEmpty;
 use tui::{
     buffer::Buffer,
     layout::Rect,
@@ -20,6 +21,8 @@ pub enum LabelLimit {
 #[derive(Debug, Clone)]
 pub struct PipeGauge<'a> {
     block: Option<Block<'a>>,
+    /// Characters to use for the progress bar
+    progress_chars: &'a NonEmpty<[char]>,
     ratio: f64,
     start_label: Option<Line<'a>>,
     inner_label: Option<Line<'a>>,
@@ -28,8 +31,8 @@ pub struct PipeGauge<'a> {
     hide_parts: LabelLimit,
 }
 
-impl Default for PipeGauge<'_> {
-    fn default() -> Self {
+impl<'a> PipeGauge<'a> {
+    pub fn new(progress_chars: &'a NonEmpty<[char]>) -> Self {
         Self {
             block: None,
             ratio: 0.0,
@@ -38,11 +41,10 @@ impl Default for PipeGauge<'_> {
             label_style: Style::default(),
             gauge_style: Style::default(),
             hide_parts: LabelLimit::default(),
+            progress_chars,
         }
     }
-}
 
-impl<'a> PipeGauge<'a> {
     /// The ratio, a value from 0.0 to 1.0 (any other greater or less will be
     /// clamped) represents the portion of the pipe gauge to fill.
     ///
@@ -196,11 +198,9 @@ impl Widget for PipeGauge<'_> {
                     gauge_area.width,
                 );
 
-                let pipe_end =
-                    start + (f64::from(end.saturating_sub(start)) * self.ratio).floor() as u16;
-                for col in start..pipe_end {
+                for (char, col) in progress_bar(self.progress_chars, start, end, self.ratio) {
                     if let Some(cell) = buf.cell_mut((col, row)) {
-                        cell.set_symbol("|").set_style(Style {
+                        cell.set_char(char).set_style(Style {
                             fg: self.gauge_style.fg,
                             bg: None,
                             add_modifier: self.gauge_style.add_modifier,
@@ -221,3 +221,90 @@ impl Widget for PipeGauge<'_> {
         }
     }
 }
+
+/// Returns an iterator over characters of the progress bar, and their positions
+///
+/// # Arguments
+///
+/// - `chars`: The characters to use for the progress bar
+/// - `start`: Start position
+/// - `end`: End position
+/// - `ratio`: How full the progress bar is
+fn progress_bar(
+    chars: &NonEmpty<[char]>, bar_start: u16, end: u16, ratio: f64,
+) -> impl Iterator<Item = (char, u16)> {
+    let bar_len = f64::from(end.saturating_sub(bar_start)) * ratio;
+
+    // Length of the bar, without accounting for the partial final character
+    let bar_len_truncated = bar_len.floor();
+
+    // The final progress character to display.
+    // This might be `None` if we can't display even the minimum segment, in
+    // which case we won't display anything at all.
+    //
+    // This might happen when, for example, we have 5 progress characters: [1, 2, 3, 4, .],
+    // 10 cells, and our progress is 50.1%. We will display 5 full cells:
+    //
+    // 50.1%: .....
+    //
+    // If it was 50.2% progress, we would display 5 full cells, and 1 cell with the first character:
+    //
+    // 50.2%: .....1
+    //             ^ extra
+    let final_progress_char = {
+        // The ratio of a single progress bar character that we lost due to truncation
+        //
+        // This ratio will be displayed as a "partial" character
+        let final_char_ratio = (bar_len - bar_len_truncated).clamp(0.0, 1.0);
+
+        let char_index = (final_char_ratio * chars.len().get() as f64).floor() as usize;
+
+        // 0-based indexing
+        char_index.checked_sub(1).and_then(|it| chars.get(it))
+    };
+
+    let bar_end = bar_start + bar_len_truncated as u16;
+
+    (bar_start..bar_end)
+        .map(move |pos| (*chars.last(), pos))
+        .chain(final_progress_char.map(|ch| (*ch, bar_end)))
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn progress_bar() {
+        let bars = (0..11)
+            .map(|i| {
+                let fill = i as f64 * 0.1;
+                super::progress_bar(
+                    &NonEmpty::<Vec<_>>::from(['1', '2', '3', '4', '.']),
+                    0,
+                    2,
+                    fill,
+                )
+                .map(|(ch, _)| ch)
+                .collect::<Vec<_>>()
+            })
+            .collect::<Vec<_>>();
+
+        assert_eq!(
+            bars,
+            vec![
+                vec![],
+                vec!['1'],
+                vec!['2'],
+                vec!['3'],
+                vec!['4'],
+                vec!['.'],
+                vec!['.', '1'],
+                vec!['.', '2'],
+                vec!['.', '3'],
+                vec!['.', '4'],
+                vec!['.', '.']
+            ]
+        );
+    }
+}

src/canvas/components/pipe_gauge.rs

@@ -61,7 +61,7 @@ impl Painter {
                 avg_loc.width -= 2;
 
                 f.render_widget(
-                    PipeGauge::default()
+                    PipeGauge::new(&self.styles.progress_bar_chars.0)
                         .gauge_style(style)
                         .label_style(style)
                         .inner_label(inner)
@@ -128,7 +128,7 @@ impl Painter {
 
                     for ((start_label, inner_label, ratio, style), row) in chunk.zip(rows.iter()) {
                         f.render_widget(
-                            PipeGauge::default()
+                            PipeGauge::new(&self.styles.progress_bar_chars.0)
                                 .gauge_style(style)
                                 .label_style(style)
                                 .inner_label(inner_label)

src/canvas/widgets/cpu_basic.rs

@@ -73,7 +73,7 @@ impl Painter {
         };
 
         draw_widgets.push(
-            PipeGauge::default()
+            PipeGauge::new(&self.styles.progress_bar_chars.0)
                 .ratio(ram_percentage / 100.0)
                 .start_label("RAM")
                 .inner_label(ram_label)
@@ -86,7 +86,7 @@ impl Painter {
             let swap_label = memory_label(swap_harvest, app_state.basic_mode_use_percent);
 
             draw_widgets.push(
-                PipeGauge::default()
+                PipeGauge::new(&self.styles.progress_bar_chars.0)
                     .ratio(swap_percentage / 100.0)
                     .start_label("SWP")
                     .inner_label(swap_label)
@@ -103,7 +103,7 @@ impl Painter {
                     memory_label(cache_harvest, app_state.basic_mode_use_percent);
 
                 draw_widgets.push(
-                    PipeGauge::default()
+                    PipeGauge::new(&self.styles.progress_bar_chars.0)
                         .ratio(cache_percentage / 100.0)
                         .start_label("CHE")
                         .inner_label(cache_fraction_label)
@@ -121,7 +121,7 @@ impl Painter {
                     memory_label(arc_harvest, app_state.basic_mode_use_percent);
 
                 draw_widgets.push(
-                    PipeGauge::default()
+                    PipeGauge::new(&self.styles.progress_bar_chars.0)
                         .ratio(arc_percentage / 100.0)
                         .start_label("ARC")
                         .inner_label(arc_fraction_label)
@@ -152,7 +152,7 @@ impl Painter {
                 };
 
                 draw_widgets.push(
-                    PipeGauge::default()
+                    PipeGauge::new(&self.styles.progress_bar_chars.0)
                         .ratio(percentage / 100.0)
                         .start_label("GPU")
                         .inner_label(label)

src/canvas/widgets/mem_basic.rs

@@ -25,7 +25,9 @@ use utils::{opt, set_colour, set_colour_list, set_style};
 use widgets::WidgetStyle;
 
 use super::Config;
-use crate::options::{OptionError, OptionResult, args::BottomArgs};
+use crate::options::{
+    OptionError, OptionResult, args::BottomArgs, config::style::widgets::ProgressBarChars,
+};
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
 #[cfg_attr(feature = "generate_schema", derive(schemars::JsonSchema))]
@@ -127,6 +129,7 @@ pub struct Styles {
     #[cfg(target_os = "linux")]
     pub(crate) thread_text_style: Style,
     pub(crate) border_type: BorderType,
+    pub(crate) progress_bar_chars: ProgressBarChars,
 }
 
 impl Default for Styles {

src/options/config/style.rs

@@ -4,7 +4,7 @@ use tui::{
 };
 
 use super::color;
-use crate::options::config::style::Styles;
+use crate::options::config::style::{Styles, widgets::ProgressBarChars};
 
 impl Styles {
     pub(crate) fn default_palette() -> Self {
@@ -69,6 +69,7 @@ impl Styles {
             border_type: BorderType::Plain,
             #[cfg(target_os = "linux")]
             thread_text_style: color!(Color::Green),
+            progress_bar_chars: ProgressBarChars::default(),
         }
     }
 

src/options/config/style/themes/default.rs

@@ -4,7 +4,7 @@ use tui::{
 };
 
 use super::{color, hex};
-use crate::options::config::style::{Styles, themes::hex_colour};
+use crate::options::config::style::{Styles, themes::hex_colour, widgets::ProgressBarChars};
 
 impl Styles {
     pub(crate) fn gruvbox_palette() -> Self {
@@ -69,6 +69,7 @@ impl Styles {
             border_type: BorderType::Plain,
             #[cfg(target_os = "linux")]
             thread_text_style: hex!("#458588"),
+            progress_bar_chars: ProgressBarChars::default(),
         }
     }
 
@@ -134,6 +135,7 @@ impl Styles {
             border_type: BorderType::Plain,
             #[cfg(target_os = "linux")]
             thread_text_style: hex!("#458588"),
+            progress_bar_chars: ProgressBarChars::default(),
         }
     }
 }

src/options/config/style/themes/gruvbox.rs

@@ -4,7 +4,7 @@ use tui::{
 };
 
 use super::{color, hex};
-use crate::options::config::style::{Styles, themes::hex_colour};
+use crate::options::config::style::{Styles, themes::hex_colour, widgets::ProgressBarChars};
 
 impl Styles {
     pub(crate) fn nord_palette() -> Self {
@@ -57,6 +57,7 @@ impl Styles {
             border_type: BorderType::Plain,
             #[cfg(target_os = "linux")]
             thread_text_style: hex!("#a3be8c"),
+            progress_bar_chars: ProgressBarChars::default(),
         }
     }
 
@@ -110,6 +111,7 @@ impl Styles {
             border_type: BorderType::Plain,
             #[cfg(target_os = "linux")]
             thread_text_style: hex!("#a3be8c"),
+            progress_bar_chars: ProgressBarChars::default(),
         }
     }
 }