StringView.h 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. //===--- StringView.h -------------------------------------------*- 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. // FIXME: Use std::string_view instead when we support C++17.
  10. //
  11. //===----------------------------------------------------------------------===//
  12. #ifndef LLVM_DEMANGLE_STRINGVIEW_H
  13. #define LLVM_DEMANGLE_STRINGVIEW_H
  14. #include "DemangleConfig.h"
  15. #include <algorithm>
  16. #include <cassert>
  17. #include <cstring>
  18. DEMANGLE_NAMESPACE_BEGIN
  19. class StringView {
  20. const char *First;
  21. const char *Last;
  22. public:
  23. static const size_t npos = ~size_t(0);
  24. template <size_t N>
  25. StringView(const char (&Str)[N]) : First(Str), Last(Str + N - 1) {}
  26. StringView(const char *First_, const char *Last_)
  27. : First(First_), Last(Last_) {}
  28. StringView(const char *First_, size_t Len)
  29. : First(First_), Last(First_ + Len) {}
  30. StringView(const char *Str) : First(Str), Last(Str + std::strlen(Str)) {}
  31. StringView() : First(nullptr), Last(nullptr) {}
  32. StringView substr(size_t Pos, size_t Len = npos) const {
  33. assert(Pos <= size());
  34. return StringView(begin() + Pos, std::min(Len, size() - Pos));
  35. }
  36. size_t find(char C, size_t From = 0) const {
  37. size_t FindBegin = std::min(From, size());
  38. // Avoid calling memchr with nullptr.
  39. if (FindBegin < size()) {
  40. // Just forward to memchr, which is faster than a hand-rolled loop.
  41. if (const void *P = ::memchr(First + FindBegin, C, size() - FindBegin))
  42. return size_t(static_cast<const char *>(P) - First);
  43. }
  44. return npos;
  45. }
  46. StringView dropFront(size_t N = 1) const {
  47. if (N >= size())
  48. N = size();
  49. return StringView(First + N, Last);
  50. }
  51. StringView dropBack(size_t N = 1) const {
  52. if (N >= size())
  53. N = size();
  54. return StringView(First, Last - N);
  55. }
  56. char front() const {
  57. assert(!empty());
  58. return *begin();
  59. }
  60. char back() const {
  61. assert(!empty());
  62. return *(end() - 1);
  63. }
  64. char popFront() {
  65. assert(!empty());
  66. return *First++;
  67. }
  68. bool consumeFront(char C) {
  69. if (!startsWith(C))
  70. return false;
  71. *this = dropFront(1);
  72. return true;
  73. }
  74. bool consumeFront(StringView S) {
  75. if (!startsWith(S))
  76. return false;
  77. *this = dropFront(S.size());
  78. return true;
  79. }
  80. bool startsWith(char C) const { return !empty() && *begin() == C; }
  81. bool startsWith(StringView Str) const {
  82. if (Str.size() > size())
  83. return false;
  84. return std::equal(Str.begin(), Str.end(), begin());
  85. }
  86. const char &operator[](size_t Idx) const { return *(begin() + Idx); }
  87. const char *begin() const { return First; }
  88. const char *end() const { return Last; }
  89. size_t size() const { return static_cast<size_t>(Last - First); }
  90. bool empty() const { return First == Last; }
  91. };
  92. inline bool operator==(const StringView &LHS, const StringView &RHS) {
  93. return LHS.size() == RHS.size() &&
  94. std::equal(LHS.begin(), LHS.end(), RHS.begin());
  95. }
  96. DEMANGLE_NAMESPACE_END
  97. #endif