vortex_array/arrays/primitive/compute/
mask.rs

1use vortex_error::VortexResult;
2use vortex_mask::Mask;
3
4use crate::arrays::PrimitiveVTable;
5use crate::arrays::primitive::PrimitiveArray;
6use crate::compute::{MaskKernel, MaskKernelAdapter};
7use crate::vtable::ValidityHelper;
8use crate::{ArrayRef, IntoArray, register_kernel};
9
10impl MaskKernel for PrimitiveVTable {
11    fn mask(&self, array: &PrimitiveArray, mask: &Mask) -> VortexResult<ArrayRef> {
12        let validity = array.validity().mask(mask)?;
13        Ok(
14            PrimitiveArray::from_byte_buffer(array.byte_buffer().clone(), array.ptype(), validity)
15                .into_array(),
16        )
17    }
18}
19
20register_kernel!(MaskKernelAdapter(PrimitiveVTable).lift());
21
22#[cfg(test)]
23mod test {
24    use crate::arrays::PrimitiveArray;
25    use crate::compute::conformance::mask::test_mask;
26
27    #[test]
28    fn test_mask_non_nullable_array() {
29        let non_nullable_array = PrimitiveArray::from_iter([1, 2, 3, 4, 5]);
30        test_mask(non_nullable_array.as_ref());
31    }
32}