Line | Branch | Exec | Source |
---|---|---|---|
1 | /* | ||
2 | Copyright 2022, 2023, 2025 Joel Svensson svenssonjoel@yahoo.se | ||
3 | Copyright 2022, 2023 Benjamin Vedder | ||
4 | |||
5 | This program is free software: you can redistribute it and/or modify | ||
6 | it under the terms of the GNU General Public License as published by | ||
7 | the Free Software Foundation, either version 3 of the License, or | ||
8 | (at your option) any later version. | ||
9 | |||
10 | This program is distributed in the hope that it will be useful, | ||
11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
13 | GNU General Public License for more details. | ||
14 | |||
15 | You should have received a copy of the GNU General Public License | ||
16 | along with this program. If not, see <http://www.gnu.org/licenses/>. | ||
17 | */ | ||
18 | |||
19 | #include "extensions.h" | ||
20 | #include "lbm_utils.h" | ||
21 | |||
22 | #include <math.h> | ||
23 | |||
24 | #ifdef LBM_OPT_MATH_EXTENSIONS_SIZE | ||
25 | #pragma GCC optimize ("-Os") | ||
26 | #endif | ||
27 | #ifdef LBM_OPT_MATH_EXTENSIONS_SIZE_AGGRESSIVE | ||
28 | #pragma GCC optimize ("-Oz") | ||
29 | #endif | ||
30 | |||
31 | |||
32 | #ifdef __STRICT_ANSI__ | ||
33 | #define isnanf isnan | ||
34 | #define isinff isinf | ||
35 | #endif | ||
36 | |||
37 | // Math | ||
38 | |||
39 | 114 | static lbm_value ext_sin(lbm_value *args, lbm_uint argn) { | |
40 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 57 times.
|
114 | LBM_CHECK_ARGN_NUMBER(1); |
41 | 57 | return lbm_enc_float(sinf(lbm_dec_as_float(args[0]))); | |
42 | } | ||
43 | |||
44 | 114 | static lbm_value ext_cos(lbm_value *args, lbm_uint argn) { | |
45 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 57 times.
|
114 | LBM_CHECK_ARGN_NUMBER(1); |
46 | 57 | return lbm_enc_float(cosf(lbm_dec_as_float(args[0]))); | |
47 | } | ||
48 | |||
49 | 113 | static lbm_value ext_tan(lbm_value *args, lbm_uint argn) { | |
50 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
51 | 56 | return lbm_enc_float(tanf(lbm_dec_as_float(args[0]))); | |
52 | } | ||
53 | |||
54 | 113 | static lbm_value ext_asin(lbm_value *args, lbm_uint argn) { | |
55 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
56 | 56 | return lbm_enc_float(asinf(lbm_dec_as_float(args[0]))); | |
57 | } | ||
58 | |||
59 | 113 | static lbm_value ext_acos(lbm_value *args, lbm_uint argn) { | |
60 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
61 | 56 | return lbm_enc_float(acosf(lbm_dec_as_float(args[0]))); | |
62 | } | ||
63 | |||
64 | 113 | static lbm_value ext_atan(lbm_value *args, lbm_uint argn) { | |
65 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
66 | 56 | return lbm_enc_float(atanf(lbm_dec_as_float(args[0]))); | |
67 | } | ||
68 | |||
69 | 114 | static lbm_value ext_atan2(lbm_value *args, lbm_uint argn) { | |
70 |
2/2✓ Branch 0 taken 58 times.
✓ Branch 1 taken 56 times.
|
114 | LBM_CHECK_ARGN_NUMBER(2); |
71 | 56 | return lbm_enc_float(atan2f(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1]))); | |
72 | } | ||
73 | |||
74 | 114 | static lbm_value ext_pow(lbm_value *args, lbm_uint argn) { | |
75 |
2/2✓ Branch 0 taken 58 times.
✓ Branch 1 taken 56 times.
|
114 | LBM_CHECK_ARGN_NUMBER(2); |
76 | 56 | return lbm_enc_float(powf(lbm_dec_as_float(args[0]), lbm_dec_as_float(args[1]))); | |
77 | } | ||
78 | |||
79 | 114 | static lbm_value ext_exp(lbm_value *args, lbm_uint argn) { | |
80 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 57 times.
|
114 | LBM_CHECK_ARGN_NUMBER(1); |
81 | 57 | return lbm_enc_float(expf(lbm_dec_as_float(args[0]))); | |
82 | } | ||
83 | |||
84 | 171 | static lbm_value ext_sqrt(lbm_value *args, lbm_uint argn) { | |
85 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 114 times.
|
171 | LBM_CHECK_ARGN_NUMBER(1); |
86 | 114 | return lbm_enc_float(sqrtf(lbm_dec_as_float(args[0]))); | |
87 | } | ||
88 | |||
89 | 113 | static lbm_value ext_log(lbm_value *args, lbm_uint argn) { | |
90 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
91 | 56 | return lbm_enc_float(logf(lbm_dec_as_float(args[0]))); | |
92 | } | ||
93 | |||
94 | 169 | static lbm_value ext_log10(lbm_value *args, lbm_uint argn) { | |
95 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 112 times.
|
169 | LBM_CHECK_ARGN_NUMBER(1); |
96 | 112 | return lbm_enc_float(log10f(lbm_dec_as_float(args[0]))); | |
97 | } | ||
98 | |||
99 | 113 | static lbm_value ext_floor(lbm_value *args, lbm_uint argn) { | |
100 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
101 | 56 | return lbm_enc_float(floorf(lbm_dec_as_float(args[0]))); | |
102 | } | ||
103 | |||
104 | 113 | static lbm_value ext_ceil(lbm_value *args, lbm_uint argn) { | |
105 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
106 | 56 | return lbm_enc_float(ceilf(lbm_dec_as_float(args[0]))); | |
107 | } | ||
108 | |||
109 | 113 | static lbm_value ext_round(lbm_value *args, lbm_uint argn) { | |
110 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 56 times.
|
113 | LBM_CHECK_ARGN_NUMBER(1); |
111 | 56 | return lbm_enc_float(roundf(lbm_dec_as_float(args[0]))); | |
112 | } | ||
113 | |||
114 | 337 | static lbm_value ext_deg2rad(lbm_value *args, lbm_uint argn) { | |
115 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 336 times.
|
337 | LBM_CHECK_NUMBER_ALL(); |
116 | |||
117 |
2/2✓ Branch 0 taken 224 times.
✓ Branch 1 taken 112 times.
|
336 | if (argn == 1) { |
118 | 224 | return lbm_enc_float(DEG2RAD_f(lbm_dec_as_float(args[0]))); | |
119 | } else { | ||
120 | 112 | lbm_value out_list = ENC_SYM_NIL; | |
121 |
2/2✓ Branch 0 taken 168 times.
✓ Branch 1 taken 112 times.
|
280 | for (int i = (int)(argn - 1);i >= 0;i--) { |
122 | 168 | out_list = lbm_cons(lbm_enc_float(DEG2RAD_f(lbm_dec_as_float(args[i]))), out_list); | |
123 | } | ||
124 | 112 | return out_list; | |
125 | } | ||
126 | } | ||
127 | |||
128 | 169 | static lbm_value ext_rad2deg(lbm_value *args, lbm_uint argn) { | |
129 |
2/2✓ Branch 0 taken 1 times.
✓ Branch 1 taken 168 times.
|
169 | LBM_CHECK_NUMBER_ALL(); |
130 | |||
131 |
2/2✓ Branch 0 taken 56 times.
✓ Branch 1 taken 112 times.
|
168 | if (argn == 1) { |
132 | 56 | return lbm_enc_float(RAD2DEG_f(lbm_dec_as_float(args[0]))); | |
133 | } else { | ||
134 | 112 | lbm_value out_list = ENC_SYM_NIL; | |
135 |
2/2✓ Branch 0 taken 168 times.
✓ Branch 1 taken 112 times.
|
280 | for (int i = (int)(argn - 1);i >= 0;i--) { |
136 | 168 | out_list = lbm_cons(lbm_enc_float(RAD2DEG_f(lbm_dec_as_float(args[i]))), out_list); | |
137 | } | ||
138 | 112 | return out_list; | |
139 | } | ||
140 | } | ||
141 | |||
142 | |||
143 | 339 | static lbm_value ext_is_nan(lbm_value *args, lbm_uint argn) { | |
144 | 339 | lbm_value res = ENC_SYM_EERROR; | |
145 |
2/2✓ Branch 0 taken 283 times.
✓ Branch 1 taken 56 times.
|
339 | if (argn == 1) { |
146 | 283 | res = ENC_SYM_TERROR; | |
147 |
2/2✓ Branch 0 taken 282 times.
✓ Branch 1 taken 1 times.
|
283 | if (lbm_is_number(args[0])) { |
148 | 282 | lbm_uint t = lbm_type_of(args[0]); | |
149 |
3/3✓ Branch 0 taken 56 times.
✓ Branch 1 taken 170 times.
✓ Branch 2 taken 56 times.
|
282 | switch(t) { |
150 | 56 | case LBM_TYPE_DOUBLE: | |
151 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
|
56 | if (isnan(lbm_dec_double(args[0]))) { |
152 | ✗ | res = ENC_SYM_TRUE; | |
153 | } else { | ||
154 | 56 | res = ENC_SYM_NIL; | |
155 | } | ||
156 | 56 | break; | |
157 | 170 | case LBM_TYPE_FLOAT: | |
158 |
2/2✓ Branch 0 taken 57 times.
✓ Branch 1 taken 113 times.
|
170 | if (isnanf(lbm_dec_float(args[0]))) { |
159 | 57 | res = ENC_SYM_TRUE; | |
160 | } else { | ||
161 | 113 | res = ENC_SYM_NIL; | |
162 | } | ||
163 | 170 | break; | |
164 | 56 | default: | |
165 | 56 | res = ENC_SYM_NIL; | |
166 | 56 | break; | |
167 | } | ||
168 | } | ||
169 | } | ||
170 | 339 | return res; | |
171 | } | ||
172 | |||
173 | 225 | static lbm_value ext_is_inf(lbm_value *args, lbm_uint argn) { | |
174 | 225 | lbm_value res = ENC_SYM_EERROR; | |
175 |
2/2✓ Branch 0 taken 169 times.
✓ Branch 1 taken 56 times.
|
225 | if (argn == 1) { |
176 | 169 | res = ENC_SYM_TERROR; | |
177 |
2/2✓ Branch 0 taken 168 times.
✓ Branch 1 taken 1 times.
|
169 | if (lbm_is_number(args[0])) { |
178 | 168 | lbm_uint t = lbm_type_of(args[0]); | |
179 |
3/3✓ Branch 0 taken 56 times.
✓ Branch 1 taken 56 times.
✓ Branch 2 taken 56 times.
|
168 | switch(t) { |
180 | 56 | case LBM_TYPE_DOUBLE: | |
181 |
2/4✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 56 times.
|
56 | if (isinf(lbm_dec_double(args[0]))) { |
182 | ✗ | res = ENC_SYM_TRUE; | |
183 | } else { | ||
184 | 56 | res = ENC_SYM_NIL; | |
185 | } | ||
186 | 56 | break; | |
187 | 56 | case LBM_TYPE_FLOAT: | |
188 |
2/4✗ Branch 0 not taken.
✓ Branch 1 taken 56 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 56 times.
|
56 | if (isinff(lbm_dec_float(args[0]))) { |
189 | ✗ | res = ENC_SYM_TRUE; | |
190 | } else { | ||
191 | 56 | res = ENC_SYM_NIL; | |
192 | } | ||
193 | 56 | break; | |
194 | 56 | default: | |
195 | 56 | res = ENC_SYM_NIL; | |
196 | 56 | break; | |
197 | } | ||
198 | } | ||
199 | } | ||
200 | 225 | return res; | |
201 | } | ||
202 | |||
203 | |||
204 | 44260 | void lbm_math_extensions_init(void) { | |
205 | |||
206 | 44260 | lbm_add_extension("sin", ext_sin); | |
207 | 44260 | lbm_add_extension("cos", ext_cos); | |
208 | 44260 | lbm_add_extension("tan", ext_tan); | |
209 | 44260 | lbm_add_extension("asin", ext_asin); | |
210 | 44260 | lbm_add_extension("acos", ext_acos); | |
211 | 44260 | lbm_add_extension("atan", ext_atan); | |
212 | 44260 | lbm_add_extension("atan2", ext_atan2); | |
213 | 44260 | lbm_add_extension("pow", ext_pow); | |
214 | 44260 | lbm_add_extension("exp", ext_exp); | |
215 | 44260 | lbm_add_extension("sqrt", ext_sqrt); | |
216 | 44260 | lbm_add_extension("log", ext_log); | |
217 | 44260 | lbm_add_extension("log10", ext_log10); | |
218 | 44260 | lbm_add_extension("floor", ext_floor); | |
219 | 44260 | lbm_add_extension("ceil", ext_ceil); | |
220 | 44260 | lbm_add_extension("round", ext_round); | |
221 | 44260 | lbm_add_extension("deg2rad", ext_deg2rad); | |
222 | 44260 | lbm_add_extension("rad2deg", ext_rad2deg); | |
223 | 44260 | lbm_add_extension("is-nan", ext_is_nan); | |
224 | 44260 | lbm_add_extension("is-inf", ext_is_inf); | |
225 | 44260 | } | |
226 |