cega 0.2.2

CGA and EGA binary image file processing library
Documentation 
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

pub mod args;

use crate::{ColorPalette, Grid, ImageType, Palette, RawGrid};

pub type CharPalette = Palette<char>;
pub type CGACharPalette = [char; 4];
pub type EGACharPalette = [char; 16];

pub const ANSIOPEN: &str = "\x1b[";
pub const ANSIRESET: &str = "\x1b[0m";
pub const DISABLEWRAPPING: &str = "\x1b[?7l";
pub const ENABLEWRAPPING: &str = "\x1b[?7h";

pub const CGACHAR: CGACharPalette = [' ', '*', '+', '▒'];
pub const EGACHAR: EGACharPalette = [
    ' ', '.', ':', '-', '=', '+', '*', '▒', '▓', '•', '#', '‖', '%', '@', '⁌', '█',
];

pub fn default_char_palette(image_type: ImageType) -> CharPalette {
    match image_type {
        ImageType::CGA => CGACHAR.to_vec(),
        ImageType::EGA => EGACHAR.to_vec(),
    }
}

#[derive(Clone, Debug)]
pub enum TerminalMode {
    Ascii,
    ColoredAscii,
    Pixels,         //full ansi_bg color pixels
    HorizontalHalf, // half left blocks + bg color for 2x density
}
impl TerminalMode {
    pub fn from_short(short: &str) -> Result<TerminalMode, String> {
        match short {
            "a" => Ok(TerminalMode::Ascii),
            "c" => Ok(TerminalMode::ColoredAscii),
            "p" => Ok(TerminalMode::Pixels),
            "h" => Ok(TerminalMode::HorizontalHalf),
            _ => Err("possible values: a, c, p, h".to_string()),
        }
    }
    pub fn adjusted_index(&self, index: usize, i: usize) -> usize {
        match self {
            TerminalMode::HorizontalHalf => (index * 2) + (i % 2),
            _ => index,
        }
    }
}

#[allow(dead_code)]
pub struct TerminalPalette {
    pub mode: TerminalMode,
    pub terminal: Palette<String>,
}

impl TerminalPalette {
    pub fn new(mode: TerminalMode, chars: CharPalette, colors: ColorPalette) -> Self {
        if chars.len() != colors.len() {
            panic!("Incompatible character and color palette lengths");
        }
        let term = match mode {
            TerminalMode::Ascii => chars.iter().map(|m| m.to_string()).collect(),
            TerminalMode::ColoredAscii => chars
                .iter()
                .zip(colors.iter())
                .map(|(ch, co)| ansi_codes(co.ansi_fg(), ch))
                .collect(),
            TerminalMode::Pixels => colors
                .iter()
                .map(|co| ansi_codes(co.ansi_bg(), &' '))
                .collect(),
            TerminalMode::HorizontalHalf => colors
                .iter()
                .flat_map(|co| half_ansi_codes(co.ansi_fg(), co.ansi_bg(), '▌'))
                .collect(),
        };

        Self {
            mode,
            terminal: term,
        }
    }
    pub fn adjusted_get(&self, index: usize, i: usize) -> String {
        self.terminal[self.mode.adjusted_index(index, i)].to_owned()
    }

    pub fn apply(&self, image_data: RawGrid) -> Grid<String> {
        image_data
            .iter()
            .map(|row| {
                row.iter()
                    .enumerate()
                    .map(|(i, index)| self.adjusted_get(*index as usize, i).to_owned())
                    .collect::<Vec<String>>()
            })
            .collect::<Grid<String>>()
    }
}

pub fn disable_wrapping(string: String) -> String {
    format!("{}{}{}", DISABLEWRAPPING, string, ENABLEWRAPPING)
}

pub fn to_string(grid: Grid<String>) -> String {
    grid.iter()
        .map(|row| row.join(""))
        .collect::<Vec<_>>()
        .join("\n")
}

pub fn ansi_codes(co: u8, ch: &char) -> String {
    format!("{}{}m{}{}", ANSIOPEN, co, ch, ANSIRESET)
}
pub fn half_ansi_codes(fg: u8, bg: u8, ch: char) -> [String; 2] {
    [
        format!("{}{};", ANSIOPEN, fg),
        format!("{}m{}{}", bg, ch, ANSIRESET),
    ]
}