Commit | Line | Data |
---|---|---|
05aa7e19 JG |
1 | // Formatting library for C++ - color support |
2 | // | |
3 | // Copyright (c) 2018 - present, Victor Zverovich and fmt contributors | |
4 | // All rights reserved. | |
5 | // | |
6 | // For the license information refer to format.h. | |
7 | ||
8 | #ifndef FMT_COLOR_H_ | |
9 | #define FMT_COLOR_H_ | |
10 | ||
11 | #include "format.h" | |
12 | ||
13 | // __declspec(deprecated) is broken in some MSVC versions. | |
14 | #if FMT_MSC_VER | |
15 | # define FMT_DEPRECATED_NONMSVC | |
16 | #else | |
17 | # define FMT_DEPRECATED_NONMSVC FMT_DEPRECATED | |
18 | #endif | |
19 | ||
20 | FMT_BEGIN_NAMESPACE | |
21 | FMT_MODULE_EXPORT_BEGIN | |
22 | ||
23 | enum class color : uint32_t { | |
24 | alice_blue = 0xF0F8FF, // rgb(240,248,255) | |
25 | antique_white = 0xFAEBD7, // rgb(250,235,215) | |
26 | aqua = 0x00FFFF, // rgb(0,255,255) | |
27 | aquamarine = 0x7FFFD4, // rgb(127,255,212) | |
28 | azure = 0xF0FFFF, // rgb(240,255,255) | |
29 | beige = 0xF5F5DC, // rgb(245,245,220) | |
30 | bisque = 0xFFE4C4, // rgb(255,228,196) | |
31 | black = 0x000000, // rgb(0,0,0) | |
32 | blanched_almond = 0xFFEBCD, // rgb(255,235,205) | |
33 | blue = 0x0000FF, // rgb(0,0,255) | |
34 | blue_violet = 0x8A2BE2, // rgb(138,43,226) | |
35 | brown = 0xA52A2A, // rgb(165,42,42) | |
36 | burly_wood = 0xDEB887, // rgb(222,184,135) | |
37 | cadet_blue = 0x5F9EA0, // rgb(95,158,160) | |
38 | chartreuse = 0x7FFF00, // rgb(127,255,0) | |
39 | chocolate = 0xD2691E, // rgb(210,105,30) | |
40 | coral = 0xFF7F50, // rgb(255,127,80) | |
41 | cornflower_blue = 0x6495ED, // rgb(100,149,237) | |
42 | cornsilk = 0xFFF8DC, // rgb(255,248,220) | |
43 | crimson = 0xDC143C, // rgb(220,20,60) | |
44 | cyan = 0x00FFFF, // rgb(0,255,255) | |
45 | dark_blue = 0x00008B, // rgb(0,0,139) | |
46 | dark_cyan = 0x008B8B, // rgb(0,139,139) | |
47 | dark_golden_rod = 0xB8860B, // rgb(184,134,11) | |
48 | dark_gray = 0xA9A9A9, // rgb(169,169,169) | |
49 | dark_green = 0x006400, // rgb(0,100,0) | |
50 | dark_khaki = 0xBDB76B, // rgb(189,183,107) | |
51 | dark_magenta = 0x8B008B, // rgb(139,0,139) | |
52 | dark_olive_green = 0x556B2F, // rgb(85,107,47) | |
53 | dark_orange = 0xFF8C00, // rgb(255,140,0) | |
54 | dark_orchid = 0x9932CC, // rgb(153,50,204) | |
55 | dark_red = 0x8B0000, // rgb(139,0,0) | |
56 | dark_salmon = 0xE9967A, // rgb(233,150,122) | |
57 | dark_sea_green = 0x8FBC8F, // rgb(143,188,143) | |
58 | dark_slate_blue = 0x483D8B, // rgb(72,61,139) | |
59 | dark_slate_gray = 0x2F4F4F, // rgb(47,79,79) | |
60 | dark_turquoise = 0x00CED1, // rgb(0,206,209) | |
61 | dark_violet = 0x9400D3, // rgb(148,0,211) | |
62 | deep_pink = 0xFF1493, // rgb(255,20,147) | |
63 | deep_sky_blue = 0x00BFFF, // rgb(0,191,255) | |
64 | dim_gray = 0x696969, // rgb(105,105,105) | |
65 | dodger_blue = 0x1E90FF, // rgb(30,144,255) | |
66 | fire_brick = 0xB22222, // rgb(178,34,34) | |
67 | floral_white = 0xFFFAF0, // rgb(255,250,240) | |
68 | forest_green = 0x228B22, // rgb(34,139,34) | |
69 | fuchsia = 0xFF00FF, // rgb(255,0,255) | |
70 | gainsboro = 0xDCDCDC, // rgb(220,220,220) | |
71 | ghost_white = 0xF8F8FF, // rgb(248,248,255) | |
72 | gold = 0xFFD700, // rgb(255,215,0) | |
73 | golden_rod = 0xDAA520, // rgb(218,165,32) | |
74 | gray = 0x808080, // rgb(128,128,128) | |
75 | green = 0x008000, // rgb(0,128,0) | |
76 | green_yellow = 0xADFF2F, // rgb(173,255,47) | |
77 | honey_dew = 0xF0FFF0, // rgb(240,255,240) | |
78 | hot_pink = 0xFF69B4, // rgb(255,105,180) | |
79 | indian_red = 0xCD5C5C, // rgb(205,92,92) | |
80 | indigo = 0x4B0082, // rgb(75,0,130) | |
81 | ivory = 0xFFFFF0, // rgb(255,255,240) | |
82 | khaki = 0xF0E68C, // rgb(240,230,140) | |
83 | lavender = 0xE6E6FA, // rgb(230,230,250) | |
84 | lavender_blush = 0xFFF0F5, // rgb(255,240,245) | |
85 | lawn_green = 0x7CFC00, // rgb(124,252,0) | |
86 | lemon_chiffon = 0xFFFACD, // rgb(255,250,205) | |
87 | light_blue = 0xADD8E6, // rgb(173,216,230) | |
88 | light_coral = 0xF08080, // rgb(240,128,128) | |
89 | light_cyan = 0xE0FFFF, // rgb(224,255,255) | |
90 | light_golden_rod_yellow = 0xFAFAD2, // rgb(250,250,210) | |
91 | light_gray = 0xD3D3D3, // rgb(211,211,211) | |
92 | light_green = 0x90EE90, // rgb(144,238,144) | |
93 | light_pink = 0xFFB6C1, // rgb(255,182,193) | |
94 | light_salmon = 0xFFA07A, // rgb(255,160,122) | |
95 | light_sea_green = 0x20B2AA, // rgb(32,178,170) | |
96 | light_sky_blue = 0x87CEFA, // rgb(135,206,250) | |
97 | light_slate_gray = 0x778899, // rgb(119,136,153) | |
98 | light_steel_blue = 0xB0C4DE, // rgb(176,196,222) | |
99 | light_yellow = 0xFFFFE0, // rgb(255,255,224) | |
100 | lime = 0x00FF00, // rgb(0,255,0) | |
101 | lime_green = 0x32CD32, // rgb(50,205,50) | |
102 | linen = 0xFAF0E6, // rgb(250,240,230) | |
103 | magenta = 0xFF00FF, // rgb(255,0,255) | |
104 | maroon = 0x800000, // rgb(128,0,0) | |
105 | medium_aquamarine = 0x66CDAA, // rgb(102,205,170) | |
106 | medium_blue = 0x0000CD, // rgb(0,0,205) | |
107 | medium_orchid = 0xBA55D3, // rgb(186,85,211) | |
108 | medium_purple = 0x9370DB, // rgb(147,112,219) | |
109 | medium_sea_green = 0x3CB371, // rgb(60,179,113) | |
110 | medium_slate_blue = 0x7B68EE, // rgb(123,104,238) | |
111 | medium_spring_green = 0x00FA9A, // rgb(0,250,154) | |
112 | medium_turquoise = 0x48D1CC, // rgb(72,209,204) | |
113 | medium_violet_red = 0xC71585, // rgb(199,21,133) | |
114 | midnight_blue = 0x191970, // rgb(25,25,112) | |
115 | mint_cream = 0xF5FFFA, // rgb(245,255,250) | |
116 | misty_rose = 0xFFE4E1, // rgb(255,228,225) | |
117 | moccasin = 0xFFE4B5, // rgb(255,228,181) | |
118 | navajo_white = 0xFFDEAD, // rgb(255,222,173) | |
119 | navy = 0x000080, // rgb(0,0,128) | |
120 | old_lace = 0xFDF5E6, // rgb(253,245,230) | |
121 | olive = 0x808000, // rgb(128,128,0) | |
122 | olive_drab = 0x6B8E23, // rgb(107,142,35) | |
123 | orange = 0xFFA500, // rgb(255,165,0) | |
124 | orange_red = 0xFF4500, // rgb(255,69,0) | |
125 | orchid = 0xDA70D6, // rgb(218,112,214) | |
126 | pale_golden_rod = 0xEEE8AA, // rgb(238,232,170) | |
127 | pale_green = 0x98FB98, // rgb(152,251,152) | |
128 | pale_turquoise = 0xAFEEEE, // rgb(175,238,238) | |
129 | pale_violet_red = 0xDB7093, // rgb(219,112,147) | |
130 | papaya_whip = 0xFFEFD5, // rgb(255,239,213) | |
131 | peach_puff = 0xFFDAB9, // rgb(255,218,185) | |
132 | peru = 0xCD853F, // rgb(205,133,63) | |
133 | pink = 0xFFC0CB, // rgb(255,192,203) | |
134 | plum = 0xDDA0DD, // rgb(221,160,221) | |
135 | powder_blue = 0xB0E0E6, // rgb(176,224,230) | |
136 | purple = 0x800080, // rgb(128,0,128) | |
137 | rebecca_purple = 0x663399, // rgb(102,51,153) | |
138 | red = 0xFF0000, // rgb(255,0,0) | |
139 | rosy_brown = 0xBC8F8F, // rgb(188,143,143) | |
140 | royal_blue = 0x4169E1, // rgb(65,105,225) | |
141 | saddle_brown = 0x8B4513, // rgb(139,69,19) | |
142 | salmon = 0xFA8072, // rgb(250,128,114) | |
143 | sandy_brown = 0xF4A460, // rgb(244,164,96) | |
144 | sea_green = 0x2E8B57, // rgb(46,139,87) | |
145 | sea_shell = 0xFFF5EE, // rgb(255,245,238) | |
146 | sienna = 0xA0522D, // rgb(160,82,45) | |
147 | silver = 0xC0C0C0, // rgb(192,192,192) | |
148 | sky_blue = 0x87CEEB, // rgb(135,206,235) | |
149 | slate_blue = 0x6A5ACD, // rgb(106,90,205) | |
150 | slate_gray = 0x708090, // rgb(112,128,144) | |
151 | snow = 0xFFFAFA, // rgb(255,250,250) | |
152 | spring_green = 0x00FF7F, // rgb(0,255,127) | |
153 | steel_blue = 0x4682B4, // rgb(70,130,180) | |
154 | tan = 0xD2B48C, // rgb(210,180,140) | |
155 | teal = 0x008080, // rgb(0,128,128) | |
156 | thistle = 0xD8BFD8, // rgb(216,191,216) | |
157 | tomato = 0xFF6347, // rgb(255,99,71) | |
158 | turquoise = 0x40E0D0, // rgb(64,224,208) | |
159 | violet = 0xEE82EE, // rgb(238,130,238) | |
160 | wheat = 0xF5DEB3, // rgb(245,222,179) | |
161 | white = 0xFFFFFF, // rgb(255,255,255) | |
162 | white_smoke = 0xF5F5F5, // rgb(245,245,245) | |
163 | yellow = 0xFFFF00, // rgb(255,255,0) | |
164 | yellow_green = 0x9ACD32 // rgb(154,205,50) | |
165 | }; // enum class color | |
166 | ||
167 | enum class terminal_color : uint8_t { | |
168 | black = 30, | |
169 | red, | |
170 | green, | |
171 | yellow, | |
172 | blue, | |
173 | magenta, | |
174 | cyan, | |
175 | white, | |
176 | bright_black = 90, | |
177 | bright_red, | |
178 | bright_green, | |
179 | bright_yellow, | |
180 | bright_blue, | |
181 | bright_magenta, | |
182 | bright_cyan, | |
183 | bright_white | |
184 | }; | |
185 | ||
186 | enum class emphasis : uint8_t { | |
187 | bold = 1, | |
188 | faint = 1 << 1, | |
189 | italic = 1 << 2, | |
190 | underline = 1 << 3, | |
191 | blink = 1 << 4, | |
192 | reverse = 1 << 5, | |
193 | conceal = 1 << 6, | |
194 | strikethrough = 1 << 7, | |
195 | }; | |
196 | ||
197 | // rgb is a struct for red, green and blue colors. | |
198 | // Using the name "rgb" makes some editors show the color in a tooltip. | |
199 | struct rgb { | |
200 | FMT_CONSTEXPR rgb() : r(0), g(0), b(0) {} | |
201 | FMT_CONSTEXPR rgb(uint8_t r_, uint8_t g_, uint8_t b_) : r(r_), g(g_), b(b_) {} | |
202 | FMT_CONSTEXPR rgb(uint32_t hex) | |
203 | : r((hex >> 16) & 0xFF), g((hex >> 8) & 0xFF), b(hex & 0xFF) {} | |
204 | FMT_CONSTEXPR rgb(color hex) | |
205 | : r((uint32_t(hex) >> 16) & 0xFF), | |
206 | g((uint32_t(hex) >> 8) & 0xFF), | |
207 | b(uint32_t(hex) & 0xFF) {} | |
208 | uint8_t r; | |
209 | uint8_t g; | |
210 | uint8_t b; | |
211 | }; | |
212 | ||
213 | FMT_BEGIN_DETAIL_NAMESPACE | |
214 | ||
215 | // color is a struct of either a rgb color or a terminal color. | |
216 | struct color_type { | |
217 | FMT_CONSTEXPR color_type() FMT_NOEXCEPT : is_rgb(), value{} {} | |
218 | FMT_CONSTEXPR color_type(color rgb_color) FMT_NOEXCEPT : is_rgb(true), | |
219 | value{} { | |
220 | value.rgb_color = static_cast<uint32_t>(rgb_color); | |
221 | } | |
222 | FMT_CONSTEXPR color_type(rgb rgb_color) FMT_NOEXCEPT : is_rgb(true), value{} { | |
223 | value.rgb_color = (static_cast<uint32_t>(rgb_color.r) << 16) | | |
224 | (static_cast<uint32_t>(rgb_color.g) << 8) | rgb_color.b; | |
225 | } | |
226 | FMT_CONSTEXPR color_type(terminal_color term_color) FMT_NOEXCEPT : is_rgb(), | |
227 | value{} { | |
228 | value.term_color = static_cast<uint8_t>(term_color); | |
229 | } | |
230 | bool is_rgb; | |
231 | union color_union { | |
232 | uint8_t term_color; | |
233 | uint32_t rgb_color; | |
234 | } value; | |
235 | }; | |
236 | ||
237 | FMT_END_DETAIL_NAMESPACE | |
238 | ||
239 | /** A text style consisting of foreground and background colors and emphasis. */ | |
240 | class text_style { | |
241 | public: | |
242 | FMT_CONSTEXPR text_style(emphasis em = emphasis()) FMT_NOEXCEPT | |
243 | : set_foreground_color(), | |
244 | set_background_color(), | |
245 | ems(em) {} | |
246 | ||
247 | FMT_CONSTEXPR text_style& operator|=(const text_style& rhs) { | |
248 | if (!set_foreground_color) { | |
249 | set_foreground_color = rhs.set_foreground_color; | |
250 | foreground_color = rhs.foreground_color; | |
251 | } else if (rhs.set_foreground_color) { | |
252 | if (!foreground_color.is_rgb || !rhs.foreground_color.is_rgb) | |
253 | FMT_THROW(format_error("can't OR a terminal color")); | |
254 | foreground_color.value.rgb_color |= rhs.foreground_color.value.rgb_color; | |
255 | } | |
256 | ||
257 | if (!set_background_color) { | |
258 | set_background_color = rhs.set_background_color; | |
259 | background_color = rhs.background_color; | |
260 | } else if (rhs.set_background_color) { | |
261 | if (!background_color.is_rgb || !rhs.background_color.is_rgb) | |
262 | FMT_THROW(format_error("can't OR a terminal color")); | |
263 | background_color.value.rgb_color |= rhs.background_color.value.rgb_color; | |
264 | } | |
265 | ||
266 | ems = static_cast<emphasis>(static_cast<uint8_t>(ems) | | |
267 | static_cast<uint8_t>(rhs.ems)); | |
268 | return *this; | |
269 | } | |
270 | ||
271 | friend FMT_CONSTEXPR text_style operator|(text_style lhs, | |
272 | const text_style& rhs) { | |
273 | return lhs |= rhs; | |
274 | } | |
275 | ||
276 | FMT_DEPRECATED_NONMSVC FMT_CONSTEXPR text_style& operator&=( | |
277 | const text_style& rhs) { | |
278 | return and_assign(rhs); | |
279 | } | |
280 | ||
281 | FMT_DEPRECATED_NONMSVC friend FMT_CONSTEXPR text_style | |
282 | operator&(text_style lhs, const text_style& rhs) { | |
283 | return lhs.and_assign(rhs); | |
284 | } | |
285 | ||
286 | FMT_CONSTEXPR bool has_foreground() const FMT_NOEXCEPT { | |
287 | return set_foreground_color; | |
288 | } | |
289 | FMT_CONSTEXPR bool has_background() const FMT_NOEXCEPT { | |
290 | return set_background_color; | |
291 | } | |
292 | FMT_CONSTEXPR bool has_emphasis() const FMT_NOEXCEPT { | |
293 | return static_cast<uint8_t>(ems) != 0; | |
294 | } | |
295 | FMT_CONSTEXPR detail::color_type get_foreground() const FMT_NOEXCEPT { | |
296 | FMT_ASSERT(has_foreground(), "no foreground specified for this style"); | |
297 | return foreground_color; | |
298 | } | |
299 | FMT_CONSTEXPR detail::color_type get_background() const FMT_NOEXCEPT { | |
300 | FMT_ASSERT(has_background(), "no background specified for this style"); | |
301 | return background_color; | |
302 | } | |
303 | FMT_CONSTEXPR emphasis get_emphasis() const FMT_NOEXCEPT { | |
304 | FMT_ASSERT(has_emphasis(), "no emphasis specified for this style"); | |
305 | return ems; | |
306 | } | |
307 | ||
308 | private: | |
309 | FMT_CONSTEXPR text_style(bool is_foreground, | |
310 | detail::color_type text_color) FMT_NOEXCEPT | |
311 | : set_foreground_color(), | |
312 | set_background_color(), | |
313 | ems() { | |
314 | if (is_foreground) { | |
315 | foreground_color = text_color; | |
316 | set_foreground_color = true; | |
317 | } else { | |
318 | background_color = text_color; | |
319 | set_background_color = true; | |
320 | } | |
321 | } | |
322 | ||
323 | // DEPRECATED! | |
324 | FMT_CONSTEXPR text_style& and_assign(const text_style& rhs) { | |
325 | if (!set_foreground_color) { | |
326 | set_foreground_color = rhs.set_foreground_color; | |
327 | foreground_color = rhs.foreground_color; | |
328 | } else if (rhs.set_foreground_color) { | |
329 | if (!foreground_color.is_rgb || !rhs.foreground_color.is_rgb) | |
330 | FMT_THROW(format_error("can't AND a terminal color")); | |
331 | foreground_color.value.rgb_color &= rhs.foreground_color.value.rgb_color; | |
332 | } | |
333 | ||
334 | if (!set_background_color) { | |
335 | set_background_color = rhs.set_background_color; | |
336 | background_color = rhs.background_color; | |
337 | } else if (rhs.set_background_color) { | |
338 | if (!background_color.is_rgb || !rhs.background_color.is_rgb) | |
339 | FMT_THROW(format_error("can't AND a terminal color")); | |
340 | background_color.value.rgb_color &= rhs.background_color.value.rgb_color; | |
341 | } | |
342 | ||
343 | ems = static_cast<emphasis>(static_cast<uint8_t>(ems) & | |
344 | static_cast<uint8_t>(rhs.ems)); | |
345 | return *this; | |
346 | } | |
347 | ||
348 | friend FMT_CONSTEXPR_DECL text_style fg(detail::color_type foreground) | |
349 | FMT_NOEXCEPT; | |
350 | ||
351 | friend FMT_CONSTEXPR_DECL text_style bg(detail::color_type background) | |
352 | FMT_NOEXCEPT; | |
353 | ||
354 | detail::color_type foreground_color; | |
355 | detail::color_type background_color; | |
356 | bool set_foreground_color; | |
357 | bool set_background_color; | |
358 | emphasis ems; | |
359 | }; | |
360 | ||
361 | /** Creates a text style from the foreground (text) color. */ | |
362 | FMT_CONSTEXPR inline text_style fg(detail::color_type foreground) FMT_NOEXCEPT { | |
363 | return text_style(true, foreground); | |
364 | } | |
365 | ||
366 | /** Creates a text style from the background color. */ | |
367 | FMT_CONSTEXPR inline text_style bg(detail::color_type background) FMT_NOEXCEPT { | |
368 | return text_style(false, background); | |
369 | } | |
370 | ||
371 | FMT_CONSTEXPR inline text_style operator|(emphasis lhs, | |
372 | emphasis rhs) FMT_NOEXCEPT { | |
373 | return text_style(lhs) | rhs; | |
374 | } | |
375 | ||
376 | FMT_BEGIN_DETAIL_NAMESPACE | |
377 | ||
378 | template <typename Char> struct ansi_color_escape { | |
379 | FMT_CONSTEXPR ansi_color_escape(detail::color_type text_color, | |
380 | const char* esc) FMT_NOEXCEPT { | |
381 | // If we have a terminal color, we need to output another escape code | |
382 | // sequence. | |
383 | if (!text_color.is_rgb) { | |
384 | bool is_background = esc == string_view("\x1b[48;2;"); | |
385 | uint32_t value = text_color.value.term_color; | |
386 | // Background ASCII codes are the same as the foreground ones but with | |
387 | // 10 more. | |
388 | if (is_background) value += 10u; | |
389 | ||
390 | size_t index = 0; | |
391 | buffer[index++] = static_cast<Char>('\x1b'); | |
392 | buffer[index++] = static_cast<Char>('['); | |
393 | ||
394 | if (value >= 100u) { | |
395 | buffer[index++] = static_cast<Char>('1'); | |
396 | value %= 100u; | |
397 | } | |
398 | buffer[index++] = static_cast<Char>('0' + value / 10u); | |
399 | buffer[index++] = static_cast<Char>('0' + value % 10u); | |
400 | ||
401 | buffer[index++] = static_cast<Char>('m'); | |
402 | buffer[index++] = static_cast<Char>('\0'); | |
403 | return; | |
404 | } | |
405 | ||
406 | for (int i = 0; i < 7; i++) { | |
407 | buffer[i] = static_cast<Char>(esc[i]); | |
408 | } | |
409 | rgb color(text_color.value.rgb_color); | |
410 | to_esc(color.r, buffer + 7, ';'); | |
411 | to_esc(color.g, buffer + 11, ';'); | |
412 | to_esc(color.b, buffer + 15, 'm'); | |
413 | buffer[19] = static_cast<Char>(0); | |
414 | } | |
415 | FMT_CONSTEXPR ansi_color_escape(emphasis em) FMT_NOEXCEPT { | |
416 | uint8_t em_codes[num_emphases] = {}; | |
417 | if (has_emphasis(em, emphasis::bold)) em_codes[0] = 1; | |
418 | if (has_emphasis(em, emphasis::faint)) em_codes[1] = 2; | |
419 | if (has_emphasis(em, emphasis::italic)) em_codes[2] = 3; | |
420 | if (has_emphasis(em, emphasis::underline)) em_codes[3] = 4; | |
421 | if (has_emphasis(em, emphasis::blink)) em_codes[4] = 5; | |
422 | if (has_emphasis(em, emphasis::reverse)) em_codes[5] = 7; | |
423 | if (has_emphasis(em, emphasis::conceal)) em_codes[6] = 8; | |
424 | if (has_emphasis(em, emphasis::strikethrough)) em_codes[7] = 9; | |
425 | ||
426 | size_t index = 0; | |
427 | for (size_t i = 0; i < num_emphases; ++i) { | |
428 | if (!em_codes[i]) continue; | |
429 | buffer[index++] = static_cast<Char>('\x1b'); | |
430 | buffer[index++] = static_cast<Char>('['); | |
431 | buffer[index++] = static_cast<Char>('0' + em_codes[i]); | |
432 | buffer[index++] = static_cast<Char>('m'); | |
433 | } | |
434 | buffer[index++] = static_cast<Char>(0); | |
435 | } | |
436 | FMT_CONSTEXPR operator const Char*() const FMT_NOEXCEPT { return buffer; } | |
437 | ||
438 | FMT_CONSTEXPR const Char* begin() const FMT_NOEXCEPT { return buffer; } | |
439 | FMT_CONSTEXPR_CHAR_TRAITS const Char* end() const FMT_NOEXCEPT { | |
440 | return buffer + std::char_traits<Char>::length(buffer); | |
441 | } | |
442 | ||
443 | private: | |
444 | static constexpr size_t num_emphases = 8; | |
445 | Char buffer[7u + 3u * num_emphases + 1u]; | |
446 | ||
447 | static FMT_CONSTEXPR void to_esc(uint8_t c, Char* out, | |
448 | char delimiter) FMT_NOEXCEPT { | |
449 | out[0] = static_cast<Char>('0' + c / 100); | |
450 | out[1] = static_cast<Char>('0' + c / 10 % 10); | |
451 | out[2] = static_cast<Char>('0' + c % 10); | |
452 | out[3] = static_cast<Char>(delimiter); | |
453 | } | |
454 | static FMT_CONSTEXPR bool has_emphasis(emphasis em, | |
455 | emphasis mask) FMT_NOEXCEPT { | |
456 | return static_cast<uint8_t>(em) & static_cast<uint8_t>(mask); | |
457 | } | |
458 | }; | |
459 | ||
460 | template <typename Char> | |
461 | FMT_CONSTEXPR ansi_color_escape<Char> make_foreground_color( | |
462 | detail::color_type foreground) FMT_NOEXCEPT { | |
463 | return ansi_color_escape<Char>(foreground, "\x1b[38;2;"); | |
464 | } | |
465 | ||
466 | template <typename Char> | |
467 | FMT_CONSTEXPR ansi_color_escape<Char> make_background_color( | |
468 | detail::color_type background) FMT_NOEXCEPT { | |
469 | return ansi_color_escape<Char>(background, "\x1b[48;2;"); | |
470 | } | |
471 | ||
472 | template <typename Char> | |
473 | FMT_CONSTEXPR ansi_color_escape<Char> make_emphasis(emphasis em) FMT_NOEXCEPT { | |
474 | return ansi_color_escape<Char>(em); | |
475 | } | |
476 | ||
477 | template <typename Char> | |
478 | inline void fputs(const Char* chars, FILE* stream) FMT_NOEXCEPT { | |
479 | std::fputs(chars, stream); | |
480 | } | |
481 | ||
482 | template <> | |
483 | inline void fputs<wchar_t>(const wchar_t* chars, FILE* stream) FMT_NOEXCEPT { | |
484 | std::fputws(chars, stream); | |
485 | } | |
486 | ||
487 | template <typename Char> inline void reset_color(FILE* stream) FMT_NOEXCEPT { | |
488 | fputs("\x1b[0m", stream); | |
489 | } | |
490 | ||
491 | template <> inline void reset_color<wchar_t>(FILE* stream) FMT_NOEXCEPT { | |
492 | fputs(L"\x1b[0m", stream); | |
493 | } | |
494 | ||
495 | template <typename Char> | |
496 | inline void reset_color(buffer<Char>& buffer) FMT_NOEXCEPT { | |
497 | auto reset_color = string_view("\x1b[0m"); | |
498 | buffer.append(reset_color.begin(), reset_color.end()); | |
499 | } | |
500 | ||
501 | template <typename Char> | |
502 | void vformat_to(buffer<Char>& buf, const text_style& ts, | |
503 | basic_string_view<Char> format_str, | |
504 | basic_format_args<buffer_context<type_identity_t<Char>>> args) { | |
505 | bool has_style = false; | |
506 | if (ts.has_emphasis()) { | |
507 | has_style = true; | |
508 | auto emphasis = detail::make_emphasis<Char>(ts.get_emphasis()); | |
509 | buf.append(emphasis.begin(), emphasis.end()); | |
510 | } | |
511 | if (ts.has_foreground()) { | |
512 | has_style = true; | |
513 | auto foreground = detail::make_foreground_color<Char>(ts.get_foreground()); | |
514 | buf.append(foreground.begin(), foreground.end()); | |
515 | } | |
516 | if (ts.has_background()) { | |
517 | has_style = true; | |
518 | auto background = detail::make_background_color<Char>(ts.get_background()); | |
519 | buf.append(background.begin(), background.end()); | |
520 | } | |
521 | detail::vformat_to(buf, format_str, args, {}); | |
522 | if (has_style) detail::reset_color<Char>(buf); | |
523 | } | |
524 | ||
525 | FMT_END_DETAIL_NAMESPACE | |
526 | ||
527 | template <typename S, typename Char = char_t<S>> | |
528 | void vprint(std::FILE* f, const text_style& ts, const S& format, | |
529 | basic_format_args<buffer_context<type_identity_t<Char>>> args) { | |
530 | basic_memory_buffer<Char> buf; | |
531 | detail::vformat_to(buf, ts, to_string_view(format), args); | |
532 | buf.push_back(Char(0)); | |
533 | detail::fputs(buf.data(), f); | |
534 | } | |
535 | ||
536 | /** | |
537 | \rst | |
538 | Formats a string and prints it to the specified file stream using ANSI | |
539 | escape sequences to specify text formatting. | |
540 | ||
541 | **Example**:: | |
542 | ||
543 | fmt::print(fmt::emphasis::bold | fg(fmt::color::red), | |
544 | "Elapsed time: {0:.2f} seconds", 1.23); | |
545 | \endrst | |
546 | */ | |
547 | template <typename S, typename... Args, | |
548 | FMT_ENABLE_IF(detail::is_string<S>::value)> | |
549 | void print(std::FILE* f, const text_style& ts, const S& format_str, | |
550 | const Args&... args) { | |
551 | vprint(f, ts, format_str, | |
552 | fmt::make_args_checked<Args...>(format_str, args...)); | |
553 | } | |
554 | ||
555 | /** | |
556 | \rst | |
557 | Formats a string and prints it to stdout using ANSI escape sequences to | |
558 | specify text formatting. | |
559 | ||
560 | **Example**:: | |
561 | ||
562 | fmt::print(fmt::emphasis::bold | fg(fmt::color::red), | |
563 | "Elapsed time: {0:.2f} seconds", 1.23); | |
564 | \endrst | |
565 | */ | |
566 | template <typename S, typename... Args, | |
567 | FMT_ENABLE_IF(detail::is_string<S>::value)> | |
568 | void print(const text_style& ts, const S& format_str, const Args&... args) { | |
569 | return print(stdout, ts, format_str, args...); | |
570 | } | |
571 | ||
572 | template <typename S, typename Char = char_t<S>> | |
573 | inline std::basic_string<Char> vformat( | |
574 | const text_style& ts, const S& format_str, | |
575 | basic_format_args<buffer_context<type_identity_t<Char>>> args) { | |
576 | basic_memory_buffer<Char> buf; | |
577 | detail::vformat_to(buf, ts, to_string_view(format_str), args); | |
578 | return fmt::to_string(buf); | |
579 | } | |
580 | ||
581 | /** | |
582 | \rst | |
583 | Formats arguments and returns the result as a string using ANSI | |
584 | escape sequences to specify text formatting. | |
585 | ||
586 | **Example**:: | |
587 | ||
588 | #include <fmt/color.h> | |
589 | std::string message = fmt::format(fmt::emphasis::bold | fg(fmt::color::red), | |
590 | "The answer is {}", 42); | |
591 | \endrst | |
592 | */ | |
593 | template <typename S, typename... Args, typename Char = char_t<S>> | |
594 | inline std::basic_string<Char> format(const text_style& ts, const S& format_str, | |
595 | const Args&... args) { | |
596 | return fmt::vformat(ts, to_string_view(format_str), | |
597 | fmt::make_args_checked<Args...>(format_str, args...)); | |
598 | } | |
599 | ||
600 | /** | |
601 | Formats a string with the given text_style and writes the output to ``out``. | |
602 | */ | |
603 | template <typename OutputIt, typename Char, | |
604 | FMT_ENABLE_IF(detail::is_output_iterator<OutputIt, Char>::value)> | |
605 | OutputIt vformat_to( | |
606 | OutputIt out, const text_style& ts, basic_string_view<Char> format_str, | |
607 | basic_format_args<buffer_context<type_identity_t<Char>>> args) { | |
608 | auto&& buf = detail::get_buffer<Char>(out); | |
609 | detail::vformat_to(buf, ts, format_str, args); | |
610 | return detail::get_iterator(buf); | |
611 | } | |
612 | ||
613 | /** | |
614 | \rst | |
615 | Formats arguments with the given text_style, writes the result to the output | |
616 | iterator ``out`` and returns the iterator past the end of the output range. | |
617 | ||
618 | **Example**:: | |
619 | ||
620 | std::vector<char> out; | |
621 | fmt::format_to(std::back_inserter(out), | |
622 | fmt::emphasis::bold | fg(fmt::color::red), "{}", 42); | |
623 | \endrst | |
624 | */ | |
625 | template <typename OutputIt, typename S, typename... Args, | |
626 | bool enable = detail::is_output_iterator<OutputIt, char_t<S>>::value&& | |
627 | detail::is_string<S>::value> | |
628 | inline auto format_to(OutputIt out, const text_style& ts, const S& format_str, | |
629 | Args&&... args) -> | |
630 | typename std::enable_if<enable, OutputIt>::type { | |
631 | return vformat_to(out, ts, to_string_view(format_str), | |
632 | fmt::make_args_checked<Args...>(format_str, args...)); | |
633 | } | |
634 | ||
635 | FMT_MODULE_EXPORT_END | |
636 | FMT_END_NAMESPACE | |
637 | ||
638 | #endif // FMT_COLOR_H_ |