vortex_alp/alp_rd/
slice.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use std::ops::Range;
5
6use vortex_array::ArrayRef;
7use vortex_array::ExecutionCtx;
8use vortex_array::IntoArray;
9use vortex_array::arrays::SliceKernel;
10use vortex_error::VortexResult;
11
12use crate::alp_rd::ALPRDArray;
13use crate::alp_rd::ALPRDVTable;
14
15impl SliceKernel for ALPRDVTable {
16    fn slice(
17        array: &Self::Array,
18        range: Range<usize>,
19        _ctx: &mut ExecutionCtx,
20    ) -> VortexResult<Option<ArrayRef>> {
21        let left_parts_exceptions = array
22            .left_parts_patches()
23            .map(|patches| patches.slice(range.clone()))
24            .transpose()?
25            .flatten();
26
27        // SAFETY: slicing components does not change the encoded values
28        Ok(Some(unsafe {
29            ALPRDArray::new_unchecked(
30                array.dtype().clone(),
31                array.left_parts().slice(range.clone())?,
32                array.left_parts_dictionary().clone(),
33                array.right_parts().slice(range)?,
34                array.right_bit_width(),
35                left_parts_exceptions,
36            )
37            .into_array()
38        }))
39    }
40}
vortex_alp/alp_rd/slice.rs

vortex_alp/alp_rd/
slice.rs
