lib/Support/Twine.cpp

   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 }