style.h

Go to the documentation of this file.

#ifndef KOUTIL_TERM_STYLE_H
#define KOUTIL_TERM_STYLE_H
 
#include "koutil/util/utils.h"
#include <array>
#include <cstdint>
#include <ostream>
#include <string_view>
#include <type_traits>
 
namespace koutil::term {
 
enum class Style : std::uint8_t {
    NONE         = 0, 
    BOLD         = 1 << 0, 
    DIM          = 1 << 1, 
    ITALIC       = 1 << 2, 
    UNDERLINE    = 1 << 3, 
    BLINK        = 1 << 4, 
    INVERSE      = 1 << 5, 
    HIDDEN       = 1 << 6, 
    STRIKETHOUGH = 1 << 7, 
};
 
constexpr Style operator+(Style a, Style b) {
    using enum_type = std::underlying_type_t<Style>;
 
    return static_cast<Style>(static_cast<enum_type>(a) | static_cast<enum_type>(b));
}
 
constexpr Style operator-(Style a, Style b) {
    using enum_type = std::underlying_type_t<Style>;
    return static_cast<Style>(static_cast<enum_type>(a) ^ (static_cast<enum_type>(b) & static_cast<enum_type>(a)));
}
 
constexpr Style operator&(Style a, Style b) {
    using enum_type = std::underlying_type_t<Style>;
    return static_cast<Style>(static_cast<enum_type>(a) & static_cast<enum_type>(b));
}
 
constexpr bool style_contains(Style style, Style flags) { return (style & flags) == flags; }
 
std::ostream& operator<<(std::ostream& stream, Style style) {
    using namespace std::string_view_literals;
    if (style == Style::NONE) {
        return stream;
    }
 
    /*
    00 -> "22"
    01 -> "22;1"
    10 -> "22;2"
    11 -> "1;2"
    */
    constexpr auto bold_dim_lookup = std::to_array<std::string_view>({ "22"sv, "22;1"sv, "22;2"sv, "1;2"sv });
 
    stream << util::ESC << '[' << bold_dim_lookup[static_cast<int>(style & (Style::BOLD + Style::DIM))];
 
    if (style_contains(style, Style::ITALIC)) {
        stream << ";3";
    } else {
        stream << ";23";
    }
 
    if (style_contains(style, Style::UNDERLINE)) {
        stream << ";4";
    } else {
        stream << ";24";
    }
 
    if (style_contains(style, Style::BLINK)) {
        stream << ";5";
    } else {
        stream << ";25";
    }
 
    if (style_contains(style, Style::INVERSE)) {
        stream << ";7";
    } else {
        stream << ";27";
    }
 
    if (style_contains(style, Style::HIDDEN)) {
        stream << ";8";
    } else {
        stream << ";28";
    }
 
    if (style_contains(style, Style::STRIKETHOUGH)) {
        stream << ";9m";
    } else {
        stream << ";29m";
    }
 
    return stream;
}
 
std::ostream& reset_all(std::ostream& stream) {
    stream << util::ESC << "[0m";
    return stream;
}
 
}
 
#endif