Support "pre-image" for pruning predicate evaluation

datafusion/optimizer/Cargo.toml

@@ -49,6 +49,7 @@ chrono = { workspace = true }
 datafusion-common = { workspace = true, default-features = true }
 datafusion-expr = { workspace = true }
 datafusion-expr-common = { workspace = true }
+datafusion-functions = { workspace = true }
 datafusion-physical-expr = { workspace = true }
 indexmap = { workspace = true }
 itertools = { workspace = true }

datafusion/optimizer/src/simplify_expressions/expr_simplifier.rs

@@ -49,7 +49,6 @@ use datafusion_physical_expr::{create_physical_expr, execution_props::ExecutionP
 
 use super::inlist_simplifier::ShortenInListSimplifier;
 use super::utils::*;
-use crate::analyzer::type_coercion::TypeCoercionRewriter;
 use crate::simplify_expressions::guarantees::GuaranteeRewriter;
 use crate::simplify_expressions::regex::simplify_regex_expr;
 use crate::simplify_expressions::unwrap_cast::{
@@ -58,6 +57,13 @@ use crate::simplify_expressions::unwrap_cast::{
     unwrap_cast_in_comparison_for_binary,
 };
 use crate::simplify_expressions::SimplifyInfo;
+use crate::{
+    analyzer::type_coercion::TypeCoercionRewriter,
+    simplify_expressions::unwrap_date_part::{
+        is_date_part_expr_and_support_unwrap_date_part_in_comparison_for_binary,
+        unwrap_date_part_in_comparison_for_binary,
+    },
+};
 use datafusion_expr_common::casts::try_cast_literal_to_type;
 use indexmap::IndexSet;
 use regex::Regex;
@@ -1956,6 +1962,36 @@ impl<S: SimplifyInfo> TreeNodeRewriter for Simplifier<'_, S> {
                 }))
             }
 
+            // =======================================
+            // unwrap_date_part_in_comparison
+            // =======================================
+            //
+            // For case:
+            // date_part(expr as data_type) op literal
+            Expr::BinaryExpr(BinaryExpr { left, op, right })
+                if is_date_part_expr_and_support_unwrap_date_part_in_comparison_for_binary(
+                    info, &left, op, &right,
+                ) && op.supports_propagation() =>
+            {
+                unwrap_date_part_in_comparison_for_binary(info, *left, *right, op)?
+            }
+            // literal op date_part(literal, expression)
+            // -->
+            // date_part(literal, expression) op_swap literal
+            Expr::BinaryExpr(BinaryExpr { left, op, right })
+                if is_date_part_expr_and_support_unwrap_date_part_in_comparison_for_binary(
+                    info, &right, op, &left,
+                ) && op.supports_propagation()
+                    && op.swap().is_some() =>
+            {
+                unwrap_date_part_in_comparison_for_binary(
+                    info,
+                    *right,
+                    *left,
+                    op.swap().unwrap(),
+                )?
+            }
+
             // no additional rewrites possible
             expr => Transformed::no(expr),
         })

datafusion/optimizer/src/simplify_expressions/mod.rs

@@ -25,6 +25,7 @@ mod regex;
 pub mod simplify_exprs;
 mod simplify_predicates;
 mod unwrap_cast;
+mod unwrap_date_part;
 mod utils;
 
 // backwards compatibility