]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // (C) Copyright Gennadiy Rozental 2001. |
2 | // Distributed under the Boost Software License, Version 1.0. | |
3 | // (See accompanying file LICENSE_1_0.txt or copy at | |
4 | // http://www.boost.org/LICENSE_1_0.txt) | |
5 | ||
6 | // See http://www.boost.org/libs/test for the library home page. | |
7 | // | |
8 | // File : $RCSfile$ | |
9 | // | |
10 | // Version : $Revision$ | |
11 | // | |
12 | // Description : contains definition for setcolor iostream manipulator | |
13 | // *************************************************************************** | |
14 | ||
15 | #ifndef BOOST_TEST_UTILS_SETCOLOR_HPP | |
16 | #define BOOST_TEST_UTILS_SETCOLOR_HPP | |
17 | ||
18 | // Boost.Test | |
19 | #include <boost/test/detail/config.hpp> | |
20 | ||
92f5a8d4 TL |
21 | #include <boost/core/ignore_unused.hpp> |
22 | ||
7c673cae FG |
23 | // STL |
24 | #include <iostream> | |
25 | #include <cstdio> | |
26 | ||
27 | #include <boost/test/detail/suppress_warnings.hpp> | |
28 | ||
11fdf7f2 TL |
29 | #ifdef _WIN32 |
30 | #include <windows.h> | |
31 | ||
32 | #if defined(__MINGW32__) && !defined(COMMON_LVB_UNDERSCORE) | |
33 | // mingw badly mimicking windows.h | |
34 | #define COMMON_LVB_UNDERSCORE 0x8000 | |
35 | #endif | |
36 | #endif | |
37 | ||
7c673cae FG |
38 | //____________________________________________________________________________// |
39 | ||
40 | namespace boost { | |
41 | namespace unit_test { | |
42 | namespace utils { | |
43 | ||
44 | // ************************************************************************** // | |
45 | // ************** term_attr ************** // | |
46 | // ************************************************************************** // | |
47 | ||
48 | struct term_attr { enum _ { | |
49 | NORMAL = 0, | |
50 | BRIGHT = 1, | |
51 | DIM = 2, | |
52 | UNDERLINE = 4, | |
53 | BLINK = 5, | |
54 | REVERSE = 7, | |
55 | CROSSOUT = 9 | |
56 | }; }; | |
57 | ||
58 | // ************************************************************************** // | |
59 | // ************** term_color ************** // | |
60 | // ************************************************************************** // | |
61 | ||
62 | struct term_color { enum _ { | |
63 | BLACK = 0, | |
64 | RED = 1, | |
65 | GREEN = 2, | |
66 | YELLOW = 3, | |
67 | BLUE = 4, | |
68 | MAGENTA = 5, | |
69 | CYAN = 6, | |
70 | WHITE = 7, | |
71 | ORIGINAL = 9 | |
72 | }; }; | |
73 | ||
74 | // ************************************************************************** // | |
75 | // ************** setcolor ************** // | |
76 | // ************************************************************************** // | |
77 | ||
11fdf7f2 | 78 | #ifndef _WIN32 |
7c673cae FG |
79 | class setcolor { |
80 | public: | |
81 | // Constructor | |
92f5a8d4 TL |
82 | explicit setcolor( bool is_color_output = false, |
83 | term_attr::_ attr = term_attr::NORMAL, | |
7c673cae FG |
84 | term_color::_ fg = term_color::ORIGINAL, |
85 | term_color::_ bg = term_color::ORIGINAL ) | |
92f5a8d4 | 86 | : m_is_color_output(is_color_output) |
7c673cae | 87 | { |
92f5a8d4 TL |
88 | m_command_size = std::sprintf( m_control_command, "%c[%c;3%c;4%cm", |
89 | 0x1B, | |
90 | static_cast<char>(attr + '0'), | |
91 | static_cast<char>(fg + '0'), | |
92 | static_cast<char>(bg + '0')); | |
7c673cae FG |
93 | } |
94 | ||
95 | friend std::ostream& | |
96 | operator<<( std::ostream& os, setcolor const& sc ) | |
97 | { | |
92f5a8d4 | 98 | if (sc.m_is_color_output && (&os == &std::cout || &os == &std::cerr)) { |
11fdf7f2 TL |
99 | return os.write( sc.m_control_command, sc.m_command_size ); |
100 | } | |
101 | return os; | |
7c673cae FG |
102 | } |
103 | ||
104 | private: | |
105 | // Data members | |
92f5a8d4 | 106 | bool m_is_color_output; |
7c673cae FG |
107 | char m_control_command[13]; |
108 | int m_command_size; | |
109 | }; | |
110 | ||
11fdf7f2 TL |
111 | #else |
112 | ||
113 | class setcolor { | |
114 | ||
115 | protected: | |
116 | void set_console_color(std::ostream& os, WORD *attributes = NULL) const { | |
92f5a8d4 TL |
117 | if (!m_is_color_output) { |
118 | return; | |
119 | } | |
11fdf7f2 TL |
120 | DWORD console_type; |
121 | if (&os == &std::cout) { | |
122 | console_type = STD_OUTPUT_HANDLE; | |
123 | } | |
124 | else if (&os == &std::cerr) { | |
125 | console_type = STD_ERROR_HANDLE; | |
126 | } | |
127 | else { | |
128 | return; | |
129 | } | |
130 | HANDLE hConsole = GetStdHandle(console_type); | |
131 | ||
132 | if(hConsole == INVALID_HANDLE_VALUE || hConsole == NULL ) | |
133 | return; | |
134 | ||
135 | if(attributes != NULL) { | |
136 | SetConsoleTextAttribute(hConsole, *attributes); | |
137 | return; | |
138 | } | |
139 | ||
140 | CONSOLE_SCREEN_BUFFER_INFO consoleInfo; | |
141 | GetConsoleScreenBufferInfo(hConsole, &consoleInfo); | |
92f5a8d4 | 142 | saved_attributes = consoleInfo.wAttributes; |
11fdf7f2 TL |
143 | |
144 | WORD fg_attr = 0; | |
145 | switch(m_fg) | |
146 | { | |
147 | case term_color::WHITE: | |
148 | fg_attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; | |
149 | break; | |
150 | case term_color::BLACK: | |
151 | fg_attr = 0; | |
152 | break; | |
153 | case term_color::RED: | |
154 | fg_attr = FOREGROUND_RED; | |
155 | break; | |
156 | case term_color::GREEN: | |
157 | fg_attr = FOREGROUND_GREEN; | |
158 | break; | |
159 | case term_color::CYAN: | |
160 | fg_attr = FOREGROUND_GREEN | FOREGROUND_BLUE; | |
161 | break; | |
162 | case term_color::MAGENTA: | |
163 | fg_attr = FOREGROUND_RED | FOREGROUND_BLUE; | |
164 | break; | |
165 | case term_color::BLUE: | |
166 | fg_attr = FOREGROUND_BLUE; | |
167 | break; | |
168 | case term_color::YELLOW: | |
169 | fg_attr = FOREGROUND_RED | FOREGROUND_GREEN; | |
170 | break; | |
171 | case term_color::ORIGINAL: | |
172 | default: | |
173 | fg_attr = saved_attributes & (FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE); | |
174 | break; | |
175 | } | |
176 | ||
177 | WORD bg_attr = 0; | |
178 | switch(m_bg) | |
179 | { | |
180 | case term_color::BLACK: | |
181 | bg_attr = 0; | |
182 | break; | |
183 | case term_color::WHITE: | |
184 | bg_attr = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE; | |
185 | break; | |
186 | case term_color::RED: | |
187 | bg_attr = BACKGROUND_RED; | |
188 | break; | |
189 | case term_color::GREEN: | |
190 | bg_attr = BACKGROUND_GREEN; | |
191 | break; | |
192 | case term_color::BLUE: | |
193 | bg_attr = BACKGROUND_BLUE; | |
194 | break; | |
195 | case term_color::ORIGINAL: | |
196 | default: | |
197 | bg_attr = saved_attributes & (BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE); | |
198 | break; | |
199 | } | |
200 | ||
201 | WORD text_attr = 0; | |
202 | switch(m_attr) | |
203 | { | |
204 | case term_attr::BRIGHT: | |
205 | text_attr = FOREGROUND_INTENSITY; | |
206 | break; | |
207 | case term_attr::UNDERLINE: | |
208 | text_attr = COMMON_LVB_UNDERSCORE; | |
209 | break; | |
210 | default: | |
211 | break; | |
212 | } | |
213 | ||
92f5a8d4 | 214 | SetConsoleTextAttribute(hConsole, fg_attr | bg_attr | text_attr); |
11fdf7f2 TL |
215 | return; |
216 | } | |
217 | ||
218 | public: | |
219 | // Constructor | |
220 | explicit setcolor( | |
92f5a8d4 | 221 | bool is_color_output = false, |
11fdf7f2 TL |
222 | term_attr::_ attr = term_attr::NORMAL, |
223 | term_color::_ fg = term_color::ORIGINAL, | |
224 | term_color::_ bg = term_color::ORIGINAL ) | |
92f5a8d4 TL |
225 | : m_is_color_output(is_color_output) |
226 | , m_attr(attr) | |
11fdf7f2 TL |
227 | , m_fg(fg) |
228 | , m_bg(bg) | |
229 | {} | |
230 | ||
231 | friend std::ostream& | |
232 | operator<<( std::ostream& os, setcolor const& sc ) | |
233 | { | |
234 | sc.set_console_color(os); | |
235 | return os; | |
236 | } | |
237 | ||
238 | private: | |
92f5a8d4 | 239 | bool m_is_color_output; |
11fdf7f2 TL |
240 | term_attr::_ m_attr; |
241 | term_color::_ m_fg; | |
242 | term_color::_ m_bg; | |
243 | ||
244 | protected: | |
245 | // Data members | |
246 | mutable WORD saved_attributes; | |
11fdf7f2 TL |
247 | }; |
248 | ||
249 | #endif | |
7c673cae FG |
250 | // ************************************************************************** // |
251 | // ************** scope_setcolor ************** // | |
252 | // ************************************************************************** // | |
253 | ||
11fdf7f2 TL |
254 | #ifndef _WIN32 |
255 | ||
7c673cae FG |
256 | struct scope_setcolor { |
257 | scope_setcolor() : m_os( 0 ) {} | |
92f5a8d4 TL |
258 | explicit scope_setcolor( bool is_color_output, |
259 | std::ostream& os, | |
7c673cae FG |
260 | term_attr::_ attr = term_attr::NORMAL, |
261 | term_color::_ fg = term_color::ORIGINAL, | |
262 | term_color::_ bg = term_color::ORIGINAL ) | |
263 | : m_os( &os ) | |
92f5a8d4 | 264 | , m_is_color_output( is_color_output ) |
7c673cae | 265 | { |
92f5a8d4 | 266 | os << setcolor( is_color_output, attr, fg, bg ); |
7c673cae FG |
267 | } |
268 | ~scope_setcolor() | |
269 | { | |
270 | if( m_os ) | |
92f5a8d4 | 271 | *m_os << setcolor( m_is_color_output ); |
7c673cae FG |
272 | } |
273 | private: | |
11fdf7f2 TL |
274 | scope_setcolor(const scope_setcolor& r); |
275 | scope_setcolor& operator=(const scope_setcolor& r); | |
7c673cae FG |
276 | // Data members |
277 | std::ostream* m_os; | |
92f5a8d4 | 278 | bool m_is_color_output; |
7c673cae FG |
279 | }; |
280 | ||
11fdf7f2 TL |
281 | #else |
282 | ||
283 | struct scope_setcolor : setcolor { | |
284 | scope_setcolor() : m_os( 0 ) {} | |
285 | explicit scope_setcolor( | |
92f5a8d4 | 286 | bool is_color_output, |
11fdf7f2 TL |
287 | std::ostream& os, |
288 | term_attr::_ attr = term_attr::NORMAL, | |
289 | term_color::_ fg = term_color::ORIGINAL, | |
290 | term_color::_ bg = term_color::ORIGINAL ) | |
92f5a8d4 TL |
291 | : setcolor(is_color_output, attr, fg, bg) |
292 | , m_os( &os ) | |
11fdf7f2 TL |
293 | { |
294 | os << *this; | |
295 | } | |
296 | ||
297 | ~scope_setcolor() | |
298 | { | |
299 | if (m_os) { | |
300 | set_console_color(*m_os, &this->saved_attributes); | |
301 | } | |
302 | } | |
303 | private: | |
304 | scope_setcolor(const scope_setcolor& r); | |
305 | scope_setcolor& operator=(const scope_setcolor& r); | |
306 | // Data members | |
307 | std::ostream* m_os; | |
308 | }; | |
309 | ||
310 | ||
311 | #endif | |
312 | ||
92f5a8d4 TL |
313 | #define BOOST_TEST_SCOPE_SETCOLOR( is_color_output, os, attr, color ) \ |
314 | utils::scope_setcolor const sc(is_color_output, os, utils::attr, utils::color); \ | |
315 | boost::ignore_unused( sc ) \ | |
7c673cae FG |
316 | /**/ |
317 | ||
318 | } // namespace utils | |
319 | } // namespace unit_test | |
320 | } // namespace boost | |
321 | ||
322 | #include <boost/test/detail/enable_warnings.hpp> | |
323 | ||
324 | #endif // BOOST_TEST_UTILS_SETCOLOR_HPP |