Added "str()" method to raw_string_ostream.

str() mirrors the same method in std::ostringstream.
str() flushes the buffered stream contents to string and returns a reference to the string.

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

diff --git a/include/llvm/Support/raw_ostream.h b/include/llvm/Support/raw_ostream.h
index 824d209189c7e24516b0900afde490d2a53626a2..3cb8a3e317e6143283e293a3195a4f07b96c58e5 100644 (file)
--- a/include/llvm/Support/raw_ostream.h
+++ b/include/llvm/Support/raw_ostream.h
@@ -219,6 +219,13 @@ public:
   raw_string_ostream(std::string &O) : OS(O) {}
   ~raw_string_ostream();
   
+  /// str - Flushes the stream contents to the target string and returns
+  ///  the strings reference.
+  std::string& str() {
+    flush();
+    return OS;
+  }
+  
   /// flush_impl - The is the piece of the class that is implemented by
   /// subclasses.  This outputs the currently buffered data and resets the
   /// buffer to empty.