Make struct cast implementation pluggable

encodings/bytebool/src/compute.rs

@@ -32,7 +32,7 @@ impl CastReduce for ByteBool {
 
         let Some(new_validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

encodings/fastlanes/src/bitpacking/compute/cast.rs

@@ -42,7 +42,7 @@ impl CastReduce for BitPacked {
         }
         let Some(new_validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

encodings/fsst/src/compute/cast.rs

@@ -50,7 +50,7 @@ impl CastReduce for FSST {
         let codes = array.codes();
         let Some(new_codes_validity) = codes
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), codes.len())?
+            .trivially_cast_nullability(dtype.nullability(), codes.len())?
         else {
             return Ok(None);
         };

encodings/pco/src/compute/cast.rs

@@ -31,7 +31,7 @@ impl CastReduce for Pco {
         let unsliced_validity =
             child_to_validity(array.slots()[0].as_ref(), array.dtype().nullability());
         let Some(new_validity) =
-            unsliced_validity.trivial_cast_nullability(dtype.nullability(), array.len())?
+            unsliced_validity.trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

vortex-array/public-api.lock

@@ -4662,14 +4662,6 @@ impl vortex_array::arrays::slice::SliceReduce for vortex_array::arrays::Struct
 
 pub fn vortex_array::arrays::Struct::slice(vortex_array::ArrayView<'_, Self>, core::ops::range::Range<usize>) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 
-impl vortex_array::scalar_fn::fns::cast::CastKernel for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
-impl vortex_array::scalar_fn::fns::cast::CastReduce for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
 impl vortex_array::scalar_fn::fns::mask::MaskReduce for vortex_array::arrays::Struct
 
 pub fn vortex_array::arrays::Struct::mask(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::ArrayRef) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
@@ -6782,14 +6774,6 @@ impl vortex_array::arrays::slice::SliceReduce for vortex_array::arrays::Struct
 
 pub fn vortex_array::arrays::Struct::slice(vortex_array::ArrayView<'_, Self>, core::ops::range::Range<usize>) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 
-impl vortex_array::scalar_fn::fns::cast::CastKernel for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
-impl vortex_array::scalar_fn::fns::cast::CastReduce for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
 impl vortex_array::scalar_fn::fns::mask::MaskReduce for vortex_array::arrays::Struct
 
 pub fn vortex_array::arrays::Struct::mask(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::ArrayRef) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
@@ -13586,9 +13570,13 @@ impl vortex_array::optimizer::kernels::ArrayKernels
 
 pub fn vortex_array::optimizer::kernels::ArrayKernels::empty() -> Self
 
+pub fn vortex_array::optimizer::kernels::ArrayKernels::find_execute_parent(&self, vortex_session::registry::Id, vortex_session::registry::Id) -> core::option::Option<alloc::sync::Arc<[vortex_array::optimizer::kernels::ExecuteParentFn]>>
+
 pub fn vortex_array::optimizer::kernels::ArrayKernels::find_reduce_parent(&self, vortex_session::registry::Id, vortex_session::registry::Id) -> core::option::Option<alloc::sync::Arc<[vortex_array::optimizer::kernels::ReduceParentFn]>>
 
-pub fn vortex_array::optimizer::kernels::ArrayKernels::register_reduce_parent<I: core::iter::traits::collect::IntoIterator<Item = vortex_array::optimizer::kernels::ReduceParentFn>>(&self, vortex_session::registry::Id, vortex_session::registry::Id, I)
+pub fn vortex_array::optimizer::kernels::ArrayKernels::register_execute_parent(&self, vortex_session::registry::Id, vortex_session::registry::Id, &[vortex_array::optimizer::kernels::ExecuteParentFn])
+
+pub fn vortex_array::optimizer::kernels::ArrayKernels::register_reduce_parent(&self, vortex_session::registry::Id, vortex_session::registry::Id, &[vortex_array::optimizer::kernels::ReduceParentFn])
 
 impl core::default::Default for vortex_array::optimizer::kernels::ArrayKernels
 
@@ -13612,6 +13600,8 @@ impl<S: vortex_session::SessionExt> vortex_array::optimizer::kernels::ArrayKerne
 
 pub fn S::kernels(&self) -> vortex_session::Ref<'_, vortex_array::optimizer::kernels::ArrayKernels>
 
+pub type vortex_array::optimizer::kernels::ExecuteParentFn = fn(&vortex_array::ArrayRef, &vortex_array::ArrayRef, usize, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
+
 pub type vortex_array::optimizer::kernels::ReduceParentFn = fn(&vortex_array::ArrayRef, &vortex_array::ArrayRef, usize) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 
 pub mod vortex_array::optimizer::rules
@@ -16242,10 +16232,6 @@ impl vortex_array::scalar_fn::fns::cast::CastKernel for vortex_array::arrays::Pr
 
 pub fn vortex_array::arrays::Primitive::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Primitive>, &vortex_array::dtype::DType, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 
-impl vortex_array::scalar_fn::fns::cast::CastKernel for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
 impl vortex_array::scalar_fn::fns::cast::CastKernel for vortex_array::arrays::VarBin
 
 pub fn vortex_array::arrays::VarBin::cast(vortex_array::ArrayView<'_, vortex_array::arrays::VarBin>, &vortex_array::dtype::DType, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
@@ -16294,10 +16280,6 @@ impl vortex_array::scalar_fn::fns::cast::CastReduce for vortex_array::arrays::Pr
 
 pub fn vortex_array::arrays::Primitive::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Primitive>, &vortex_array::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
 
-impl vortex_array::scalar_fn::fns::cast::CastReduce for vortex_array::arrays::Struct
-
-pub fn vortex_array::arrays::Struct::cast(vortex_array::ArrayView<'_, vortex_array::arrays::Struct>, &vortex_array::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
-
 impl vortex_array::scalar_fn::fns::cast::CastReduce for vortex_array::arrays::VarBin
 
 pub fn vortex_array::arrays::VarBin::cast(vortex_array::ArrayView<'_, vortex_array::arrays::VarBin>, &vortex_array::dtype::DType) -> vortex_error::VortexResult<core::option::Option<vortex_array::ArrayRef>>
@@ -19578,10 +19560,10 @@ pub fn vortex_array::validity::Validity::to_array(&self, usize) -> vortex_array:
 
 pub fn vortex_array::validity::Validity::to_mask(&self, usize, &mut vortex_array::ExecutionCtx) -> vortex_error::VortexResult<vortex_mask::Mask>
 
-pub fn vortex_array::validity::Validity::trivial_cast_nullability(self, vortex_array::dtype::Nullability, usize) -> vortex_error::VortexResult<core::option::Option<vortex_array::validity::Validity>>
-
 pub fn vortex_array::validity::Validity::trivial_into_non_nullable(self, usize) -> vortex_error::VortexResult<core::option::Option<vortex_array::validity::Validity>>
 
+pub fn vortex_array::validity::Validity::trivially_cast_nullability(self, vortex_array::dtype::Nullability, usize) -> vortex_error::VortexResult<core::option::Option<vortex_array::validity::Validity>>
+
 pub fn vortex_array::validity::Validity::union_nullability(self, vortex_array::dtype::Nullability) -> Self
 
 impl vortex_array::validity::Validity

vortex-array/src/arrays/bool/compute/cast.rs

@@ -22,7 +22,7 @@ impl CastReduce for Bool {
 
         let Some(new_validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

vortex-array/src/arrays/decimal/compute/cast.rs

@@ -40,7 +40,7 @@ impl CastReduce for Decimal {
 
         let Some(new_validity) = array
             .validity()?
-            .trivial_cast_nullability(*to_nullability, array.len())?
+            .trivially_cast_nullability(*to_nullability, array.len())?
         else {
             return Ok(None);
         };

vortex-array/src/arrays/fixed_size_list/compute/cast.rs

@@ -39,7 +39,7 @@ impl CastReduce for FixedSizeList {
 
         let Some(validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

vortex-array/src/arrays/list/compute/cast.rs

@@ -23,7 +23,7 @@ impl CastReduce for List {
 
         let Some(validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

vortex-array/src/arrays/listview/compute/cast.rs

@@ -42,7 +42,7 @@ impl CastReduce for ListView {
         };
         let Some(validity) = array
             .validity()?
-            .trivial_cast_nullability(dtype.nullability(), array.len())?
+            .trivially_cast_nullability(dtype.nullability(), array.len())?
         else {
             return Ok(None);
         };

vortex-array/src/arrays/primitive/compute/cast.rs

@@ -36,7 +36,7 @@ impl CastReduce for Primitive {
 
         let Some(new_validity) = array
             .validity()?
-            .trivial_cast_nullability(*new_nullability, array.len())?
+            .trivially_cast_nullability(*new_nullability, array.len())?
         else {
             return Ok(None);
         };