vortex_array/arrays/varbinview/compute/
mask.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use std::sync::Arc;
5
6use vortex_error::VortexResult;
7
8use crate::ArrayRef;
9use crate::IntoArray;
10use crate::array::ArrayView;
11use crate::arrays::VarBinView;
12use crate::arrays::VarBinViewArray;
13use crate::scalar_fn::fns::mask::MaskReduce;
14use crate::validity::Validity;
15
16impl MaskReduce for VarBinView {
17    fn mask(array: ArrayView<'_, VarBinView>, mask: &ArrayRef) -> VortexResult<Option<ArrayRef>> {
18        // SAFETY: masking the validity does not affect the invariants
19        unsafe {
20            Ok(Some(
21                VarBinViewArray::new_handle_unchecked(
22                    array.views_handle().clone(),
23                    Arc::clone(array.data_buffers()),
24                    array.dtype().as_nullable(),
25                    array.validity()?.and(Validity::Array(mask.clone()))?,
26                )
27                .into_array(),
28            ))
29        }
30    }
31}
32
33#[cfg(test)]
34mod tests {
35    use crate::IntoArray;
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"]).into_array(),
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            .into_array(),
54        );
55    }
56}
vortex_array/arrays/varbinview/compute/mask.rs

vortex_array/arrays/varbinview/compute/
mask.rs
