1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
//! # ksl::builtin::bimath
//!
//! Built-in math functions which expects two number arguments.
//!
//! - ArcTan2
//! - Div
//! - Max
//! - Min
//! - Mod
//! - NDiv
//! - Power
//! - Quot
//! - Rem
//! - Sub
use crate::{Environment, builtin::BiMathType, eval::apply::eval_apply, is_number_eq, value::Value};
pub(crate) fn builtin(args: &[Value], ftype: BiMathType, env: Environment) -> Result<Value, std::sync::Arc<str>> {
if let [v1, v2] = args {
match (eval_apply(v1, env.clone())?, eval_apply(v2, env)?) {
(Value::Number(n1), Value::Number(n2)) => match ftype {
BiMathType::ArcTan2 => Ok(Value::Number(n2.atan2(n1))),
BiMathType::Div => {
if is_number_eq(n2, 0.0) {
Err(std::sync::Arc::from(concat!(
"Error[ksl::builtin::Div]: ",
"Division by zero."
)))
} else {
Ok(Value::Number(n1 / n2))
}
}
BiMathType::Max => Ok(Value::Number(n1.max(n2))),
BiMathType::Min => Ok(Value::Number(n1.min(n2))),
BiMathType::Mod => {
if is_number_eq(n2, 0.0) {
Err(std::sync::Arc::from(concat!(
"Error[ksl::builtin::Mod]: ",
"Division by zero."
)))
} else if is_number_eq(n1.trunc(), n1) && is_number_eq(n2.trunc(), n2) {
Ok(Value::Number(n1 - n2 * (n1 / n2).floor()))
} else {
Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::Mod]: ",
"Expected two integers, but got `({}, {})`."
),
n1, n2
)))
}
}
BiMathType::NDiv => {
if is_number_eq(n2, 0.0) {
Err(std::sync::Arc::from(concat!(
"Error[ksl::builtin::NDiv]: ",
"Division by zero."
)))
} else if is_number_eq(n1.trunc(), n1) && is_number_eq(n2.trunc(), n2) {
Ok(Value::Number((n1 / n2).floor()))
} else {
Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::NDiv]: ",
"Expected two integers, but got `({}, {})`."
),
n1, n2
)))
}
}
BiMathType::Power => Ok(Value::Number(n1.powf(n2))),
BiMathType::Quot => {
if is_number_eq(n2, 0.0) {
Err(std::sync::Arc::from(concat!(
"Error[ksl::builtin::Quot]: ",
"Division by zero."
)))
} else if is_number_eq(n1.trunc(), n1) && is_number_eq(n2.trunc(), n2) {
Ok(Value::Number((n1 / n2).trunc()))
} else {
Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::Quot]: ",
"Expected two integers, but got `({}, {})`."
),
n1, n2
)))
}
}
BiMathType::Rem => {
if is_number_eq(n2, 0.0) {
Err(std::sync::Arc::from(concat!(
"Error[ksl::builtin::Rem]: ",
"Division by zero."
)))
} else if is_number_eq(n1.trunc(), n1) && is_number_eq(n2.trunc(), n2) {
Ok(Value::Number(n1 - n2 * (n1 / n2).trunc()))
} else {
Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::Rem]: ",
"Expected two integers, but got `({}, {})`."
),
n1, n2
)))
}
}
BiMathType::Sub => Ok(Value::Number(n1 - n2)),
},
(Value::Number(_), e) | (e, _) => Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::{:?}]: ",
"Expected a number, but got: `{}`."
),
ftype, e
))),
}
} else {
Err(std::sync::Arc::from(format!(
concat!(
"Error[ksl::builtin::{:?}]: ",
"Expected {} parameters, but {} were passed."
),
ftype,
2,
args.len()
)))
}
}