| //===---------------------AnsiTerminal.h ------------------------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #define ANSI_FG_COLOR_BLACK 30 |
| #define ANSI_FG_COLOR_RED 31 |
| #define ANSI_FG_COLOR_GREEN 32 |
| #define ANSI_FG_COLOR_YELLOW 33 |
| #define ANSI_FG_COLOR_BLUE 34 |
| #define ANSI_FG_COLOR_PURPLE 35 |
| #define ANSI_FG_COLOR_CYAN 36 |
| #define ANSI_FG_COLOR_WHITE 37 |
| |
| #define ANSI_BG_COLOR_BLACK 40 |
| #define ANSI_BG_COLOR_RED 41 |
| #define ANSI_BG_COLOR_GREEN 42 |
| #define ANSI_BG_COLOR_YELLOW 43 |
| #define ANSI_BG_COLOR_BLUE 44 |
| #define ANSI_BG_COLOR_PURPLE 45 |
| #define ANSI_BG_COLOR_CYAN 46 |
| #define ANSI_BG_COLOR_WHITE 47 |
| |
| #define ANSI_SPECIAL_FRAMED 51 |
| #define ANSI_SPECIAL_ENCIRCLED 52 |
| |
| #define ANSI_CTRL_NORMAL 0 |
| #define ANSI_CTRL_BOLD 1 |
| #define ANSI_CTRL_FAINT 2 |
| #define ANSI_CTRL_ITALIC 3 |
| #define ANSI_CTRL_UNDERLINE 4 |
| #define ANSI_CTRL_SLOW_BLINK 5 |
| #define ANSI_CTRL_FAST_BLINK 6 |
| #define ANSI_CTRL_IMAGE_NEGATIVE 7 |
| #define ANSI_CTRL_CONCEAL 8 |
| #define ANSI_CTRL_CROSSED_OUT 9 |
| |
| #define ANSI_ESC_START "\033[" |
| #define ANSI_ESC_END "m" |
| |
| #define ANSI_STR(s) #s |
| #define ANSI_DEF_STR(s) ANSI_STR(s) |
| |
| #define ANSI_ESCAPE1(s) ANSI_ESC_START ANSI_DEF_STR(s) ANSI_ESC_END |
| |
| #define ANSI_1_CTRL(ctrl1) "\033["##ctrl1 ANSI_ESC_END |
| #define ANSI_2_CTRL(ctrl1, ctrl2) "\033["##ctrl1 ";"##ctrl2 ANSI_ESC_END |
| |
| #include "llvm/ADT/ArrayRef.h" |
| #include "llvm/ADT/STLExtras.h" |
| #include "llvm/ADT/StringRef.h" |
| |
| #include <string> |
| |
| namespace lldb_utility { |
| |
| namespace ansi { |
| |
| inline std::string FormatAnsiTerminalCodes(llvm::StringRef format, |
| bool do_color = true) { |
| // Convert "${ansi.XXX}" tokens to ansi values or clear them if do_color is |
| // false. |
| static const struct { |
| const char *name; |
| const char *value; |
| } g_color_tokens[] = { |
| #define _TO_STR2(_val) #_val |
| #define _TO_STR(_val) _TO_STR2(_val) |
| {"fg.black}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_BLACK) ANSI_ESC_END}, |
| {"fg.red}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_RED) ANSI_ESC_END}, |
| {"fg.green}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_GREEN) ANSI_ESC_END}, |
| {"fg.yellow}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_YELLOW) ANSI_ESC_END}, |
| {"fg.blue}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_BLUE) ANSI_ESC_END}, |
| {"fg.purple}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_PURPLE) ANSI_ESC_END}, |
| {"fg.cyan}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_CYAN) ANSI_ESC_END}, |
| {"fg.white}", ANSI_ESC_START _TO_STR(ANSI_FG_COLOR_WHITE) ANSI_ESC_END}, |
| {"bg.black}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_BLACK) ANSI_ESC_END}, |
| {"bg.red}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_RED) ANSI_ESC_END}, |
| {"bg.green}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_GREEN) ANSI_ESC_END}, |
| {"bg.yellow}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_YELLOW) ANSI_ESC_END}, |
| {"bg.blue}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_BLUE) ANSI_ESC_END}, |
| {"bg.purple}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_PURPLE) ANSI_ESC_END}, |
| {"bg.cyan}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_CYAN) ANSI_ESC_END}, |
| {"bg.white}", ANSI_ESC_START _TO_STR(ANSI_BG_COLOR_WHITE) ANSI_ESC_END}, |
| {"normal}", ANSI_ESC_START _TO_STR(ANSI_CTRL_NORMAL) ANSI_ESC_END}, |
| {"bold}", ANSI_ESC_START _TO_STR(ANSI_CTRL_BOLD) ANSI_ESC_END}, |
| {"faint}", ANSI_ESC_START _TO_STR(ANSI_CTRL_FAINT) ANSI_ESC_END}, |
| {"italic}", ANSI_ESC_START _TO_STR(ANSI_CTRL_ITALIC) ANSI_ESC_END}, |
| {"underline}", ANSI_ESC_START _TO_STR(ANSI_CTRL_UNDERLINE) ANSI_ESC_END}, |
| {"slow-blink}", |
| ANSI_ESC_START _TO_STR(ANSI_CTRL_SLOW_BLINK) ANSI_ESC_END}, |
| {"fast-blink}", |
| ANSI_ESC_START _TO_STR(ANSI_CTRL_FAST_BLINK) ANSI_ESC_END}, |
| {"negative}", |
| ANSI_ESC_START _TO_STR(ANSI_CTRL_IMAGE_NEGATIVE) ANSI_ESC_END}, |
| {"conceal}", ANSI_ESC_START _TO_STR(ANSI_CTRL_CONCEAL) ANSI_ESC_END}, |
| {"crossed-out}", |
| ANSI_ESC_START _TO_STR(ANSI_CTRL_CROSSED_OUT) ANSI_ESC_END}, |
| #undef _TO_STR |
| #undef _TO_STR2 |
| }; |
| auto codes = llvm::makeArrayRef(g_color_tokens); |
| |
| static const char tok_hdr[] = "${ansi."; |
| |
| std::string fmt; |
| while (!format.empty()) { |
| llvm::StringRef left, right; |
| std::tie(left, right) = format.split(tok_hdr); |
| |
| fmt.append(left); |
| |
| if (left == format && right.empty()) { |
| // The header was not found. Just exit. |
| break; |
| } |
| |
| bool found_code = false; |
| for (const auto &code : codes) { |
| if (!right.consume_front(code.name)) |
| continue; |
| |
| if (do_color) |
| fmt.append(code.value); |
| found_code = true; |
| break; |
| } |
| format = right; |
| // If we haven't found a valid replacement value, we just copy the string |
| // to the result without any modifications. |
| if (!found_code) |
| fmt.append(tok_hdr); |
| } |
| return fmt; |
| } |
| } |
| } |