SmallString.h revision 983c7fe847dd3f46945f0117ab19345b9c68e88f
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/ADT/StringRef.h" 19 20namespace llvm { 21 22/// SmallString - A SmallString is just a SmallVector with methods and accessors 23/// that make it work better as a string (e.g. operator+ etc). 24template<unsigned InternalLen> 25class SmallString : public SmallVector<char, InternalLen> { 26public: 27 // Default ctor - Initialize to empty. 28 SmallString() {} 29 30 // Initialize with a range. 31 template<typename ItTy> 32 SmallString(ItTy S, ItTy E) : SmallVector<char, InternalLen>(S, E) {} 33 34 // Copy ctor. 35 SmallString(const SmallString &RHS) : SmallVector<char, InternalLen>(RHS) {} 36 37 38 // Extra methods. 39 StringRef str() const { return StringRef(this->begin(), this->size()); } 40 41 // Implicit conversion to StringRef. 42 operator StringRef() const { return str(); } 43 44 const char *c_str() { 45 this->push_back(0); 46 this->pop_back(); 47 return this->data(); 48 } 49 50 // Extra operators. 51 const SmallString &operator=(StringRef RHS) { 52 this->clear(); 53 return *this += RHS; 54 } 55 56 SmallString &operator+=(StringRef RHS) { 57 this->append(RHS.begin(), RHS.end()); 58 return *this; 59 } 60 SmallString &operator+=(char C) { 61 this->push_back(C); 62 return *this; 63 } 64}; 65 66} 67 68#endif 69