Add pruning support to between exprs (#2596)

Author: onursatici

vortex-expr/src/between.rs

@@ -8,7 +8,7 @@ use vortex_dtype::DType;
 use vortex_dtype::DType::Bool;
 use vortex_error::VortexResult;
 
-use crate::{ExprRef, VortexExpr};
+use crate::{BinaryExpr, ExprRef, VortexExpr};
 
 #[derive(Debug, Eq, Hash)]
 #[allow(clippy::derived_hash_with_manual_eq)]
@@ -33,6 +33,20 @@ impl Between {
             options,
         })
     }
+
+    pub fn to_binary_expr(&self) -> ExprRef {
+        let lhs = BinaryExpr::new_expr(
+            self.lower.clone(),
+            self.options.lower_strict.to_operator().into(),
+            self.arr.clone(),
+        );
+        let rhs = BinaryExpr::new_expr(
+            self.arr.clone(),
+            self.options.upper_strict.to_operator().into(),
+            self.upper.clone(),
+        );
+        BinaryExpr::new_expr(lhs, crate::Operator::And, rhs)
+    }
 }
 
 impl Display for Between {

vortex-expr/src/operators.rs

@@ -73,3 +73,16 @@ impl Operator {
         }
     }
 }
+
+impl From<compute::Operator> for Operator {
+    fn from(cmp_operator: compute::Operator) -> Self {
+        match cmp_operator {
+            compute::Operator::Eq => Operator::Eq,
+            compute::Operator::NotEq => Operator::NotEq,
+            compute::Operator::Gt => Operator::Gt,
+            compute::Operator::Gte => Operator::Gte,
+            compute::Operator::Lt => Operator::Lt,
+            compute::Operator::Lte => Operator::Lte,
+        }
+    }
+}

vortex-expr/src/pruning.rs

@@ -13,6 +13,7 @@ use vortex_dtype::{FieldName, Nullability};
 use vortex_error::{VortexExpect as _, VortexResult};
 use vortex_scalar::Scalar;
 
+use crate::between::Between;
 use crate::{
     BinaryExpr, ExprRef, GetItem, Identity, Literal, Not, Operator, VortexExprExt, and, eq,
     get_item, gt, ident, lit, not, or,
@@ -226,6 +227,10 @@ fn convert_to_pruning_expression(expr: &ExprRef) -> PruningPredicateStats {
         };
     }
 
+    if let Some(between_expr) = expr.as_any().downcast_ref::<Between>() {
+        return convert_to_pruning_expression(&between_expr.to_binary_expr());
+    }
+
     not_prunable()
 }