1 //===-- Twine.cpp - Fast Temporary String Concatenation -------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm/ADT/Twine.h" 11 #include "llvm/ADT/SmallString.h" 12 #include "llvm/Support/Debug.h" 13 #include "llvm/Support/raw_ostream.h" 14 using namespace llvm; 15 16 std::string Twine::str() const { 17 // If we're storing only a std::string, just return it. 18 if (LHSKind == StdStringKind && RHSKind == EmptyKind) 19 return *LHS.stdString; 20 21 // Otherwise, flatten and copy the contents first. 22 SmallString<256> Vec; 23 return toStringRef(Vec).str(); 24 } 25 26 void Twine::toVector(SmallVectorImpl<char> &Out) const { 27 raw_svector_ostream OS(Out); 28 print(OS); 29 } 30 31 StringRef Twine::toNullTerminatedStringRef(SmallVectorImpl<char> &Out) const { 32 if (isUnary()) { 33 switch (getLHSKind()) { 34 case CStringKind: 35 // Already null terminated, yay! 36 return StringRef(LHS.cString); 37 case StdStringKind: { 38 const std::string *str = LHS.stdString; 39 return StringRef(str->c_str(), str->size()); 40 } 41 default: 42 break; 43 } 44 } 45 toVector(Out); 46 Out.push_back(0); 47 Out.pop_back(); 48 return StringRef(Out.data(), Out.size()); 49 } 50 51 void Twine::printOneChild(raw_ostream &OS, Child Ptr, 52 NodeKind Kind) const { 53 switch (Kind) { 54 case Twine::NullKind: break; 55 case Twine::EmptyKind: break; 56 case Twine::TwineKind: 57 Ptr.twine->print(OS); 58 break; 59 case Twine::CStringKind: 60 OS << Ptr.cString; 61 break; 62 case Twine::StdStringKind: 63 OS << *Ptr.stdString; 64 break; 65 case Twine::StringRefKind: 66 OS << *Ptr.stringRef; 67 break; 68 case Twine::SmallStringKind: 69 OS << *Ptr.smallString; 70 break; 71 case Twine::CharKind: 72 OS << Ptr.character; 73 break; 74 case Twine::DecUIKind: 75 OS << Ptr.decUI; 76 break; 77 case Twine::DecIKind: 78 OS << Ptr.decI; 79 break; 80 case Twine::DecULKind: 81 OS << *Ptr.decUL; 82 break; 83 case Twine::DecLKind: 84 OS << *Ptr.decL; 85 break; 86 case Twine::DecULLKind: 87 OS << *Ptr.decULL; 88 break; 89 case Twine::DecLLKind: 90 OS << *Ptr.decLL; 91 break; 92 case Twine::UHexKind: 93 OS.write_hex(*Ptr.uHex); 94 break; 95 } 96 } 97 98 void Twine::printOneChildRepr(raw_ostream &OS, Child Ptr, 99 NodeKind Kind) const { 100 switch (Kind) { 101 case Twine::NullKind: 102 OS << "null"; break; 103 case Twine::EmptyKind: 104 OS << "empty"; break; 105 case Twine::TwineKind: 106 OS << "rope:"; 107 Ptr.twine->printRepr(OS); 108 break; 109 case Twine::CStringKind: 110 OS << "cstring:\"" 111 << Ptr.cString << "\""; 112 break; 113 case Twine::StdStringKind: 114 OS << "std::string:\"" 115 << Ptr.stdString << "\""; 116 break; 117 case Twine::StringRefKind: 118 OS << "stringref:\"" 119 << Ptr.stringRef << "\""; 120 break; 121 case Twine::SmallStringKind: 122 OS << "smallstring:\"" << *Ptr.smallString << "\""; 123 break; 124 case Twine::CharKind: 125 OS << "char:\"" << Ptr.character << "\""; 126 break; 127 case Twine::DecUIKind: 128 OS << "decUI:\"" << Ptr.decUI << "\""; 129 break; 130 case Twine::DecIKind: 131 OS << "decI:\"" << Ptr.decI << "\""; 132 break; 133 case Twine::DecULKind: 134 OS << "decUL:\"" << *Ptr.decUL << "\""; 135 break; 136 case Twine::DecLKind: 137 OS << "decL:\"" << *Ptr.decL << "\""; 138 break; 139 case Twine::DecULLKind: 140 OS << "decULL:\"" << *Ptr.decULL << "\""; 141 break; 142 case Twine::DecLLKind: 143 OS << "decLL:\"" << *Ptr.decLL << "\""; 144 break; 145 case Twine::UHexKind: 146 OS << "uhex:\"" << Ptr.uHex << "\""; 147 break; 148 } 149 } 150 151 void Twine::print(raw_ostream &OS) const { 152 printOneChild(OS, LHS, getLHSKind()); 153 printOneChild(OS, RHS, getRHSKind()); 154 } 155 156 void Twine::printRepr(raw_ostream &OS) const { 157 OS << "(Twine "; 158 printOneChildRepr(OS, LHS, getLHSKind()); 159 OS << " "; 160 printOneChildRepr(OS, RHS, getRHSKind()); 161 OS << ")"; 162 } 163 164 void Twine::dump() const { 165 print(dbgs()); 166 } 167 168 void Twine::dumpRepr() const { 169 printRepr(dbgs()); 170 } 171