polars_arrow/legacy/array/
slice.rs

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
use crate::array::Array;

/// Utility trait to slice concrete arrow arrays whilst keeping their
/// concrete type. E.g. don't return `Box<dyn Array>`.
pub trait SlicedArray {
    /// Slices this [`Array`].
    /// # Implementation
    /// This operation is `O(1)` over `len`.
    /// # Panic
    /// This function panics iff `offset + length > self.len()`.
    fn slice_typed(&self, offset: usize, length: usize) -> Self
    where
        Self: Sized;

    /// Slices the [`Array`].
    /// # Implementation
    /// This operation is `O(1)`.
    ///
    /// # Safety
    /// The caller must ensure that `offset + length <= self.len()`
    unsafe fn slice_typed_unchecked(&self, offset: usize, length: usize) -> Self
    where
        Self: Sized;
}

impl<T: Array + Clone> SlicedArray for T {
    fn slice_typed(&self, offset: usize, length: usize) -> Self {
        let mut arr = self.clone();
        arr.slice(offset, length);
        arr
    }

    unsafe fn slice_typed_unchecked(&self, offset: usize, length: usize) -> Self {
        let mut arr = self.clone();
        arr.slice_unchecked(offset, length);
        arr
    }
}