git上发现一个很有趣的代码

freeeyes

这段代码非常有趣。
可以给printf上色，纯属娱乐，不过非常好玩。在这记录一下。

1. /*----------------------------------------------
2. File: printfc.hh -- printf with color
3. Date: 2014/2/24 16:52:35
4. Author: js2854@gmail.com
5. ----------------------------------------------*/
7. #ifndef __PRINTFC_H__
8. #define __PRINTFC_H__
10. #include <stdio.h>
12. #define MAX_STR_LEN                2048
14. #ifndef WIN32
15. #include <stdarg.h> //vsnprintf,va_start,va_end
16. #include <string.h> //strok
17. const typedef enum _ForeColor
18. {
19.         FG_BLACK                = 30, //黑色
20.         FG_RED                        = 31, //红色
21.         FG_GREEN                = 32, //绿色
22.         FG_YELLOW                = 33, //黄色
23.         FG_BLUE                        = 34, //蓝色
24.         FG_PURPLE                = 35, //紫色
25.         FG_DARKGREEN        = 36, //深绿色
26.         FG_WHITE                = 37, //白色
27. }ForeColor;
29. const typedef enum _BackColor
30. {
31.         BG_BLACK                = 40, //黑色
32.         BG_DARKRED                = 41, //深红色
33.         BG_GREEN                = 42, //绿色
34.         BG_YELLOW                = 43, //黄色
35.         BG_BLUE                        = 44, //蓝色
36.         BG_PURPLE                = 45, //紫色
37.         BG_DARKGREEN        = 46, //深绿色
38.         BG_WHITE                = 47, //白色
39. }BackColor;
41. #else
42. #include <windows.h>
43. const typedef enum _ForeColor
44. {
45.         FG_BLACK                = 0, //黑色
46.         FG_RED                        = FOREGROUND_INTENSITY | FOREGROUND_RED, //红色
47.         FG_GREEN                = FOREGROUND_INTENSITY | FOREGROUND_GREEN, //绿色
48.         FG_YELLOW                = FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN, //黄色
49.         FG_BLUE                        = FOREGROUND_INTENSITY | FOREGROUND_BLUE, //蓝色
50.         FG_PURPLE                = FOREGROUND_RED | FOREGROUND_BLUE, //紫色
51.         FG_DARKGREEN        = FOREGROUND_GREEN, //深绿色
52.         FG_WHITE                = FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE, //白色
53. }ForeColor;
55. const typedef enum _BackColor
56. {
57.         BG_BLACK                = 0, //黑色
58.         BG_DARKRED                = BACKGROUND_RED, //深红色
59.         BG_GREEN                = BACKGROUND_INTENSITY | BACKGROUND_GREEN, //绿色
60.         BG_YELLOW                = BACKGROUND_INTENSITY | BACKGROUND_RED | BACKGROUND_GREEN, //黄色
61.         BG_BLUE                        = BACKGROUND_INTENSITY | BACKGROUND_BLUE, //蓝色
62.         BG_PURPLE                = BACKGROUND_RED | BACKGROUND_BLUE, //紫色
63.         BG_DARKGREEN        = BACKGROUND_GREEN, //深绿色
64.         BG_WHITE                = BACKGROUND_INTENSITY | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE, //白色
65. }BackColor;
67. #endif
69. int printfc(ForeColor fc, const char* format_str, ...)
70. {
71.         int len = 0;
72. #ifndef WIN32
73.         printf("\e[%dm", fc);
74. #else
75.         CONSOLE_SCREEN_BUFFER_INFO oldInfo = {0};
76.         HANDLE hStd = ::GetStdHandle(STD_OUTPUT_HANDLE);
77.         if (hStd != INVALID_HANDLE_VALUE && GetConsoleScreenBufferInfo(hStd, &oldInfo))
78.         {
79.                 SetConsoleTextAttribute(hStd, fc);
80. #endif
81.        
82.                 va_list p_list;
83.                 va_start(p_list, format_str);
84.                 len = vprintf(format_str, p_list);
85.                 va_end(p_list);
87. #ifndef WIN32
88.                 printf("\e[0m");//关闭所有属性
89. #else
90.                 SetConsoleTextAttribute(hStd, oldInfo.wAttributes);
91.         }
92. #endif
94.         return len;
95. }
97. int printfc(ForeColor fc, BackColor bc, const char* format_str, ...)
98. {
99.         int len = 0;
100. #ifndef WIN32
101.         static char text[MAX_STR_LEN] = {0};
103.         memset(text, 0, sizeof(text));
105.         va_list p_list;
106.         va_start(p_list, format_str);
107.         len = vsnprintf(text, sizeof(text), format_str, p_list);
108.         va_end(p_list);
110.         const char *split = "\n";
111.         char *p = strtok(text, split);
112.         bool last_is_lf = (text[len-1] == '\n');
113.         while(p != NULL)
114.         {
115.                 printf("\e[%d;%dm%s\e[0m", fc, bc, p);
116.                 p = strtok(NULL, split);
118.                 if (p != NULL || last_is_lf) printf("\n");
119.         }
120. #else
121.         CONSOLE_SCREEN_BUFFER_INFO oldInfo = {0};
122.         HANDLE hStd = ::GetStdHandle(STD_OUTPUT_HANDLE);
123.         if (hStd != INVALID_HANDLE_VALUE && GetConsoleScreenBufferInfo(hStd, &oldInfo))
124.         {
125.                 SetConsoleTextAttribute(hStd, fc|bc);
126.                
127.                 va_list p_list;
128.                 va_start(p_list, format_str);
129.                 len = vprintf(format_str, p_list);
130.                 va_end(p_list);
131.                
132.                 SetConsoleTextAttribute(hStd, oldInfo.wAttributes);
133.         }
134. #endif
135.        
136.         return len;
137. }
139. #endif //__PRINTFC_H__

复制代码