Skip to main content

vortex_array/arrays/map/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::array::ArrayView;
8use crate::arrays::ListView;
9use crate::arrays::map::Map;
10use crate::arrays::map::MapArrayExt;
11use crate::arrays::map::MapArraySlotsExt;
12use crate::arrays::map::compute::rebuild_map_from_array;
13use crate::executor::ExecutionCtx;
14use crate::scalar_fn::fns::mask::MaskKernel;
15use crate::scalar_fn::fns::mask::MaskReduce;
16
17impl MaskReduce for Map {
18    const VALIDITY_IS_METADATA_ONLY: bool = true;
19
20    fn mask(array: ArrayView<'_, Self>, mask: &ArrayRef) -> VortexResult<Option<ArrayRef>> {
21        let Some(entries) =
22            <ListView as MaskReduce>::mask(array.entries().as_::<ListView>(), mask)?
23        else {
24            return Ok(None);
25        };
26
27        rebuild_map_from_array(array.map_dtype().clone(), entries).map(Some)
28    }
29}
30
31impl MaskKernel for Map {
32    fn mask(
33        array: ArrayView<'_, Self>,
34        mask: &ArrayRef,
35        _ctx: &mut ExecutionCtx,
36    ) -> VortexResult<Option<ArrayRef>> {
37        <Self as MaskReduce>::mask(array, mask)
38    }
39}