1
0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-10-20 03:23:01 +02:00

Factor out common condition into an easier to understand helper function (NFC).

llvm-svn: 324935
This commit is contained in:
Adrian Prantl 2018-02-12 21:11:14 +00:00
parent f51e269bcc
commit 4d423c8fcd
2 changed files with 12 additions and 2 deletions

View File

@ -20,9 +20,17 @@ static cl::opt<cl::boolOrDefault>
cl::desc("use colored syntax highlighting (default=autodetect)"),
cl::init(cl::BOU_UNSET));
bool WithColor::colorsEnabled(raw_ostream &OS) {
switch (UseColor) {
case cl::BOU_UNSET: return OS.has_colors();
case cl::BOU_TRUE: return true;
case cl::BOU_FALSE: return false;
}
}
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) {
if (colorsEnabled(OS)) {
switch (Type) {
case Address: OS.changeColor(raw_ostream::YELLOW); break;
case String: OS.changeColor(raw_ostream::GREEN); break;
@ -38,6 +46,6 @@ WithColor::WithColor(raw_ostream &OS, enum HighlightColor Type) : OS(OS) {
}
WithColor::~WithColor() {
if (UseColor == cl::BOU_UNSET ? OS.has_colors() : UseColor == cl::BOU_TRUE)
if (colorsEnabled(OS))
OS.resetColor();
}

View File

@ -34,6 +34,8 @@ enum HighlightColor {
/// specific color.
class WithColor {
raw_ostream &OS;
/// Determine whether colors should be displayed.
bool colorsEnabled(raw_ostream &OS);
public:
/// To be used like this: WithColor(OS, syntax::String) << "text";