console_color.cpp

Go to the documentation of this file.

#include "console_color.h"
#include <iostream>
 
namespace pesieve {
    namespace util {
 
        bool get_current_color(int descriptor, WORD &color) {
            CONSOLE_SCREEN_BUFFER_INFO info;
            if (!GetConsoleScreenBufferInfo(GetStdHandle(descriptor), &info))
                return false;
            color = info.wAttributes;
            return true;
        }
 
    }
};
 
void pesieve::util::print_in_color(int color, const std::string &text, bool is_error)
{
    int descriptor = is_error ? STD_ERROR_HANDLE : STD_OUTPUT_HANDLE;
    std::ostream &stream = is_error ? std::cerr : std::cout;
 
    WORD old_color = 7; //default
    get_current_color(descriptor, old_color);
 
    HANDLE hConsole = GetStdHandle(descriptor);
    FlushConsoleInputBuffer(hConsole);
    SetConsoleTextAttribute(hConsole, color); // back to default color
 
    stream << text;
 
    FlushConsoleInputBuffer(hConsole);
    SetConsoleTextAttribute(hConsole, old_color); // back to default color
    FlushConsoleInputBuffer(hConsole);
 
    stream.flush();
}