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/Support/raw_ostream.h"
  12 using namespace llvm;
  13
  14 std::string Twine::str() const {
  15   // FIXME: This should probably use the toVector implementation, once that is
  16   // efficient.
  17   std::string Res;
  18   raw_string_ostream OS(Res);
  19   print(OS);
  20   OS.flush();
  21   return Res;
  22 }
  23
  24 void Twine::toVector(SmallVectorImpl<char> &Out) const {
  25   // FIXME: This is very inefficient, since we are creating a large raw_ostream
  26   // buffer -- hitting malloc, which we were supposed to avoid -- all when we
  27   // have this pretty little small vector available.
  28   //
  29   // The best way to fix this is to make raw_svector_ostream do the right thing
  30   // and be efficient, by augmenting the base raw_ostream with the ability to
  31   // have the buffer managed by a concrete implementation.
  32   raw_svector_ostream OS(Out);
  33   print(OS);
  34 }
  35
  36 void Twine::printOneChild(raw_ostream &OS, const void *Ptr,
  37                           NodeKind Kind) const {
  38   switch (Kind) {
  39   case Twine::NullKind: break;
  40   case Twine::EmptyKind: break;
  41   case Twine::TwineKind:
  42     static_cast<const Twine*>(Ptr)->print(OS);
  43     break;
  44   case Twine::CStringKind:
  45     OS << static_cast<const char*>(Ptr);
  46     break;
  47   case Twine::StdStringKind:
  48     OS << *static_cast<const std::string*>(Ptr);
  49     break;
  50   case Twine::StringRefKind:
  51     OS << *static_cast<const StringRef*>(Ptr);
  52     break;
  53   case Twine::DecUIKind:
  54     OS << *static_cast<const unsigned int*>(Ptr);
  55     break;
  56   case Twine::DecIKind:
  57     OS << *static_cast<const int*>(Ptr);
  58     break;
  59   case Twine::DecULKind:
  60     OS << *static_cast<const unsigned long*>(Ptr);
  61     break;
  62   case Twine::DecLKind:
  63     OS << *static_cast<const long*>(Ptr);
  64     break;
  65   case Twine::DecULLKind:
  66     OS << *static_cast<const unsigned long long*>(Ptr);
  67     break;
  68   case Twine::DecLLKind:
  69     OS << *static_cast<const long long*>(Ptr);
  70     break;
  71   case Twine::UHexKind:
  72     OS.write_hex(*static_cast<const uint64_t*>(Ptr));
  73     break;
  74   }
  75 }
  76
  77 void Twine::printOneChildRepr(raw_ostream &OS, const void *Ptr,
  78                               NodeKind Kind) const {
  79   switch (Kind) {
  80   case Twine::NullKind:
  81     OS << "null"; break;
  82   case Twine::EmptyKind:
  83     OS << "empty"; break;
  84   case Twine::TwineKind:
  85     OS << "rope:";
  86     static_cast<const Twine*>(Ptr)->printRepr(OS);
  87     break;
  88   case Twine::CStringKind:
  89     OS << "cstring:\""
  90        << static_cast<const char*>(Ptr) << "\"";
  91     break;
  92   case Twine::StdStringKind:
  93     OS << "std::string:\""
  94        << static_cast<const std::string*>(Ptr) << "\"";
  95     break;
  96   case Twine::StringRefKind:
  97     OS << "stringref:\""
  98        << static_cast<const StringRef*>(Ptr) << "\"";
  99     break;
 100   case Twine::DecUIKind:
 101     OS << "decUI:\"" << *static_cast<const unsigned int*>(Ptr) << "\"";
 102     break;
 103   case Twine::DecIKind:
 104     OS << "decI:\"" << *static_cast<const int*>(Ptr) << "\"";
 105     break;
 106   case Twine::DecULKind:
 107     OS << "decUL:\"" << *static_cast<const unsigned long*>(Ptr) << "\"";
 108     break;
 109   case Twine::DecLKind:
 110     OS << "decL:\"" << *static_cast<const long*>(Ptr) << "\"";
 111     break;
 112   case Twine::DecULLKind:
 113     OS << "decULL:\"" << *static_cast<const unsigned long long*>(Ptr) << "\"";
 114     break;
 115   case Twine::DecLLKind:
 116     OS << "decLL:\"" << *static_cast<const long long*>(Ptr) << "\"";
 117     break;
 118   case Twine::UHexKind:
 119     OS << "uhex:\"" << static_cast<const uint64_t*>(Ptr) << "\"";
 120     break;
 121   }
 122 }
 123
 124 void Twine::print(raw_ostream &OS) const {
 125   printOneChild(OS, LHS, getLHSKind());
 126   printOneChild(OS, RHS, getRHSKind());
 127 }
 128
 129 void Twine::printRepr(raw_ostream &OS) const {
 130   OS << "(Twine ";
 131   printOneChildRepr(OS, LHS, getLHSKind());
 132   OS << " ";
 133   printOneChildRepr(OS, RHS, getRHSKind());
 134   OS << ")";
 135 }
 136
 137 void Twine::dump() const {
 138   print(llvm::errs());
 139 }
 140
 141 void Twine::dumpRepr() const {
 142   printRepr(llvm::errs());
 143 }