| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // Copied from strings/stringpiece.cc with modifications | 4 // Copied from strings/stringpiece.cc with modifications |
| 5 | 5 |
| 6 #include <algorithm> | 6 #include <algorithm> |
| 7 #include <ostream> | 7 #include <ostream> |
| 8 | 8 |
| 9 #include "base/string_piece.h" | 9 #include "base/string_piece.h" |
| 10 | 10 |
| 11 namespace base { | 11 namespace base { |
| 12 | 12 |
| 13 // MSVC doesn't like complex extern templates and DLLs. | |
| 14 #if !defined(COMPILER_MSVC) | |
| 15 namespace internal { | |
| 16 template class StringPieceDetail<std::string>; | |
| 17 template class StringPieceDetail<string16>; | |
| 18 } // namespace internal | |
| 19 | |
| 20 template class BasicStringPiece<std::string>; | |
| 21 template class BasicStringPiece<string16>; | |
| 22 #endif | |
| 23 | |
| 24 typedef StringPiece::size_type size_type; | 13 typedef StringPiece::size_type size_type; |
| 25 | 14 |
| 26 bool operator==(const StringPiece& x, const StringPiece& y) { | 15 bool operator==(const StringPiece& x, const StringPiece& y) { |
| 27 if (x.size() != y.size()) | 16 if (x.size() != y.size()) |
| 28 return false; | 17 return false; |
| 29 | 18 |
| 30 return StringPiece::wordmemcmp(x.data(), y.data(), x.size()) == 0; | 19 return StringPiece::wordmemcmp(x.data(), y.data(), x.size()) == 0; |
| 31 } | 20 } |
| 32 | 21 |
| 33 void BasicStringPiece<std::string>::CopyToString(std::string* target) const { | 22 void StringPiece::CopyToString(std::string* target) const { |
| 34 target->assign(!empty() ? data() : "", size()); | 23 target->assign(!empty() ? data() : "", size()); |
| 35 } | 24 } |
| 36 | 25 |
| 37 void BasicStringPiece<std::string>::AppendToString(std::string* target) const { | 26 void StringPiece::AppendToString(std::string* target) const { |
| 38 if (!empty()) | 27 if (!empty()) |
| 39 target->append(data(), size()); | 28 target->append(data(), size()); |
| 40 } | 29 } |
| 41 | 30 |
| 42 size_type BasicStringPiece<std::string>::copy(char* buf, | 31 size_type StringPiece::copy(char* buf, size_type n, size_type pos) const { |
| 43 size_type n, | |
| 44 size_type pos) const { | |
| 45 size_type ret = std::min(length_ - pos, n); | 32 size_type ret = std::min(length_ - pos, n); |
| 46 memcpy(buf, ptr_ + pos, ret); | 33 memcpy(buf, ptr_ + pos, ret); |
| 47 return ret; | 34 return ret; |
| 48 } | 35 } |
| 49 | 36 |
| 50 size_type BasicStringPiece<std::string>::find(const BasicStringPiece& s, | 37 size_type StringPiece::find(const StringPiece& s, size_type pos) const { |
| 51 size_type pos) const { | |
| 52 if (pos > length_) | 38 if (pos > length_) |
| 53 return npos; | 39 return npos; |
| 54 | 40 |
| 55 const char* result = std::search(ptr_ + pos, ptr_ + length_, | 41 const char* result = std::search(ptr_ + pos, ptr_ + length_, |
| 56 s.ptr_, s.ptr_ + s.length_); | 42 s.ptr_, s.ptr_ + s.length_); |
| 57 const size_type xpos = result - ptr_; | 43 const size_type xpos = result - ptr_; |
| 58 return xpos + s.length_ <= length_ ? xpos : npos; | 44 return xpos + s.length_ <= length_ ? xpos : npos; |
| 59 } | 45 } |
| 60 | 46 |
| 61 size_type BasicStringPiece<std::string>::find(char c, size_type pos) const { | 47 size_type StringPiece::find(char c, size_type pos) const { |
| 62 if (pos >= length_) | 48 if (pos >= length_) |
| 63 return npos; | 49 return npos; |
| 64 | 50 |
| 65 const char* result = std::find(ptr_ + pos, ptr_ + length_, c); | 51 const char* result = std::find(ptr_ + pos, ptr_ + length_, c); |
| 66 return result != ptr_ + length_ ? static_cast<size_t>(result - ptr_) : npos; | 52 return result != ptr_ + length_ ? static_cast<size_t>(result - ptr_) : npos; |
| 67 } | 53 } |
| 68 | 54 |
| 69 size_type BasicStringPiece<std::string>::rfind(const BasicStringPiece& s, | 55 size_type StringPiece::rfind(const StringPiece& s, size_type pos) const { |
| 70 size_type pos) const { | |
| 71 if (length_ < s.length_) | 56 if (length_ < s.length_) |
| 72 return npos; | 57 return npos; |
| 73 | 58 |
| 74 if (s.empty()) | 59 if (s.empty()) |
| 75 return std::min(length_, pos); | 60 return std::min(length_, pos); |
| 76 | 61 |
| 77 const char* last = ptr_ + std::min(length_ - s.length_, pos) + s.length_; | 62 const char* last = ptr_ + std::min(length_ - s.length_, pos) + s.length_; |
| 78 const char* result = std::find_end(ptr_, last, s.ptr_, s.ptr_ + s.length_); | 63 const char* result = std::find_end(ptr_, last, s.ptr_, s.ptr_ + s.length_); |
| 79 return result != last ? static_cast<size_t>(result - ptr_) : npos; | 64 return result != last ? static_cast<size_t>(result - ptr_) : npos; |
| 80 } | 65 } |
| 81 | 66 |
| 82 size_type BasicStringPiece<std::string>::rfind(char c, size_type pos) const { | 67 size_type StringPiece::rfind(char c, size_type pos) const { |
| 83 if (length_ == 0) | 68 if (length_ == 0) |
| 84 return npos; | 69 return npos; |
| 85 | 70 |
| 86 for (size_type i = std::min(pos, length_ - 1); ; --i) { | 71 for (size_type i = std::min(pos, length_ - 1); ; --i) { |
| 87 if (ptr_[i] == c) | 72 if (ptr_[i] == c) |
| 88 return i; | 73 return i; |
| 89 if (i == 0) | 74 if (i == 0) |
| 90 break; | 75 break; |
| 91 } | 76 } |
| 92 return npos; | 77 return npos; |
| 93 } | 78 } |
| 94 | 79 |
| 95 // For each character in characters_wanted, sets the index corresponding | 80 // For each character in characters_wanted, sets the index corresponding |
| 96 // to the ASCII code of that character to 1 in table. This is used by | 81 // to the ASCII code of that character to 1 in table. This is used by |
| 97 // the find_.*_of methods below to tell whether or not a character is in | 82 // the find_.*_of methods below to tell whether or not a character is in |
| 98 // the lookup table in constant time. | 83 // the lookup table in constant time. |
| 99 // The argument `table' must be an array that is large enough to hold all | 84 // The argument `table' must be an array that is large enough to hold all |
| 100 // the possible values of an unsigned char. Thus it should be be declared | 85 // the possible values of an unsigned char. Thus it should be be declared |
| 101 // as follows: | 86 // as follows: |
| 102 // bool table[UCHAR_MAX + 1] | 87 // bool table[UCHAR_MAX + 1] |
| 103 static inline void BuildLookupTable(const StringPiece& characters_wanted, | 88 static inline void BuildLookupTable(const StringPiece& characters_wanted, |
| 104 bool* table) { | 89 bool* table) { |
| 105 const size_type length = characters_wanted.length(); | 90 const size_type length = characters_wanted.length(); |
| 106 const char* const data = characters_wanted.data(); | 91 const char* const data = characters_wanted.data(); |
| 107 for (size_type i = 0; i < length; ++i) { | 92 for (size_type i = 0; i < length; ++i) { |
| 108 table[static_cast<unsigned char>(data[i])] = true; | 93 table[static_cast<unsigned char>(data[i])] = true; |
| 109 } | 94 } |
| 110 } | 95 } |
| 111 | 96 |
| 112 size_type BasicStringPiece<std::string>::find_first_of( | 97 size_type StringPiece::find_first_of(const StringPiece& s, |
| 113 const BasicStringPiece& s, size_type pos) const { | 98 size_type pos) const { |
| 114 if (length_ == 0 || s.length_ == 0) | 99 if (length_ == 0 || s.length_ == 0) |
| 115 return npos; | 100 return npos; |
| 116 | 101 |
| 117 // Avoid the cost of BuildLookupTable() for a single-character search. | 102 // Avoid the cost of BuildLookupTable() for a single-character search. |
| 118 if (s.length_ == 1) | 103 if (s.length_ == 1) |
| 119 return find_first_of(s.ptr_[0], pos); | 104 return find_first_of(s.ptr_[0], pos); |
| 120 | 105 |
| 121 bool lookup[UCHAR_MAX + 1] = { false }; | 106 bool lookup[UCHAR_MAX + 1] = { false }; |
| 122 BuildLookupTable(s, lookup); | 107 BuildLookupTable(s, lookup); |
| 123 for (size_type i = pos; i < length_; ++i) { | 108 for (size_type i = pos; i < length_; ++i) { |
| 124 if (lookup[static_cast<unsigned char>(ptr_[i])]) { | 109 if (lookup[static_cast<unsigned char>(ptr_[i])]) { |
| 125 return i; | 110 return i; |
| 126 } | 111 } |
| 127 } | 112 } |
| 128 return npos; | 113 return npos; |
| 129 } | 114 } |
| 130 | 115 |
| 131 size_type BasicStringPiece<std::string>::find_first_not_of( | 116 size_type StringPiece::find_first_not_of(const StringPiece& s, |
| 132 const BasicStringPiece& s, size_type pos) const { | 117 size_type pos) const { |
| 133 if (length_ == 0) | 118 if (length_ == 0) |
| 134 return npos; | 119 return npos; |
| 135 | 120 |
| 136 if (s.length_ == 0) | 121 if (s.length_ == 0) |
| 137 return 0; | 122 return 0; |
| 138 | 123 |
| 139 // Avoid the cost of BuildLookupTable() for a single-character search. | 124 // Avoid the cost of BuildLookupTable() for a single-character search. |
| 140 if (s.length_ == 1) | 125 if (s.length_ == 1) |
| 141 return find_first_not_of(s.ptr_[0], pos); | 126 return find_first_not_of(s.ptr_[0], pos); |
| 142 | 127 |
| 143 bool lookup[UCHAR_MAX + 1] = { false }; | 128 bool lookup[UCHAR_MAX + 1] = { false }; |
| 144 BuildLookupTable(s, lookup); | 129 BuildLookupTable(s, lookup); |
| 145 for (size_type i = pos; i < length_; ++i) { | 130 for (size_type i = pos; i < length_; ++i) { |
| 146 if (!lookup[static_cast<unsigned char>(ptr_[i])]) { | 131 if (!lookup[static_cast<unsigned char>(ptr_[i])]) { |
| 147 return i; | 132 return i; |
| 148 } | 133 } |
| 149 } | 134 } |
| 150 return npos; | 135 return npos; |
| 151 } | 136 } |
| 152 | 137 |
| 153 size_type BasicStringPiece<std::string>::find_first_not_of( | 138 size_type StringPiece::find_first_not_of(char c, size_type pos) const { |
| 154 char c, size_type pos) const { | |
| 155 if (length_ == 0) | 139 if (length_ == 0) |
| 156 return npos; | 140 return npos; |
| 157 | 141 |
| 158 for (; pos < length_; ++pos) { | 142 for (; pos < length_; ++pos) { |
| 159 if (ptr_[pos] != c) { | 143 if (ptr_[pos] != c) { |
| 160 return pos; | 144 return pos; |
| 161 } | 145 } |
| 162 } | 146 } |
| 163 return npos; | 147 return npos; |
| 164 } | 148 } |
| 165 | 149 |
| 166 size_type BasicStringPiece<std::string>::find_last_of(const StringPiece& s, | 150 size_type StringPiece::find_last_of(const StringPiece& s, size_type pos) const { |
| 167 size_type pos) const { | |
| 168 if (length_ == 0 || s.length_ == 0) | 151 if (length_ == 0 || s.length_ == 0) |
| 169 return npos; | 152 return npos; |
| 170 | 153 |
| 171 // Avoid the cost of BuildLookupTable() for a single-character search. | 154 // Avoid the cost of BuildLookupTable() for a single-character search. |
| 172 if (s.length_ == 1) | 155 if (s.length_ == 1) |
| 173 return find_last_of(s.ptr_[0], pos); | 156 return find_last_of(s.ptr_[0], pos); |
| 174 | 157 |
| 175 bool lookup[UCHAR_MAX + 1] = { false }; | 158 bool lookup[UCHAR_MAX + 1] = { false }; |
| 176 BuildLookupTable(s, lookup); | 159 BuildLookupTable(s, lookup); |
| 177 for (size_type i = std::min(pos, length_ - 1); ; --i) { | 160 for (size_type i = std::min(pos, length_ - 1); ; --i) { |
| 178 if (lookup[static_cast<unsigned char>(ptr_[i])]) | 161 if (lookup[static_cast<unsigned char>(ptr_[i])]) |
| 179 return i; | 162 return i; |
| 180 if (i == 0) | 163 if (i == 0) |
| 181 break; | 164 break; |
| 182 } | 165 } |
| 183 return npos; | 166 return npos; |
| 184 } | 167 } |
| 185 | 168 |
| 186 size_type BasicStringPiece<std::string>::find_last_not_of( | 169 size_type StringPiece::find_last_not_of(const StringPiece& s, |
| 187 const BasicStringPiece& s, size_type pos) const { | 170 size_type pos) const { |
| 188 if (length_ == 0) | 171 if (length_ == 0) |
| 189 return npos; | 172 return npos; |
| 190 | 173 |
| 191 size_type i = std::min(pos, length_ - 1); | 174 size_type i = std::min(pos, length_ - 1); |
| 192 if (s.length_ == 0) | 175 if (s.length_ == 0) |
| 193 return i; | 176 return i; |
| 194 | 177 |
| 195 // Avoid the cost of BuildLookupTable() for a single-character search. | 178 // Avoid the cost of BuildLookupTable() for a single-character search. |
| 196 if (s.length_ == 1) | 179 if (s.length_ == 1) |
| 197 return find_last_not_of(s.ptr_[0], pos); | 180 return find_last_not_of(s.ptr_[0], pos); |
| 198 | 181 |
| 199 bool lookup[UCHAR_MAX + 1] = { false }; | 182 bool lookup[UCHAR_MAX + 1] = { false }; |
| 200 BuildLookupTable(s, lookup); | 183 BuildLookupTable(s, lookup); |
| 201 for (; ; --i) { | 184 for (; ; --i) { |
| 202 if (!lookup[static_cast<unsigned char>(ptr_[i])]) | 185 if (!lookup[static_cast<unsigned char>(ptr_[i])]) |
| 203 return i; | 186 return i; |
| 204 if (i == 0) | 187 if (i == 0) |
| 205 break; | 188 break; |
| 206 } | 189 } |
| 207 return npos; | 190 return npos; |
| 208 } | 191 } |
| 209 | 192 |
| 210 size_type BasicStringPiece<std::string>::find_last_not_of(char c, | 193 size_type StringPiece::find_last_not_of(char c, size_type pos) const { |
| 211 size_type pos) const { | |
| 212 if (length_ == 0) | 194 if (length_ == 0) |
| 213 return npos; | 195 return npos; |
| 214 | 196 |
| 215 for (size_type i = std::min(pos, length_ - 1); ; --i) { | 197 for (size_type i = std::min(pos, length_ - 1); ; --i) { |
| 216 if (ptr_[i] != c) | 198 if (ptr_[i] != c) |
| 217 return i; | 199 return i; |
| 218 if (i == 0) | 200 if (i == 0) |
| 219 break; | 201 break; |
| 220 } | 202 } |
| 221 return npos; | 203 return npos; |
| 222 } | 204 } |
| 223 | 205 |
| 224 BasicStringPiece<std::string> BasicStringPiece<std::string>::substr( | 206 StringPiece StringPiece::substr(size_type pos, size_type n) const { |
| 225 size_type pos, size_type n) const { | |
| 226 if (pos > length_) pos = length_; | 207 if (pos > length_) pos = length_; |
| 227 if (n > length_ - pos) n = length_ - pos; | 208 if (n > length_ - pos) n = length_ - pos; |
| 228 return StringPiece(ptr_ + pos, n); | 209 return StringPiece(ptr_ + pos, n); |
| 229 } | 210 } |
| 230 | 211 |
| 212 const StringPiece::size_type StringPiece::npos = size_type(-1); |
| 213 |
| 231 } // namespace base | 214 } // namespace base |
| OLD | NEW |