mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-11-22 11:40:05 +00:00
137953538a
Previously we required _PRINT to enable _COLOR, which forbids outputting colored disassembly into a string in library. This commit will allow library users to request enabling ANSI color escape sequences.
126 lines
3.2 KiB
C++
126 lines
3.2 KiB
C++
// Copyright (c) 2015-2016 The Khronos Group Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include "print.h"
|
|
|
|
#if defined(SPIRV_ANDROID) || defined(SPIRV_LINUX) || defined(SPIRV_MAC) || \
|
|
defined(SPIRV_FREEBSD)
|
|
namespace libspirv {
|
|
|
|
clr::reset::operator const char*() { return "\x1b[0m"; }
|
|
|
|
clr::grey::operator const char*() { return "\x1b[1;30m"; }
|
|
|
|
clr::red::operator const char*() { return "\x1b[31m"; }
|
|
|
|
clr::green::operator const char*() { return "\x1b[32m"; }
|
|
|
|
clr::yellow::operator const char*() { return "\x1b[33m"; }
|
|
|
|
clr::blue::operator const char*() { return "\x1b[34m"; }
|
|
|
|
} // namespace libspirv
|
|
#elif defined(SPIRV_WINDOWS)
|
|
#include <windows.h>
|
|
|
|
namespace libspirv {
|
|
|
|
static void SetConsoleForegroundColorPrimary(HANDLE hConsole, WORD color) {
|
|
// Get screen buffer information from console handle
|
|
CONSOLE_SCREEN_BUFFER_INFO bufInfo;
|
|
GetConsoleScreenBufferInfo(hConsole, &bufInfo);
|
|
|
|
// Get background color
|
|
color = WORD(color | (bufInfo.wAttributes & 0xfff0));
|
|
|
|
// Set foreground color
|
|
SetConsoleTextAttribute(hConsole, color);
|
|
}
|
|
|
|
static void SetConsoleForegroundColor(WORD color) {
|
|
SetConsoleForegroundColorPrimary(GetStdHandle(STD_OUTPUT_HANDLE), color);
|
|
SetConsoleForegroundColorPrimary(GetStdHandle(STD_ERROR_HANDLE), color);
|
|
}
|
|
|
|
clr::reset::operator const char*() {
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(0xf);
|
|
return "";
|
|
}
|
|
return "\x1b[0m";
|
|
}
|
|
|
|
clr::grey::operator const char*() {
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(FOREGROUND_INTENSITY);
|
|
return "";
|
|
}
|
|
return "\x1b[1;30m";
|
|
}
|
|
|
|
clr::red::operator const char*() {
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(FOREGROUND_RED);
|
|
return "";
|
|
}
|
|
return "\x1b[31m";
|
|
}
|
|
|
|
clr::green::operator const char*() {
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(FOREGROUND_GREEN);
|
|
return "";
|
|
}
|
|
return "\x1b[32m";
|
|
}
|
|
|
|
clr::yellow::operator const char*() {
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(FOREGROUND_RED | FOREGROUND_GREEN);
|
|
return "";
|
|
}
|
|
return "\x1b[33m";
|
|
}
|
|
|
|
clr::blue::operator const char*() {
|
|
// Blue all by itself is hard to see against a black background (the
|
|
// default on command shell), or a medium blue background (the default
|
|
// on PowerShell). So increase its intensity.
|
|
|
|
if (isPrint) {
|
|
SetConsoleForegroundColor(FOREGROUND_BLUE | FOREGROUND_INTENSITY);
|
|
return "";
|
|
}
|
|
return "\x1b[94m";
|
|
}
|
|
|
|
} // namespace libspirv
|
|
#else
|
|
namespace libspirv {
|
|
|
|
clr::reset::operator const char*() { return ""; }
|
|
|
|
clr::grey::operator const char*() { return ""; }
|
|
|
|
clr::red::operator const char*() { return ""; }
|
|
|
|
clr::green::operator const char*() { return ""; }
|
|
|
|
clr::yellow::operator const char*() { return ""; }
|
|
|
|
clr::blue::operator const char*() { return ""; }
|
|
|
|
} // namespace libspirv
|
|
#endif
|