Lines
100 %
Functions
15.15 %
use conjure_core::ast::{Expression, ReturnType};
use conjure_core::metadata::Metadata;
use conjure_core::rule_engine::{
register_rule, register_rule_set, ApplicationError, ApplicationError::*, ApplicationResult,
Reduction,
};
use uniplate::Uniplate;
use crate::ast::{Atom, Literal, SymbolTable};
use crate::{into_matrix_expr, matrix_expr};
use super::utils::is_all_constant;
register_rule_set!("Bubble", ("Base"));
// Bubble reduction rules
/*
Reduce bubbles with a boolean expression to a conjunction with their condition.
e.g. (a / b = c) @ (b != 0) => (a / b = c) & (b != 0)
*/
#[register_rule(("Bubble", 8900))]
fn expand_bubble(expr: &Expression, _: &SymbolTable) -> ApplicationResult {
match expr {
Expression::Bubble(_, a, b) if a.return_type() == Some(ReturnType::Bool) => {
Ok(Reduction::pure(Expression::And(
Metadata::new(),
Box::new(matrix_expr![*a.clone(), *b.clone()]),
)))
}
_ => Err(ApplicationError::RuleNotApplicable),
Bring bubbles with a non-boolean expression higher up the tree.
E.g. ((a / b) @ (b != 0)) = c => (a / b = c) @ (b != 0)
fn bubble_up(expr: &Expression, _: &SymbolTable) -> ApplicationResult {
let mut sub = expr.children();
let mut bubbled_conditions = vec![];
for e in sub.iter_mut() {
if let Expression::Bubble(_, a, b) = e {
if a.return_type() != Some(ReturnType::Bool) {
bubbled_conditions.push(*b.clone());
*e = *a.clone();
if bubbled_conditions.is_empty() {
return Err(ApplicationError::RuleNotApplicable);
Ok(Reduction::pure(Expression::Bubble(
Box::new(expr.with_children(sub)),
Box::new(Expression::And(
Box::new(into_matrix_expr![bubbled_conditions]),
)),
// Bubble applications
/// Converts an unsafe division to a safe division with a bubble condition.
///
/// ```text
/// a / b => (a / b) @ (b != 0)
/// ```
/// Division by zero is undefined and therefore not allowed, so we add a condition to check for it.
/// This condition is brought up the tree and expanded into a conjunction with the first
/// boolean-type expression it is paired with.
#[register_rule(("Bubble", 6000))]
fn div_to_bubble(expr: &Expression, _: &SymbolTable) -> ApplicationResult {
if is_all_constant(expr) {
return Err(RuleNotApplicable);
if let Expression::UnsafeDiv(_, a, b) = expr {
// bubble bottom up
if !a.is_safe() || !b.is_safe() {
return Ok(Reduction::pure(Expression::Bubble(
Box::new(Expression::SafeDiv(Metadata::new(), a.clone(), b.clone())),
Box::new(Expression::Neq(
b.clone(),
Box::new(Expression::from(0)),
)));
Err(ApplicationError::RuleNotApplicable)
/// Converts an unsafe mod to a safe mod with a bubble condition.
/// a % b => (a % b) @ (b != 0)
/// Mod zero is undefined and therefore not allowed, so we add a condition to check for it.
fn mod_to_bubble(expr: &Expression, _: &SymbolTable) -> ApplicationResult {
if let Expression::UnsafeMod(_, a, b) = expr {
Box::new(Expression::SafeMod(Metadata::new(), a.clone(), b.clone())),
/// Converts an unsafe pow to a safe pow with a bubble condition.
/// a**b => (a ** b) @ ((a!=0 \/ b!=0) /\ b>=0
/// Pow is only defined when `(a!=0 \/ b!=0) /\ b>=0`, so we add a condition to check for it.
fn pow_to_bubble(expr: &Expression, _: &SymbolTable) -> ApplicationResult {
if let Expression::UnsafePow(_, a, b) = expr.clone() {
Box::new(Expression::SafePow(Metadata::new(), a.clone(), b.clone())),
Box::new(matrix_expr![
Expression::Or(
Expression::Neq(
a,
Box::new(Atom::Literal(Literal::Int(0)).into()),
),
]),
Expression::Geq(
b,