grass/src/builtin/math.rs
2020-02-16 18:18:57 -05:00

58 lines
2.1 KiB
Rust

use std::collections::BTreeMap;
use super::Builtin;
use crate::units::Unit;
use crate::value::{Number, Value};
pub(crate) fn register(f: &mut BTreeMap<String, Builtin>) {
decl!(f "percentage", |args, _| {
max_args!(args, 1);
let num = match arg!(args, 0, "number").eval() {
Value::Dimension(n, Unit::None) => n * Number::from(100),
_ => todo!("expected unitless number in builtin function `percentage()`")
};
Ok(Value::Dimension(num, Unit::Percent))
});
decl!(f "round", |args, _| {
max_args!(args, 1);
match arg!(args, 0, "number").eval() {
Value::Dimension(n, u) => Ok(Value::Dimension(n.round(), u)),
v => return Err(format!("$number: {} is not a number.", v).into()),
}
});
decl!(f "ceil", |args, _| {
max_args!(args, 1);
match arg!(args, 0, "number").eval() {
Value::Dimension(n, u) => Ok(Value::Dimension(n.ceil(), u)),
v => return Err(format!("$number: {} is not a number.", v).into()),
}
});
decl!(f "floor", |args, _| {
max_args!(args, 1);
match arg!(args, 0, "number").eval() {
Value::Dimension(n, u) => Ok(Value::Dimension(n.floor(), u)),
v => return Err(format!("$number: {} is not a number.", v).into()),
}
});
decl!(f "abs", |args, _| {
max_args!(args, 1);
match arg!(args, 0, "number").eval() {
Value::Dimension(n, u) => Ok(Value::Dimension(n.abs(), u)),
v => return Err(format!("$number: {} is not a number.", v).into()),
}
});
decl!(f "comparable", |args, _| {
max_args!(args, 2);
let unit1 = match arg!(args, 0, "number1").eval() {
Value::Dimension(_, u) => u,
v => return Err(format!("$number1: {} is not a number.", v).into()),
};
let unit2 = match arg!(args, 1, "number2").eval() {
Value::Dimension(_, u) => u,
v => return Err(format!("$number2: {} is not a number.", v).into()),
};
Ok(Value::bool(unit1.comparable(&unit2)))
});
}