Colour.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
#include <ostream>
#include "Environment.hpp"
 
namespace stc::colour {
 
enum class FourBitColour {
    BLACK = 30,
    RED = 31,
    GREEN = 32,
    YELLOW = 33,
    BLUE = 34,
    MAGENTA = 35,
    CYAN = 36,
    WHITE = 37,
 
    BRIGHT_BLACK = 90,
    BRIGHT_RED = 91,
    BRIGHT_GREEN = 92,
    BRIGHT_YELLOW = 93,
    BRIGHT_BLUE = 94,
    BRIGHT_MAGENTA = 95,
    BRIGHT_CYAN = 96,
    BRIGHT_WHITE = 97,
};
 
enum class Typography {
    BOLD = 1,
    FAINT = 2,
    ITALIC = 3,
    UNDERLINE = 4,
    SLOW_BLINK = 5,
 
    RESET_INTENSITY = 22,
    NO_ITALIC = 23,
    NO_UNDERLINE = 24,
    NO_BLINKING = 25
};
 
namespace _detail {
 
const int FOREGROUND = 38;
const int BACKGROUND = 48;
 
// forced: iword defaults to 0
const int MODE_AUTO = 0;
const int MODE_FORCE = 1;
 
 
inline int getStreamConfigIdx() {
    static int idx = std::ios::xalloc();
    return idx;
}
 
template <typename CharT>
static bool shouldPrintColour(std::basic_ostream<CharT>& ss) {
    auto iword = ss.iword(getStreamConfigIdx());
    if (iword == MODE_FORCE) {
        return true;
    }
 
    return isCppStreamTTY(ss);
}
 
template <int Mode>
struct Colouriser {
    template <FourBitColour Colour, typename CharT>
    static constexpr std::basic_ostream<CharT>& fourBit(std::basic_ostream<CharT>& stream) {
        if (shouldPrintColour(stream)) {
            stream << "\033["
                << (Mode == _detail::FOREGROUND ? static_cast<int>(Colour) : (static_cast<int>(Colour) + 10))
                << "m";
        }
        return stream;
    }
 
    template <uint8_t code, typename CharT>
    static constexpr std::basic_ostream<CharT>& eightBit(std::basic_ostream<CharT>& stream) {
        if (shouldPrintColour(stream)) {
            stream << "\033[" << Mode << ";5;" << code << "m";
        }
        return stream;
    }
 
    template <uint8_t r, uint8_t g, uint8_t b, typename CharT>
    static constexpr std::basic_ostream<CharT>& truecolour(std::basic_ostream<CharT>& stream) {
        if (shouldPrintColour(stream)) {
            stream << "\033[" << Mode << ";2;" << r << ";" << g << ";" << b << "m";
        }
        return stream;
    }
};
 
}
 
using FgColour = _detail::Colouriser<_detail::FOREGROUND>;
using BgColour = _detail::Colouriser<_detail::BACKGROUND>;
 
template <Typography feature, typename CharT>
static constexpr std::basic_ostream<CharT>& use(std::basic_ostream<CharT>& stream) {
    if (_detail::shouldPrintColour(stream)) {
        stream << "\033[" << static_cast<int>(feature) << "m";
    }
    return stream;
}
 
template <bool val = true, typename CharT>
static constexpr std::basic_ostream<CharT>& force(std::basic_ostream<CharT>& stream) {
    stream.iword(_detail::getStreamConfigIdx()) = val ? _detail::MODE_FORCE : _detail::MODE_AUTO;
 
    return stream;
}
 
template <typename CharT>
static constexpr std::basic_ostream<CharT>& reset(std::basic_ostream<CharT>& stream) {
    if (_detail::shouldPrintColour(stream)) {
        stream << "\033[0m";
    }
    return stream;
}
 
template <FourBitColour Colour, typename CharT>
static constexpr std::basic_ostream<CharT>& fg(std::basic_ostream<CharT>& stream) {
    return FgColour::fourBit<Colour, CharT>(stream);
}
 
template <uint8_t code, typename CharT>
static constexpr std::basic_ostream<CharT>& fg(std::basic_ostream<CharT>& stream) {
    return FgColour::eightBit<code, CharT>(stream);
}
 
template <uint8_t r, uint8_t g, uint8_t b, typename CharT>
static constexpr std::basic_ostream<CharT>& fg(std::basic_ostream<CharT>& stream) {
    return FgColour::truecolour<r, g, b, CharT>(stream);
}
 
template <FourBitColour Colour, typename CharT>
static constexpr std::basic_ostream<CharT>& bg(std::basic_ostream<CharT>& stream) {
    return BgColour::fourBit<Colour, CharT>(stream);
}
 
template <uint8_t code, typename CharT>
static constexpr std::basic_ostream<CharT>& bg(std::basic_ostream<CharT>& stream) {
    return BgColour::eightBit<code, CharT>(stream);
}
 
template <uint8_t r, uint8_t g, uint8_t b, typename CharT>
static constexpr std::basic_ostream<CharT>& bg(std::basic_ostream<CharT>& stream) {
    return BgColour::truecolour<r, g, b, CharT>(stream);
}
 
 
}