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

vortex_array/arrays/varbinview/compute/
mask.rs
