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   raw_svector_ostream OS(Out);
  23   print(OS);
  24 }
  25
  26 void Twine::printOneChild(raw_ostream &OS, const void *Ptr,
  27                           NodeKind Kind) const {
  28   switch (Kind) {
  29   case Twine::NullKind: break;
  30   case Twine::EmptyKind: break;
  31   case Twine::CStringKind:
  32     OS << static_cast<const char*>(Ptr);
  33     break;
  34   case Twine::StdStringKind:
  35     OS << *static_cast<const std::string*>(Ptr);
  36     break;
  37   case Twine::StringRefKind:
  38     OS << *static_cast<const StringRef*>(Ptr);
  39     break;
  40   case Twine::TwineKind:
  41     static_cast<const Twine*>(Ptr)->print(OS);
  42     break;
  43   }
  44 }
  45
  46 void Twine::printOneChildRepr(raw_ostream &OS, const void *Ptr,
  47                               NodeKind Kind) const {
  48   switch (Kind) {
  49   case Twine::NullKind:
  50     OS << "null"; break;
  51   case Twine::EmptyKind:
  52     OS << "empty"; break;
  53   case Twine::CStringKind:
  54     OS << "cstring:\""
  55        << static_cast<const char*>(Ptr) << "\"";
  56     break;
  57   case Twine::StdStringKind:
  58     OS << "std::string:\""
  59        << *static_cast<const std::string*>(Ptr) << "\"";
  60     break;
  61   case Twine::StringRefKind:
  62     OS << "stringref:\""
  63        << *static_cast<const StringRef*>(Ptr) << "\"";
  64     break;
  65   case Twine::TwineKind:
  66     OS << "rope:";
  67     static_cast<const Twine*>(Ptr)->printRepr(OS);
  68     break;
  69   }
  70 }
  71
  72 void Twine::print(raw_ostream &OS) const {
  73   printOneChild(OS, LHS, getLHSKind());
  74   printOneChild(OS, RHS, getRHSKind());
  75 }
  76
  77 void Twine::printRepr(raw_ostream &OS) const {
  78   OS << "(Twine ";
  79   printOneChildRepr(OS, LHS, getLHSKind());
  80   OS << " ";
  81   printOneChildRepr(OS, RHS, getRHSKind());
  82   OS << ")";
  83 }
  84
  85 void Twine::dump() const {
  86   print(llvm::errs());
  87 }
  88
  89 void Twine::dumpRepr() const {
  90   printRepr(llvm::errs());
  91 }