Twine.cpp revision 7dc7ac3cb20b7ef8e6febe0ac3bc430230f29893 
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"
14using namespace llvm;
15
16std::string Twine::str() const {
17  SmallString<256> Vec;
18  return toStringRef(Vec).str();
19}
20
21void Twine::toVector(SmallVectorImpl<char> &Out) const {
22  raw_svector_ostream OS(Out);
23  print(OS);
24}
25
26StringRef Twine::toStringRef(SmallVectorImpl<char> &Out) const {
27  if (isSingleStringRef())
28    return getSingleStringRef();
29  toVector(Out);
30  return StringRef(Out.data(), Out.size());
31}
32
33StringRef Twine::toNullTerminatedStringRef(SmallVectorImpl<char> &Out) const {
34  if (isSingleStringRef()) {
35    StringRef sr = getSingleStringRef();
36    if (*(sr.begin() + sr.size()) == 0)
37      return sr;
38  }
39  toVector(Out);
40  Out.push_back(0);
41  Out.pop_back();
42  return StringRef(Out.data(), Out.size());
43}
44
45void Twine::printOneChild(raw_ostream &OS, const void *Ptr,
46                          NodeKind Kind) const {
47  switch (Kind) {
48  case Twine::NullKind: break;
49  case Twine::EmptyKind: break;
50  case Twine::TwineKind:
51    static_cast<const Twine*>(Ptr)->print(OS);
52    break;
53  case Twine::CStringKind:
54    OS << static_cast<const char*>(Ptr);
55    break;
56  case Twine::StdStringKind:
57    OS << *static_cast<const std::string*>(Ptr);
58    break;
59  case Twine::StringRefKind:
60    OS << *static_cast<const StringRef*>(Ptr);
61    break;
62  case Twine::DecUIKind:
63    OS << (unsigned)(uintptr_t)Ptr;
64    break;
65  case Twine::DecIKind:
66    OS << (int)(intptr_t)Ptr;
67    break;
68  case Twine::DecULKind:
69    OS << *static_cast<const unsigned long*>(Ptr);
70    break;
71  case Twine::DecLKind:
72    OS << *static_cast<const long*>(Ptr);
73    break;
74  case Twine::DecULLKind:
75    OS << *static_cast<const unsigned long long*>(Ptr);
76    break;
77  case Twine::DecLLKind:
78    OS << *static_cast<const long long*>(Ptr);
79    break;
80  case Twine::UHexKind:
81    OS.write_hex(*static_cast<const uint64_t*>(Ptr));
82    break;
83  }
84}
85
86void Twine::printOneChildRepr(raw_ostream &OS, const void *Ptr,
87                              NodeKind Kind) const {
88  switch (Kind) {
89  case Twine::NullKind:
90    OS << "null"; break;
91  case Twine::EmptyKind:
92    OS << "empty"; break;
93  case Twine::TwineKind:
94    OS << "rope:";
95    static_cast<const Twine*>(Ptr)->printRepr(OS);
96    break;
97  case Twine::CStringKind:
98    OS << "cstring:\""
99       << static_cast<const char*>(Ptr) << "\"";
100    break;
101  case Twine::StdStringKind:
102    OS << "std::string:\""
103       << static_cast<const std::string*>(Ptr) << "\"";
104    break;
105  case Twine::StringRefKind:
106    OS << "stringref:\""
107       << static_cast<const StringRef*>(Ptr) << "\"";
108    break;
109  case Twine::DecUIKind:
110    OS << "decUI:\"" << (unsigned)(uintptr_t)Ptr << "\"";
111    break;
112  case Twine::DecIKind:
113    OS << "decI:\"" << (int)(intptr_t)Ptr << "\"";
114    break;
115  case Twine::DecULKind:
116    OS << "decUL:\"" << *static_cast<const unsigned long*>(Ptr) << "\"";
117    break;
118  case Twine::DecLKind:
119    OS << "decL:\"" << *static_cast<const long*>(Ptr) << "\"";
120    break;
121  case Twine::DecULLKind:
122    OS << "decULL:\"" << *static_cast<const unsigned long long*>(Ptr) << "\"";
123    break;
124  case Twine::DecLLKind:
125    OS << "decLL:\"" << *static_cast<const long long*>(Ptr) << "\"";
126    break;
127  case Twine::UHexKind:
128    OS << "uhex:\"" << static_cast<const uint64_t*>(Ptr) << "\"";
129    break;
130  }
131}
132
133void Twine::print(raw_ostream &OS) const {
134  printOneChild(OS, LHS, getLHSKind());
135  printOneChild(OS, RHS, getRHSKind());
136}
137
138void Twine::printRepr(raw_ostream &OS) const {
139  OS << "(Twine ";
140  printOneChildRepr(OS, LHS, getLHSKind());
141  OS << " ";
142  printOneChildRepr(OS, RHS, getRHSKind());
143  OS << ")";
144}
145
146void Twine::dump() const {
147  print(llvm::dbgs());
148}
149
150void Twine::dumpRepr() const {
151  printRepr(llvm::dbgs());
152}
153