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

like "llvm_cerr << std::flush";

llvm-svn: 31926

1 files changed, 7 insertions, 0 deletions

diff --git a/llvm/include/llvm/Support/Streams.h b/llvm/include/llvm/Support/Streams.h
index 5af96fdff3a..138a0701497 100644
--- a/llvm/include/llvm/Support/Streams.h
+++ b/llvm/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;