use crate::eval::coercion::to_number;
use crate::eval::functions::check_arity;
use crate::types::{ErrorKind, Value};
pub fn log_fn(args: &[Value]) -> Value {
if let Some(err) = check_arity(args, 1, 2) {
return err;
}
let n = match to_number(args[0].clone()) {
Err(e) => return e,
Ok(v) => v,
};
if n <= 0.0 {
return Value::Error(ErrorKind::Num);
}
let result = if args.len() == 2 {
let base = match to_number(args[1].clone()) {
Err(e) => return e,
Ok(v) => v,
};
if base <= 0.0 {
return Value::Error(ErrorKind::Num);
}
if base == 1.0 {
return Value::Error(ErrorKind::DivByZero);
}
n.log(base)
} else {
n.log10()
};
if !result.is_finite() {
return Value::Error(ErrorKind::Num);
}
Value::Number(result)
}
pub fn log10_fn(args: &[Value]) -> Value {
if let Some(err) = check_arity(args, 1, 1) {
return err;
}
let n = match to_number(args[0].clone()) {
Err(e) => return e,
Ok(v) => v,
};
if n <= 0.0 {
return Value::Error(ErrorKind::Num);
}
let result = n.log10();
if !result.is_finite() {
return Value::Error(ErrorKind::Num);
}
Value::Number(result)
}
pub fn ln_fn(args: &[Value]) -> Value {
if let Some(err) = check_arity(args, 1, 1) {
return err;
}
let n = match to_number(args[0].clone()) {
Err(e) => return e,
Ok(v) => v,
};
if n <= 0.0 {
return Value::Error(ErrorKind::Num);
}
let result = n.ln();
if !result.is_finite() {
return Value::Error(ErrorKind::Num);
}
Value::Number(result)
}
#[cfg(test)]
mod tests;