1 //
2 // Copyright (c) 2019-2020 Advanced Micro Devices, Inc. All rights reserved.
3 //
4 // Permission is hereby granted, free of charge, to any person obtaining a copy
5 // of this software and associated documentation files (the "Software"), to deal
6 // in the Software without restriction, including without limitation the rights
7 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 // copies of the Software, and to permit persons to whom the Software is
9 // furnished to do so, subject to the following conditions:
10 //
11 // The above copyright notice and this permission notice shall be included in
12 // all copies or substantial portions of the Software.
13 //
14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
20 // THE SOFTWARE.
21 //
22
23 #include "Common.h"
24
ReadFile(std::vector<char> & out,const wchar_t * fileName)25 void ReadFile(std::vector<char>& out, const wchar_t* fileName)
26 {
27 std::ifstream file(fileName, std::ios::ate | std::ios::binary);
28 assert(file.is_open());
29 size_t fileSize = (size_t)file.tellg();
30 if(fileSize > 0)
31 {
32 out.resize(fileSize);
33 file.seekg(0);
34 file.read(out.data(), fileSize);
35 }
36 else
37 out.clear();
38 }
39
SaveFile(const wchar_t * filePath,const void * data,size_t dataSize)40 void SaveFile(const wchar_t* filePath, const void* data, size_t dataSize)
41 {
42 FILE* f = nullptr;
43 _wfopen_s(&f, filePath, L"wb");
44 if(f)
45 {
46 fwrite(data, 1, dataSize, f);
47 fclose(f);
48 }
49 else
50 assert(0);
51 }
52
SetConsoleColor(CONSOLE_COLOR color)53 void SetConsoleColor(CONSOLE_COLOR color)
54 {
55 WORD attr = 0;
56 switch(color)
57 {
58 case CONSOLE_COLOR::INFO:
59 attr = FOREGROUND_INTENSITY;;
60 break;
61 case CONSOLE_COLOR::NORMAL:
62 attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE;
63 break;
64 case CONSOLE_COLOR::WARNING:
65 attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
66 break;
67 case CONSOLE_COLOR::ERROR_:
68 attr = FOREGROUND_RED | FOREGROUND_INTENSITY;
69 break;
70 default:
71 assert(0);
72 }
73
74 HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
75 SetConsoleTextAttribute(out, attr);
76 }
77
PrintMessage(CONSOLE_COLOR color,const char * msg)78 void PrintMessage(CONSOLE_COLOR color, const char* msg)
79 {
80 if(color != CONSOLE_COLOR::NORMAL)
81 SetConsoleColor(color);
82
83 printf("%s\n", msg);
84
85 if (color != CONSOLE_COLOR::NORMAL)
86 SetConsoleColor(CONSOLE_COLOR::NORMAL);
87 }
88
PrintMessage(CONSOLE_COLOR color,const wchar_t * msg)89 void PrintMessage(CONSOLE_COLOR color, const wchar_t* msg)
90 {
91 if(color != CONSOLE_COLOR::NORMAL)
92 SetConsoleColor(color);
93
94 wprintf(L"%s\n", msg);
95
96 if (color != CONSOLE_COLOR::NORMAL)
97 SetConsoleColor(CONSOLE_COLOR::NORMAL);
98 }
99
100 static const size_t CONSOLE_SMALL_BUF_SIZE = 256;
101
PrintMessageV(CONSOLE_COLOR color,const char * format,va_list argList)102 void PrintMessageV(CONSOLE_COLOR color, const char* format, va_list argList)
103 {
104 size_t dstLen = (size_t)::_vscprintf(format, argList);
105 if(dstLen)
106 {
107 bool useSmallBuf = dstLen < CONSOLE_SMALL_BUF_SIZE;
108 char smallBuf[CONSOLE_SMALL_BUF_SIZE];
109 std::vector<char> bigBuf(useSmallBuf ? 0 : dstLen + 1);
110 char* bufPtr = useSmallBuf ? smallBuf : bigBuf.data();
111 ::vsprintf_s(bufPtr, dstLen + 1, format, argList);
112 PrintMessage(color, bufPtr);
113 }
114 }
115
PrintMessageV(CONSOLE_COLOR color,const wchar_t * format,va_list argList)116 void PrintMessageV(CONSOLE_COLOR color, const wchar_t* format, va_list argList)
117 {
118 size_t dstLen = (size_t)::_vcwprintf(format, argList);
119 if(dstLen)
120 {
121 bool useSmallBuf = dstLen < CONSOLE_SMALL_BUF_SIZE;
122 wchar_t smallBuf[CONSOLE_SMALL_BUF_SIZE];
123 std::vector<wchar_t> bigBuf(useSmallBuf ? 0 : dstLen + 1);
124 wchar_t* bufPtr = useSmallBuf ? smallBuf : bigBuf.data();
125 ::vswprintf_s(bufPtr, dstLen + 1, format, argList);
126 PrintMessage(color, bufPtr);
127 }
128 }
129
PrintMessageF(CONSOLE_COLOR color,const char * format,...)130 void PrintMessageF(CONSOLE_COLOR color, const char* format, ...)
131 {
132 va_list argList;
133 va_start(argList, format);
134 PrintMessageV(color, format, argList);
135 va_end(argList);
136 }
137
PrintMessageF(CONSOLE_COLOR color,const wchar_t * format,...)138 void PrintMessageF(CONSOLE_COLOR color, const wchar_t* format, ...)
139 {
140 va_list argList;
141 va_start(argList, format);
142 PrintMessageV(color, format, argList);
143 va_end(argList);
144 }
145
PrintWarningF(const char * format,...)146 void PrintWarningF(const char* format, ...)
147 {
148 va_list argList;
149 va_start(argList, format);
150 PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
151 va_end(argList);
152 }
153
PrintWarningF(const wchar_t * format,...)154 void PrintWarningF(const wchar_t* format, ...)
155 {
156 va_list argList;
157 va_start(argList, format);
158 PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
159 va_end(argList);
160 }
161
PrintErrorF(const char * format,...)162 void PrintErrorF(const char* format, ...)
163 {
164 va_list argList;
165 va_start(argList, format);
166 PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
167 va_end(argList);
168 }
169
PrintErrorF(const wchar_t * format,...)170 void PrintErrorF(const wchar_t* format, ...)
171 {
172 va_list argList;
173 va_start(argList, format);
174 PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
175 va_end(argList);
176 }
177