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
use crate::prelude::*;
use crate::series::IsSorted;
use crate::utils::{CustomIterTools, NoNull};
impl<T> ChunkReverse<T> for ChunkedArray<T>
where
T: PolarsNumericType,
{
fn reverse(&self) -> ChunkedArray<T> {
let mut out = if let Ok(slice) = self.cont_slice() {
let ca: NoNull<ChunkedArray<T>> = slice.iter().rev().copied().collect_trusted();
ca.into_inner()
} else {
self.into_iter().rev().collect_trusted()
};
out.rename(self.name());
match self.is_sorted2() {
IsSorted::Ascending => out.set_sorted2(IsSorted::Descending),
IsSorted::Descending => out.set_sorted2(IsSorted::Ascending),
_ => {}
}
out
}
}
macro_rules! impl_reverse {
($arrow_type:ident, $ca_type:ident) => {
impl ChunkReverse<$arrow_type> for $ca_type {
fn reverse(&self) -> Self {
let mut ca: Self = self.into_iter().rev().collect_trusted();
ca.rename(self.name());
ca
}
}
};
}
impl_reverse!(BooleanType, BooleanChunked);
impl_reverse!(Utf8Type, Utf8Chunked);
#[cfg(feature = "dtype-binary")]
impl_reverse!(BinaryType, BinaryChunked);
impl_reverse!(ListType, ListChunked);
#[cfg(feature = "object")]
impl<T: PolarsObject> ChunkReverse<ObjectType<T>> for ObjectChunked<T> {
fn reverse(&self) -> Self {
unsafe { self.take_unchecked((0..self.len()).rev().into()) }
}
}