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