research/argus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add sugar for creating comparison ops

Browse source
This commit is contained in:
Anand Balakrishnan 2023-03-20 10:26:04 -07:00
parent 55894d55fb
commit 79384d436d
No known key found for this signature in database
1 changed files with 53 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

54
argus-core/src/expr/num_ops.rs
View file

@ -1,4 +1,4 @@
use super::{internal_macros, NumExpr};
use super::{internal_macros, BoolExpr, NumExpr};
use std::ops::{Add, Div, Mul, Neg};
impl Neg for NumExpr {
@ -85,3 +85,55 @@ impl Div for NumExpr {
}
internal_macros::forward_box_binop! {impl Div, div for NumExpr, NumExpr }
use super::Ordering;
impl NumExpr {
pub fn less_than(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::Less { strict: true },
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
pub fn less_than_eq(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::Less { strict: false },
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
pub fn greater_than(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::Greater { strict: true },
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
pub fn greater_than_eq(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::Greater { strict: false },
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
pub fn equal(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::Eq,
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
pub fn not_equal(self, rhs: Self) -> BoolExpr {
BoolExpr::Cmp {
op: Ordering::NotEq,
lhs: Box::new(self),
rhs: Box::new(rhs),
}
}
}