vortex_array/arrays/dict/compute/
like.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5
6use super::{DictArray, DictVTable};
7use crate::arrays::ConstantArray;
8use crate::compute::{LikeKernel, LikeKernelAdapter, LikeOptions, like};
9use crate::{Array, ArrayRef, IntoArray, register_kernel};
10
11impl LikeKernel for DictVTable {
12    fn like(
13        &self,
14        array: &DictArray,
15        pattern: &dyn Array,
16        options: LikeOptions,
17    ) -> VortexResult<Option<ArrayRef>> {
18        // if we have more values than codes, it is faster to canonicalise first.
19        if array.values().len() > array.codes().len() {
20            return Ok(None);
21        }
22        if let Some(pattern) = pattern.as_constant() {
23            let pattern = ConstantArray::new(pattern, array.values().len()).into_array();
24            let values = like(array.values(), &pattern, options)?;
25
26            // SAFETY: LIKE preserves the len of the values, so codes are still pointing at
27            //  valid positions.
28            unsafe {
29                Ok(Some(
30                    DictArray::new_unchecked(array.codes().clone(), values).into_array(),
31                ))
32            }
33        } else {
34            Ok(None)
35        }
36    }
37}
38
39register_kernel!(LikeKernelAdapter(DictVTable).lift());
vortex_array/arrays/dict/compute/like.rs

vortex_array/arrays/dict/compute/
like.rs
