From 6be1f01935f23d92c967024ee718a7f48748283c Mon Sep 17 00:00:00 2001 From: Jonas Devlieghere Date: Sun, 15 Apr 2018 08:44:15 +0000 Subject: [Support] Extend WithColor helpers Although printing warnings and errors to stderr is by far the most common case, this patch makes it possible to specify any stream. llvm-svn: 330094 --- llvm/lib/Support/WithColor.cpp | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) (limited to 'llvm/lib/Support') diff --git a/llvm/lib/Support/WithColor.cpp b/llvm/lib/Support/WithColor.cpp index 29fa16493f7..5a1364ff3cb 100644 --- a/llvm/lib/Support/WithColor.cpp +++ b/llvm/lib/Support/WithColor.cpp @@ -59,16 +59,22 @@ WithColor::WithColor(raw_ostream &OS, HighlightColor Color) : OS(OS) { } } -raw_ostream &WithColor::error() { - return WithColor(errs(), HighlightColor::Error).get() << "error: "; +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) { + return WithColor(OS, HighlightColor::Error).get() << "error: "; } -raw_ostream &WithColor::warning() { - return WithColor(errs(), HighlightColor::Warning).get() << "warning: "; +raw_ostream &WithColor::warning(raw_ostream &OS) { + return WithColor(OS, HighlightColor::Warning).get() << "warning: "; } -raw_ostream &WithColor::note() { - return WithColor(errs(), HighlightColor::Note).get() << "note: "; +raw_ostream &WithColor::note(raw_ostream &OS) { + return WithColor(OS, HighlightColor::Note).get() << "note: "; } WithColor::~WithColor() { -- cgit v1.2.3