forked from OSchip/llvm-project
41 lines
1.5 KiB
C++
41 lines
1.5 KiB
C++
//===- SyntaxHighlighting.cpp ---------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "SyntaxHighlighting.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
using namespace dwarf;
|
|
using namespace syntax;
|
|
|
|
static cl::opt<cl::boolOrDefault>
|
|
UseColor("color",
|
|
cl::desc("use colored syntax highlighting (default=autodetect)"),
|
|
cl::init(cl::BOU_UNSET));
|
|
|
|
WithColor::WithColor(raw_ostream &OS, enum HighlightColor Type) : OS(OS) {
|
|
// Detect color from terminal type unless the user passed the --color option.
|
|
if (UseColor == cl::BOU_UNSET ? OS.has_colors() : UseColor == cl::BOU_TRUE) {
|
|
switch (Type) {
|
|
case Address: OS.changeColor(raw_ostream::YELLOW); break;
|
|
case String: OS.changeColor(raw_ostream::GREEN); break;
|
|
case Tag: OS.changeColor(raw_ostream::BLUE); break;
|
|
case Attribute: OS.changeColor(raw_ostream::CYAN); break;
|
|
case Enumerator: OS.changeColor(raw_ostream::MAGENTA); break;
|
|
case Macro: OS.changeColor(raw_ostream::RED); break;
|
|
}
|
|
}
|
|
}
|
|
|
|
WithColor::~WithColor() {
|
|
if (UseColor == cl::BOU_UNSET ? OS.has_colors() : UseColor == cl::BOU_TRUE)
|
|
OS.resetColor();
|
|
}
|