mirror of
https://github.com/RPCS3/llvm.git
synced 2025-04-13 11:30:21 +00:00
52107f0189
Several tools prefix the error/warning/note output with the name of the tool. One such tool is LLD for example. This commit adds as an optional 'Prefix' argument to the convenience helpers. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@330526 91177308-0d34-0410-b5e6-96231b3b80d8
87 lines
2.5 KiB
C++
87 lines
2.5 KiB
C++
//===- WithColor.cpp ------------------------------------------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/WithColor.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
|
|
static cl::opt<cl::boolOrDefault>
|
|
UseColor("color",
|
|
cl::desc("use colored syntax highlighting (default=autodetect)"),
|
|
cl::init(cl::BOU_UNSET));
|
|
|
|
bool WithColor::colorsEnabled(raw_ostream &OS) {
|
|
if (UseColor == cl::BOU_UNSET)
|
|
return OS.has_colors();
|
|
return UseColor == cl::BOU_TRUE;
|
|
}
|
|
|
|
WithColor::WithColor(raw_ostream &OS, HighlightColor Color) : OS(OS) {
|
|
// Detect color from terminal type unless the user passed the --color option.
|
|
if (colorsEnabled(OS)) {
|
|
switch (Color) {
|
|
case HighlightColor::Address:
|
|
OS.changeColor(raw_ostream::YELLOW);
|
|
break;
|
|
case HighlightColor::String:
|
|
OS.changeColor(raw_ostream::GREEN);
|
|
break;
|
|
case HighlightColor::Tag:
|
|
OS.changeColor(raw_ostream::BLUE);
|
|
break;
|
|
case HighlightColor::Attribute:
|
|
OS.changeColor(raw_ostream::CYAN);
|
|
break;
|
|
case HighlightColor::Enumerator:
|
|
OS.changeColor(raw_ostream::MAGENTA);
|
|
break;
|
|
case HighlightColor::Macro:
|
|
OS.changeColor(raw_ostream::RED);
|
|
break;
|
|
case HighlightColor::Error:
|
|
OS.changeColor(raw_ostream::RED, true);
|
|
break;
|
|
case HighlightColor::Warning:
|
|
OS.changeColor(raw_ostream::MAGENTA, true);
|
|
break;
|
|
case HighlightColor::Note:
|
|
OS.changeColor(raw_ostream::BLACK, true);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
raw_ostream &WithColor::error() { return error(errs()); }
|
|
|
|
raw_ostream &WithColor::warning() { return warning(errs()); }
|
|
|
|
raw_ostream &WithColor::note() { return note(errs()); }
|
|
|
|
raw_ostream &WithColor::error(raw_ostream &OS, StringRef Prefix) {
|
|
OS << Prefix;
|
|
return WithColor(OS, HighlightColor::Error).get() << "error: ";
|
|
}
|
|
|
|
raw_ostream &WithColor::warning(raw_ostream &OS, StringRef Prefix) {
|
|
OS << Prefix;
|
|
return WithColor(OS, HighlightColor::Warning).get() << "warning: ";
|
|
}
|
|
|
|
raw_ostream &WithColor::note(raw_ostream &OS, StringRef Prefix) {
|
|
OS << Prefix;
|
|
return WithColor(OS, HighlightColor::Note).get() << "note: ";
|
|
}
|
|
|
|
WithColor::~WithColor() {
|
|
if (colorsEnabled(OS))
|
|
OS.resetColor();
|
|
}
|