vortex-alp 0.65.0

Vortex ALP array
Documentation 
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

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

use std::ops::Range;

use vortex_array::ArrayRef;
use vortex_array::ExecutionCtx;
use vortex_array::IntoArray;
use vortex_array::arrays::slice::SliceKernel;
use vortex_error::VortexResult;

use crate::alp_rd::ALPRD;
use crate::alp_rd::ALPRDArray;

impl SliceKernel for ALPRD {
    fn slice(
        array: &Self::Array,
        range: Range<usize>,
        _ctx: &mut ExecutionCtx,
    ) -> VortexResult<Option<ArrayRef>> {
        let left_parts_exceptions = array
            .left_parts_patches()
            .map(|patches| patches.slice(range.clone()))
            .transpose()?
            .flatten();

        // SAFETY: slicing components does not change the encoded values
        Ok(Some(unsafe {
            ALPRDArray::new_unchecked(
                array.dtype().clone(),
                array.left_parts().slice(range.clone())?,
                array.left_parts_dictionary().clone(),
                array.right_parts().slice(range)?,
                array.right_bit_width(),
                left_parts_exceptions,
            )
            .into_array()
        }))
    }
}