vortex_array/arrays/primitive/compute/
mask.rs

1use vortex_error::VortexResult;
2use vortex_mask::Mask;
3
4use crate::arrays::PrimitiveEncoding;
5use crate::arrays::primitive::PrimitiveArray;
6use crate::compute::MaskFn;
7use crate::variants::PrimitiveArrayTrait as _;
8use crate::{Array, ArrayRef};
9
10impl MaskFn<&PrimitiveArray> for PrimitiveEncoding {
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
20#[cfg(test)]
21mod test {
22    use crate::arrays::PrimitiveArray;
23    use crate::compute::conformance::mask::test_mask;
24
25    #[test]
26    fn test_mask_non_nullable_array() {
27        let non_nullable_array = PrimitiveArray::from_iter([1, 2, 3, 4, 5]);
28        test_mask(&non_nullable_array);
29    }
30}