1 //===-- StringExtras.cpp - Implement the StringExtras header --------------===// 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 implements the StringExtras.h header 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "llvm/ADT/StringExtras.h" 15 #include "llvm/ADT/SmallVector.h" 16 #include <cstring> 17 using namespace llvm; 18 19 /// getToken - This function extracts one token from source, ignoring any 20 /// leading characters that appear in the Delimiters string, and ending the 21 /// token at any of the characters that appear in the Delimiters string. If 22 /// there are no tokens in the source string, an empty string is returned. 23 /// The Source source string is updated in place to remove the returned string 24 /// and any delimiter prefix from it. 25 std::string llvm::getToken(std::string &Source, const char *Delimiters) { 26 size_t NumDelimiters = std::strlen(Delimiters); 27 28 // Figure out where the token starts. 29 std::string::size_type Start = 30 Source.find_first_not_of(Delimiters, 0, NumDelimiters); 31 if (Start == std::string::npos) Start = Source.size(); 32 33 // Find the next occurance of the delimiter. 34 std::string::size_type End = 35 Source.find_first_of(Delimiters, Start, NumDelimiters); 36 if (End == std::string::npos) End = Source.size(); 37 38 // Create the return token. 39 std::string Result = std::string(Source.begin()+Start, Source.begin()+End); 40 41 // Erase the token that we read in. 42 Source.erase(Source.begin(), Source.begin()+End); 43 44 return Result; 45 } 46 47 /// SplitString - Split up the specified string according to the specified 48 /// delimiters, appending the result fragments to the output list. 49 void llvm::SplitString(const std::string &Source, 50 std::vector<std::string> &OutFragments, 51 const char *Delimiters) { 52 std::string S = Source; 53 54 std::string S2 = getToken(S, Delimiters); 55 while (!S2.empty()) { 56 OutFragments.push_back(S2); 57 S2 = getToken(S, Delimiters); 58 } 59 } 60 61 void llvm::StringRef::split(SmallVectorImpl<StringRef> &A, 62 StringRef Separators, int MaxSplit, 63 bool KeepEmpty) const { 64 StringRef rest = *this; 65 66 // rest.data() is used to distinguish cases like "a," that splits into 67 // "a" + "" and "a" that splits into "a" + 0. 68 for (int splits = 0; 69 rest.data() != NULL && (MaxSplit < 0 || splits < MaxSplit); 70 ++splits) { 71 std::pair<llvm::StringRef, llvm::StringRef> p = rest.split(Separators); 72 73 if (p.first.size() != 0 || KeepEmpty) 74 A.push_back(p.first); 75 rest = p.second; 76 } 77 // If we have a tail left, add it. 78 if (rest.data() != NULL && (rest.size() != 0 || KeepEmpty)) 79 A.push_back(rest); 80 } 81