Add method that will take a function pointer so that it can handle things

like "llvm_cerr << std::flush";


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@31926 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 7 insertions, 0 deletions

diff --git a/include/llvm/Support/Streams.h b/include/llvm/Support/Streams.h
index 5af96fdff3..138a070149 100644
--- a/include/llvm/Support/Streams.h
+++ b/include/llvm/Support/Streams.h
@@ -29,6 +29,13 @@ namespace llvm {
     llvm_ostream() : Stream(0) {}
     llvm_ostream(std::ostream &OStream) : Stream(&OStream) {}
 
+    std::ostream* stream() const { return Stream; }
+
+    inline llvm_ostream &operator << (std::ostream& (*Func)(std::ostream&)) {
+      *Stream << Func;
+      return *this;
+    }
+      
     template <typename Ty>
     llvm_ostream &operator << (const Ty &Thing) {
       if (Stream) *Stream << Thing;