vortex_array/arrays/list/compute/
mask.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5use vortex_mask::Mask;
6
7use crate::arrays::{ListArray, ListVTable};
8use crate::compute::{MaskKernel, MaskKernelAdapter};
9use crate::vtable::ValidityHelper;
10use crate::{ArrayRef, IntoArray, register_kernel};
11
12impl MaskKernel for ListVTable {
13    fn mask(&self, array: &ListArray, mask: &Mask) -> VortexResult<ArrayRef> {
14        ListArray::try_new(
15            array.elements().clone(),
16            array.offsets().clone(),
17            array.validity().mask(mask)?,
18        )
19        .map(|a| a.into_array())
20    }
21}
22
23register_kernel!(MaskKernelAdapter(ListVTable).lift());