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   std::string Res;
  16   raw_string_ostream OS(Res);
  17   print(OS);
  18   return Res;
  19 }
  20
  21 void Twine::toVector(SmallVectorImpl<char> &Out) const {
  22   // FIXME: This is very inefficient, since we are creating a large raw_ostream
  23   // buffer -- hitting malloc, which we were supposed to avoid -- all when we
  24   // have this pretty little small vector available.
  25   //
  26   // The best way to fix this is to make raw_svector_ostream do the right thing
  27   // and be efficient, by augmenting the base raw_ostream with the ability to
  28   // have the buffer managed by a concrete implementation.
  29   raw_svector_ostream OS(Out);
  30   print(OS);
  31 }
  32
  33 void Twine::printOneChild(raw_ostream &OS, const void *Ptr,
  34                           NodeKind Kind) const {
  35   switch (Kind) {
  36   case Twine::NullKind: break;
  37   case Twine::EmptyKind: break;
  38   case Twine::TwineKind:
  39     static_cast<const Twine*>(Ptr)->print(OS);
  40     break;
  41   case Twine::CStringKind:
  42     OS << static_cast<const char*>(Ptr);
  43     break;
  44   case Twine::StdStringKind:
  45     OS << *static_cast<const std::string*>(Ptr);
  46     break;
  47   case Twine::StringRefKind:
  48     OS << *static_cast<const StringRef*>(Ptr);
  49     break;
  50   case Twine::UDec32Kind:
  51     OS << *static_cast<const uint32_t*>(Ptr);
  52     break;
  53   case Twine::SDec32Kind:
  54     OS << *static_cast<const int32_t*>(Ptr);
  55     break;
  56   case Twine::UDec64Kind:
  57     OS << *static_cast<const uint64_t*>(Ptr);
  58     break;
  59   case Twine::SDec64Kind:
  60     OS << *static_cast<const int64_t*>(Ptr);
  61     break;
  62   case Twine::UHexKind:
  63     // FIXME: Add raw_ostream functionality for this.
  64     OS << ::utohexstr(*static_cast<const uint64_t*>(Ptr));
  65     break;
  66   }
  67 }
  68
  69 void Twine::printOneChildRepr(raw_ostream &OS, const void *Ptr,
  70                               NodeKind Kind) const {
  71   switch (Kind) {
  72   case Twine::NullKind:
  73     OS << "null"; break;
  74   case Twine::EmptyKind:
  75     OS << "empty"; break;
  76   case Twine::TwineKind:
  77     OS << "rope:";
  78     static_cast<const Twine*>(Ptr)->printRepr(OS);
  79     break;
  80   case Twine::CStringKind:
  81     OS << "cstring:\""
  82        << static_cast<const char*>(Ptr) << "\"";
  83     break;
  84   case Twine::StdStringKind:
  85     OS << "std::string:\""
  86        << static_cast<const std::string*>(Ptr) << "\"";
  87     break;
  88   case Twine::StringRefKind:
  89     OS << "stringref:\""
  90        << static_cast<const StringRef*>(Ptr) << "\"";
  91     break;
  92   case Twine::UDec32Kind:
  93     OS << "udec32:" << static_cast<const uint64_t*>(Ptr) << "\"";
  94     break;
  95   case Twine::SDec32Kind:
  96     OS << "sdec32:" << static_cast<const int64_t*>(Ptr) << "\"";
  97     break;
  98   case Twine::UDec64Kind:
  99     OS << "udec64:" << static_cast<const uint64_t*>(Ptr) << "\"";
 100     break;
 101   case Twine::SDec64Kind:
 102     OS << "sdec64:" << static_cast<const int64_t*>(Ptr) << "\"";
 103     break;
 104   case Twine::UHexKind:
 105     OS << "uhex:" << static_cast<const uint64_t*>(Ptr) << "\"";
 106     break;
 107   }
 108 }
 109
 110 void Twine::print(raw_ostream &OS) const {
 111   printOneChild(OS, LHS, getLHSKind());
 112   printOneChild(OS, RHS, getRHSKind());
 113 }
 114
 115 void Twine::printRepr(raw_ostream &OS) const {
 116   OS << "(Twine ";
 117   printOneChildRepr(OS, LHS, getLHSKind());
 118   OS << " ";
 119   printOneChildRepr(OS, RHS, getRHSKind());
 120   OS << ")";
 121 }
 122
 123 void Twine::dump() const {
 124   print(llvm::errs());
 125 }
 126
 127 void Twine::dumpRepr() const {
 128   printRepr(llvm::errs());
 129 }