Add stats rewrite session API

Signed-off-by: Nicholas Gates <nick@nickgates.com>

Author: gatesn

vortex-array/public-api.lock

@@ -12778,10 +12778,14 @@ pub fn vortex_array::expr::Expression::children(&self) -> &alloc::sync::Arc<allo
 
 pub fn vortex_array::expr::Expression::display_tree(&self) -> impl core::fmt::Display
 
+pub fn vortex_array::expr::Expression::falsify(&self, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
 pub fn vortex_array::expr::Expression::fmt_sql(&self, &mut core::fmt::Formatter<'_>) -> core::fmt::Result
 
 pub fn vortex_array::expr::Expression::return_dtype(&self, &vortex_array::dtype::DType) -> vortex_error::VortexResult<vortex_array::dtype::DType>
 
+pub fn vortex_array::expr::Expression::satisfy(&self, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
 pub fn vortex_array::expr::Expression::scalar_fn(&self) -> &vortex_array::scalar_fn::ScalarFnRef
 
 pub fn vortex_array::expr::Expression::stat_expression(&self, vortex_array::expr::stats::Stat, &dyn vortex_array::expr::pruning::StatsCatalog) -> core::option::Option<vortex_array::expr::Expression>
@@ -19774,6 +19778,68 @@ pub fn vortex_array::stats::expr::sum(vortex_array::expr::Expression) -> vortex_
 
 pub mod vortex_array::stats::flatbuffers
 
+pub mod vortex_array::stats::rewrite
+
+pub struct vortex_array::stats::rewrite::StatsRewriteCtx<'a>
+
+impl<'a> vortex_array::stats::StatsRewriteCtx<'a>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::falsify(&self, &vortex_array::expr::Expression) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::new(&'a vortex_session::VortexSession) -> Self
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::satisfy(&self, &vortex_array::expr::Expression) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::session(&self) -> &'a vortex_session::VortexSession
+
+pub trait vortex_array::stats::rewrite::StatsRewriteRule: core::fmt::Debug + core::marker::Send + core::marker::Sync + 'static
+
+pub fn vortex_array::stats::rewrite::StatsRewriteRule::falsify(&self, &vortex_array::expr::Expression, &vortex_array::stats::StatsRewriteCtx<'_>) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::rewrite::StatsRewriteRule::satisfy(&self, &vortex_array::expr::Expression, &vortex_array::stats::StatsRewriteCtx<'_>) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::rewrite::StatsRewriteRule::scalar_fn_id(&self) -> vortex_array::scalar_fn::ScalarFnId
+
+pub fn vortex_array::stats::rewrite::falsify(&vortex_array::expr::Expression, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::rewrite::satisfy(&vortex_array::expr::Expression, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub type vortex_array::stats::rewrite::StatsRewriteRuleRef = alloc::sync::Arc<dyn vortex_array::stats::StatsRewriteRule>
+
+pub mod vortex_array::stats::session
+
+pub struct vortex_array::stats::session::StatsRewriteSession
+
+impl vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::register<R: vortex_array::stats::StatsRewriteRule>(&self, R)
+
+pub fn vortex_array::stats::StatsRewriteSession::register_ref(&self, vortex_array::stats::StatsRewriteRuleRef)
+
+pub fn vortex_array::stats::StatsRewriteSession::rules_for(&self, vortex_array::scalar_fn::ScalarFnId) -> alloc::vec::Vec<vortex_array::stats::StatsRewriteRuleRef>
+
+impl core::default::Default for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::default() -> vortex_array::stats::StatsRewriteSession
+
+impl core::fmt::Debug for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::fmt(&self, &mut core::fmt::Formatter<'_>) -> core::fmt::Result
+
+impl vortex_session::SessionVar for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::as_any(&self) -> &dyn core::any::Any
+
+pub fn vortex_array::stats::StatsRewriteSession::as_any_mut(&mut self) -> &mut dyn core::any::Any
+
+pub trait vortex_array::stats::session::StatsRewriteSessionExt: vortex_session::SessionExt
+
+pub fn vortex_array::stats::session::StatsRewriteSessionExt::stats_rewrites(&self) -> vortex_session::Ref<'_, vortex_array::stats::StatsRewriteSession>
+
+impl<S: vortex_session::SessionExt> vortex_array::stats::StatsRewriteSessionExt for S
+
+pub fn S::stats_rewrites(&self) -> vortex_session::Ref<'_, vortex_array::stats::StatsRewriteSession>
+
 pub struct vortex_array::stats::ArrayStats
 
 impl vortex_array::stats::ArrayStats
@@ -19834,6 +19900,42 @@ pub fn vortex_array::stats::MutTypedStatsSetRef<'_, '_>::is_empty(&self) -> bool
 
 pub fn vortex_array::stats::MutTypedStatsSetRef<'_, '_>::len(&self) -> usize
 
+pub struct vortex_array::stats::StatsRewriteCtx<'a>
+
+impl<'a> vortex_array::stats::StatsRewriteCtx<'a>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::falsify(&self, &vortex_array::expr::Expression) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::new(&'a vortex_session::VortexSession) -> Self
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::satisfy(&self, &vortex_array::expr::Expression) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteCtx<'a>::session(&self) -> &'a vortex_session::VortexSession
+
+pub struct vortex_array::stats::StatsRewriteSession
+
+impl vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::register<R: vortex_array::stats::StatsRewriteRule>(&self, R)
+
+pub fn vortex_array::stats::StatsRewriteSession::register_ref(&self, vortex_array::stats::StatsRewriteRuleRef)
+
+pub fn vortex_array::stats::StatsRewriteSession::rules_for(&self, vortex_array::scalar_fn::ScalarFnId) -> alloc::vec::Vec<vortex_array::stats::StatsRewriteRuleRef>
+
+impl core::default::Default for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::default() -> vortex_array::stats::StatsRewriteSession
+
+impl core::fmt::Debug for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::fmt(&self, &mut core::fmt::Formatter<'_>) -> core::fmt::Result
+
+impl vortex_session::SessionVar for vortex_array::stats::StatsRewriteSession
+
+pub fn vortex_array::stats::StatsRewriteSession::as_any(&self) -> &dyn core::any::Any
+
+pub fn vortex_array::stats::StatsRewriteSession::as_any_mut(&mut self) -> &mut dyn core::any::Any
+
 pub struct vortex_array::stats::StatsSet
 
 impl vortex_array::stats::StatsSet
@@ -20002,14 +20104,34 @@ pub fn vortex_array::stats::TypedStatsSetRef<'_, '_>::len(&self) -> usize
 
 pub const vortex_array::stats::PRUNING_STATS: &[vortex_array::expr::stats::Stat]
 
+pub trait vortex_array::stats::StatsRewriteRule: core::fmt::Debug + core::marker::Send + core::marker::Sync + 'static
+
+pub fn vortex_array::stats::StatsRewriteRule::falsify(&self, &vortex_array::expr::Expression, &vortex_array::stats::StatsRewriteCtx<'_>) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteRule::satisfy(&self, &vortex_array::expr::Expression, &vortex_array::stats::StatsRewriteCtx<'_>) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
+pub fn vortex_array::stats::StatsRewriteRule::scalar_fn_id(&self) -> vortex_array::scalar_fn::ScalarFnId
+
+pub trait vortex_array::stats::StatsRewriteSessionExt: vortex_session::SessionExt
+
+pub fn vortex_array::stats::StatsRewriteSessionExt::stats_rewrites(&self) -> vortex_session::Ref<'_, vortex_array::stats::StatsRewriteSession>
+
+impl<S: vortex_session::SessionExt> vortex_array::stats::StatsRewriteSessionExt for S
+
+pub fn S::stats_rewrites(&self) -> vortex_session::Ref<'_, vortex_array::stats::StatsRewriteSession>
+
 pub fn vortex_array::stats::as_stat_bitset_bytes(&[vortex_array::expr::stats::Stat]) -> alloc::vec::Vec<u8>
 
+pub fn vortex_array::stats::falsify(&vortex_array::expr::Expression, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
 pub fn vortex_array::stats::min_max(vortex_array::expr::Expression) -> vortex_array::expr::Expression
 
 pub fn vortex_array::stats::nan_count(vortex_array::expr::Expression) -> vortex_array::expr::Expression
 
 pub fn vortex_array::stats::null_count(vortex_array::expr::Expression) -> vortex_array::expr::Expression
 
+pub fn vortex_array::stats::satisfy(&vortex_array::expr::Expression, &vortex_session::VortexSession) -> vortex_error::VortexResult<core::option::Option<vortex_array::expr::Expression>>
+
 pub fn vortex_array::stats::stat(vortex_array::expr::Expression, vortex_array::aggregate_fn::AggregateFnRef) -> vortex_array::expr::Expression
 
 pub fn vortex_array::stats::stats_from_bitset_bytes(&[u8]) -> alloc::vec::Vec<vortex_array::expr::stats::Stat>
@@ -20018,6 +20140,8 @@ pub fn vortex_array::stats::sum(vortex_array::expr::Expression) -> vortex_array:
 
 pub type vortex_array::stats::StatsArray = [(vortex_array::expr::stats::Stat, vortex_array::expr::stats::Precision<vortex_array::scalar::ScalarValue>); 4]
 
+pub type vortex_array::stats::StatsRewriteRuleRef = alloc::sync::Arc<dyn vortex_array::stats::StatsRewriteRule>
+
 pub mod vortex_array::stream
 
 pub struct vortex_array::stream::ArrayStreamAdapter<S>

vortex-array/src/expr/expression.rs

@@ -12,6 +12,7 @@ use std::sync::Arc;
 use itertools::Itertools;
 use vortex_error::VortexResult;
 use vortex_error::vortex_ensure;
+use vortex_session::VortexSession;
 
 use crate::dtype::DType;
 use crate::expr::StatsCatalog;
@@ -135,6 +136,22 @@ impl Expression {
         self.scalar_fn().stat_falsification(self, catalog)
     }
 
+    /// Returns an expression that proves this predicate is definitely false from stats.
+    ///
+    /// If the returned expression evaluates to `true` for a stats scope, this expression is
+    /// guaranteed to be false for every row in that scope. `false` and `null` are unknown.
+    pub fn falsify(&self, session: &VortexSession) -> VortexResult<Option<Expression>> {
+        crate::stats::rewrite::falsify(self, session)
+    }
+
+    /// Returns an expression that proves this predicate is definitely true from stats.
+    ///
+    /// If the returned expression evaluates to `true` for a stats scope, this expression is
+    /// guaranteed to be true for every row in that scope. `false` and `null` are unknown.
+    pub fn satisfy(&self, session: &VortexSession) -> VortexResult<Option<Expression>> {
+        crate::stats::rewrite::satisfy(self, session)
+    }
+
     /// Returns an expression representing the zoned statistic for the given stat, if available.
     ///
     /// The [`StatsCatalog`] returns expressions that can be evaluated using the zone map as a

vortex-array/src/stats/mod.rs

@@ -17,9 +17,13 @@ pub use stats_set::*;
 mod array;
 pub mod expr;
 pub mod flatbuffers;
+pub mod rewrite;
+pub mod session;
 mod stats_set;
 
 pub use array::*;
+pub use rewrite::*;
+pub use session::*;
 use vortex_error::VortexExpect;
 
 use crate::expr::stats::Stat;