color support

1 files changed, 45 insertions, 0 deletions

diff --git a/Color.cc b/Color.cc
new file mode 100644
index 0000000..55fa3d5
--- /dev/null
+++ b/Color.cc
@@ -0,0 +1,45 @@
+#ifdef _MSC_VER
+#include <Windows.h>
+#endif
+
+#include <map>
+
+#include "Color.hh"
+
+namespace pcc {
+#if _MSC_VER
+static const std::map<Color, WORD> ColorMap = {
+    {Color::BLACK, 0},
+    {Color::RED, FOREGROUND_INTENSITY | FOREGROUND_RED},
+    {Color::GREEN, FOREGROUND_INTENSITY | FOREGROUND_GREEN},
+    {Color::BLUE, FOREGROUND_INTENSITY | FOREGROUND_BLUE},
+    {Color::YELLOW, FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN},
+    {Color::CYAN, FOREGROUND_INTENSITY | FOREGROUND_GREEN | FOREGROUND_BLUE},
+    {Color::MAGENTA, FOREGROUND_INTENSITY | FOREGROUND_BLUE | FOREGROUND_RED},
+    {Color::WHITE, FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN |
+                       FOREGROUND_BLUE},
+    {Color::DEFAULT, FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE}};
+#else
+static const std::map<Color, const char*> ColorMap = {
+    {Color::BLACK, "\e[30;1m"},   {Color::RED, "\e[31;1m"},
+    {Color::GREEN, "\e[32;1m"},   {Color::BLUE, "\e[34;1m"},
+    {Color::YELLOW, "\e[33;1m"},  {Color::CYAN, "\e[36;1m"},
+    {Color::MAGENTA, "\e[35;1m"}, {Color::WHITE, "\e[37;1m"},
+    {Color::DEFAULT, "\e[0m"}};
+#endif // _MSC_VER
+
+std::ostream& operator<<(std::ostream& out, Color color) {
+#ifdef _MSC_VER
+    HANDLE h = GetStdHandle(STD_ERROR_HANDLE);
+    CONSOLE_SCREEN_BUFFER_INFO info;
+    GetConsoleScreenBufferInfo(h, &info);
+    WORD attr = info.wAttributes;
+    attr &= ~(FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN |
+              FOREGROUND_BLUE);
+    SetConsoleTextAttribute(h, attr | ColorMap.at(color));
+#else
+    out << ColorMap.at(color);
+#endif // _MSC_VER
+    return out;
+}
+} // namespace pcc