ledcat 0.2.0

Control lots of LED's over lots of protocols
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use crate::device::*;
use std::io::Write;

pub struct AnsiDisplay {
    width: usize,
    height: usize,

    initial_frame: bool,
}

impl Output for AnsiDisplay {
    fn output_frame(&mut self, frame: &[Pixel]) -> io::Result<()> {
        // A buffer is used so frames can be written in one go, significantly improving
        // performance.
        let mut buf = Vec::new();

        // Clear the screen so there is no other stuff on it.
        if self.initial_frame {
            write!(buf, "\x1b[3J\x1b[H\x1b[2J")?;
            self.initial_frame = false;
        }
        write!(buf, "\x1b[1;1H")?;

        // Two pixels are rendered at once using the Upper Half Block character. The top half is
        // colored with the foreground color while the lower half uses the background. This neat
        // trick allows us to render square pixels with a higher density than combining two
        // rectangular characters.
        for y in 0..self.height / 2 + (self.height & 1) {
            for x in 0..self.width {
                let pix_hi = &frame[y * 2 * self.width + x];
                let pix_lo = frame.get((y * 2 + 1) * self.width + x);
                // Set the background color.
                if let Some(pix_lo) = pix_lo {
                    write!(buf, "\x1b[48;2;{};{};{}m", pix_lo.r, pix_lo.g, pix_lo.b)?;
                } else {
                    write!(buf, "\x1b[48;2;0m")?;
                }
                // Set the foreground color.
                write!(buf, "\x1b[38;2;{};{};{}m", pix_hi.r, pix_hi.g, pix_hi.b)?;
                write!(buf, "\u{2580}")?;
            }
            // Reset to the default background color and jump to the next line.
            writeln!(buf, "\x1b[0m")?;
        }

        io::stdout().write_all(&buf)
    }
}

pub fn command<'a, 'b>() -> clap::App<'a, 'b> {
    clap::SubCommand::with_name("show").about("Visualize 2D frames using a text based display")
}

pub fn from_command(_: &clap::ArgMatches, gargs: &GlobalArgs) -> io::Result<FromCommand> {
    let (width, height) = gargs.dimensions_2d()?;
    Ok(FromCommand::Output(Box::new(AnsiDisplay {
        width,
        height,
        initial_frame: true,
    })))
}