vortex_array/arrays/bool/compute/
mask.rs

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