File: StringView.h

package info (click to toggle)
llvm-toolchain-13 1%3A13.0.1-11
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,418,840 kB
  • sloc: cpp: 5,290,826; ansic: 996,570; asm: 544,593; python: 188,212; objc: 72,027; lisp: 30,291; f90: 25,395; sh: 24,898; javascript: 9,780; pascal: 9,398; perl: 7,484; ml: 5,432; awk: 3,523; makefile: 2,913; xml: 953; cs: 573; fortran: 539
file content (99 lines) | stat: -rw-r--r-- 2,654 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
//===-- Standalone implementation std::string_view --------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIBC_UTILS_CPP_STRINGVIEW_H
#define LLVM_LIBC_UTILS_CPP_STRINGVIEW_H

#include <stddef.h>

namespace __llvm_libc {
namespace cpp {

// This is very simple alternate of the std::string_view class. There is no
// bounds check performed in any of the methods. The callers are expected to
// do the checks before invoking the methods.
//
// This class will be extended as needed in future.
class StringView {
private:
  const char *Data;
  size_t Len;

public:
  StringView() : Data(nullptr), Len(0) {}

  // Assumes Str is a null-terminated string. The length of the string does
  // not include the terminating null character.
  explicit StringView(const char *Str) : Data(Str), Len(0) {
    if (Str == nullptr)
      return;
    for (const char *D = Data; *D != '\0'; ++D, ++Len)
      ;
    if (Len == 0)
      Data = nullptr;
  }

  explicit StringView(const char *Str, size_t N)
      : Data(N ? Str : nullptr), Len(Str == nullptr ? 0 : N) {}

  const char *data() const { return Data; }

  size_t size() { return Len; }

  const char &operator[](size_t Index) const { return Data[Index]; }

  StringView remove_prefix(size_t N) const {
    if (N >= Len)
      return StringView();
    return StringView(Data + N, Len - N);
  }

  StringView remove_suffix(size_t N) const {
    if (N >= Len)
      return StringView();
    return StringView(Data, Len - N);
  }

  // An equivalent method is not available in std::string_view.
  StringView trim(char C) const {
    if (Len == 0)
      return StringView();

    const char *NewStart = Data;
    size_t PrefixLen = 0;
    for (; PrefixLen < Len; ++NewStart, ++PrefixLen) {
      if (*NewStart != C)
        break;
    }

    size_t SuffixLen = 0;
    const char *NewEnd = Data + Len - 1;
    for (; SuffixLen < Len; --NewEnd, ++SuffixLen) {
      if (*NewEnd != C)
        break;
    }

    return remove_prefix(PrefixLen).remove_suffix(SuffixLen);
  }

  // An equivalent method is not available in std::string_view.
  bool equals(StringView Other) const {
    if (Len != Other.Len)
      return false;
    for (size_t I = 0; I < Len; ++I) {
      if (Data[I] != Other.Data[I])
        return false;
    }
    return true;
  }
};

} // namespace cpp
} // namespace __llvm_libc

#endif // LLVM_LIBC_UTILS_CPP_STRINGVIEW_H