use std::collections::HashSet;
use church_encoded::boolean::{False, True};
use environment::Binding;
use term::{app, lam, var, Term};
pub fn default_bindings() -> HashSet<Binding> {
binds! {
Zero => zero(),
zero => zero(),
One => one(),
one => one(),
SUCC => succ(),
succ => succ(),
PRED => pred(),
pred => pred(),
IsZero => is_zero(),
is_zero => is_zero(),
}
}
pub fn zero() -> Term {
lam("f", lam("a", var("a")))
}
pub fn one() -> Term {
lam("f", lam("a", app(var("f"), var("a"))))
}
macro_rules! church_num {
($($typ:ty),*) => {
$(
impl From<$typ> for Term {
fn from(val: $typ) -> Self {
lam("f", lam("a", (0..val).fold(var("a"), |acc, _| app(var("f"), acc))))
}
}
)*
}
}
church_num!(u8, u16, u32, u64, u128, usize);
pub fn succ() -> Term {
lam(
"n",
lam(
"f",
lam("a", app(var("f"), app![var("n"), var("f"), var("a")])),
),
)
}
pub fn pred() -> Term {
lam(
"n",
lam(
"f",
lam(
"a",
app![
var("n"),
lam("g", lam("h", app(var("h"), app(var("g"), var("f"))))),
lam("u", var("a")),
lam("u", var("u"))
],
),
),
)
}
pub fn is_zero() -> Term {
lam("n", app![var("n"), lam("a", False()), True()])
}
#[cfg(test)]
mod tests;