/src/regex/regex-automata/src/util/memchr.rs
Line | Count | Source |
1 | | /*! |
2 | | This module defines simple wrapper routines for the memchr functions from the |
3 | | `memchr` crate. Basically, when the `memchr` crate is available, we use it, |
4 | | otherwise we use a naive implementation which is still pretty fast. |
5 | | */ |
6 | | |
7 | | pub(crate) use self::inner::*; |
8 | | |
9 | | #[cfg(feature = "perf-literal-substring")] |
10 | | pub(super) mod inner { |
11 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
12 | 77.5k | pub(crate) fn memchr(n1: u8, haystack: &[u8]) -> Option<usize> { |
13 | 77.5k | memchr::memchr(n1, haystack) |
14 | 77.5k | } |
15 | | |
16 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
17 | 79.9k | pub(crate) fn memchr2(n1: u8, n2: u8, haystack: &[u8]) -> Option<usize> { |
18 | 79.9k | memchr::memchr2(n1, n2, haystack) |
19 | 79.9k | } |
20 | | |
21 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
22 | 56.3k | pub(crate) fn memchr3( |
23 | 56.3k | n1: u8, |
24 | 56.3k | n2: u8, |
25 | 56.3k | n3: u8, |
26 | 56.3k | haystack: &[u8], |
27 | 56.3k | ) -> Option<usize> { |
28 | 56.3k | memchr::memchr3(n1, n2, n3, haystack) |
29 | 56.3k | } |
30 | | |
31 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
32 | 9.43k | pub(crate) fn memrchr(n1: u8, haystack: &[u8]) -> Option<usize> { |
33 | 9.43k | memchr::memrchr(n1, haystack) |
34 | 9.43k | } |
35 | | |
36 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
37 | 21.3k | pub(crate) fn memrchr2(n1: u8, n2: u8, haystack: &[u8]) -> Option<usize> { |
38 | 21.3k | memchr::memrchr2(n1, n2, haystack) |
39 | 21.3k | } |
40 | | |
41 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
42 | 5.86k | pub(crate) fn memrchr3( |
43 | 5.86k | n1: u8, |
44 | 5.86k | n2: u8, |
45 | 5.86k | n3: u8, |
46 | 5.86k | haystack: &[u8], |
47 | 5.86k | ) -> Option<usize> { |
48 | 5.86k | memchr::memrchr3(n1, n2, n3, haystack) |
49 | 5.86k | } |
50 | | } |
51 | | |
52 | | #[cfg(not(feature = "perf-literal-substring"))] |
53 | | pub(super) mod inner { |
54 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
55 | | pub(crate) fn memchr(n1: u8, haystack: &[u8]) -> Option<usize> { |
56 | | haystack.iter().position(|&b| b == n1) |
57 | | } |
58 | | |
59 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
60 | | pub(crate) fn memchr2(n1: u8, n2: u8, haystack: &[u8]) -> Option<usize> { |
61 | | haystack.iter().position(|&b| b == n1 || b == n2) |
62 | | } |
63 | | |
64 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
65 | | pub(crate) fn memchr3( |
66 | | n1: u8, |
67 | | n2: u8, |
68 | | n3: u8, |
69 | | haystack: &[u8], |
70 | | ) -> Option<usize> { |
71 | | haystack.iter().position(|&b| b == n1 || b == n2 || b == n3) |
72 | | } |
73 | | |
74 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
75 | | pub(crate) fn memrchr(n1: u8, haystack: &[u8]) -> Option<usize> { |
76 | | haystack.iter().rposition(|&b| b == n1) |
77 | | } |
78 | | |
79 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
80 | | pub(crate) fn memrchr2(n1: u8, n2: u8, haystack: &[u8]) -> Option<usize> { |
81 | | haystack.iter().rposition(|&b| b == n1 || b == n2) |
82 | | } |
83 | | |
84 | | #[cfg_attr(feature = "perf-inline", inline(always))] |
85 | | pub(crate) fn memrchr3( |
86 | | n1: u8, |
87 | | n2: u8, |
88 | | n3: u8, |
89 | | haystack: &[u8], |
90 | | ) -> Option<usize> { |
91 | | haystack.iter().rposition(|&b| b == n1 || b == n2 || b == n3) |
92 | | } |
93 | | } |