Teach raw_ostream to accept SmallString.

Saves adding .str() call to any raw_ostream << SmallString usage
and a small step towards making .str() consistent in the ADTs by
removing one of the SmallString::str() use cases, discussion at

http://lists.cs.uiuc.edu/pipermail/llvm-commits/Week-of-Mon-20141013/240026.html

I'll update the Phabricator patch http://reviews.llvm.org/D6372
for review of the Twine SmallString support, it's more complex
than this one.



git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@231763 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Yaron Keren
2015-03-10 07:33:23 +00:00
parent 3a3665fd38
commit 1132d0cc5b
6 changed files with 17 additions and 12 deletions

View File

@@ -14,6 +14,7 @@
#ifndef LLVM_SUPPORT_RAW_OSTREAM_H #ifndef LLVM_SUPPORT_RAW_OSTREAM_H
#define LLVM_SUPPORT_RAW_OSTREAM_H #define LLVM_SUPPORT_RAW_OSTREAM_H
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h" #include "llvm/ADT/StringRef.h"
#include "llvm/Support/Compiler.h" #include "llvm/Support/Compiler.h"
#include "llvm/Support/DataTypes.h" #include "llvm/Support/DataTypes.h"
@@ -185,6 +186,10 @@ public:
return write(Str.data(), Str.length()); return write(Str.data(), Str.length());
} }
raw_ostream &operator<<(const llvm::SmallVectorImpl<char> &Str) {
return write(Str.data(), Str.size());
}
raw_ostream &operator<<(unsigned long N); raw_ostream &operator<<(unsigned long N);
raw_ostream &operator<<(long N); raw_ostream &operator<<(long N);
raw_ostream &operator<<(unsigned long long N); raw_ostream &operator<<(unsigned long long N);

View File

@@ -1010,7 +1010,7 @@ static void WriteConstantInternal(raw_ostream &Out, const Constant *CV,
(StrVal[1] >= '0' && StrVal[1] <= '9'))) { (StrVal[1] >= '0' && StrVal[1] <= '9'))) {
// Reparse stringized version! // Reparse stringized version!
if (APFloat(APFloat::IEEEdouble, StrVal).convertToDouble() == Val) { if (APFloat(APFloat::IEEEdouble, StrVal).convertToDouble() == Val) {
Out << StrVal.str(); Out << StrVal;
return; return;
} }
} }

View File

@@ -2291,7 +2291,7 @@ void APInt::dump() const {
this->toStringUnsigned(U); this->toStringUnsigned(U);
this->toStringSigned(S); this->toStringSigned(S);
dbgs() << "APInt(" << BitWidth << "b, " dbgs() << "APInt(" << BitWidth << "b, "
<< U.str() << "u " << S.str() << "s)"; << U << "u " << S << "s)";
} }
void APInt::print(raw_ostream &OS, bool isSigned) const { void APInt::print(raw_ostream &OS, bool isSigned) const {

View File

@@ -975,7 +975,7 @@ static bool TestCodeGenerator(BugDriver &BD, Module *Test, Module *Safe,
} }
if (BD.writeProgramToFile(SafeModuleBC.str(), SafeModuleFD, Safe)) { if (BD.writeProgramToFile(SafeModuleBC.str(), SafeModuleFD, Safe)) {
errs() << "Error writing bitcode to `" << SafeModuleBC.str() errs() << "Error writing bitcode to `" << SafeModuleBC
<< "'\nExiting."; << "'\nExiting.";
exit(1); exit(1);
} }
@@ -1050,7 +1050,7 @@ bool BugDriver::debugCodeGenerator(std::string *Error) {
} }
if (writeProgramToFile(TestModuleBC.str(), TestModuleFD, ToCodeGen)) { if (writeProgramToFile(TestModuleBC.str(), TestModuleFD, ToCodeGen)) {
errs() << "Error writing bitcode to `" << TestModuleBC.str() errs() << "Error writing bitcode to `" << TestModuleBC
<< "'\nExiting."; << "'\nExiting.";
exit(1); exit(1);
} }
@@ -1068,7 +1068,7 @@ bool BugDriver::debugCodeGenerator(std::string *Error) {
} }
if (writeProgramToFile(SafeModuleBC.str(), SafeModuleFD, ToNotCodeGen)) { if (writeProgramToFile(SafeModuleBC.str(), SafeModuleFD, ToNotCodeGen)) {
errs() << "Error writing bitcode to `" << SafeModuleBC.str() errs() << "Error writing bitcode to `" << SafeModuleBC
<< "'\nExiting."; << "'\nExiting.";
exit(1); exit(1);
} }
@@ -1079,17 +1079,17 @@ bool BugDriver::debugCodeGenerator(std::string *Error) {
outs() << "You can reproduce the problem with the command line: \n"; outs() << "You can reproduce the problem with the command line: \n";
if (isExecutingJIT()) { if (isExecutingJIT()) {
outs() << " lli -load " << SharedObject << " " << TestModuleBC.str(); outs() << " lli -load " << SharedObject << " " << TestModuleBC;
} else { } else {
outs() << " llc " << TestModuleBC.str() << " -o " << TestModuleBC.str() outs() << " llc " << TestModuleBC << " -o " << TestModuleBC
<< ".s\n"; << ".s\n";
outs() << " gcc " << SharedObject << " " << TestModuleBC.str() outs() << " gcc " << SharedObject << " " << TestModuleBC.str()
<< ".s -o " << TestModuleBC.str() << ".exe"; << ".s -o " << TestModuleBC << ".exe";
#if defined (HAVE_LINK_R) #if defined (HAVE_LINK_R)
outs() << " -Wl,-R."; outs() << " -Wl,-R.";
#endif #endif
outs() << "\n"; outs() << "\n";
outs() << " " << TestModuleBC.str() << ".exe"; outs() << " " << TestModuleBC << ".exe";
} }
for (unsigned i = 0, e = InputArgv.size(); i != e; ++i) for (unsigned i = 0, e = InputArgv.size(); i != e; ++i)
outs() << " " << InputArgv[i]; outs() << " " << InputArgv[i];

View File

@@ -727,7 +727,7 @@ public:
OS.flush(); OS.flush();
// Emit the string. // Emit the string.
O.indent(6) << "AsmString = \"" << OutString.str() << "\";\n"; O.indent(6) << "AsmString = \"" << OutString << "\";\n";
O.indent(6) << "break;\n"; O.indent(6) << "break;\n";
O.indent(4) << '}'; O.indent(4) << '}';

View File

@@ -188,7 +188,7 @@ EmitMatcher(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
<< " children in Scope"; << " children in Scope";
} }
OS << '\n' << TmpBuf.str(); OS << '\n' << TmpBuf;
CurrentIdx += ChildSize; CurrentIdx += ChildSize;
} }
@@ -342,7 +342,7 @@ EmitMatcher(const Matcher *N, unsigned Indent, unsigned CurrentIdx,
if (!OmitComments) if (!OmitComments)
OS << "// ->" << CurrentIdx+ChildSize; OS << "// ->" << CurrentIdx+ChildSize;
OS << '\n'; OS << '\n';
OS << TmpBuf.str(); OS << TmpBuf;
CurrentIdx += ChildSize; CurrentIdx += ChildSize;
} }