CastFn

Signed-off-by: Nicholas Gates <[email protected]>

Author: gatesn

vortex-array/src/arrays/scalar_fn/vtable/mod.rs

@@ -12,22 +12,25 @@ use vortex_buffer::BufferHandle;
 use vortex_dtype::DType;
 use vortex_error::VortexExpect;
 use vortex_error::VortexResult;
-use vortex_error::{vortex_bail, vortex_ensure};
+use vortex_error::vortex_bail;
+use vortex_error::vortex_ensure;
 use vortex_vector::Vector;
 
+use crate::Array;
+use crate::ArrayRef;
+use crate::IntoArray;
 use crate::arrays::scalar_fn::array::ScalarFnArray;
 use crate::arrays::scalar_fn::metadata::ScalarFnMetadata;
 use crate::execution::ExecutionCtx;
 use crate::expr::functions;
 use crate::expr::functions::scalar::ScalarFn;
 use crate::serde::ArrayChildren;
+use crate::vtable;
 use crate::vtable::ArrayId;
 use crate::vtable::ArrayVTable;
 use crate::vtable::ArrayVTableExt;
 use crate::vtable::NotSupported;
 use crate::vtable::VTable;
-use crate::{vtable, IntoArray};
-use crate::{Array, ArrayRef};
 
 vtable!(ScalarFn);
 

vortex-array/src/expr/expression.rs

@@ -17,8 +17,7 @@ use vortex_error::VortexResult;
 use vortex_vector::Vector;
 use vortex_vector::VectorOps;
 
-use crate::expr::display::DisplayTreeExpr;
-use crate::expr::stats::Stat;
+use crate::ArrayRef;
 use crate::expr::ChildName;
 use crate::expr::ExecutionArgs;
 use crate::expr::ExprId;
@@ -27,7 +26,8 @@ use crate::expr::ExpressionView;
 use crate::expr::Root;
 use crate::expr::StatsCatalog;
 use crate::expr::VTable;
-use crate::ArrayRef;
+use crate::expr::display::DisplayTreeExpr;
+use crate::expr::stats::Stat;
 
 /// A node in a Vortex expression tree.
 ///

vortex-array/src/expr/exprs/scalar_fn.rs

@@ -1,31 +1,32 @@
 // SPDX-License-Identifier: Apache-2.0
 // SPDX-FileCopyrightText: Copyright the Vortex contributors
 
-use itertools::Itertools;
 use std::fmt::Debug;
 use std::fmt::Formatter;
 use std::marker::PhantomData;
 use std::sync::Arc;
+
+use itertools::Itertools;
 use vortex_dtype::DType;
+use vortex_error::VortexResult;
 use vortex_error::vortex_bail;
 use vortex_error::vortex_ensure;
-use vortex_error::VortexResult;
 use vortex_session::SessionVar;
 use vortex_vector::Vector;
 
-use crate::expr::functions;
-use crate::expr::functions::scalar::ScalarFn;
-use crate::expr::functions::ScalarFnVTable;
-use crate::expr::stats::Stat;
-use crate::expr::transform::rules::Matcher;
+use crate::ArrayRef;
 use crate::expr::ChildName;
 use crate::expr::ExecutionArgs;
 use crate::expr::ExprId;
 use crate::expr::Expression;
 use crate::expr::ExpressionView;
 use crate::expr::StatsCatalog;
 use crate::expr::VTable;
-use crate::ArrayRef;
+use crate::expr::functions;
+use crate::expr::functions::ScalarFnVTable;
+use crate::expr::functions::scalar::ScalarFn;
+use crate::expr::stats::Stat;
+use crate::expr::transform::rules::Matcher;
 
 /// An expression that wraps arbitrary scalar functions.
 ///

vortex-array/src/expr/functions/scalar.rs

@@ -13,15 +13,16 @@ use vortex_error::VortexResult;
 use vortex_utils::debug_with::DebugWith;
 use vortex_vector::Datum;
 
-use crate::expr::functions::execution::ExecutionCtx;
+use crate::expr::Expression;
+use crate::expr::StatsCatalog;
+use crate::expr::functions::ArgName;
 use crate::expr::functions::Arity;
 use crate::expr::functions::FunctionId;
 use crate::expr::functions::NullHandling;
 use crate::expr::functions::ScalarFnVTable;
-use crate::expr::functions::{ArgName, VTable};
+use crate::expr::functions::VTable;
+use crate::expr::functions::execution::ExecutionCtx;
 use crate::expr::stats::Stat;
-use crate::expr::Expression;
-use crate::expr::StatsCatalog;
 
 /// An instance of a scalar function bound to some invocation options.
 pub struct ScalarFn {

vortex-array/src/expr/functions/vtable.rs

@@ -12,18 +12,18 @@ use std::sync::Arc;
 
 use arcref::ArcRef;
 use vortex_dtype::DType;
-use vortex_error::vortex_bail;
 use vortex_error::VortexExpect;
 use vortex_error::VortexResult;
+use vortex_error::vortex_bail;
 use vortex_vector::Datum;
 
-use crate::expr::functions::execution::ExecutionCtx;
-use crate::expr::functions::scalar::ScalarFn;
+use crate::expr::Expression;
+use crate::expr::StatsCatalog;
 use crate::expr::functions::ArgName;
 use crate::expr::functions::FunctionId;
+use crate::expr::functions::execution::ExecutionCtx;
+use crate::expr::functions::scalar::ScalarFn;
 use crate::expr::stats::Stat;
-use crate::expr::Expression;
-use crate::expr::StatsCatalog;
 
 /// A non-object-safe vtable trait for scalar function types.
 ///

vortex-array/src/expr/session/mod.rs

@@ -4,34 +4,34 @@
 mod rewrite;
 
 pub use rewrite::RewriteRuleRegistry;
-use vortex_session::registry::Registry;
 use vortex_session::Ref;
 use vortex_session::SessionExt;
+use vortex_session::registry::Registry;
 
+use crate::expr::ExprVTable;
+use crate::expr::VTable;
 use crate::expr::exprs::between::Between;
 use crate::expr::exprs::binary::Binary;
 use crate::expr::exprs::cast::Cast;
-use crate::expr::exprs::get_item::transform::PackGetItemRule;
 use crate::expr::exprs::get_item::GetItem;
+use crate::expr::exprs::get_item::transform::PackGetItemRule;
 use crate::expr::exprs::is_null::IsNull;
 use crate::expr::exprs::like::Like;
 use crate::expr::exprs::list_contains::ListContains;
 use crate::expr::exprs::literal::Literal;
-use crate::expr::exprs::merge::transform::RemoveMergeRule;
 use crate::expr::exprs::merge::Merge;
+use crate::expr::exprs::merge::transform::RemoveMergeRule;
 use crate::expr::exprs::not::Not;
 use crate::expr::exprs::pack::Pack;
 use crate::expr::exprs::root::Root;
-use crate::expr::exprs::select::transform::RemoveSelectRule;
 use crate::expr::exprs::select::Select;
+use crate::expr::exprs::select::transform::RemoveSelectRule;
 use crate::expr::transform::rules::Any;
 use crate::expr::transform::rules::Exact;
 use crate::expr::transform::rules::ParentReduceRule;
 use crate::expr::transform::rules::ReduceRule;
 use crate::expr::transform::rules::RuleContext;
 use crate::expr::transform::rules::TypedRuleContext;
-use crate::expr::ExprVTable;
-use crate::expr::VTable;
 
 /// Registry of expression vtables.
 pub type ExprRegistry = Registry<ExprVTable>;

vortex-array/src/expr/vtable.rs

@@ -12,19 +12,19 @@ use std::sync::Arc;
 
 use arcref::ArcRef;
 use vortex_dtype::DType;
-use vortex_error::vortex_bail;
-use vortex_error::vortex_err;
 use vortex_error::VortexExpect;
 use vortex_error::VortexResult;
+use vortex_error::vortex_bail;
+use vortex_error::vortex_err;
 use vortex_vector::Vector;
 use vortex_vector::VectorOps;
 
-use crate::expr::expression::Expression;
-use crate::expr::stats::Stat;
+use crate::ArrayRef;
 use crate::expr::ExprId;
 use crate::expr::ExpressionView;
 use crate::expr::StatsCatalog;
-use crate::ArrayRef;
+use crate::expr::expression::Expression;
+use crate::expr::stats::Stat;
 
 ///
 /// This trait defines the interface for expression vtables, including methods for
@@ -493,8 +493,8 @@ mod tests {
     use crate::expr::exprs::root::root;
     use crate::expr::exprs::select::select;
     use crate::expr::exprs::select::select_exclude;
-    use crate::expr::proto::deserialize_expr_proto;
     use crate::expr::proto::ExprSerializeProtoExt;
+    use crate::expr::proto::deserialize_expr_proto;
     use crate::expr::session::ExprRegistry;
     use crate::expr::session::ExprSession;
 

vortex-array/src/scalar_fns/cast/mod.rs

@@ -3,16 +3,26 @@
 
 mod array;
 
-use crate::expr::functions::{ArgName, Arity, ExecutionCtx, FunctionId, NullHandling, VTable};
-use crate::expr::stats::Stat;
-use crate::expr::{Expression, StatsCatalog};
-use crate::scalar_fns::BuiltinScalarFns;
 use prost::Message;
 use vortex_dtype::DType;
-use vortex_error::{vortex_err, vortex_panic, VortexExpect, VortexResult};
+use vortex_error::VortexExpect;
+use vortex_error::VortexResult;
+use vortex_error::vortex_err;
+use vortex_error::vortex_panic;
 use vortex_proto::expr as pb;
 use vortex_vector::Datum;
 
+use crate::expr::Expression;
+use crate::expr::StatsCatalog;
+use crate::expr::functions::ArgName;
+use crate::expr::functions::Arity;
+use crate::expr::functions::ExecutionCtx;
+use crate::expr::functions::FunctionId;
+use crate::expr::functions::NullHandling;
+use crate::expr::functions::VTable;
+use crate::expr::stats::Stat;
+use crate::scalar_fns::BuiltinScalarFns;
+
 pub struct CastFn;
 impl VTable for CastFn {
     type Options = DType;

vortex-array/src/scalar_fns/mod.rs

@@ -9,13 +9,16 @@
 //! This set of functions should cover the basics, and in general leans towards the semantics of
 //! the equivalent Arrow compute function.
 
-use crate::arrays::ScalarFnArrayExt;
-use crate::expr::{Expression, ScalarFnExprExt};
-use crate::scalar_fns::cast::CastFn;
-use crate::{Array, ArrayRef};
 use vortex_dtype::DType;
 use vortex_error::VortexResult;
 
+use crate::Array;
+use crate::ArrayRef;
+use crate::arrays::ScalarFnArrayExt;
+use crate::expr::Expression;
+use crate::expr::ScalarFnExprExt;
+use crate::scalar_fns::cast::CastFn;
+
 mod cast;
 
 /// A collection of built-in scalar functions that can be applied to expressions or arrays.

vortex-compute/src/cast/bool.rs

@@ -4,15 +4,15 @@
 use num_traits::One;
 use num_traits::Zero;
 use vortex_buffer::Buffer;
-use vortex_dtype::match_each_native_ptype;
 use vortex_dtype::DType;
-use vortex_error::vortex_bail;
+use vortex_dtype::match_each_native_ptype;
 use vortex_error::VortexResult;
+use vortex_error::vortex_bail;
+use vortex_vector::Vector;
+use vortex_vector::VectorOps;
 use vortex_vector::bool::BoolVector;
 use vortex_vector::null::NullVector;
 use vortex_vector::primitive::PVector;
-use vortex_vector::Vector;
-use vortex_vector::VectorOps;
 
 use crate::cast::Cast;