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    const VALIDITY_IS_METADATA_ONLY: bool = true;
18
19    fn mask(array: ArrayView<'_, VarBinView>, mask: &ArrayRef) -> VortexResult<Option<ArrayRef>> {
20        // SAFETY: masking the validity does not affect the invariants
21        unsafe {
22            Ok(Some(
23                VarBinViewArray::new_handle_unchecked(
24                    array.views_handle().clone(),
25                    Arc::clone(array.data_buffers()),
26                    array.dtype().as_nullable(),
27                    array.validity()?.and(Validity::Array(mask.clone()))?,
28                )
29                .into_array(),
30            ))
31        }
32    }
33}
34
35#[cfg(test)]
36mod tests {
37    use crate::IntoArray;
38    use crate::VortexSessionExecute;
39    use crate::array_session;
40    use crate::arrays::VarBinViewArray;
41    use crate::compute::conformance::mask::test_mask_conformance;
42
43    #[test]
44    fn take_mask_var_bin_view_array() {
45        test_mask_conformance(
46            &VarBinViewArray::from_iter_str(["one", "two", "three", "four", "five"]).into_array(),
47            &mut array_session().create_execution_ctx(),
48        );
49
50        test_mask_conformance(
51            &VarBinViewArray::from_iter_nullable_str([
52                Some("one"),
53                None,
54                Some("three"),
55                Some("four"),
56                Some("five"),
57            ])
58            .into_array(),
59            &mut array_session().create_execution_ctx(),
60        );
61    }
62}
vortex_array/arrays/varbinview/compute/mask.rs

vortex_array/arrays/varbinview/compute/
mask.rs
