SmallString.h revision 3a54b3dc87a581c203b18050b4f787b4ca28a12c
1//===- llvm/ADT/SmallString.h - 'Normally small' strings --------*- C++ -*-===// 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// This file defines the SmallString class. 11// 12//===----------------------------------------------------------------------===// 13 14#ifndef LLVM_ADT_SMALLSTRING_H 15#define LLVM_ADT_SMALLSTRING_H 16 17#include "llvm/ADT/SmallVector.h" 18#include "llvm/Support/DataTypes.h" 19#include <cstring> 20 21namespace llvm { 22 23/// SmallString - A SmallString is just a SmallVector with methods and accessors 24/// that make it work better as a string (e.g. operator+ etc). 25template<unsigned InternalLen> 26class SmallString : public SmallVector<char, InternalLen> { 27public: 28 // Default ctor - Initialize to empty. 29 SmallString() {} 30 31 // Initialize with a range. 32 template<typename ItTy> 33 SmallString(ItTy S, ItTy E) : SmallVector<char, InternalLen>(S, E) {} 34 35 // Copy ctor. 36 SmallString(const SmallString &RHS) : SmallVector<char, InternalLen>(RHS) {} 37 38 39 // Extra methods. 40 const char *c_str() const { 41 SmallString *This = const_cast<SmallString*>(this); 42 // Ensure that there is a \0 at the end of the string. 43 This->reserve(this->size()+1); 44 This->End[0] = 0; 45 return this->begin(); 46 } 47 48 // Extra operators. 49 const SmallString &operator=(const char *RHS) { 50 this->clear(); 51 return *this += RHS; 52 } 53 54 SmallString &operator+=(const char *RHS) { 55 this->append(RHS, RHS+strlen(RHS)); 56 return *this; 57 } 58 SmallString &operator+=(char C) { 59 this->push_back(C); 60 return *this; 61 } 62 63 SmallString &append_uint_32(uint32_t N) { 64 char Buffer[20]; 65 char *BufPtr = Buffer+20; 66 67 if (N == 0) *--BufPtr = '0'; // Handle special case. 68 69 while (N) { 70 *--BufPtr = '0' + char(N % 10); 71 N /= 10; 72 } 73 this->append(BufPtr, Buffer+20); 74 return *this; 75 } 76 77 SmallString &append_uint(uint64_t N) { 78 if (N == uint32_t(N)) 79 return append_uint_32(uint32_t(N)); 80 81 char Buffer[40]; 82 char *BufPtr = Buffer+40; 83 84 if (N == 0) *--BufPtr = '0'; // Handle special case... 85 86 while (N) { 87 *--BufPtr = '0' + char(N % 10); 88 N /= 10; 89 } 90 91 this->append(BufPtr, Buffer+40); 92 return *this; 93 } 94 95 SmallString &append_sint(int64_t N) { 96 // TODO, wrong for minint64. 97 if (N < 0) { 98 this->push_back('-'); 99 N = -N; 100 } 101 return append_uint(N); 102 } 103 104}; 105 106 107} 108 109#endif 110