LLVM 19.0.0git
StringRef.h
Go to the documentation of this file.
1//===- StringRef.h - Constant String Reference Wrapper ----------*- C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#ifndef LLVM_ADT_STRINGREF_H
10#define LLVM_ADT_STRINGREF_H
11
16#include <algorithm>
17#include <cassert>
18#include <cstddef>
19#include <cstring>
20#include <limits>
21#include <string>
22#include <string_view>
23#include <type_traits>
24#include <utility>
25
26namespace llvm {
27
28 class APInt;
29 class hash_code;
30 template <typename T> class SmallVectorImpl;
31 class StringRef;
32
33 /// Helper functions for StringRef::getAsInteger.
34 bool getAsUnsignedInteger(StringRef Str, unsigned Radix,
35 unsigned long long &Result);
36
37 bool getAsSignedInteger(StringRef Str, unsigned Radix, long long &Result);
38
39 bool consumeUnsignedInteger(StringRef &Str, unsigned Radix,
40 unsigned long long &Result);
41 bool consumeSignedInteger(StringRef &Str, unsigned Radix, long long &Result);
42
43 /// StringRef - Represent a constant reference to a string, i.e. a character
44 /// array and a length, which need not be null terminated.
45 ///
46 /// This class does not own the string data, it is expected to be used in
47 /// situations where the character data resides in some other buffer, whose
48 /// lifetime extends past that of the StringRef. For this reason, it is not in
49 /// general safe to store a StringRef.
51 public:
52 static constexpr size_t npos = ~size_t(0);
53
54 using iterator = const char *;
55 using const_iterator = const char *;
56 using size_type = size_t;
57
58 private:
59 /// The start of the string, in an external buffer.
60 const char *Data = nullptr;
61
62 /// The length of the string.
63 size_t Length = 0;
64
65 // Workaround memcmp issue with null pointers (undefined behavior)
66 // by providing a specialized version
67 static int compareMemory(const char *Lhs, const char *Rhs, size_t Length) {
68 if (Length == 0) { return 0; }
69 return ::memcmp(Lhs,Rhs,Length);
70 }
71
72 public:
73 /// @name Constructors
74 /// @{
75
76 /// Construct an empty string ref.
77 /*implicit*/ StringRef() = default;
78
79 /// Disable conversion from nullptr. This prevents things like
80 /// if (S == nullptr)
81 StringRef(std::nullptr_t) = delete;
82
83 /// Construct a string ref from a cstring.
84 /*implicit*/ constexpr StringRef(const char *Str)
85 : Data(Str), Length(Str ?
86 // GCC 7 doesn't have constexpr char_traits. Fall back to __builtin_strlen.
87#if defined(_GLIBCXX_RELEASE) && _GLIBCXX_RELEASE < 8
88 __builtin_strlen(Str)
89#else
90 std::char_traits<char>::length(Str)
91#endif
92 : 0) {
93 }
94
95 /// Construct a string ref from a pointer and length.
96 /*implicit*/ constexpr StringRef(const char *data, size_t length)
97 : Data(data), Length(length) {}
98
99 /// Construct a string ref from an std::string.
100 /*implicit*/ StringRef(const std::string &Str)
101 : Data(Str.data()), Length(Str.length()) {}
102
103 /// Construct a string ref from an std::string_view.
104 /*implicit*/ constexpr StringRef(std::string_view Str)
105 : Data(Str.data()), Length(Str.size()) {}
106
107 /// @}
108 /// @name Iterators
109 /// @{
110
111 iterator begin() const { return Data; }
112
113 iterator end() const { return Data + Length; }
114
115 const unsigned char *bytes_begin() const {
116 return reinterpret_cast<const unsigned char *>(begin());
117 }
118 const unsigned char *bytes_end() const {
119 return reinterpret_cast<const unsigned char *>(end());
120 }
122 return make_range(bytes_begin(), bytes_end());
123 }
124
125 /// @}
126 /// @name String Operations
127 /// @{
128
129 /// data - Get a pointer to the start of the string (which may not be null
130 /// terminated).
131 [[nodiscard]] constexpr const char *data() const { return Data; }
132
133 /// empty - Check if the string is empty.
134 [[nodiscard]] constexpr bool empty() const { return Length == 0; }
135
136 /// size - Get the string size.
137 [[nodiscard]] constexpr size_t size() const { return Length; }
138
139 /// front - Get the first character in the string.
140 [[nodiscard]] char front() const {
141 assert(!empty());
142 return Data[0];
143 }
144
145 /// back - Get the last character in the string.
146 [[nodiscard]] char back() const {
147 assert(!empty());
148 return Data[Length-1];
149 }
150
151 // copy - Allocate copy in Allocator and return StringRef to it.
152 template <typename Allocator>
153 [[nodiscard]] StringRef copy(Allocator &A) const {
154 // Don't request a length 0 copy from the allocator.
155 if (empty())
156 return StringRef();
157 char *S = A.template Allocate<char>(Length);
158 std::copy(begin(), end(), S);
159 return StringRef(S, Length);
160 }
161
162 /// equals - Check for string equality, this is more efficient than
163 /// compare() when the relative ordering of inequal strings isn't needed.
164 [[nodiscard]] bool equals(StringRef RHS) const {
165 return (Length == RHS.Length &&
166 compareMemory(Data, RHS.Data, RHS.Length) == 0);
167 }
168
169 /// Check for string equality, ignoring case.
170 [[nodiscard]] bool equals_insensitive(StringRef RHS) const {
171 return Length == RHS.Length && compare_insensitive(RHS) == 0;
172 }
173
174 /// compare - Compare two strings; the result is negative, zero, or positive
175 /// if this string is lexicographically less than, equal to, or greater than
176 /// the \p RHS.
177 [[nodiscard]] int compare(StringRef RHS) const {
178 // Check the prefix for a mismatch.
179 if (int Res = compareMemory(Data, RHS.Data, std::min(Length, RHS.Length)))
180 return Res < 0 ? -1 : 1;
181
182 // Otherwise the prefixes match, so we only need to check the lengths.
183 if (Length == RHS.Length)
184 return 0;
185 return Length < RHS.Length ? -1 : 1;
186 }
187
188 /// Compare two strings, ignoring case.
189 [[nodiscard]] int compare_insensitive(StringRef RHS) const;
190
191 /// compare_numeric - Compare two strings, treating sequences of digits as
192 /// numbers.
193 [[nodiscard]] int compare_numeric(StringRef RHS) const;
194
195 /// Determine the edit distance between this string and another
196 /// string.
197 ///
198 /// \param Other the string to compare this string against.
199 ///
200 /// \param AllowReplacements whether to allow character
201 /// replacements (change one character into another) as a single
202 /// operation, rather than as two operations (an insertion and a
203 /// removal).
204 ///
205 /// \param MaxEditDistance If non-zero, the maximum edit distance that
206 /// this routine is allowed to compute. If the edit distance will exceed
207 /// that maximum, returns \c MaxEditDistance+1.
208 ///
209 /// \returns the minimum number of character insertions, removals,
210 /// or (if \p AllowReplacements is \c true) replacements needed to
211 /// transform one of the given strings into the other. If zero,
212 /// the strings are identical.
213 [[nodiscard]] unsigned edit_distance(StringRef Other,
214 bool AllowReplacements = true,
215 unsigned MaxEditDistance = 0) const;
216
217 [[nodiscard]] unsigned
218 edit_distance_insensitive(StringRef Other, bool AllowReplacements = true,
219 unsigned MaxEditDistance = 0) const;
220
221 /// str - Get the contents as an std::string.
222 [[nodiscard]] std::string str() const {
223 if (!Data) return std::string();
224 return std::string(Data, Length);
225 }
226
227 /// @}
228 /// @name Operator Overloads
229 /// @{
230
231 [[nodiscard]] char operator[](size_t Index) const {
232 assert(Index < Length && "Invalid index!");
233 return Data[Index];
234 }
235
236 /// Disallow accidental assignment from a temporary std::string.
237 ///
238 /// The declaration here is extra complicated so that `stringRef = {}`
239 /// and `stringRef = "abc"` continue to select the move assignment operator.
240 template <typename T>
241 std::enable_if_t<std::is_same<T, std::string>::value, StringRef> &
242 operator=(T &&Str) = delete;
243
244 /// @}
245 /// @name Type Conversions
246 /// @{
247
248 constexpr operator std::string_view() const {
249 return std::string_view(data(), size());
250 }
251
252 /// @}
253 /// @name String Predicates
254 /// @{
255
256 /// Check if this string starts with the given \p Prefix.
257 [[nodiscard]] bool starts_with(StringRef Prefix) const {
258 return Length >= Prefix.Length &&
259 compareMemory(Data, Prefix.Data, Prefix.Length) == 0;
260 }
261 [[nodiscard]] bool starts_with(char Prefix) const {
262 return !empty() && front() == Prefix;
263 }
264
265 /// Check if this string starts with the given \p Prefix, ignoring case.
266 [[nodiscard]] bool starts_with_insensitive(StringRef Prefix) const;
267
268 /// Check if this string ends with the given \p Suffix.
269 [[nodiscard]] bool ends_with(StringRef Suffix) const {
270 return Length >= Suffix.Length &&
271 compareMemory(end() - Suffix.Length, Suffix.Data, Suffix.Length) ==
272 0;
273 }
274 [[nodiscard]] bool ends_with(char Suffix) const {
275 return !empty() && back() == Suffix;
276 }
277
278 /// Check if this string ends with the given \p Suffix, ignoring case.
279 [[nodiscard]] bool ends_with_insensitive(StringRef Suffix) const;
280
281 /// @}
282 /// @name String Searching
283 /// @{
284
285 /// Search for the first character \p C in the string.
286 ///
287 /// \returns The index of the first occurrence of \p C, or npos if not
288 /// found.
289 [[nodiscard]] size_t find(char C, size_t From = 0) const {
290 return std::string_view(*this).find(C, From);
291 }
292
293 /// Search for the first character \p C in the string, ignoring case.
294 ///
295 /// \returns The index of the first occurrence of \p C, or npos if not
296 /// found.
297 [[nodiscard]] size_t find_insensitive(char C, size_t From = 0) const;
298
299 /// Search for the first character satisfying the predicate \p F
300 ///
301 /// \returns The index of the first character satisfying \p F starting from
302 /// \p From, or npos if not found.
303 [[nodiscard]] size_t find_if(function_ref<bool(char)> F,
304 size_t From = 0) const {
305 StringRef S = drop_front(From);
306 while (!S.empty()) {
307 if (F(S.front()))
308 return size() - S.size();
309 S = S.drop_front();
310 }
311 return npos;
312 }
313
314 /// Search for the first character not satisfying the predicate \p F
315 ///
316 /// \returns The index of the first character not satisfying \p F starting
317 /// from \p From, or npos if not found.
318 [[nodiscard]] size_t find_if_not(function_ref<bool(char)> F,
319 size_t From = 0) const {
320 return find_if([F](char c) { return !F(c); }, From);
321 }
322
323 /// Search for the first string \p Str in the string.
324 ///
325 /// \returns The index of the first occurrence of \p Str, or npos if not
326 /// found.
327 [[nodiscard]] size_t find(StringRef Str, size_t From = 0) const;
328
329 /// Search for the first string \p Str in the string, ignoring case.
330 ///
331 /// \returns The index of the first occurrence of \p Str, or npos if not
332 /// found.
333 [[nodiscard]] size_t find_insensitive(StringRef Str, size_t From = 0) const;
334
335 /// Search for the last character \p C in the string.
336 ///
337 /// \returns The index of the last occurrence of \p C, or npos if not
338 /// found.
339 [[nodiscard]] size_t rfind(char C, size_t From = npos) const {
340 size_t I = std::min(From, Length);
341 while (I) {
342 --I;
343 if (Data[I] == C)
344 return I;
345 }
346 return npos;
347 }
348
349 /// Search for the last character \p C in the string, ignoring case.
350 ///
351 /// \returns The index of the last occurrence of \p C, or npos if not
352 /// found.
353 [[nodiscard]] size_t rfind_insensitive(char C, size_t From = npos) const;
354
355 /// Search for the last string \p Str in the string.
356 ///
357 /// \returns The index of the last occurrence of \p Str, or npos if not
358 /// found.
359 [[nodiscard]] size_t rfind(StringRef Str) const;
360
361 /// Search for the last string \p Str in the string, ignoring case.
362 ///
363 /// \returns The index of the last occurrence of \p Str, or npos if not
364 /// found.
365 [[nodiscard]] size_t rfind_insensitive(StringRef Str) const;
366
367 /// Find the first character in the string that is \p C, or npos if not
368 /// found. Same as find.
369 [[nodiscard]] size_t find_first_of(char C, size_t From = 0) const {
370 return find(C, From);
371 }
372
373 /// Find the first character in the string that is in \p Chars, or npos if
374 /// not found.
375 ///
376 /// Complexity: O(size() + Chars.size())
377 [[nodiscard]] size_t find_first_of(StringRef Chars, size_t From = 0) const;
378
379 /// Find the first character in the string that is not \p C or npos if not
380 /// found.
381 [[nodiscard]] size_t find_first_not_of(char C, size_t From = 0) const;
382
383 /// Find the first character in the string that is not in the string
384 /// \p Chars, or npos if not found.
385 ///
386 /// Complexity: O(size() + Chars.size())
387 [[nodiscard]] size_t find_first_not_of(StringRef Chars,
388 size_t From = 0) const;
389
390 /// Find the last character in the string that is \p C, or npos if not
391 /// found.
392 [[nodiscard]] size_t find_last_of(char C, size_t From = npos) const {
393 return rfind(C, From);
394 }
395
396 /// Find the last character in the string that is in \p C, or npos if not
397 /// found.
398 ///
399 /// Complexity: O(size() + Chars.size())
400 [[nodiscard]] size_t find_last_of(StringRef Chars,
401 size_t From = npos) const;
402
403 /// Find the last character in the string that is not \p C, or npos if not
404 /// found.
405 [[nodiscard]] size_t find_last_not_of(char C, size_t From = npos) const;
406
407 /// Find the last character in the string that is not in \p Chars, or
408 /// npos if not found.
409 ///
410 /// Complexity: O(size() + Chars.size())
411 [[nodiscard]] size_t find_last_not_of(StringRef Chars,
412 size_t From = npos) const;
413
414 /// Return true if the given string is a substring of *this, and false
415 /// otherwise.
416 [[nodiscard]] bool contains(StringRef Other) const {
417 return find(Other) != npos;
418 }
419
420 /// Return true if the given character is contained in *this, and false
421 /// otherwise.
422 [[nodiscard]] bool contains(char C) const {
423 return find_first_of(C) != npos;
424 }
425
426 /// Return true if the given string is a substring of *this, and false
427 /// otherwise.
428 [[nodiscard]] bool contains_insensitive(StringRef Other) const {
429 return find_insensitive(Other) != npos;
430 }
431
432 /// Return true if the given character is contained in *this, and false
433 /// otherwise.
434 [[nodiscard]] bool contains_insensitive(char C) const {
435 return find_insensitive(C) != npos;
436 }
437
438 /// @}
439 /// @name Helpful Algorithms
440 /// @{
441
442 /// Return the number of occurrences of \p C in the string.
443 [[nodiscard]] size_t count(char C) const {
444 size_t Count = 0;
445 for (size_t I = 0; I != Length; ++I)
446 if (Data[I] == C)
447 ++Count;
448 return Count;
449 }
450
451 /// Return the number of non-overlapped occurrences of \p Str in
452 /// the string.
453 size_t count(StringRef Str) const;
454
455 /// Parse the current string as an integer of the specified radix. If
456 /// \p Radix is specified as zero, this does radix autosensing using
457 /// extended C rules: 0 is octal, 0x is hex, 0b is binary.
458 ///
459 /// If the string is invalid or if only a subset of the string is valid,
460 /// this returns true to signify the error. The string is considered
461 /// erroneous if empty or if it overflows T.
462 template <typename T> bool getAsInteger(unsigned Radix, T &Result) const {
463 if constexpr (std::numeric_limits<T>::is_signed) {
464 long long LLVal;
465 if (getAsSignedInteger(*this, Radix, LLVal) ||
466 static_cast<T>(LLVal) != LLVal)
467 return true;
468 Result = LLVal;
469 } else {
470 unsigned long long ULLVal;
471 // The additional cast to unsigned long long is required to avoid the
472 // Visual C++ warning C4805: '!=' : unsafe mix of type 'bool' and type
473 // 'unsigned __int64' when instantiating getAsInteger with T = bool.
474 if (getAsUnsignedInteger(*this, Radix, ULLVal) ||
475 static_cast<unsigned long long>(static_cast<T>(ULLVal)) != ULLVal)
476 return true;
477 Result = ULLVal;
478 }
479 return false;
480 }
481
482 /// Parse the current string as an integer of the specified radix. If
483 /// \p Radix is specified as zero, this does radix autosensing using
484 /// extended C rules: 0 is octal, 0x is hex, 0b is binary.
485 ///
486 /// If the string does not begin with a number of the specified radix,
487 /// this returns true to signify the error. The string is considered
488 /// erroneous if empty or if it overflows T.
489 /// The portion of the string representing the discovered numeric value
490 /// is removed from the beginning of the string.
491 template <typename T> bool consumeInteger(unsigned Radix, T &Result) {
492 if constexpr (std::numeric_limits<T>::is_signed) {
493 long long LLVal;
494 if (consumeSignedInteger(*this, Radix, LLVal) ||
495 static_cast<long long>(static_cast<T>(LLVal)) != LLVal)
496 return true;
497 Result = LLVal;
498 } else {
499 unsigned long long ULLVal;
500 if (consumeUnsignedInteger(*this, Radix, ULLVal) ||
501 static_cast<unsigned long long>(static_cast<T>(ULLVal)) != ULLVal)
502 return true;
503 Result = ULLVal;
504 }
505 return false;
506 }
507
508 /// Parse the current string as an integer of the specified \p Radix, or of
509 /// an autosensed radix if the \p Radix given is 0. The current value in
510 /// \p Result is discarded, and the storage is changed to be wide enough to
511 /// store the parsed integer.
512 ///
513 /// \returns true if the string does not solely consist of a valid
514 /// non-empty number in the appropriate base.
515 ///
516 /// APInt::fromString is superficially similar but assumes the
517 /// string is well-formed in the given radix.
518 bool getAsInteger(unsigned Radix, APInt &Result) const;
519
520 /// Parse the current string as an integer of the specified \p Radix. If
521 /// \p Radix is specified as zero, this does radix autosensing using
522 /// extended C rules: 0 is octal, 0x is hex, 0b is binary.
523 ///
524 /// If the string does not begin with a number of the specified radix,
525 /// this returns true to signify the error. The string is considered
526 /// erroneous if empty.
527 /// The portion of the string representing the discovered numeric value
528 /// is removed from the beginning of the string.
529 bool consumeInteger(unsigned Radix, APInt &Result);
530
531 /// Parse the current string as an IEEE double-precision floating
532 /// point value. The string must be a well-formed double.
533 ///
534 /// If \p AllowInexact is false, the function will fail if the string
535 /// cannot be represented exactly. Otherwise, the function only fails
536 /// in case of an overflow or underflow, or an invalid floating point
537 /// representation.
538 bool getAsDouble(double &Result, bool AllowInexact = true) const;
539
540 /// @}
541 /// @name String Operations
542 /// @{
543
544 // Convert the given ASCII string to lowercase.
545 [[nodiscard]] std::string lower() const;
546
547 /// Convert the given ASCII string to uppercase.
548 [[nodiscard]] std::string upper() const;
549
550 /// @}
551 /// @name Substring Operations
552 /// @{
553
554 /// Return a reference to the substring from [Start, Start + N).
555 ///
556 /// \param Start The index of the starting character in the substring; if
557 /// the index is npos or greater than the length of the string then the
558 /// empty substring will be returned.
559 ///
560 /// \param N The number of characters to included in the substring. If N
561 /// exceeds the number of characters remaining in the string, the string
562 /// suffix (starting with \p Start) will be returned.
563 [[nodiscard]] constexpr StringRef substr(size_t Start,
564 size_t N = npos) const {
565 Start = std::min(Start, Length);
566 return StringRef(Data + Start, std::min(N, Length - Start));
567 }
568
569 /// Return a StringRef equal to 'this' but with only the first \p N
570 /// elements remaining. If \p N is greater than the length of the
571 /// string, the entire string is returned.
572 [[nodiscard]] StringRef take_front(size_t N = 1) const {
573 if (N >= size())
574 return *this;
575 return drop_back(size() - N);
576 }
577
578 /// Return a StringRef equal to 'this' but with only the last \p N
579 /// elements remaining. If \p N is greater than the length of the
580 /// string, the entire string is returned.
581 [[nodiscard]] StringRef take_back(size_t N = 1) const {
582 if (N >= size())
583 return *this;
584 return drop_front(size() - N);
585 }
586
587 /// Return the longest prefix of 'this' such that every character
588 /// in the prefix satisfies the given predicate.
589 [[nodiscard]] StringRef take_while(function_ref<bool(char)> F) const {
590 return substr(0, find_if_not(F));
591 }
592
593 /// Return the longest prefix of 'this' such that no character in
594 /// the prefix satisfies the given predicate.
595 [[nodiscard]] StringRef take_until(function_ref<bool(char)> F) const {
596 return substr(0, find_if(F));
597 }
598
599 /// Return a StringRef equal to 'this' but with the first \p N elements
600 /// dropped.
601 [[nodiscard]] StringRef drop_front(size_t N = 1) const {
602 assert(size() >= N && "Dropping more elements than exist");
603 return substr(N);
604 }
605
606 /// Return a StringRef equal to 'this' but with the last \p N elements
607 /// dropped.
608 [[nodiscard]] StringRef drop_back(size_t N = 1) const {
609 assert(size() >= N && "Dropping more elements than exist");
610 return substr(0, size()-N);
611 }
612
613 /// Return a StringRef equal to 'this', but with all characters satisfying
614 /// the given predicate dropped from the beginning of the string.
615 [[nodiscard]] StringRef drop_while(function_ref<bool(char)> F) const {
616 return substr(find_if_not(F));
617 }
618
619 /// Return a StringRef equal to 'this', but with all characters not
620 /// satisfying the given predicate dropped from the beginning of the string.
621 [[nodiscard]] StringRef drop_until(function_ref<bool(char)> F) const {
622 return substr(find_if(F));
623 }
624
625 /// Returns true if this StringRef has the given prefix and removes that
626 /// prefix.
628 if (!starts_with(Prefix))
629 return false;
630
631 *this = substr(Prefix.size());
632 return true;
633 }
634
635 /// Returns true if this StringRef has the given prefix, ignoring case,
636 /// and removes that prefix.
638 if (!starts_with_insensitive(Prefix))
639 return false;
640
641 *this = substr(Prefix.size());
642 return true;
643 }
644
645 /// Returns true if this StringRef has the given suffix and removes that
646 /// suffix.
647 bool consume_back(StringRef Suffix) {
648 if (!ends_with(Suffix))
649 return false;
650
651 *this = substr(0, size() - Suffix.size());
652 return true;
653 }
654
655 /// Returns true if this StringRef has the given suffix, ignoring case,
656 /// and removes that suffix.
658 if (!ends_with_insensitive(Suffix))
659 return false;
660
661 *this = substr(0, size() - Suffix.size());
662 return true;
663 }
664
665 /// Return a reference to the substring from [Start, End).
666 ///
667 /// \param Start The index of the starting character in the substring; if
668 /// the index is npos or greater than the length of the string then the
669 /// empty substring will be returned.
670 ///
671 /// \param End The index following the last character to include in the
672 /// substring. If this is npos or exceeds the number of characters
673 /// remaining in the string, the string suffix (starting with \p Start)
674 /// will be returned. If this is less than \p Start, an empty string will
675 /// be returned.
676 [[nodiscard]] StringRef slice(size_t Start, size_t End) const {
677 Start = std::min(Start, Length);
678 End = std::clamp(End, Start, Length);
679 return StringRef(Data + Start, End - Start);
680 }
681
682 /// Split into two substrings around the first occurrence of a separator
683 /// character.
684 ///
685 /// If \p Separator is in the string, then the result is a pair (LHS, RHS)
686 /// such that (*this == LHS + Separator + RHS) is true and RHS is
687 /// maximal. If \p Separator is not in the string, then the result is a
688 /// pair (LHS, RHS) where (*this == LHS) and (RHS == "").
689 ///
690 /// \param Separator The character to split on.
691 /// \returns The split substrings.
692 [[nodiscard]] std::pair<StringRef, StringRef> split(char Separator) const {
693 return split(StringRef(&Separator, 1));
694 }
695
696 /// Split into two substrings around the first occurrence of a separator
697 /// string.
698 ///
699 /// If \p Separator is in the string, then the result is a pair (LHS, RHS)
700 /// such that (*this == LHS + Separator + RHS) is true and RHS is
701 /// maximal. If \p Separator is not in the string, then the result is a
702 /// pair (LHS, RHS) where (*this == LHS) and (RHS == "").
703 ///
704 /// \param Separator - The string to split on.
705 /// \return - The split substrings.
706 [[nodiscard]] std::pair<StringRef, StringRef>
707 split(StringRef Separator) const {
708 size_t Idx = find(Separator);
709 if (Idx == npos)
710 return std::make_pair(*this, StringRef());
711 return std::make_pair(slice(0, Idx), slice(Idx + Separator.size(), npos));
712 }
713
714 /// Split into two substrings around the last occurrence of a separator
715 /// string.
716 ///
717 /// If \p Separator is in the string, then the result is a pair (LHS, RHS)
718 /// such that (*this == LHS + Separator + RHS) is true and RHS is
719 /// minimal. If \p Separator is not in the string, then the result is a
720 /// pair (LHS, RHS) where (*this == LHS) and (RHS == "").
721 ///
722 /// \param Separator - The string to split on.
723 /// \return - The split substrings.
724 [[nodiscard]] std::pair<StringRef, StringRef>
725 rsplit(StringRef Separator) const {
726 size_t Idx = rfind(Separator);
727 if (Idx == npos)
728 return std::make_pair(*this, StringRef());
729 return std::make_pair(slice(0, Idx), slice(Idx + Separator.size(), npos));
730 }
731
732 /// Split into substrings around the occurrences of a separator string.
733 ///
734 /// Each substring is stored in \p A. If \p MaxSplit is >= 0, at most
735 /// \p MaxSplit splits are done and consequently <= \p MaxSplit + 1
736 /// elements are added to A.
737 /// If \p KeepEmpty is false, empty strings are not added to \p A. They
738 /// still count when considering \p MaxSplit
739 /// An useful invariant is that
740 /// Separator.join(A) == *this if MaxSplit == -1 and KeepEmpty == true
741 ///
742 /// \param A - Where to put the substrings.
743 /// \param Separator - The string to split on.
744 /// \param MaxSplit - The maximum number of times the string is split.
745 /// \param KeepEmpty - True if empty substring should be added.
747 StringRef Separator, int MaxSplit = -1,
748 bool KeepEmpty = true) const;
749
750 /// Split into substrings around the occurrences of a separator character.
751 ///
752 /// Each substring is stored in \p A. If \p MaxSplit is >= 0, at most
753 /// \p MaxSplit splits are done and consequently <= \p MaxSplit + 1
754 /// elements are added to A.
755 /// If \p KeepEmpty is false, empty strings are not added to \p A. They
756 /// still count when considering \p MaxSplit
757 /// An useful invariant is that
758 /// Separator.join(A) == *this if MaxSplit == -1 and KeepEmpty == true
759 ///
760 /// \param A - Where to put the substrings.
761 /// \param Separator - The string to split on.
762 /// \param MaxSplit - The maximum number of times the string is split.
763 /// \param KeepEmpty - True if empty substring should be added.
764 void split(SmallVectorImpl<StringRef> &A, char Separator, int MaxSplit = -1,
765 bool KeepEmpty = true) const;
766
767 /// Split into two substrings around the last occurrence of a separator
768 /// character.
769 ///
770 /// If \p Separator is in the string, then the result is a pair (LHS, RHS)
771 /// such that (*this == LHS + Separator + RHS) is true and RHS is
772 /// minimal. If \p Separator is not in the string, then the result is a
773 /// pair (LHS, RHS) where (*this == LHS) and (RHS == "").
774 ///
775 /// \param Separator - The character to split on.
776 /// \return - The split substrings.
777 [[nodiscard]] std::pair<StringRef, StringRef> rsplit(char Separator) const {
778 return rsplit(StringRef(&Separator, 1));
779 }
780
781 /// Return string with consecutive \p Char characters starting from the
782 /// the left removed.
783 [[nodiscard]] StringRef ltrim(char Char) const {
784 return drop_front(std::min(Length, find_first_not_of(Char)));
785 }
786
787 /// Return string with consecutive characters in \p Chars starting from
788 /// the left removed.
789 [[nodiscard]] StringRef ltrim(StringRef Chars = " \t\n\v\f\r") const {
790 return drop_front(std::min(Length, find_first_not_of(Chars)));
791 }
792
793 /// Return string with consecutive \p Char characters starting from the
794 /// right removed.
795 [[nodiscard]] StringRef rtrim(char Char) const {
796 return drop_back(Length - std::min(Length, find_last_not_of(Char) + 1));
797 }
798
799 /// Return string with consecutive characters in \p Chars starting from
800 /// the right removed.
801 [[nodiscard]] StringRef rtrim(StringRef Chars = " \t\n\v\f\r") const {
802 return drop_back(Length - std::min(Length, find_last_not_of(Chars) + 1));
803 }
804
805 /// Return string with consecutive \p Char characters starting from the
806 /// left and right removed.
807 [[nodiscard]] StringRef trim(char Char) const {
808 return ltrim(Char).rtrim(Char);
809 }
810
811 /// Return string with consecutive characters in \p Chars starting from
812 /// the left and right removed.
813 [[nodiscard]] StringRef trim(StringRef Chars = " \t\n\v\f\r") const {
814 return ltrim(Chars).rtrim(Chars);
815 }
816
817 /// Detect the line ending style of the string.
818 ///
819 /// If the string contains a line ending, return the line ending character
820 /// sequence that is detected. Otherwise return '\n' for unix line endings.
821 ///
822 /// \return - The line ending character sequence.
823 [[nodiscard]] StringRef detectEOL() const {
824 size_t Pos = find('\r');
825 if (Pos == npos) {
826 // If there is no carriage return, assume unix
827 return "\n";
828 }
829 if (Pos + 1 < Length && Data[Pos + 1] == '\n')
830 return "\r\n"; // Windows
831 if (Pos > 0 && Data[Pos - 1] == '\n')
832 return "\n\r"; // You monster!
833 return "\r"; // Classic Mac
834 }
835 /// @}
836 };
837
838 /// A wrapper around a string literal that serves as a proxy for constructing
839 /// global tables of StringRefs with the length computed at compile time.
840 /// In order to avoid the invocation of a global constructor, StringLiteral
841 /// should *only* be used in a constexpr context, as such:
842 ///
843 /// constexpr StringLiteral S("test");
844 ///
845 class StringLiteral : public StringRef {
846 private:
847 constexpr StringLiteral(const char *Str, size_t N) : StringRef(Str, N) {
848 }
849
850 public:
851 template <size_t N>
852 constexpr StringLiteral(const char (&Str)[N])
853#if defined(__clang__) && __has_attribute(enable_if)
854#pragma clang diagnostic push
855#pragma clang diagnostic ignored "-Wgcc-compat"
856 __attribute((enable_if(__builtin_strlen(Str) == N - 1,
857 "invalid string literal")))
858#pragma clang diagnostic pop
859#endif
860 : StringRef(Str, N - 1) {
861 }
862
863 // Explicit construction for strings like "foo\0bar".
864 template <size_t N>
865 static constexpr StringLiteral withInnerNUL(const char (&Str)[N]) {
866 return StringLiteral(Str, N - 1);
867 }
868 };
869
870 /// @name StringRef Comparison Operators
871 /// @{
872
874 return LHS.equals(RHS);
875 }
876
877 inline bool operator!=(StringRef LHS, StringRef RHS) { return !(LHS == RHS); }
878
880 return LHS.compare(RHS) < 0;
881 }
882
884 return LHS.compare(RHS) <= 0;
885 }
886
888 return LHS.compare(RHS) > 0;
889 }
890
892 return LHS.compare(RHS) >= 0;
893 }
894
895 inline std::string &operator+=(std::string &buffer, StringRef string) {
896 return buffer.append(string.data(), string.size());
897 }
898
899 /// @}
900
901 /// Compute a hash_code for a StringRef.
902 [[nodiscard]] hash_code hash_value(StringRef S);
903
904 // Provide DenseMapInfo for StringRefs.
905 template <> struct DenseMapInfo<StringRef, void> {
906 static inline StringRef getEmptyKey() {
907 return StringRef(
908 reinterpret_cast<const char *>(~static_cast<uintptr_t>(0)), 0);
909 }
910
911 static inline StringRef getTombstoneKey() {
912 return StringRef(
913 reinterpret_cast<const char *>(~static_cast<uintptr_t>(1)), 0);
914 }
915
916 static unsigned getHashValue(StringRef Val);
917
919 if (RHS.data() == getEmptyKey().data())
920 return LHS.data() == getEmptyKey().data();
921 if (RHS.data() == getTombstoneKey().data())
922 return LHS.data() == getTombstoneKey().data();
923 return LHS == RHS;
924 }
925 };
926
927} // end namespace llvm
928
929#endif // LLVM_ADT_STRINGREF_H
BlockVerifier::State From
static GCRegistry::Add< ErlangGC > A("erlang", "erlang-compatible garbage collector")
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
#define LLVM_GSL_POINTER
LLVM_GSL_POINTER - Apply this to non-owning classes like StringRef to enable lifetime warnings.
Definition: Compiler.h:326
static constexpr size_t npos
static Error split(StringRef Str, char Separator, std::pair< StringRef, StringRef > &Split)
Checked version of split, to ensure mandatory subparts.
Definition: DataLayout.cpp:235
Returns the sub type a function will return at a given Idx Should correspond to the result type of an ExtractValue instruction executed with just that one unsigned Idx
This file defines DenseMapInfo traits for DenseMap.
uint32_t Index
std::optional< std::vector< StOtherPiece > > Other
Definition: ELFYAML.cpp:1291
bool End
Definition: ELF_riscv.cpp:480
#define F(x, y, z)
Definition: MD5.cpp:55
#define I(x, y, z)
Definition: MD5.cpp:58
if(VerifyEach)
Basic Register Allocator
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())
static StringRef substr(StringRef Str, uint64_t Len)
DEMANGLE_NAMESPACE_BEGIN bool starts_with(std::string_view self, char C) noexcept
Value * RHS
Value * LHS
Class for arbitrary precision integers.
Definition: APInt.h:76
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
Definition: SmallVector.h:586
A wrapper around a string literal that serves as a proxy for constructing global tables of StringRefs...
Definition: StringRef.h:845
constexpr StringLiteral(const char(&Str)[N])
Definition: StringRef.h:852
static constexpr StringLiteral withInnerNUL(const char(&Str)[N])
Definition: StringRef.h:865
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:50
std::pair< StringRef, StringRef > split(char Separator) const
Split into two substrings around the first occurrence of a separator character.
Definition: StringRef.h:692
StringRef trim(StringRef Chars=" \t\n\v\f\r") const
Return string with consecutive characters in Chars starting from the left and right removed.
Definition: StringRef.h:813
bool consume_back(StringRef Suffix)
Returns true if this StringRef has the given suffix and removes that suffix.
Definition: StringRef.h:647
bool consumeInteger(unsigned Radix, T &Result)
Parse the current string as an integer of the specified radix.
Definition: StringRef.h:491
bool getAsInteger(unsigned Radix, T &Result) const
Parse the current string as an integer of the specified radix.
Definition: StringRef.h:462
iterator_range< const unsigned char * > bytes() const
Definition: StringRef.h:121
std::string str() const
str - Get the contents as an std::string.
Definition: StringRef.h:222
size_t find_if(function_ref< bool(char)> F, size_t From=0) const
Search for the first character satisfying the predicate F.
Definition: StringRef.h:303
const unsigned char * bytes_end() const
Definition: StringRef.h:118
constexpr StringRef substr(size_t Start, size_t N=npos) const
Return a reference to the substring from [Start, Start + N).
Definition: StringRef.h:563
bool starts_with(StringRef Prefix) const
Check if this string starts with the given Prefix.
Definition: StringRef.h:257
constexpr bool empty() const
empty - Check if the string is empty.
Definition: StringRef.h:134
bool contains_insensitive(StringRef Other) const
Return true if the given string is a substring of *this, and false otherwise.
Definition: StringRef.h:428
StringRef take_while(function_ref< bool(char)> F) const
Return the longest prefix of 'this' such that every character in the prefix satisfies the given predi...
Definition: StringRef.h:589
bool ends_with(char Suffix) const
Definition: StringRef.h:274
char operator[](size_t Index) const
Definition: StringRef.h:231
StringRef drop_front(size_t N=1) const
Return a StringRef equal to 'this' but with the first N elements dropped.
Definition: StringRef.h:601
bool contains_insensitive(char C) const
Return true if the given character is contained in *this, and false otherwise.
Definition: StringRef.h:434
iterator begin() const
Definition: StringRef.h:111
std::pair< StringRef, StringRef > rsplit(char Separator) const
Split into two substrings around the last occurrence of a separator character.
Definition: StringRef.h:777
StringRef drop_until(function_ref< bool(char)> F) const
Return a StringRef equal to 'this', but with all characters not satisfying the given predicate droppe...
Definition: StringRef.h:621
size_t size_type
Definition: StringRef.h:56
char back() const
back - Get the last character in the string.
Definition: StringRef.h:146
StringRef slice(size_t Start, size_t End) const
Return a reference to the substring from [Start, End).
Definition: StringRef.h:676
constexpr size_t size() const
size - Get the string size.
Definition: StringRef.h:137
char front() const
front - Get the first character in the string.
Definition: StringRef.h:140
bool starts_with(char Prefix) const
Definition: StringRef.h:261
size_t find_last_of(char C, size_t From=npos) const
Find the last character in the string that is C, or npos if not found.
Definition: StringRef.h:392
constexpr const char * data() const
data - Get a pointer to the start of the string (which may not be null terminated).
Definition: StringRef.h:131
StringRef ltrim(char Char) const
Return string with consecutive Char characters starting from the the left removed.
Definition: StringRef.h:783
bool contains(StringRef Other) const
Return true if the given string is a substring of *this, and false otherwise.
Definition: StringRef.h:416
bool consume_front(StringRef Prefix)
Returns true if this StringRef has the given prefix and removes that prefix.
Definition: StringRef.h:627
StringRef detectEOL() const
Detect the line ending style of the string.
Definition: StringRef.h:823
size_t find_first_of(char C, size_t From=0) const
Find the first character in the string that is C, or npos if not found.
Definition: StringRef.h:369
StringRef()=default
Construct an empty string ref.
size_t rfind(char C, size_t From=npos) const
Search for the last character C in the string.
Definition: StringRef.h:339
iterator end() const
Definition: StringRef.h:113
StringRef rtrim(char Char) const
Return string with consecutive Char characters starting from the right removed.
Definition: StringRef.h:795
bool contains(char C) const
Return true if the given character is contained in *this, and false otherwise.
Definition: StringRef.h:422
StringRef(std::nullptr_t)=delete
Disable conversion from nullptr.
StringRef take_back(size_t N=1) const
Return a StringRef equal to 'this' but with only the last N elements remaining.
Definition: StringRef.h:581
StringRef take_front(size_t N=1) const
Return a StringRef equal to 'this' but with only the first N elements remaining.
Definition: StringRef.h:572
bool equals(StringRef RHS) const
equals - Check for string equality, this is more efficient than compare() when the relative ordering ...
Definition: StringRef.h:164
StringRef take_until(function_ref< bool(char)> F) const
Return the longest prefix of 'this' such that no character in the prefix satisfies the given predicat...
Definition: StringRef.h:595
size_t find(char C, size_t From=0) const
Search for the first character C in the string.
Definition: StringRef.h:289
constexpr StringRef(const char *data, size_t length)
Construct a string ref from a pointer and length.
Definition: StringRef.h:96
StringRef trim(char Char) const
Return string with consecutive Char characters starting from the left and right removed.
Definition: StringRef.h:807
size_t count(char C) const
Return the number of occurrences of C in the string.
Definition: StringRef.h:443
bool consume_back_insensitive(StringRef Suffix)
Returns true if this StringRef has the given suffix, ignoring case, and removes that suffix.
Definition: StringRef.h:657
StringRef copy(Allocator &A) const
Definition: StringRef.h:153
bool ends_with(StringRef Suffix) const
Check if this string ends with the given Suffix.
Definition: StringRef.h:269
std::pair< StringRef, StringRef > rsplit(StringRef Separator) const
Split into two substrings around the last occurrence of a separator string.
Definition: StringRef.h:725
constexpr StringRef(const char *Str)
Construct a string ref from a cstring.
Definition: StringRef.h:84
std::pair< StringRef, StringRef > split(StringRef Separator) const
Split into two substrings around the first occurrence of a separator string.
Definition: StringRef.h:707
StringRef ltrim(StringRef Chars=" \t\n\v\f\r") const
Return string with consecutive characters in Chars starting from the left removed.
Definition: StringRef.h:789
std::enable_if_t< std::is_same< T, std::string >::value, StringRef > & operator=(T &&Str)=delete
Disallow accidental assignment from a temporary std::string.
StringRef rtrim(StringRef Chars=" \t\n\v\f\r") const
Return string with consecutive characters in Chars starting from the right removed.
Definition: StringRef.h:801
StringRef drop_while(function_ref< bool(char)> F) const
Return a StringRef equal to 'this', but with all characters satisfying the given predicate dropped fr...
Definition: StringRef.h:615
const unsigned char * bytes_begin() const
Definition: StringRef.h:115
int compare(StringRef RHS) const
compare - Compare two strings; the result is negative, zero, or positive if this string is lexicograp...
Definition: StringRef.h:177
StringRef drop_back(size_t N=1) const
Return a StringRef equal to 'this' but with the last N elements dropped.
Definition: StringRef.h:608
bool equals_insensitive(StringRef RHS) const
Check for string equality, ignoring case.
Definition: StringRef.h:170
bool consume_front_insensitive(StringRef Prefix)
Returns true if this StringRef has the given prefix, ignoring case, and removes that prefix.
Definition: StringRef.h:637
StringRef(const std::string &Str)
Construct a string ref from an std::string.
Definition: StringRef.h:100
constexpr StringRef(std::string_view Str)
Construct a string ref from an std::string_view.
Definition: StringRef.h:104
size_t find_if_not(function_ref< bool(char)> F, size_t From=0) const
Search for the first character not satisfying the predicate F.
Definition: StringRef.h:318
An efficient, type-erasing, non-owning reference to a callable.
A range adaptor for a pair of iterators.
This provides a very simple, boring adaptor for a begin and end iterator into a range type.
This is an optimization pass for GlobalISel generic memory operations.
Definition: AddressRanges.h:18
@ Length
Definition: DWP.cpp:456
bool operator<(int64_t V1, const APSInt &V2)
Definition: APSInt.h:361
auto find(R &&Range, const T &Val)
Provide wrappers to std::find which take ranges instead of having to pass begin/end explicitly.
Definition: STLExtras.h:1742
bool getAsSignedInteger(StringRef Str, unsigned Radix, long long &Result)
Definition: StringRef.cpp:496
hash_code hash_value(const FixedPointSemantics &Val)
Definition: APFixedPoint.h:128
auto size(R &&Range, std::enable_if_t< std::is_base_of< std::random_access_iterator_tag, typename std::iterator_traits< decltype(Range.begin())>::iterator_category >::value, void > *=nullptr)
Get the size of a range.
Definition: STLExtras.h:1680
bool operator!=(uint64_t V1, const APInt &V2)
Definition: APInt.h:2043
bool operator>=(int64_t V1, const APSInt &V2)
Definition: APSInt.h:360
iterator_range< T > make_range(T x, T y)
Convenience function for iterating over sub-ranges.
std::string & operator+=(std::string &buffer, StringRef string)
Definition: StringRef.h:895
bool operator==(const AddressRangeValuePair &LHS, const AddressRangeValuePair &RHS)
bool consumeUnsignedInteger(StringRef &Str, unsigned Radix, unsigned long long &Result)
Definition: StringRef.cpp:408
bool operator>(int64_t V1, const APSInt &V2)
Definition: APSInt.h:362
auto find_if_not(R &&Range, UnaryPredicate P)
Definition: STLExtras.h:1754
bool consumeSignedInteger(StringRef &Str, unsigned Radix, long long &Result)
Definition: StringRef.cpp:456
auto count(R &&Range, const E &Element)
Wrapper function around std::count to count the number of times an element Element occurs in the give...
Definition: STLExtras.h:1914
auto find_if(R &&Range, UnaryPredicate P)
Provide wrappers to std::find_if which take ranges instead of having to pass begin/end explicitly.
Definition: STLExtras.h:1749
bool getAsUnsignedInteger(StringRef Str, unsigned Radix, unsigned long long &Result)
Helper functions for StringRef::getAsInteger.
Definition: StringRef.cpp:486
bool operator<=(int64_t V1, const APSInt &V2)
Definition: APSInt.h:359
Implement std::hash so that hash_code can be used in STL containers.
Definition: BitVector.h:858
#define N
static bool isEqual(StringRef LHS, StringRef RHS)
Definition: StringRef.h:918
static unsigned getHashValue(StringRef Val)
static StringRef getTombstoneKey()
Definition: StringRef.h:911
An information struct used to provide DenseMap with the various necessary components for a given valu...
Definition: DenseMapInfo.h:50