1 | //! Popular color palettes for [`anstyle::AnsiColor`] |
2 | //! |
3 | //! Based on [wikipedia](https://en.wikipedia.org/wiki/ANSI_escape_code#3-bit_and_4-bit) |
4 | use anstyle::RgbColor as Rgb; |
5 | |
6 | /// A color palette for rendering 4-bit [`anstyle::AnsiColor`] |
7 | #[allow (clippy::exhaustive_structs)] |
8 | #[derive (Copy, Clone, PartialEq, Eq, Debug)] |
9 | pub struct Palette(pub RawPalette); |
10 | type RawPalette = [Rgb; 16]; |
11 | |
12 | impl Palette { |
13 | /// Look up the [`anstyle::RgbColor`] in the palette |
14 | pub const fn get(&self, color: anstyle::AnsiColor) -> Rgb { |
15 | let color = anstyle::Ansi256Color::from_ansi(color); |
16 | *self.get_ansi256_ref(color) |
17 | } |
18 | const fn get_ansi256_ref(&self, color: anstyle::Ansi256Color) -> &Rgb { |
19 | let index = color.index() as usize; |
20 | &self.0[index] |
21 | } |
22 | |
23 | pub(crate) const fn rgb_from_ansi(&self, color: anstyle::AnsiColor) -> anstyle::RgbColor { |
24 | self.get(color) |
25 | } |
26 | |
27 | pub(crate) const fn rgb_from_index(&self, index: u8) -> Option<anstyle::RgbColor> { |
28 | let index = index as usize; |
29 | if index < self.0.len() { |
30 | Some(self.0[index]) |
31 | } else { |
32 | None |
33 | } |
34 | } |
35 | |
36 | pub(crate) const fn find_match(&self, color: anstyle::RgbColor) -> anstyle::AnsiColor { |
37 | let mut best_index = 0; |
38 | let mut best_distance = crate::distance(color, self.0[best_index]); |
39 | |
40 | let mut index = best_index + 1; |
41 | while index < self.0.len() { |
42 | let distance = crate::distance(color, self.0[index]); |
43 | if distance < best_distance { |
44 | best_index = index; |
45 | best_distance = distance; |
46 | } |
47 | |
48 | index += 1; |
49 | } |
50 | |
51 | if let Some(color) = anstyle::Ansi256Color(best_index as u8).into_ansi() { |
52 | color |
53 | } else { |
54 | // Panic |
55 | #[allow (clippy::no_effect)] |
56 | ["best_index is out of bounds" ][best_index]; |
57 | // Make compiler happy |
58 | anstyle::AnsiColor::Black |
59 | } |
60 | } |
61 | } |
62 | |
63 | impl Default for Palette { |
64 | fn default() -> Self { |
65 | DEFAULT |
66 | } |
67 | } |
68 | |
69 | impl std::ops::Index<anstyle::AnsiColor> for Palette { |
70 | type Output = Rgb; |
71 | |
72 | #[inline ] |
73 | fn index(&self, color: anstyle::AnsiColor) -> &Rgb { |
74 | let color: Ansi256Color = anstyle::Ansi256Color::from_ansi(color); |
75 | self.get_ansi256_ref(color) |
76 | } |
77 | } |
78 | |
79 | impl From<RawPalette> for Palette { |
80 | fn from(raw: RawPalette) -> Self { |
81 | Self(raw) |
82 | } |
83 | } |
84 | |
85 | /// Platform-specific default |
86 | #[cfg (not(windows))] |
87 | pub use VGA as DEFAULT; |
88 | |
89 | /// Platform-specific default |
90 | #[cfg (windows)] |
91 | pub use WIN10_CONSOLE as DEFAULT; |
92 | |
93 | /// Typical colors that are used when booting PCs and leaving them in text mode |
94 | pub const VGA: Palette = Palette([ |
95 | Rgb(0, 0, 0), |
96 | Rgb(170, 0, 0), |
97 | Rgb(0, 170, 0), |
98 | Rgb(170, 85, 0), |
99 | Rgb(0, 0, 170), |
100 | Rgb(170, 0, 170), |
101 | Rgb(0, 170, 170), |
102 | Rgb(170, 170, 170), |
103 | Rgb(85, 85, 85), |
104 | Rgb(255, 85, 85), |
105 | Rgb(85, 255, 85), |
106 | Rgb(255, 255, 85), |
107 | Rgb(85, 85, 255), |
108 | Rgb(255, 85, 255), |
109 | Rgb(85, 255, 255), |
110 | Rgb(255, 255, 255), |
111 | ]); |
112 | |
113 | /// Campbell theme, used as of Windows 10 version 1709. |
114 | pub const WIN10_CONSOLE: Palette = Palette([ |
115 | Rgb(12, 12, 12), |
116 | Rgb(197, 15, 31), |
117 | Rgb(19, 161, 14), |
118 | Rgb(193, 156, 0), |
119 | Rgb(0, 55, 218), |
120 | Rgb(136, 23, 152), |
121 | Rgb(58, 150, 221), |
122 | Rgb(204, 204, 204), |
123 | Rgb(118, 118, 118), |
124 | Rgb(231, 72, 86), |
125 | Rgb(22, 198, 12), |
126 | Rgb(249, 241, 165), |
127 | Rgb(59, 120, 255), |
128 | Rgb(180, 0, 158), |
129 | Rgb(97, 214, 214), |
130 | Rgb(242, 242, 242), |
131 | ]); |
132 | |