grass/src/builtin/math.rs
2020-02-16 22:04:54 -05:00

59 lines
2.2 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") {
Value::Dimension(n, Unit::None) => n * Number::from(100),
v @ Value::Dimension(..) => return Err(format!("$number: Expected {} to have no units.", v).into()),
v => return Err(format!("$number: {} is not a number.", v).into()),
};
Ok(Value::Dimension(num, Unit::Percent))
});
decl!(f "round", |args, _| {
max_args!(args, 1);
match arg!(args, 0, "number") {
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") {
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") {
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") {
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") {
Value::Dimension(_, u) => u,
v => return Err(format!("$number1: {} is not a number.", v).into()),
};
let unit2 = match arg!(args, 1, "number2") {
Value::Dimension(_, u) => u,
v => return Err(format!("$number2: {} is not a number.", v).into()),
};
Ok(Value::bool(unit1.comparable(&unit2)))
});
}