terminal.h

Go to the documentation of this file.

// Encapsulates terminal control (colors, cursor, ...)
// Author: Max Schwarz <max.schwarz@uni-bonn.de>

#ifndef TERMINAL_H
#define TERMINAL_H

#include <stdint.h>
#include <string>
#include <map>
#include <chrono>

namespace rosbag_fancy
{

class Terminal
{
public:
        enum SimpleColor
        {
                Black,
                Red,
                Green,
                Yellow,
                Blue,
                Magenta,
                Cyan,
                White,
                IntenseBlack,
                IntenseRed,
                IntenseGreen,
                IntenseYellow,
                IntenseBlue,
                IntenseMagenta,
                IntenseCyan,
                IntenseWhite,

                Grayscale = 0xe8
        };

        class Parser
        {
        public:
                Parser();

                void parse(char c);

                void parse(const std::string& str);

                void apply(Terminal* term);
        private:
                void parseSetAttributes(const std::string& str);

                enum State
                {
                        STATE_ESCAPE,
                        STATE_TYPE,
                        STATE_CSI
                };

                State m_state;
                std::string m_buf;

                int m_fgColor;
                int m_bgColor;
                bool m_bold;
        };

        Terminal();

        void setForegroundColor(uint32_t color);

        void setBackgroundColor(uint32_t color);

        void setCursorInvisible();

        void setCursorVisible();

        void setEcho(bool on);

        void setBold(bool on);


        void setSimpleBackground(SimpleColor color);
        void setSimpleForeground(SimpleColor color);
        void setSimplePair(SimpleColor fg, SimpleColor bg);

        void setStandardColors();

        void clearToEndOfLine();

        void moveCursorUp(int numLines);

        void moveCursorToStartOfLine();

        bool getSize(int* columns, int* rows);

        bool has256Colors() const;

        bool interactive() const
        { return m_valid; }

        void setWindowTitle(const std::string& title);
        void clearWindowTitle(const std::string& backup);


        enum SpecialKey
        {
                SK_F1 = 0x100, SK_F2, SK_F3, SK_F4,
                SK_F5, SK_F6, SK_F7, SK_F8,
                SK_F9, SK_F10, SK_F11, SK_F12,
                SK_Left, SK_Right, SK_Up, SK_Down, SK_Backspace
        };

        int readKey();

private:
        bool m_valid;
        bool m_256colors;
        bool m_truecolor;

        std::string m_bgColorStr;
        std::string m_fgColorStr;
        std::string m_opStr;
        std::string m_sgr0Str;
        std::string m_elStr;
        std::string m_upStr;
        std::string m_boldStr;

        std::map<std::string, SpecialKey> m_specialKeys;

        std::string m_currentEscapeStr;
        std::chrono::steady_clock::time_point m_escapeStartTime;
        bool m_currentEscapeAborted = false;
        unsigned int m_currentEscapeAbortIdx = 0;
};

}

#endif