diff options
Diffstat (limited to 'llvm')
| -rw-r--r-- | llvm/include/llvm/Support/WithColor.h | 8 | ||||
| -rw-r--r-- | llvm/lib/Support/WithColor.cpp | 9 |
2 files changed, 11 insertions, 6 deletions
diff --git a/llvm/include/llvm/Support/WithColor.h b/llvm/include/llvm/Support/WithColor.h index 59f05ecc3c3..ee1e05cda32 100644 --- a/llvm/include/llvm/Support/WithColor.h +++ b/llvm/include/llvm/Support/WithColor.h @@ -10,6 +10,8 @@ #ifndef LLVM_SUPPORT_WITHCOLOR_H #define LLVM_SUPPORT_WITHCOLOR_H +#include "llvm/ADT/StringRef.h" + namespace llvm { class raw_ostream; @@ -50,11 +52,11 @@ public: static raw_ostream ¬e(); /// Convenience method for printing "error: " to the given stream. - static raw_ostream &error(raw_ostream &OS); + static raw_ostream &error(raw_ostream &OS, StringRef Prefix = ""); /// Convenience method for printing "warning: " to the given stream. - static raw_ostream &warning(raw_ostream &OS); + static raw_ostream &warning(raw_ostream &OS, StringRef Prefix = ""); /// Convenience method for printing "note: " to the given stream. - static raw_ostream ¬e(raw_ostream &OS); + static raw_ostream ¬e(raw_ostream &OS, StringRef Prefix = ""); }; } // end namespace llvm diff --git a/llvm/lib/Support/WithColor.cpp b/llvm/lib/Support/WithColor.cpp index 5a1364ff3cb..c3ab35bd891 100644 --- a/llvm/lib/Support/WithColor.cpp +++ b/llvm/lib/Support/WithColor.cpp @@ -65,15 +65,18 @@ raw_ostream &WithColor::warning() { return warning(errs()); } raw_ostream &WithColor::note() { return note(errs()); } -raw_ostream &WithColor::error(raw_ostream &OS) { +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) { +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) { +raw_ostream &WithColor::note(raw_ostream &OS, StringRef Prefix) { + OS << Prefix; return WithColor(OS, HighlightColor::Note).get() << "note: "; } |
