1use haystack::{Hay, Haystack};
2use std::ops::Range;
3
4unsafe impl<T> Hay for [T] {
5 type Index = usize;
6
7 #[inline]
8 fn empty<'a>() -> &'a Self {
9 &[]
10 }
11
12 #[inline]
13 fn start_index(&self) -> usize {
14 0
15 }
16
17 #[inline]
18 fn end_index(&self) -> usize {
19 self.len()
20 }
21
22 #[inline]
23 unsafe fn slice_unchecked(&self, range: Range<usize>) -> &Self {
24 self.get_unchecked(range)
25 }
26
27 #[inline]
28 unsafe fn next_index(&self, index: Self::Index) -> Self::Index {
29 index + 1
30 }
31
32 #[inline]
33 unsafe fn prev_index(&self, index: Self::Index) -> Self::Index {
34 index - 1
35 }
36}
37
38unsafe impl<'h, T: 'h> Haystack for &'h mut [T] {
39 #[inline]
40 fn empty() -> Self {
41 &mut []
42 }
43
44 #[inline]
45 unsafe fn slice_unchecked(self, range: Range<usize>) -> Self {
46 self.get_unchecked_mut(range)
47 }
48
49 #[inline]
50 unsafe fn split_around(self, range: Range<usize>) -> [Self; 3] {
51 let (haystack, right) = self.split_at_mut(range.end);
52 let (left, middle) = haystack.split_at_mut(range.start);
53 [left, middle, right]
54 }
55
56 #[inline]
57 fn restore_range(&self, range: Range<usize>, subrange: Range<usize>) -> Range<usize> {
58 (subrange.start + range.start)..(subrange.end + range.start)
59 }
60}
61
62#[cfg(feature = "std")]
63unsafe impl<T> Haystack for Vec<T> {
64 #[inline]
65 fn empty() -> Self {
66 Vec::new()
67 }
68
69 #[inline]
70 unsafe fn slice_unchecked(mut self, range: Range<usize>) -> Self {
71 self.truncate(range.end);
72 self.drain(..range.start);
73 self
74 }
75
76 #[inline]
77 unsafe fn split_around(mut self, range: Range<usize>) -> [Self; 3] {
78 let right = self.split_off(range.end);
79 let middle = self.split_off(range.start);
80 [self, middle, right]
81 }
82
83 #[inline]
84 fn restore_range(&self, range: Range<usize>, subrange: Range<usize>) -> Range<usize> {
85 (subrange.start + range.start)..(subrange.end + range.start)
86 }
87}
88
89mod func;
90pub(crate) mod slice;