Skip to main content

vortex_array/arrays/varbinview/compute/
slice.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use std::ops::Range;
5use std::sync::Arc;
6
7use vortex_error::VortexResult;
8
9use crate::ArrayRef;
10use crate::IntoArray;
11use crate::array::ArrayView;
12use crate::arrays::VarBinView;
13use crate::arrays::VarBinViewArray;
14use crate::arrays::slice::SliceReduce;
15use crate::arrays::varbinview::BinaryView;
16
17impl SliceReduce for VarBinView {
18    fn slice(array: ArrayView<'_, Self>, range: Range<usize>) -> VortexResult<Option<ArrayRef>> {
19        let views = array
20            .views_handle()
21            .slice_typed::<BinaryView>(range.clone());
22        let data_buffers = Arc::clone(array.data_buffers());
23        let dtype = array.dtype().clone();
24        let validity = array.validity()?.slice(range)?;
25
26        // Safety:
27        // range is validated within bounds, and is shared between all children.
28        let array = unsafe {
29            VarBinViewArray::new_handle_unchecked(views, data_buffers, dtype, validity).into_array()
30        };
31
32        Ok(Some(array))
33    }
34}