vortex_array/arrays/varbinview/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::{VarBinViewArray, VarBinViewVTable};
8use crate::compute::{MaskKernel, MaskKernelAdapter};
9use crate::vtable::ValidityHelper;
10use crate::{ArrayRef, IntoArray, register_kernel};
11
12impl MaskKernel for VarBinViewVTable {
13    fn mask(&self, array: &VarBinViewArray, mask: &Mask) -> VortexResult<ArrayRef> {
14        // SAFETY: masking the validity does not affect the invariants
15        unsafe {
16            Ok(VarBinViewArray::new_unchecked(
17                array.views().clone(),
18                array.buffers().clone(),
19                array.dtype().as_nullable(),
20                array.validity().mask(mask)?,
21            )
22            .into_array())
23        }
24    }
25}
26
27register_kernel!(MaskKernelAdapter(VarBinViewVTable).lift());
28
29#[cfg(test)]
30mod tests {
31    use crate::arrays::VarBinViewArray;
32    use crate::compute::conformance::mask::test_mask_conformance;
33
34    #[test]
35    fn take_mask_var_bin_view_array() {
36        test_mask_conformance(
37            VarBinViewArray::from_iter_str(["one", "two", "three", "four", "five"]).as_ref(),
38        );
39
40        test_mask_conformance(
41            VarBinViewArray::from_iter_nullable_str([
42                Some("one"),
43                None,
44                Some("three"),
45                Some("four"),
46                Some("five"),
47            ])
48            .as_ref(),
49        );
50    }
51}
vortex_array/arrays/varbinview/compute/mask.rs

vortex_array/arrays/varbinview/compute/
mask.rs
