diff options
author | Torok Edwin <edwintorok@gmail.com> | 2009-06-04 07:09:50 +0000 |
---|---|---|
committer | Torok Edwin <edwintorok@gmail.com> | 2009-06-04 07:09:50 +0000 |
commit | e8ebb0fe1bba0fdff7475d98e1f8a04804b0b056 (patch) | |
tree | e95ec82d472d32e37882e836f167990a044b8bbd /lib/Support | |
parent | bccf4b3050907b61b9d5349601269a0474b4c0fd (diff) | |
download | external_llvm-e8ebb0fe1bba0fdff7475d98e1f8a04804b0b056.zip external_llvm-e8ebb0fe1bba0fdff7475d98e1f8a04804b0b056.tar.gz external_llvm-e8ebb0fe1bba0fdff7475d98e1f8a04804b0b056.tar.bz2 |
Add support for outputting ANSI colors to raw_fd_ostream.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@72854 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/Support')
-rw-r--r-- | lib/Support/raw_ostream.cpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/lib/Support/raw_ostream.cpp b/lib/Support/raw_ostream.cpp index 6ac37bc..42e6fda 100644 --- a/lib/Support/raw_ostream.cpp +++ b/lib/Support/raw_ostream.cpp @@ -14,6 +14,7 @@ #include "llvm/Support/raw_ostream.h" #include "llvm/Support/Format.h" #include "llvm/System/Program.h" +#include "llvm/System/Process.h" #include "llvm/ADT/SmallVector.h" #include "llvm/Config/config.h" #include "llvm/Support/Compiler.h" @@ -301,6 +302,35 @@ uint64_t raw_fd_ostream::seek(uint64_t off) { return pos; } +raw_ostream &raw_fd_ostream::changeColor(enum Colors colors, bool bold, + bool bg) { + if (sys::Process::ColorNeedsFlush()) + flush(); + const char *colorcode = + (colors == SAVEDCOLOR) ? sys::Process::OutputBold(bg) + : sys::Process::OutputColor(colors, bold, bg); + if (colorcode) { + unsigned len = strlen(colorcode); + write(colorcode, len); + // don't account colors towards output characters + pos -= len; + } + return *this; +} + +raw_ostream &raw_fd_ostream::resetColor() { + if (sys::Process::ColorNeedsFlush()) + flush(); + const char *colorcode = sys::Process::ResetColor(); + if (colorcode) { + unsigned len = strlen(colorcode); + write(colorcode, len); + // don't account colors towards output characters + pos -= len; + } + return *this; +} + //===----------------------------------------------------------------------===// // raw_stdout/err_ostream //===----------------------------------------------------------------------===// |