1 //===-- Twine.cpp - Fast Temporary String Concatenation -------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/ADT/Twine.h"
11 #include "llvm/ADT/SmallString.h"
12 #include "llvm/Support/Debug.h"
13 #include "llvm/Support/FormatVariadic.h"
14 #include "llvm/Support/raw_ostream.h"
17 std::string Twine::str() const {
18 // If we're storing only a std::string, just return it.
19 if (LHSKind == StdStringKind && RHSKind == EmptyKind)
20 return *LHS.stdString;
22 // If we're storing a formatv_object, we can avoid an extra copy by formatting
23 // it immediately and returning the result.
24 if (LHSKind == FormatvObjectKind && RHSKind == EmptyKind)
25 return LHS.formatvObject->str();
27 // Otherwise, flatten and copy the contents first.
29 return toStringRef(Vec).str();
32 void Twine::toVector(SmallVectorImpl<char> &Out) const {
33 raw_svector_ostream OS(Out);
37 StringRef Twine::toNullTerminatedStringRef(SmallVectorImpl<char> &Out) const {
39 switch (getLHSKind()) {
41 // Already null terminated, yay!
42 return StringRef(LHS.cString);
44 const std::string *str = LHS.stdString;
45 return StringRef(str->c_str(), str->size());
54 return StringRef(Out.data(), Out.size());
57 void Twine::printOneChild(raw_ostream &OS, Child Ptr,
58 NodeKind Kind) const {
60 case Twine::NullKind: break;
61 case Twine::EmptyKind: break;
62 case Twine::TwineKind:
65 case Twine::CStringKind:
68 case Twine::StdStringKind:
71 case Twine::StringRefKind:
74 case Twine::SmallStringKind:
75 OS << *Ptr.smallString;
77 case Twine::FormatvObjectKind:
78 OS << *Ptr.formatvObject;
83 case Twine::DecUIKind:
89 case Twine::DecULKind:
95 case Twine::DecULLKind:
98 case Twine::DecLLKind:
101 case Twine::UHexKind:
102 OS.write_hex(*Ptr.uHex);
107 void Twine::printOneChildRepr(raw_ostream &OS, Child Ptr,
108 NodeKind Kind) const {
110 case Twine::NullKind:
112 case Twine::EmptyKind:
113 OS << "empty"; break;
114 case Twine::TwineKind:
116 Ptr.twine->printRepr(OS);
118 case Twine::CStringKind:
120 << Ptr.cString << "\"";
122 case Twine::StdStringKind:
123 OS << "std::string:\""
124 << Ptr.stdString << "\"";
126 case Twine::StringRefKind:
128 << Ptr.stringRef << "\"";
130 case Twine::SmallStringKind:
131 OS << "smallstring:\"" << *Ptr.smallString << "\"";
133 case Twine::FormatvObjectKind:
134 OS << "formatv:\"" << *Ptr.formatvObject << "\"";
136 case Twine::CharKind:
137 OS << "char:\"" << Ptr.character << "\"";
139 case Twine::DecUIKind:
140 OS << "decUI:\"" << Ptr.decUI << "\"";
142 case Twine::DecIKind:
143 OS << "decI:\"" << Ptr.decI << "\"";
145 case Twine::DecULKind:
146 OS << "decUL:\"" << *Ptr.decUL << "\"";
148 case Twine::DecLKind:
149 OS << "decL:\"" << *Ptr.decL << "\"";
151 case Twine::DecULLKind:
152 OS << "decULL:\"" << *Ptr.decULL << "\"";
154 case Twine::DecLLKind:
155 OS << "decLL:\"" << *Ptr.decLL << "\"";
157 case Twine::UHexKind:
158 OS << "uhex:\"" << Ptr.uHex << "\"";
163 void Twine::print(raw_ostream &OS) const {
164 printOneChild(OS, LHS, getLHSKind());
165 printOneChild(OS, RHS, getRHSKind());
168 void Twine::printRepr(raw_ostream &OS) const {
170 printOneChildRepr(OS, LHS, getLHSKind());
172 printOneChildRepr(OS, RHS, getRHSKind());
176 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
177 LLVM_DUMP_METHOD void Twine::dump() const {
181 LLVM_DUMP_METHOD void Twine::dumpRepr() const {