vortex_array/arrays/bool/compute/
mask.rs

1use vortex_error::VortexResult;
2use vortex_mask::Mask;
3
4use crate::arrays::{BoolArray, BoolVTable};
5use crate::compute::{MaskKernel, MaskKernelAdapter};
6use crate::vtable::ValidityHelper;
7use crate::{ArrayRef, IntoArray, register_kernel};
8
9impl MaskKernel for BoolVTable {
10    fn mask(&self, array: &BoolArray, mask: &Mask) -> VortexResult<ArrayRef> {
11        Ok(
12            BoolArray::new(array.boolean_buffer().clone(), array.validity().mask(mask)?)
13                .into_array(),
14        )
15    }
16}
17
18register_kernel!(MaskKernelAdapter(BoolVTable).lift());