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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
#ifdef _WIN32
#include <iostream>
#include "windows.hxx"
#include "conversion.hxx"
#include "io.hxx"
#ifndef ENABLE_VIRTUAL_TERMINAL_PROCESSING
static DWORD const ENABLE_VIRTUAL_TERMINAL_PROCESSING = 4;
#endif
using namespace std;
namespace replxx {
WinAttributes WIN_ATTR;
template<typename T>
T* HandleEsc(T* p, T* end) {
if (*p == '[') {
int code = 0;
int thisBackground( WIN_ATTR._defaultBackground );
for (++p; p < end; ++p) {
char32_t c = *p;
if ('0' <= c && c <= '9') {
code = code * 10 + (c - '0');
} else if (c == 'm' || c == ';') {
switch (code) {
case 0:
WIN_ATTR._consoleAttribute = WIN_ATTR._defaultAttribute;
WIN_ATTR._consoleColor = WIN_ATTR._defaultColor | thisBackground;
break;
case 1: // BOLD
case 5: // BLINK
WIN_ATTR._consoleAttribute = (WIN_ATTR._defaultAttribute ^ FOREGROUND_INTENSITY) & INTENSITY;
break;
case 22:
WIN_ATTR._consoleAttribute = WIN_ATTR._defaultAttribute;
break;
case 30:
case 90:
WIN_ATTR._consoleColor = thisBackground;
break;
case 31:
case 91:
WIN_ATTR._consoleColor = FOREGROUND_RED | thisBackground;
break;
case 32:
case 92:
WIN_ATTR._consoleColor = FOREGROUND_GREEN | thisBackground;
break;
case 33:
case 93:
WIN_ATTR._consoleColor = FOREGROUND_RED | FOREGROUND_GREEN | thisBackground;
break;
case 34:
case 94:
WIN_ATTR._consoleColor = FOREGROUND_BLUE | thisBackground;
break;
case 35:
case 95:
WIN_ATTR._consoleColor = FOREGROUND_BLUE | FOREGROUND_RED | thisBackground;
break;
case 36:
case 96:
WIN_ATTR._consoleColor = FOREGROUND_BLUE | FOREGROUND_GREEN | thisBackground;
break;
case 37:
case 97:
WIN_ATTR._consoleColor = FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_BLUE | thisBackground;
break;
case 101:
thisBackground = BACKGROUND_RED;
break;
}
if ( ( code >= 90 ) && ( code <= 97 ) ) {
WIN_ATTR._consoleAttribute = (WIN_ATTR._defaultAttribute ^ FOREGROUND_INTENSITY) & INTENSITY;
}
code = 0;
}
if (*p == 'm') {
++p;
break;
}
}
} else {
++p;
}
auto handle = GetStdHandle(STD_OUTPUT_HANDLE);
SetConsoleTextAttribute(
handle,
WIN_ATTR._consoleAttribute | WIN_ATTR._consoleColor
);
return p;
}
int win_write( char const* str_, int size_ ) {
int count( 0 );
DWORD currentMode( 0 );
HANDLE consoleOut( GetStdHandle( STD_OUTPUT_HANDLE ) );
if ( tty::out && GetConsoleMode( consoleOut, ¤tMode ) ) {
UINT inputCodePage( GetConsoleCP() );
UINT outputCodePage( GetConsoleOutputCP() );
SetConsoleCP( 65001 );
SetConsoleOutputCP( 65001 );
DWORD nWritten( 0 );
if ( SetConsoleMode( consoleOut, currentMode | ENABLE_VIRTUAL_TERMINAL_PROCESSING ) ) {
WriteConsoleA( consoleOut, str_, size_, &nWritten, nullptr );
count = nWritten;
SetConsoleMode( consoleOut, currentMode );
} else {
char const* s( str_ );
char const* e( str_ + size_ );
while ( str_ < e ) {
if ( *str_ == 27 ) {
if ( s < str_ ) {
int toWrite( str_ - s );
WriteConsoleA( consoleOut, s, static_cast<DWORD>( toWrite ), &nWritten, nullptr );
count += nWritten;
if ( nWritten != toWrite ) {
s = str_ = nullptr;
break;
}
}
s = HandleEsc( str_ + 1, e );
int escaped( s - str_);
count += escaped;
str_ = s;
} else {
++ str_;
}
}
if ( s < str_ ) {
WriteConsoleA( consoleOut, s, static_cast<DWORD>( str_ - s ), &nWritten, nullptr );
count += nWritten;
}
}
SetConsoleCP( inputCodePage );
SetConsoleOutputCP( outputCodePage );
} else {
count = _write( 1, str_, size_ );
}
return ( count );
}
}
#endif
|
