trmclr.hpp

#ifndef __TRMCLR_H__
#define __TRMCLR_H__

#include <cstdint>
#include <ostream>
#include <sstream>
#include <limits>
#include <string>

namespace trmclr
{

struct Style
{
    Style(uint32_t value) : _value(value) {}

    operator uint32_t() const { return _value; }

    uint32_t _value;
};

enum StyleTypes
{
    FOREGROUND,
    ATTRIBUTE,
    BACKGROUND,
    N_STYLE_TYPES
};

static const uint32_t STYLE_SHIFT =  std::numeric_limits<uint32_t>::digits / N_STYLE_TYPES;

struct Attribute
{
    static const uint32_t SHIFT = STYLE_SHIFT * ATTRIBUTE;

    enum
    {
        DEFAULT     = 0x001 << SHIFT,
        BOLD        = 0x002 << SHIFT,
        DIM         = 0x004 << SHIFT,
        UNDERLINED  = 0x010 << SHIFT,
        BLINK       = 0x020 << SHIFT,
        REVERSE     = 0x080 << SHIFT,
        HIDDEN      = 0x100 << SHIFT
    };
};

struct Foreground
{
    static const uint32_t SHIFT = STYLE_SHIFT * FOREGROUND;

    enum
    {
        BLACK           = 30 << SHIFT,
        RED             = 31 << SHIFT,
        GREEN           = 32 << SHIFT,
        YELLOW          = 33 << SHIFT,
        BLUE            = 34 << SHIFT,
        MAGENTA         = 35 << SHIFT,
        CYAN            = 36 << SHIFT,
        LIGHT_GRAY      = 37 << SHIFT,
        DEFAULT         = 39 << SHIFT,
        DARK_GRAY       = 90 << SHIFT,
        LIGHT_RED       = 91 << SHIFT,
        LIGHT_GREEN     = 92 << SHIFT,
        LIGHT_YELLOW    = 93 << SHIFT,
        LIGHT_BLUE      = 94 << SHIFT,
        LIGHT_MAGENTA   = 95 << SHIFT,
        LIGHT_CYAN      = 96 << SHIFT,
        WHITE           = 97 << SHIFT
    };
};

struct Background
{
    static const uint32_t SHIFT = STYLE_SHIFT * BACKGROUND;

    enum
    {
        BLACK           =  40 << SHIFT,
        RED             =  41 << SHIFT,
        GREEN           =  42 << SHIFT,
        YELLOW          =  43 << SHIFT,
        BLUE            =  44 << SHIFT,
        MAGENTA         =  45 << SHIFT,
        CYAN            =  46 << SHIFT,
        LIGHT_GRAY      =  47 << SHIFT,
        DEFAULT         =  49 << SHIFT,
        DARK_GRAY       = 100 << SHIFT,
        LIGHT_RED       = 101 << SHIFT,
        LIGHT_GREEN     = 102 << SHIFT,
        LIGHT_YELLOW    = 103 << SHIFT,
        LIGHT_BLUE      = 104 << SHIFT,
        LIGHT_MAGENTA   = 105 << SHIFT,
        LIGHT_CYAN      = 106 << SHIFT,
        WHITE           = 107 << SHIFT
    };
};

std::ostream& operator<< (std::ostream& os, const Style& style)
{
    const uint32_t base = 1 << STYLE_SHIFT;
          uint32_t encoded = style / base;
          uint32_t decoded = style % base;
    
    os << "\x1B[" << (decoded ? decoded : Foreground::DEFAULT >> Foreground::SHIFT);

    decoded = encoded % base;

    for (uint32_t i = 0; decoded != 0; decoded >>= 1, i++)
    {
        if (decoded & 1)
        {
            os << ";" << i;
        }
    }

    encoded = encoded / base;
    decoded = encoded % base;

    os << ";" << (decoded ? decoded : Background::DEFAULT >> Background::SHIFT) << "m";

    return os;
}

} // end trmclr namespace

#endif // end __TRMCLR_H__