Add an "if" function

If the first argument is true, returns the second argument, otherwise, return the third
This commit is contained in:
Ophir LOJKINE 2022-03-14 14:59:48 +01:00
parent a1bfe7c0b6
commit a219f0b66f
3 changed files with 13 additions and 0 deletions

View File

@ -121,6 +121,15 @@ pub fn builtin_function(identifier: &str) -> Option<Function> {
Ok(Value::Float(max_float))
}
})),
"if" => Some(Function::new(|argument| {
if let [condition, if_true, if_false] = &argument.as_fixed_len_tuple(3)?[..] {
return Ok(if condition.as_boolean()? { if_true } else { if_false }.clone())
}
Err(EvalexprError::type_error(
argument.clone(),
vec![ValueType::Boolean, ValueType::Empty, ValueType::Empty],
))
})),
"len" => Some(Function::new(|argument| {
if let Ok(subject) = argument.as_string() {
Ok(Value::from(subject.len() as i64))

View File

@ -327,6 +327,7 @@
//! | `floor` | 1 | Numeric | Returns the largest integer less than or equal to a number |
//! | `round` | 1 | Numeric | Returns the nearest integer to a number. Rounds half-way cases away from 0.0 |
//! | `ceil` | 1 | Numeric | Returns the smallest integer greater than or equal to a number |
//! | `if` | 3 | Boolean, Any, Any | If the first argument is true, returns the second argument, otherwise, return the third |
//! | `math::ln` | 1 | Numeric | Returns the natural logarithm of the number |
//! | `math::log` | 2 | Numeric, Numeric | Returns the logarithm of the number with respect to an arbitrary base |
//! | `math::log2` | 1 | Numeric | Returns the base 2 logarithm of the number |

View File

@ -408,6 +408,9 @@ fn test_builtin_functions() {
assert_eq!(eval("shl(-6, 5)"), Ok(Value::Int(-192)));
assert_eq!(eval("shr(5, 1)"), Ok(Value::Int(2)));
assert_eq!(eval("shr(-6, 5)"), Ok(Value::Int(-1)));
assert_eq!(eval("if(true, -6, 5)"), Ok(Value::Int(-6)));
assert_eq!(eval("if(false, -6, 5)"), Ok(Value::Int(5)));
assert_eq!(eval("if(2-1==1, \"good\", 0)"), Ok(Value::String(String::from("good"))));
}
#[test]