1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use std::fmt;

// See https://en.wikipedia.org/wiki/Block_Elements for more fun
static CHARS: [char; 8] = ['▏', '▎', '▍', '▌', '▋', '▊', '▉', '█'];

/// A pixel precise horizontal bar
pub struct ProgressBar {
    pub part: f32,
    pub chars_len: usize,
}

impl ProgressBar {
    /// create a bar of a given char length.
    /// `part` must be in `[0,1]`.
    /// `chars_len` is the max width of the bar in characters
    pub fn new(part: f32, chars_len: usize) -> Self {
        Self { part, chars_len }
    }
}

impl fmt::Display for ProgressBar {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        let mut s = String::new();
        let mp = (self.chars_len as f32) * self.part;
        let full = mp as usize;
        for _ in 0..full {
            s.push(CHARS[7]);
        }
        let remain = (mp.fract() * 8.0).round() as usize;
        if remain > 0 {
            s.push(CHARS[remain - 1]);
        }
        f.pad(&s)
    }
}