ansi.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2022-2026 FLAK-ZOSO <mattia.marchese.2006@gmail.com>
 * SPDX-License-Identifier: GPL-3.0-only
 */
 
#pragma once
 
#include <variant>
#include <bitset>
#include <string>
 
 
#define ESC "\x1b"
#define CSI "\x1b["
#define CLS "\x1b[2J"
#define SSB "\x1b[3J"
#define HIDE_CURSOR "\x1b[?25l"
#define SHOW_CURSOR "\x1b[?25h"
#define TL "\x1b[H"
 
 
namespace sista {
    enum class ForegroundColor: int {
        BLACK = 30,
        RED = 31,
        GREEN = 32,
        YELLOW = 33,
        BLUE = 34,
        MAGENTA = 35,
        CYAN = 36,
        WHITE = 37
    };
    enum class BackgroundColor: int {
        BLACK = 40,
        RED = 41,
        GREEN = 42,
        YELLOW = 43,
        BLUE = 44,
        MAGENTA = 45,
        CYAN = 46,
        WHITE = 47
    };
    enum class Attribute: int {
        RESET = 0,
        BRIGHT = 1,
        FAINT = 2,
        ITALIC = 3,
        UNDERSCORE = 4,
        BLINK = 5,
        RAPID_BLINK = 6,
        REVERSE = 7,
        HIDDEN = 8,
        STRIKETHROUGH = 9
    };
    inline std::bitset<10> make_attr_bitset(std::initializer_list<Attribute> attrs) {
        std::bitset<10> bits;
        for (auto attr : attrs) {
            bits.set(static_cast<int>(attr));
        }
        return bits;
    }
 
    struct RGBColor {
        unsigned char red; 
        unsigned char green; 
        unsigned char blue; 
        RGBColor();
        RGBColor(unsigned char, unsigned char, unsigned char);
    };
 
    void setForegroundColor(ForegroundColor);
    void setBackgroundColor(BackgroundColor);
    void setAttribute(Attribute);
    void resetAttribute(Attribute);
 
    void resetAnsi();
 
    void setForegroundColor(const RGBColor&);
    void setBackgroundColor(const RGBColor&);
    void setForegroundColor(unsigned char, unsigned char, unsigned char);
    void setBackgroundColor(unsigned char, unsigned char, unsigned char);
    void setForegroundColor(unsigned char);
    void setBackgroundColor(unsigned char);
 
    std::string fgColorStr(ForegroundColor);
    std::string bgColorStr(BackgroundColor);
    std::string attrStr(Attribute);
    std::string fgColorStr(const RGBColor&);
    std::string bgColorStr(const RGBColor&);
    std::string fgColorStr(unsigned char, unsigned char, unsigned char);
    std::string bgColorStr(unsigned char, unsigned char, unsigned char);
 
    enum class ScreenMode: int {
        MONOCROME_TEXT_40_25 = 0,
        COLOR_TEXT_40_25 = 1,
        MONOCROME_TEXT_80_25 = 2,
        COLOR_TEXT_80_25 = 3,
        FOUR_COLORS_GRAPHICS_320_200 = 4,
        MONOCROME_GRAPHICS_320_200 = 5,
        MONOCROME_GRAPHICS_640_200 = 6,
        LINE_WRAPPING = 7,
        COLOR_GRAPHICS_320_200 = 13,
        COLOR_16_COLORS_GRAPHICS_640_200 = 14,
        MONOCROME_2_COLORS_GRAPHICS_640_350 = 15,
        COLOR_16_COLORS_GRAPHICS_640_350 = 16,
        MONOCROME_2_COLORS_GRAPHICS_640_480 = 17,
        COLOR_16_COLORS_GRAPHICS_640_480 = 18,
        COLOR_256_COLORS_GRAPHICS_320_200 = 19
    };
 
    void setScreenMode(ScreenMode);
    void unsetScreenMode(ScreenMode);
 
    void printHyperlink(std::string, std::string);
 
    struct ANSISettings {
        std::variant<ForegroundColor, RGBColor> foregroundColor;
        std::variant<BackgroundColor, RGBColor> backgroundColor;
        std::variant<Attribute, std::bitset<10>> attribute;
 
        ANSISettings();
        ANSISettings(const RGBColor&, const RGBColor&, const Attribute&);
        ANSISettings(const ForegroundColor&, const BackgroundColor&, const Attribute&);
        ANSISettings(const RGBColor&, const BackgroundColor&, const Attribute&);
        ANSISettings(const ForegroundColor&, const RGBColor&, const Attribute&);
        ANSISettings(const RGBColor&, const RGBColor&, const std::bitset<10>&);
        ANSISettings(const ForegroundColor&, const RGBColor&, const std::bitset<10>&);
        ANSISettings(const RGBColor&, const BackgroundColor&, const std::bitset<10>&);
        ANSISettings(const ForegroundColor&, const BackgroundColor&, std::initializer_list<Attribute>);
        ANSISettings(const ForegroundColor&, const RGBColor&, std::initializer_list<Attribute>);
        ANSISettings(const RGBColor&, const BackgroundColor&, std::initializer_list<Attribute>);
        ANSISettings(const ForegroundColor&, const BackgroundColor&, const std::bitset<10>&);
        ANSISettings(const RGBColor&, const RGBColor&, std::initializer_list<Attribute>);
        ANSISettings(const std::variant<ForegroundColor, RGBColor>&,
                     const std::variant<BackgroundColor, RGBColor>&,
                     const std::variant<Attribute, std::bitset<10>>&);
 
        void apply() const;
        void reset() const;
    };
};