u

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>

Author: joseph-isaacs

vortex-array/public-api.lock

@@ -1,5 +1,5 @@
  Documenting vortex-array v0.1.0 (/Users/joeisaacs/git/spiraldb/vortex/vortex-array)
-    Finished `dev` profile [unoptimized + debuginfo] target(s) in 2.75s
+    Finished `dev` profile [unoptimized + debuginfo] target(s) in 3.03s
 pub mod vortex_array
 pub mod vortex_array::accessor
 pub trait vortex_array::accessor::ArrayAccessor<Item: ?core::marker::Sized>
@@ -63,6 +63,8 @@ impl vortex_array::expr::CastReduce for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::cast(array: &vortex_array::arrays::DictArray, dtype: &vortex_dtype::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 impl vortex_array::expr::FillNullKernel for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::fill_null(array: &vortex_array::arrays::DictArray, fill_value: &vortex_scalar::scalar::Scalar, _ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
+impl vortex_array::expr::LikeReduce for vortex_array::arrays::DictVTable
+pub fn vortex_array::arrays::DictVTable::like(array: &vortex_array::arrays::DictArray, pattern: &dyn vortex_array::Array, options: vortex_array::expr::LikeOptions) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 impl vortex_array::vtable::BaseArrayVTable<vortex_array::arrays::DictVTable> for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::array_eq(array: &vortex_array::arrays::DictArray, other: &vortex_array::arrays::DictArray, precision: vortex_array::Precision) -> bool
 pub fn vortex_array::arrays::DictVTable::array_hash<H: core::hash::Hasher>(array: &vortex_array::arrays::DictArray, state: &mut H, precision: vortex_array::Precision)
@@ -569,6 +571,8 @@ impl vortex_array::expr::CastReduce for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::cast(array: &vortex_array::arrays::DictArray, dtype: &vortex_dtype::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 impl vortex_array::expr::FillNullKernel for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::fill_null(array: &vortex_array::arrays::DictArray, fill_value: &vortex_scalar::scalar::Scalar, _ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
+impl vortex_array::expr::LikeReduce for vortex_array::arrays::DictVTable
+pub fn vortex_array::arrays::DictVTable::like(array: &vortex_array::arrays::DictArray, pattern: &dyn vortex_array::Array, options: vortex_array::expr::LikeOptions) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 impl vortex_array::vtable::BaseArrayVTable<vortex_array::arrays::DictVTable> for vortex_array::arrays::DictVTable
 pub fn vortex_array::arrays::DictVTable::array_eq(array: &vortex_array::arrays::DictArray, other: &vortex_array::arrays::DictArray, precision: vortex_array::Precision) -> bool
 pub fn vortex_array::arrays::DictVTable::array_hash<H: core::hash::Hasher>(array: &vortex_array::arrays::DictArray, state: &mut H, precision: vortex_array::Precision)
@@ -4571,6 +4575,8 @@ pub trait vortex_array::expr::LikeKernel: vortex_array::vtable::VTable
 pub fn vortex_array::expr::LikeKernel::like(array: &Self::Array, pattern: &dyn vortex_array::Array, options: vortex_array::expr::LikeOptions, ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 pub trait vortex_array::expr::LikeReduce: vortex_array::vtable::VTable
 pub fn vortex_array::expr::LikeReduce::like(array: &Self::Array, pattern: &dyn vortex_array::Array, options: vortex_array::expr::LikeOptions) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
+impl vortex_array::expr::LikeReduce for vortex_array::arrays::DictVTable
+pub fn vortex_array::arrays::DictVTable::like(array: &vortex_array::arrays::DictArray, pattern: &dyn vortex_array::Array, options: vortex_array::expr::LikeOptions) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 pub trait vortex_array::expr::NotKernel: vortex_array::vtable::VTable
 pub fn vortex_array::expr::NotKernel::invert(array: &Self::Array, ctx: &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 pub trait vortex_array::expr::NotReduce: vortex_array::vtable::VTable

vortex-array/src/arrays/dict/compute/like.rs

@@ -0,0 +1,44 @@
+// SPDX-License-Identifier: Apache-2.0
+// SPDX-FileCopyrightText: Copyright the Vortex contributors
+
+use vortex_error::VortexResult;
+
+use super::DictArray;
+use super::DictVTable;
+use crate::Array;
+use crate::ArrayRef;
+use crate::IntoArray;
+use crate::arrays::ConstantArray;
+use crate::expr::LikeOptions;
+use crate::expr::LikeReduce;
+use crate::expr::like::arrow_like;
+
+impl LikeReduce for DictVTable {
+    fn like(
+        array: &DictArray,
+        pattern: &dyn Array,
+        options: LikeOptions,
+    ) -> VortexResult<Option<ArrayRef>> {
+        // If we have more values than codes, it is faster to canonicalize first.
+        if array.values().len() > array.codes().len() {
+            return Ok(None);
+        }
+        if let Some(pattern) = pattern.as_constant() {
+            let pattern = ConstantArray::new(pattern, array.values().len()).into_array();
+            let values = arrow_like(array.values(), &pattern, options)?;
+
+            // SAFETY: LIKE preserves the len of the values, so codes are still pointing at
+            //  valid positions.
+            // Preserve all_values_referenced since codes are unchanged.
+            unsafe {
+                Ok(Some(
+                    DictArray::new_unchecked(array.codes().clone(), values)
+                        .set_all_values_referenced(array.has_all_values_referenced())
+                        .into_array(),
+                ))
+            }
+        } else {
+            Ok(None)
+        }
+    }
+}

vortex-array/src/arrays/dict/compute/mod.rs

@@ -6,6 +6,7 @@ mod compare;
 mod fill_null;
 mod is_constant;
 mod is_sorted;
+mod like;
 mod min_max;
 pub(crate) mod rules;
 mod slice;

vortex-array/src/arrays/dict/compute/rules.rs

@@ -19,6 +19,7 @@ use crate::arrays::SliceReduceAdaptor;
 use crate::builtins::ArrayBuiltins;
 use crate::compute::CastReduceAdaptor;
 use crate::expr::Cast;
+use crate::expr::LikeReduceAdaptor;
 use crate::expr::Pack;
 use crate::optimizer::ArrayOptimizer;
 use crate::optimizer::rules::ArrayParentReduceRule;
@@ -27,6 +28,7 @@ use crate::optimizer::rules::ParentRuleSet;
 pub(crate) const PARENT_RULES: ParentRuleSet<DictVTable> = ParentRuleSet::new(&[
     ParentRuleSet::lift(&FilterReduceAdaptor(DictVTable)),
     ParentRuleSet::lift(&CastReduceAdaptor(DictVTable)),
+    ParentRuleSet::lift(&LikeReduceAdaptor(DictVTable)),
     ParentRuleSet::lift(&DictionaryScalarFnValuesPushDownRule),
     ParentRuleSet::lift(&DictionaryScalarFnCodesPullUpRule),
     ParentRuleSet::lift(&SliceReduceAdaptor(DictVTable)),