blob: cbb0da6f48baf3215e8b3caa50986e18c5cd6b01 [file] [log] [blame]
Gavin Howard3eb626f2018-02-14 13:54:35 -07001/*
Gavin Howardb5904bf2018-02-20 13:28:18 -07002 * *****************************************************************************
Gavin Howard3eb626f2018-02-14 13:54:35 -07003 *
Gavin Howardb5904bf2018-02-20 13:28:18 -07004 * Copyright 2018 Gavin D. Howard
Gavin Howard3eb626f2018-02-14 13:54:35 -07005 *
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
10 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
11 * AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
12 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
13 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
14 * OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
15 * PERFORMANCE OF THIS SOFTWARE.
16 *
Gavin Howardb5904bf2018-02-20 13:28:18 -070017 * *****************************************************************************
Gavin Howard3eb626f2018-02-14 13:54:35 -070018 *
19 * Code for the number type.
20 *
21 */
22
23#include <assert.h>
24#include <stdbool.h>
25#include <string.h>
26
Gavin Howard8e2cc692018-02-15 17:39:14 -070027#include <math.h>
28
29#include <limits.h>
30
Gavin Howard29493062018-03-20 19:57:37 -060031#include <status.h>
Gavin Howard3ba6c8d2018-02-15 12:23:35 -070032#include <num.h>
Gavin Howardf507f232018-02-23 21:10:24 -070033#include <vector.h>
Gavin Howardd5551672018-09-22 19:52:42 -060034#include <vm.h>
Gavin Howard3eb626f2018-02-14 13:54:35 -070035
Gavin Howard0be26ed2018-08-31 20:21:56 -060036void bc_num_zero(BcNum *n) {
Gavin Howard63738202018-09-26 15:34:20 -060037 if (!n) return;
38 memset(n->num, 0, n->cap * sizeof(char));
39 n->neg = false;
40 n->len = 0;
41 n->rdx = 0;
Gavin Howard0be26ed2018-08-31 20:21:56 -060042}
43
44void bc_num_one(BcNum *n) {
Gavin Howard63738202018-09-26 15:34:20 -060045 if (!n) return;
46 bc_num_zero(n);
47 n->len = 1;
48 n->num[0] = 1;
Gavin Howard0be26ed2018-08-31 20:21:56 -060049}
50
51void bc_num_ten(BcNum *n) {
Gavin Howard63738202018-09-26 15:34:20 -060052 if (!n) return;
53 bc_num_zero(n);
54 n->len = 2;
55 n->num[0] = 0;
56 n->num[1] = 1;
Gavin Howard0be26ed2018-08-31 20:21:56 -060057}
58
Gavin Howard2ea7dc42018-05-22 14:02:02 -060059BcStatus bc_num_subArrays(BcDigit *n1, BcDigit *n2, size_t len) {
Gavin Howard63738202018-09-26 15:34:20 -060060 size_t i, j;
61 for (i = 0; !bcg.signe && i < len; ++i) {
62 for (n1[i] -= n2[i], j = 0; !bcg.signe && n1[i + j] < 0;) {
63 n1[i + j++] += 10;
64 n1[i + j] -= 1;
65 }
66 }
67 return bcg.signe ? BC_STATUS_EXEC_SIGNAL : BC_STATUS_SUCCESS;
Gavin Howarde1e74942018-03-20 15:51:52 -060068}
69
Gavin Howard6a804cf2018-05-17 16:54:12 -060070ssize_t bc_num_compare(BcDigit *n1, BcDigit *n2, size_t len) {
Gavin Howard63738202018-09-26 15:34:20 -060071 size_t i;
72 BcDigit c = 0;
73 for (i = len - 1; !bcg.signe && i < len && !(c = n1[i] - n2[i]); --i);
74 return (c < 0 ? -1 : 1) * (ssize_t) (i + 1);
Gavin Howard08bf5292018-03-20 14:59:33 -060075}
76
Gavin Howard6a804cf2018-05-17 16:54:12 -060077ssize_t bc_num_cmp(BcNum *a, BcNum *b) {
Gavin Howard4681d1b2018-03-05 19:49:33 -070078
Gavin Howard63738202018-09-26 15:34:20 -060079 size_t i, min, a_int, b_int, diff;
80 BcDigit *max_num, *min_num;
81 bool a_max;
82 ssize_t cmp, neg = 1;
Gavin Howard4681d1b2018-03-05 19:49:33 -070083
Gavin Howard63738202018-09-26 15:34:20 -060084 if (!a) return !b ? 0 : !b->neg * -2 + 1;
85 else if (!b) return a->neg * -2 + 1;
Gavin Howard4681d1b2018-03-05 19:49:33 -070086
Gavin Howard63738202018-09-26 15:34:20 -060087 if (!a->len) return (!b->neg * -2 + 1) * !!b->len;
88 else if (!b->len) return a->neg * -2 + 1;
Gavin Howard2444f272018-09-11 12:25:53 -060089
Gavin Howard63738202018-09-26 15:34:20 -060090 if (a->neg) {
91 if (b->neg) neg = -1;
92 else return -1;
93 }
94 else if (b->neg) return 1;
Gavin Howard4681d1b2018-03-05 19:49:33 -070095
Gavin Howard63738202018-09-26 15:34:20 -060096 a_int = BC_NUM_INT(a);
97 b_int = BC_NUM_INT(b);
98 a_int -= b_int;
Gavin Howard4681d1b2018-03-05 19:49:33 -070099
Gavin Howard63738202018-09-26 15:34:20 -0600100 if (a_int) return (ssize_t) a_int;
Gavin Howard4681d1b2018-03-05 19:49:33 -0700101
Gavin Howard63738202018-09-26 15:34:20 -0600102 a_max = a->rdx > b->rdx;
Gavin Howard4681d1b2018-03-05 19:49:33 -0700103
Gavin Howard63738202018-09-26 15:34:20 -0600104 if (a_max) {
105 min = b->rdx;
106 diff = a->rdx - b->rdx;
107 max_num = a->num + diff;
108 min_num = b->num;
109 }
110 else {
111 min = a->rdx;
112 diff = b->rdx - a->rdx;
113 max_num = b->num + diff;
114 min_num = a->num;
115 }
Gavin Howard4681d1b2018-03-05 19:49:33 -0700116
Gavin Howard63738202018-09-26 15:34:20 -0600117 cmp = bc_num_compare(max_num, min_num, b_int + min);
118 if (cmp) return cmp * (!a_max * -2 + 1) * neg;
Gavin Howard021150b2018-03-10 15:40:42 -0700119
Gavin Howard63738202018-09-26 15:34:20 -0600120 for (max_num -= diff, i = diff - 1; !bcg.signe && i < diff; --i) {
121 if (max_num[i]) return neg * (!a_max * -2 + 1);
122 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700123
Gavin Howard63738202018-09-26 15:34:20 -0600124 return 0;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700125}
126
Gavin Howardd9734e52018-03-29 15:54:41 -0600127void bc_num_truncate(BcNum *n, size_t places) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700128
Gavin Howard63738202018-09-26 15:34:20 -0600129 BcDigit *ptr;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700130
Gavin Howard63738202018-09-26 15:34:20 -0600131 assert(places <= n->rdx);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700132
Gavin Howard63738202018-09-26 15:34:20 -0600133 if (!places) return;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700134
Gavin Howard63738202018-09-26 15:34:20 -0600135 ptr = n->num + places;
136 n->len -= places;
137 n->rdx -= places;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700138
Gavin Howard63738202018-09-26 15:34:20 -0600139 memmove(n->num, ptr, n->len * sizeof(BcDigit));
140 memset(n->num + n->len, 0, sizeof(BcDigit) * (n->cap - n->len));
Gavin Howard6fbdb292018-02-27 15:44:48 -0700141}
142
Gavin Howard3f68df72018-03-22 20:30:27 -0600143BcStatus bc_num_extend(BcNum *n, size_t places) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700144
Gavin Howard63738202018-09-26 15:34:20 -0600145 BcStatus s;
146 BcDigit *ptr;
147 size_t len;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700148
Gavin Howard63738202018-09-26 15:34:20 -0600149 if (!places) return BC_STATUS_SUCCESS;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700150
Gavin Howard63738202018-09-26 15:34:20 -0600151 len = n->len + places;
152 if (n->cap < len && (s = bc_num_expand(n, len))) return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700153
Gavin Howard63738202018-09-26 15:34:20 -0600154 ptr = n->num + places;
155 memmove(ptr, n->num, sizeof(BcDigit) * n->len);
156 memset(n->num, 0, sizeof(BcDigit) * places);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700157
Gavin Howard63738202018-09-26 15:34:20 -0600158 n->len += places;
159 n->rdx += places;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700160
Gavin Howard63738202018-09-26 15:34:20 -0600161 return BC_STATUS_SUCCESS;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700162}
163
Gavin Howardc0f65d72018-09-01 16:59:04 -0600164void bc_num_clean(BcNum *n) {
Gavin Howard63738202018-09-26 15:34:20 -0600165 while (n->len > 0 && !n->num[n->len - 1]) --n->len;
166 if (n->len == 0) n->rdx = n->neg = 0;
167 else if (n->len < n->rdx) n->len = n->rdx;
Gavin Howarde57525d2018-09-01 16:56:21 -0600168}
169
Gavin Howard337fe602018-09-01 15:10:59 -0600170BcStatus bc_num_retireMul(BcNum *n, size_t scale) {
171
Gavin Howard63738202018-09-26 15:34:20 -0600172 BcStatus s = BC_STATUS_SUCCESS;
Gavin Howard337fe602018-09-01 15:10:59 -0600173
Gavin Howard63738202018-09-26 15:34:20 -0600174 if (n->rdx < scale) s = bc_num_extend(n, scale - n->rdx);
175 else bc_num_truncate(n, n->rdx - scale);
Gavin Howard337fe602018-09-01 15:10:59 -0600176
Gavin Howard63738202018-09-26 15:34:20 -0600177 bc_num_clean(n);
Gavin Howard337fe602018-09-01 15:10:59 -0600178
Gavin Howard63738202018-09-26 15:34:20 -0600179 return s;
Gavin Howard337fe602018-09-01 15:10:59 -0600180}
181
Gavin Howard9c4358c2018-03-22 20:11:28 -0600182BcStatus bc_num_inv(BcNum *a, BcNum *b, size_t scale) {
183
Gavin Howard63738202018-09-26 15:34:20 -0600184 BcStatus s;
185 BcNum one;
Gavin Howard9c4358c2018-03-22 20:11:28 -0600186
Gavin Howard63738202018-09-26 15:34:20 -0600187 if ((s = bc_num_init(&one, BC_NUM_DEF_SIZE))) return s;
Gavin Howard9c4358c2018-03-22 20:11:28 -0600188
Gavin Howard63738202018-09-26 15:34:20 -0600189 bc_num_one(&one);
190 s = bc_num_div(&one, a, b, scale);
191 bc_num_free(&one);
Gavin Howard9c4358c2018-03-22 20:11:28 -0600192
Gavin Howard63738202018-09-26 15:34:20 -0600193 return s;
Gavin Howard9c4358c2018-03-22 20:11:28 -0600194}
195
Gavin Howard24434e12018-09-06 15:10:32 -0600196BcStatus bc_num_alg_a(BcNum *a, BcNum *b, BcNum *c, size_t sub) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700197
Gavin Howard63738202018-09-26 15:34:20 -0600198 BcDigit *ptr, *ptr_a, *ptr_b, *ptr_c;
199 size_t i, max, min_rdx, min_int, diff, a_int, b_int;
200 BcDigit carry;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700201
Gavin Howard63738202018-09-26 15:34:20 -0600202 // Because this function doesn't need to use scale (per the bc spec),
203 // I am hijacking it to say whether it's doing an add or a subtract.
Gavin Howard73cce1a2018-09-06 15:16:09 -0600204
Gavin Howard63738202018-09-26 15:34:20 -0600205 if (!a->len) {
206 BcStatus s = bc_num_copy(c, b);
207 if (sub) c->neg = !c->neg;
208 return s;
209 }
210 else if (!b->len) return bc_num_copy(c, a);
Gavin Howardf6964a12018-03-14 10:52:31 -0600211
Gavin Howard63738202018-09-26 15:34:20 -0600212 c->neg = a->neg;
213 memset(c->num, 0, c->cap * sizeof(BcDigit));
Gavin Howard6fbdb292018-02-27 15:44:48 -0700214
Gavin Howard63738202018-09-26 15:34:20 -0600215 c->rdx = BC_MAX(a->rdx, b->rdx);
216 min_rdx = BC_MIN(a->rdx, b->rdx);
217 c->len = 0;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700218
Gavin Howard63738202018-09-26 15:34:20 -0600219 if (a->rdx > b->rdx) {
220 diff = a->rdx - b->rdx;
221 ptr = a->num;
222 ptr_a = a->num + diff;
223 ptr_b = b->num;
224 }
225 else {
226 diff = b->rdx - a->rdx;
227 ptr = b->num;
228 ptr_a = a->num;
229 ptr_b = b->num + diff;
230 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700231
Gavin Howard63738202018-09-26 15:34:20 -0600232 for (ptr_c = c->num, i = 0; i < diff; ++i, ++c->len) ptr_c[i] = ptr[i];
Gavin Howard6fbdb292018-02-27 15:44:48 -0700233
Gavin Howard63738202018-09-26 15:34:20 -0600234 ptr_c += diff;
235 a_int = BC_NUM_INT(a);
236 b_int = BC_NUM_INT(b);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700237
Gavin Howard63738202018-09-26 15:34:20 -0600238 if (a_int > b_int) {
239 min_int = b_int;
240 max = a_int;
241 ptr = ptr_a;
242 }
243 else {
244 min_int = a_int;
245 max = b_int;
246 ptr = ptr_b;
247 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700248
Gavin Howard63738202018-09-26 15:34:20 -0600249 for (carry = 0, i = 0; !bcg.signe && i < min_rdx + min_int; ++i, ++c->len) {
250 ptr_c[i] = ptr_a[i] + ptr_b[i] + carry;
251 carry = ptr_c[i] / 10;
252 ptr_c[i] %= 10;
253 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700254
Gavin Howard63738202018-09-26 15:34:20 -0600255 for (; !bcg.signe && i < max + min_rdx; ++i, ++c->len) {
256 ptr_c[i] += ptr[i] + carry;
257 carry = ptr_c[i] / 10;
258 ptr_c[i] %= 10;
259 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700260
Gavin Howard63738202018-09-26 15:34:20 -0600261 if (bcg.signe) return BC_STATUS_EXEC_SIGNAL;
Gavin Howard2ea7dc42018-05-22 14:02:02 -0600262
Gavin Howard63738202018-09-26 15:34:20 -0600263 if (carry) c->num[c->len++] = carry;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700264
Gavin Howard63738202018-09-26 15:34:20 -0600265 return BC_STATUS_SUCCESS;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700266}
267
Gavin Howard3f68df72018-03-22 20:30:27 -0600268BcStatus bc_num_alg_s(BcNum *a, BcNum *b, BcNum *c, size_t sub) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700269
Gavin Howard63738202018-09-26 15:34:20 -0600270 BcStatus s;
271 ssize_t cmp;
272 BcNum *minuend, *subtrahend;
273 size_t start;
274 bool aneg, bneg, neg;
Gavin Howarda1c090a2018-03-05 14:20:33 -0700275
Gavin Howard63738202018-09-26 15:34:20 -0600276 // Because this function doesn't need to use scale (per the bc spec),
277 // I am hijacking it to say whether it's doing an add or a subtract.
Gavin Howard6fbdb292018-02-27 15:44:48 -0700278
Gavin Howard63738202018-09-26 15:34:20 -0600279 if (!a->len) {
280 s = bc_num_copy(c, b);
281 if (sub) c->neg = !b->neg;
282 return s;
283 }
284 else if (!b->len) return bc_num_copy(c, a);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700285
Gavin Howard63738202018-09-26 15:34:20 -0600286 aneg = a->neg;
287 bneg = b->neg;
288 a->neg = b->neg = false;
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700289
Gavin Howard63738202018-09-26 15:34:20 -0600290 cmp = bc_num_cmp(a, b);
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700291
Gavin Howard63738202018-09-26 15:34:20 -0600292 a->neg = aneg;
293 b->neg = bneg;
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700294
Gavin Howard63738202018-09-26 15:34:20 -0600295 if (!cmp) {
296 bc_num_zero(c);
297 return BC_STATUS_SUCCESS;
298 }
299 else if (cmp > 0) {
300 neg = a->neg;
301 minuend = a;
302 subtrahend = b;
303 }
304 else {
305 neg = b->neg;
306 if (sub) neg = !neg;
307 minuend = b;
308 subtrahend = a;
309 }
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700310
Gavin Howard63738202018-09-26 15:34:20 -0600311 if ((s = bc_num_copy(c, minuend))) return s;
312 c->neg = neg;
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700313
Gavin Howard63738202018-09-26 15:34:20 -0600314 if (c->rdx < subtrahend->rdx) {
315 if ((s = bc_num_extend(c, subtrahend->rdx - c->rdx))) return s;
316 start = 0;
317 }
318 else start = c->rdx - subtrahend->rdx;
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700319
Gavin Howard63738202018-09-26 15:34:20 -0600320 s = bc_num_subArrays(c->num + start, subtrahend->num, subtrahend->len);
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700321
Gavin Howard63738202018-09-26 15:34:20 -0600322 bc_num_clean(c);
Gavin Howard9a6f7a42018-03-05 14:07:06 -0700323
Gavin Howard63738202018-09-26 15:34:20 -0600324 return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700325}
326
Gavin Howard3f68df72018-03-22 20:30:27 -0600327BcStatus bc_num_alg_m(BcNum *a, BcNum *b, BcNum *c, size_t scale) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700328
Gavin Howard63738202018-09-26 15:34:20 -0600329 BcStatus s;
330 BcDigit carry;
331 size_t i, j, len;
332 bool aone = BC_NUM_ONE(a);
Gavin Howard337fe602018-09-01 15:10:59 -0600333
Gavin Howard63738202018-09-26 15:34:20 -0600334 c->rdx = a->rdx + b->rdx;
335 scale = BC_MAX(scale, a->rdx);
336 scale = BC_MAX(scale, b->rdx);
337 scale = BC_MIN(scale, c->rdx);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700338
Gavin Howard63738202018-09-26 15:34:20 -0600339 if (!a->len || !b->len) {
340 bc_num_zero(c);
341 return BC_STATUS_SUCCESS;
342 }
343 else if (aone || BC_NUM_ONE(b)) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700344
Gavin Howard63738202018-09-26 15:34:20 -0600345 if (aone) {
346 s = bc_num_copy(c, b);
347 if (a->neg) c->neg = !c->neg;
348 }
349 else {
350 s = bc_num_copy(c, a);
351 if (b->neg) c->neg = !c->neg;
352 }
Gavin Howard337fe602018-09-01 15:10:59 -0600353
Gavin Howard63738202018-09-26 15:34:20 -0600354 if (s) return s;
Gavin Howardbe261e72018-09-05 10:11:41 -0600355
Gavin Howard63738202018-09-26 15:34:20 -0600356 return bc_num_retireMul(c, scale);
357 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700358
Gavin Howard63738202018-09-26 15:34:20 -0600359 memset(c->num, 0, sizeof(BcDigit) * c->cap);
360 c->len = carry = len = 0;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700361
Gavin Howard63738202018-09-26 15:34:20 -0600362 for (i = 0; !bcg.signe && i < b->len; ++i) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700363
Gavin Howard63738202018-09-26 15:34:20 -0600364 for (j = 0; !bcg.signe && j < a->len; ++j) {
365 c->num[i + j] += a->num[j] * b->num[i] + carry;
366 carry = c->num[i + j] / 10;
367 c->num[i + j] %= 10;
368 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700369
Gavin Howard63738202018-09-26 15:34:20 -0600370 if (bcg.signe) return BC_STATUS_EXEC_SIGNAL;
Gavin Howard0dfe2922018-05-22 13:57:02 -0600371
Gavin Howard63738202018-09-26 15:34:20 -0600372 if (carry) {
373 c->num[i + j] += carry;
374 carry = 0;
375 len = BC_MAX(len, i + j + 1);
376 }
377 else len = BC_MAX(len, i + j);
378 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700379
Gavin Howard63738202018-09-26 15:34:20 -0600380 if (bcg.signe) return BC_STATUS_EXEC_SIGNAL;
Gavin Howard0dfe2922018-05-22 13:57:02 -0600381
Gavin Howard63738202018-09-26 15:34:20 -0600382 c->len = BC_MAX(len, c->rdx);
383 c->neg = !a->neg != !b->neg;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700384
Gavin Howard63738202018-09-26 15:34:20 -0600385 return bc_num_retireMul(c, scale);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700386}
387
Gavin Howard3f68df72018-03-22 20:30:27 -0600388BcStatus bc_num_alg_d(BcNum *a, BcNum *b, BcNum *c, size_t scale) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700389
Gavin Howard63738202018-09-26 15:34:20 -0600390 BcStatus s;
391 BcDigit *n, *bptr, q;
392 size_t len, end, i;
393 BcNum cp;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700394
Gavin Howard63738202018-09-26 15:34:20 -0600395 if (!b->len) return BC_STATUS_MATH_DIVIDE_BY_ZERO;
396 else if (!a->len) {
397 bc_num_zero(c);
398 return BC_STATUS_SUCCESS;
399 }
400 else if (BC_NUM_ONE(b)) {
Gavin Howardae6e7d52018-03-14 10:13:43 -0600401
Gavin Howard63738202018-09-26 15:34:20 -0600402 if ((s = bc_num_copy(c, a))) return s;
403 if (b->neg) c->neg = !c->neg;
Gavin Howardae6e7d52018-03-14 10:13:43 -0600404
Gavin Howard63738202018-09-26 15:34:20 -0600405 return bc_num_retireMul(c, scale);
406 }
Gavin Howardb651f1a2018-02-28 17:34:18 -0700407
Gavin Howard63738202018-09-26 15:34:20 -0600408 if ((s = bc_num_init(&cp, BC_NUM_MREQ(a, b, scale)))) return s;
409 if ((s = bc_num_copy(&cp, a))) goto err;
Gavin Howardb651f1a2018-02-28 17:34:18 -0700410
Gavin Howard63738202018-09-26 15:34:20 -0600411 if ((len = b->len) > cp.len) {
412 if ((s = bc_num_expand(&cp, len + 2))) goto err;
413 if ((s = bc_num_extend(&cp, len - cp.len))) goto err;
414 }
Gavin Howard021150b2018-03-10 15:40:42 -0700415
Gavin Howard63738202018-09-26 15:34:20 -0600416 if (b->rdx > cp.rdx && (s = bc_num_extend(&cp, b->rdx - cp.rdx)))
417 goto err;
Gavin Howardb651f1a2018-02-28 17:34:18 -0700418
Gavin Howard63738202018-09-26 15:34:20 -0600419 cp.rdx -= b->rdx;
Gavin Howardb651f1a2018-02-28 17:34:18 -0700420
Gavin Howard63738202018-09-26 15:34:20 -0600421 if (scale > cp.rdx && (s = bc_num_extend(&cp, scale - cp.rdx)))
422 goto err;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700423
Gavin Howard63738202018-09-26 15:34:20 -0600424 if (b->rdx == b->len) {
Gavin Howard6dbf6bf2018-09-01 15:14:48 -0600425
Gavin Howard63738202018-09-26 15:34:20 -0600426 bool zero = true;
427 for (i = 0; zero && i < len; ++i) zero = !b->num[len - i - 1];
Gavin Howard6dbf6bf2018-09-01 15:14:48 -0600428
Gavin Howard63738202018-09-26 15:34:20 -0600429 if (i == len && zero) {
430 s = BC_STATUS_MATH_DIVIDE_BY_ZERO;
431 goto err;
432 }
Gavin Howard6dbf6bf2018-09-01 15:14:48 -0600433
Gavin Howard63738202018-09-26 15:34:20 -0600434 len -= i - 1;
435 }
Gavin Howardc1a6a342018-03-05 12:10:14 -0700436
Gavin Howard63738202018-09-26 15:34:20 -0600437 if (cp.cap == cp.len && (s = bc_num_expand(&cp, cp.len + 1))) goto err;
Gavin Howardb651f1a2018-02-28 17:34:18 -0700438
Gavin Howard63738202018-09-26 15:34:20 -0600439 // We want an extra zero in front to make things simpler.
440 cp.num[cp.len++] = 0;
441 end = cp.len - len;
Gavin Howardac7656d2018-03-01 17:18:40 -0700442
Gavin Howard63738202018-09-26 15:34:20 -0600443 if ((s = bc_num_expand(c, cp.len))) goto err;
Gavin Howardac7656d2018-03-01 17:18:40 -0700444
Gavin Howard63738202018-09-26 15:34:20 -0600445 bc_num_zero(c);
446 c->rdx = cp.rdx;
447 c->len = cp.len;
448 bptr = b->num;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700449
Gavin Howard63738202018-09-26 15:34:20 -0600450 for (i = end - 1; !bcg.signe && i < end; --i) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700451
Gavin Howard63738202018-09-26 15:34:20 -0600452 n = cp.num + i;
Gavin Howardb651f1a2018-02-28 17:34:18 -0700453
Gavin Howard63738202018-09-26 15:34:20 -0600454 for (q = 0; (!s && n[len]) || bc_num_compare(n, bptr, len) >= 0; ++q)
455 s = bc_num_subArrays(n, bptr, len);
Gavin Howardb651f1a2018-02-28 17:34:18 -0700456
Gavin Howard63738202018-09-26 15:34:20 -0600457 c->num[i] = q;
458 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700459
Gavin Howard63738202018-09-26 15:34:20 -0600460 if (s) goto err;
Gavin Howard0dfe2922018-05-22 13:57:02 -0600461
Gavin Howard63738202018-09-26 15:34:20 -0600462 c->neg = !a->neg != !b->neg;
463 s = bc_num_retireMul(c, scale);
Gavin Howard5b24c3f2018-03-13 23:57:15 -0600464
Gavin Howardb651f1a2018-02-28 17:34:18 -0700465err:
Gavin Howard63738202018-09-26 15:34:20 -0600466 bc_num_free(&cp);
467 return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700468}
469
Gavin Howard3f68df72018-03-22 20:30:27 -0600470BcStatus bc_num_alg_mod(BcNum *a, BcNum *b, BcNum *c, size_t scale) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700471
Gavin Howard63738202018-09-26 15:34:20 -0600472 BcStatus s;
473 BcNum c1, c2;
474 size_t len, tscale = BC_MAX(scale + b->rdx, a->rdx);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700475
Gavin Howard63738202018-09-26 15:34:20 -0600476 if (!b->len) return BC_STATUS_MATH_DIVIDE_BY_ZERO;
Gavin Howard8b254872018-03-14 01:13:35 -0600477
Gavin Howard63738202018-09-26 15:34:20 -0600478 if (!a->len) {
479 bc_num_zero(c);
480 return BC_STATUS_SUCCESS;
481 }
Gavin Howard8b254872018-03-14 01:13:35 -0600482
Gavin Howard63738202018-09-26 15:34:20 -0600483 if (scale) scale = tscale;
484 len = BC_NUM_MREQ(a, b, scale);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700485
Gavin Howard63738202018-09-26 15:34:20 -0600486 if ((s = bc_num_init(&c1, len))) return s;
487 if ((s = bc_num_init(&c2, len))) goto c2_err;
488 if ((s = bc_num_div(a, b, &c1, scale))) goto err;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700489
Gavin Howard63738202018-09-26 15:34:20 -0600490 if ((s = bc_num_mul(&c1, b, &c2, scale))) goto err;
491 if ((s = bc_num_sub(a, &c2, c, scale))) goto err;
Gavin Howard5d149cf2018-09-06 13:46:23 -0600492
Gavin Howard63738202018-09-26 15:34:20 -0600493 if (tscale > c->rdx && c->len) s = bc_num_extend(c, tscale - c->rdx);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700494
495err:
Gavin Howard63738202018-09-26 15:34:20 -0600496 bc_num_free(&c2);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700497c2_err:
Gavin Howard63738202018-09-26 15:34:20 -0600498 bc_num_free(&c1);
499 return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700500}
501
Gavin Howard3f68df72018-03-22 20:30:27 -0600502BcStatus bc_num_alg_p(BcNum *a, BcNum *b, BcNum *c, size_t scale) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700503
Gavin Howard63738202018-09-26 15:34:20 -0600504 BcStatus s;
505 BcNum copy;
506 unsigned long pow;
507 size_t i, powrdx, resrdx;
508 bool neg, zero;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700509
Gavin Howard63738202018-09-26 15:34:20 -0600510 if (b->rdx) return BC_STATUS_MATH_NON_INTEGER;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700511
Gavin Howard63738202018-09-26 15:34:20 -0600512 if (!b->len) {
513 bc_num_one(c);
514 return BC_STATUS_SUCCESS;
515 }
516 else if (!a->len) {
517 bc_num_zero(c);
518 return BC_STATUS_SUCCESS;
519 }
520 else if (BC_NUM_ONE(b)) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700521
Gavin Howard63738202018-09-26 15:34:20 -0600522 if (!b->neg) s = bc_num_copy(c, a);
523 else s = bc_num_inv(a, c, scale);
Gavin Howard5fc40e72018-03-05 12:11:09 -0700524
Gavin Howard63738202018-09-26 15:34:20 -0600525 return s;
526 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700527
Gavin Howard63738202018-09-26 15:34:20 -0600528 neg = b->neg;
529 b->neg = false;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700530
Gavin Howard63738202018-09-26 15:34:20 -0600531 if ((s = bc_num_ulong(b, &pow))) return s;
532 if ((s = bc_num_init(&copy, a->len))) return s;
533 if ((s = bc_num_copy(&copy, a))) goto err;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700534
Gavin Howard63738202018-09-26 15:34:20 -0600535 if (!neg) scale = BC_MIN(a->rdx * pow, BC_MAX(scale, a->rdx));
Gavin Howardb29674f2018-03-22 22:24:58 -0600536
Gavin Howard63738202018-09-26 15:34:20 -0600537 b->neg = neg;
Gavin Howardb29674f2018-03-22 22:24:58 -0600538
Gavin Howard63738202018-09-26 15:34:20 -0600539 for (powrdx = a->rdx; !bcg.signe && !(pow & 1); pow >>= 1) {
540 powrdx <<= 1;
541 if ((s = bc_num_mul(&copy, &copy, &copy, powrdx))) goto err;
542 }
Gavin Howard6fb635f2018-03-03 12:45:42 -0700543
Gavin Howard63738202018-09-26 15:34:20 -0600544 if ((s = bc_num_copy(c, &copy))) goto err;
545 if (bcg.signe) {
546 s = BC_STATUS_EXEC_SIGNAL;
547 goto err;
548 }
Gavin Howard6fb635f2018-03-03 12:45:42 -0700549
Gavin Howard63738202018-09-26 15:34:20 -0600550 resrdx = powrdx;
Gavin Howard6fb635f2018-03-03 12:45:42 -0700551
Gavin Howard63738202018-09-26 15:34:20 -0600552 for (pow >>= 1; !bcg.signe && pow != 0; pow >>= 1) {
Gavin Howard1d959152018-03-03 23:33:13 -0700553
Gavin Howard63738202018-09-26 15:34:20 -0600554 powrdx <<= 1;
Gavin Howard1d959152018-03-03 23:33:13 -0700555
Gavin Howard63738202018-09-26 15:34:20 -0600556 if ((s = bc_num_mul(&copy, &copy, &copy, powrdx))) goto err;
Gavin Howard1d959152018-03-03 23:33:13 -0700557
Gavin Howard63738202018-09-26 15:34:20 -0600558 if (pow & 1) {
559 resrdx += powrdx;
560 if ((s = bc_num_mul(c, &copy, c, resrdx))) goto err;
561 }
562 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700563
Gavin Howard63738202018-09-26 15:34:20 -0600564 if (neg && (s = bc_num_inv(c, c, scale))) goto err;
565 if (bcg.signe) {
566 s = BC_STATUS_EXEC_SIGNAL;
567 goto err;
568 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700569
Gavin Howard63738202018-09-26 15:34:20 -0600570 if (c->rdx > scale) bc_num_truncate(c, c->rdx - scale);
Gavin Howard1d959152018-03-03 23:33:13 -0700571
Gavin Howard63738202018-09-26 15:34:20 -0600572 for (zero = true, i = 0; zero && i < c->len; ++i) zero = !c->num[i];
573 if (zero) bc_num_zero(c);
Gavin Howard1d959152018-03-03 23:33:13 -0700574
Gavin Howard6fbdb292018-02-27 15:44:48 -0700575err:
Gavin Howard63738202018-09-26 15:34:20 -0600576 bc_num_free(&copy);
577 return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700578}
579
Gavin Howard63738202018-09-26 15:34:20 -0600580BcStatus bc_num_binary(BcNum *a, BcNum *b, BcNum *c, size_t scale,
Gavin Howard6e0f3c52018-08-27 17:28:22 -0600581 BcNumBinaryOp op, size_t req)
Gavin Howard6fbdb292018-02-27 15:44:48 -0700582{
Gavin Howard63738202018-09-26 15:34:20 -0600583 BcStatus status;
584 BcNum num2, *ptr_a, *ptr_b;
585 bool init = false;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700586
Gavin Howard63738202018-09-26 15:34:20 -0600587 assert(a && b && c && op);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700588
Gavin Howard63738202018-09-26 15:34:20 -0600589 if (c == a) {
590 memcpy(&num2, c, sizeof(BcNum));
591 ptr_a = &num2;
592 init = true;
593 }
594 else ptr_a = a;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700595
Gavin Howard63738202018-09-26 15:34:20 -0600596 if (c == b) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700597
Gavin Howard63738202018-09-26 15:34:20 -0600598 if (c == a) {
599 ptr_b = ptr_a;
600 }
601 else {
602 memcpy(&num2, c, sizeof(BcNum));
603 ptr_b = &num2;
604 init = true;
605 }
606 }
607 else ptr_b = b;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700608
Gavin Howard63738202018-09-26 15:34:20 -0600609 if (init) status = bc_num_init(c, req);
610 else status = bc_num_expand(c, req);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700611
Gavin Howard63738202018-09-26 15:34:20 -0600612 if (status) goto err;
613 status = op(ptr_a, ptr_b, c, scale);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700614
Gavin Howard9c4358c2018-03-22 20:11:28 -0600615err:
Gavin Howard63738202018-09-26 15:34:20 -0600616 if (c == a || c == b) bc_num_free(&num2);
617 return status;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700618}
619
Gavin Howard3f68df72018-03-22 20:30:27 -0600620bool bc_num_strValid(const char *val, size_t base) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700621
Gavin Howard63738202018-09-26 15:34:20 -0600622 BcDigit b;
623 bool small, radix = false;
624 size_t i, len = strlen(val);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700625
Gavin Howard63738202018-09-26 15:34:20 -0600626 if (!len) return true;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700627
Gavin Howard63738202018-09-26 15:34:20 -0600628 small = base <= 10;
629 b = (BcDigit) (small ? base + '0' : base - 9 + 'A');
Gavin Howard6fbdb292018-02-27 15:44:48 -0700630
Gavin Howard63738202018-09-26 15:34:20 -0600631 for (i = 0; i < len; ++i) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700632
Gavin Howard63738202018-09-26 15:34:20 -0600633 BcDigit c = val[i];
Gavin Howardf6e3fb32018-08-09 13:48:59 -0600634
Gavin Howard63738202018-09-26 15:34:20 -0600635 if (c == '.') {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700636
Gavin Howard63738202018-09-26 15:34:20 -0600637 if (radix) return false;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700638
Gavin Howard63738202018-09-26 15:34:20 -0600639 radix = true;
640 continue;
641 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700642
Gavin Howard63738202018-09-26 15:34:20 -0600643 if (c < '0' || (small && c >= b) || (c > '9' && (c < 'A' || c >= b)))
644 return false;
645 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700646
Gavin Howard63738202018-09-26 15:34:20 -0600647 return true;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700648}
649
Gavin Howard3f68df72018-03-22 20:30:27 -0600650BcStatus bc_num_parseDecimal(BcNum *n, const char *val) {
Gavin Howard6fbdb292018-02-27 15:44:48 -0700651
Gavin Howard63738202018-09-26 15:34:20 -0600652 BcStatus s;
653 size_t len, i;
654 const char *ptr;
655 bool zero = true;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700656
Gavin Howard63738202018-09-26 15:34:20 -0600657 for (i = 0; val[i] == '0'; ++i);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700658
Gavin Howard63738202018-09-26 15:34:20 -0600659 val += i;
660 len = strlen(val);
661 bc_num_zero(n);
Gavin Howard6fbdb292018-02-27 15:44:48 -0700662
Gavin Howard63738202018-09-26 15:34:20 -0600663 if (len) {
664 for (i = 0; zero && i < len; ++i) zero = val[i] == '0' || val[i] == '.';
665 if ((s = bc_num_expand(n, len))) return s;
666 }
Gavin Howard9c4358c2018-03-22 20:11:28 -0600667
Gavin Howard63738202018-09-26 15:34:20 -0600668 if (zero) {
669 memset(n->num, 0, sizeof(BcDigit) * n->cap);
670 n->neg = false;
671 return BC_STATUS_SUCCESS;
672 }
Gavin Howard6fbdb292018-02-27 15:44:48 -0700673
Gavin Howard63738202018-09-26 15:34:20 -0600674 ptr = strchr(val, '.');
Gavin Howard6fbdb292018-02-27 15:44:48 -0700675
Gavin Howard63738202018-09-26 15:34:20 -0600676 // Explicitly test for NULL here to produce either a 0 or 1.
677 n->rdx = (size_t) ((ptr != NULL) * ((val + len) - (ptr + 1)));
Gavin Howard6fbdb292018-02-27 15:44:48 -0700678
Gavin Howard63738202018-09-26 15:34:20 -0600679 for (i = len - 1; i < len; ++n->len, i -= 1 + (i && val[i - 1] == '.'))
680 n->num[n->len] = val[i] - '0';
Gavin Howard6fbdb292018-02-27 15:44:48 -0700681
Gavin Howard63738202018-09-26 15:34:20 -0600682 return BC_STATUS_SUCCESS;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700683}
684
Gavin Howard3f68df72018-03-22 20:30:27 -0600685BcStatus bc_num_parseBase(BcNum *n, const char *val, BcNum *base) {
Gavin Howardede51f02018-03-02 12:30:00 -0700686
Gavin Howard63738202018-09-26 15:34:20 -0600687 BcStatus s;
688 BcNum temp, mult, result;
689 BcDigit c = '\0';
690 bool zero;
691 unsigned long v;
692 size_t i, digits, len = strlen(val);
Gavin Howardede51f02018-03-02 12:30:00 -0700693
Gavin Howard63738202018-09-26 15:34:20 -0600694 bc_num_zero(n);
Gavin Howardede51f02018-03-02 12:30:00 -0700695
Gavin Howard63738202018-09-26 15:34:20 -0600696 for (zero = true, i = 0; zero && i < len; ++i)
697 zero = (val[i] == '.' || val[i] == '0');
698 if (zero) return BC_STATUS_SUCCESS;
Gavin Howardede51f02018-03-02 12:30:00 -0700699
Gavin Howard63738202018-09-26 15:34:20 -0600700 if ((s = bc_num_init(&temp, BC_NUM_DEF_SIZE))) return s;
701 if ((s = bc_num_init(&mult, BC_NUM_DEF_SIZE))) goto mult_err;
Gavin Howardede51f02018-03-02 12:30:00 -0700702
Gavin Howard63738202018-09-26 15:34:20 -0600703 for (i = 0; i < len && (c = val[i]) != '.'; ++i) {
Gavin Howardede51f02018-03-02 12:30:00 -0700704
Gavin Howard63738202018-09-26 15:34:20 -0600705 v = (unsigned long) (c <= '9' ? c - '0' : c - 'A' + 10);
Gavin Howardfcb6ebb2018-03-09 10:41:06 -0700706
Gavin Howard63738202018-09-26 15:34:20 -0600707 if ((s = bc_num_mul(n, base, &mult, 0))) goto int_err;
708 if ((s = bc_num_ulong2num(&temp, v))) goto int_err;
709 if ((s = bc_num_add(&mult, &temp, n, 0))) goto int_err;
710 }
Gavin Howardede51f02018-03-02 12:30:00 -0700711
Gavin Howard63738202018-09-26 15:34:20 -0600712 if (i == len && !(c = val[i])) goto int_err;
713 assert(c == '.');
714 if ((s = bc_num_init(&result, base->len))) goto int_err;
Gavin Howardede51f02018-03-02 12:30:00 -0700715
Gavin Howard63738202018-09-26 15:34:20 -0600716 bc_num_zero(&result);
717 bc_num_one(&mult);
Gavin Howardede51f02018-03-02 12:30:00 -0700718
Gavin Howard63738202018-09-26 15:34:20 -0600719 for (i += 1, digits = 0; i < len && (c = val[i]); ++i, ++digits) {
Gavin Howardede51f02018-03-02 12:30:00 -0700720
Gavin Howard63738202018-09-26 15:34:20 -0600721 v = (unsigned long) (c <= '9' ? c - '0' : c - 'A' + 10);
Gavin Howardede51f02018-03-02 12:30:00 -0700722
Gavin Howard63738202018-09-26 15:34:20 -0600723 if ((s = bc_num_mul(&result, base, &result, 0))) goto err;
724 if ((s = bc_num_ulong2num(&temp, v))) goto err;
725 if ((s = bc_num_add(&result, &temp, &result, 0))) goto err;
726 if ((s = bc_num_mul(&mult, base, &mult, 0))) goto err;
727 }
Gavin Howardede51f02018-03-02 12:30:00 -0700728
Gavin Howard63738202018-09-26 15:34:20 -0600729 if ((s = bc_num_div(&result, &mult, &result, digits))) goto err;
730 if ((s = bc_num_add(n, &result, n, digits))) goto err;
Gavin Howard5b24c3f2018-03-13 23:57:15 -0600731
Gavin Howard63738202018-09-26 15:34:20 -0600732 if (n->len) {
733 if (n->rdx < digits && n->len) s = bc_num_extend(n, digits - n->rdx);
734 }
735 else bc_num_zero(n);
Gavin Howardede51f02018-03-02 12:30:00 -0700736
737err:
Gavin Howard63738202018-09-26 15:34:20 -0600738 bc_num_free(&result);
Gavin Howardede51f02018-03-02 12:30:00 -0700739int_err:
Gavin Howard63738202018-09-26 15:34:20 -0600740 bc_num_free(&mult);
Gavin Howardede51f02018-03-02 12:30:00 -0700741mult_err:
Gavin Howard63738202018-09-26 15:34:20 -0600742 bc_num_free(&temp);
743 return s;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700744}
745
Gavin Howard80977b22018-09-06 14:29:13 -0600746BcStatus bc_num_printNewline(size_t *nchars, size_t line_len) {
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600747
Gavin Howard63738202018-09-26 15:34:20 -0600748 if (*nchars == line_len - 1) {
749 if (putchar('\\') == EOF) return BC_STATUS_IO_ERR;
750 if (putchar('\n') == EOF) return BC_STATUS_IO_ERR;
751 *nchars = 0;
752 }
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600753
Gavin Howard63738202018-09-26 15:34:20 -0600754 return BC_STATUS_SUCCESS;
Gavin Howard80977b22018-09-06 14:29:13 -0600755}
756
757BcStatus bc_num_printDigits(size_t num, size_t width, bool radix,
758 size_t *nchars, size_t line_len)
759{
Gavin Howard63738202018-09-26 15:34:20 -0600760 BcStatus s;
761 size_t exp, pow;
Gavin Howard80977b22018-09-06 14:29:13 -0600762
Gavin Howard63738202018-09-26 15:34:20 -0600763 if ((s = bc_num_printNewline(nchars, line_len))) return s;
Gavin Howard1a5cac22018-09-25 12:59:21 -0600764
Gavin Howard63738202018-09-26 15:34:20 -0600765 if (putchar(radix ? '.' : ' ') == EOF) return BC_STATUS_IO_ERR;
766 ++(*nchars);
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600767
Gavin Howard63738202018-09-26 15:34:20 -0600768 if ((s = bc_num_printNewline(nchars, line_len))) return s;
Gavin Howard80977b22018-09-06 14:29:13 -0600769
Gavin Howard63738202018-09-26 15:34:20 -0600770 for (exp = 0, pow = 1; exp < width - 1; ++exp, pow *= 10);
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600771
Gavin Howard63738202018-09-26 15:34:20 -0600772 for (exp = 0; exp < width; pow /= 10, ++(*nchars), ++exp) {
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600773
Gavin Howard63738202018-09-26 15:34:20 -0600774 size_t div;
Gavin Howardf6e3fb32018-08-09 13:48:59 -0600775
Gavin Howard63738202018-09-26 15:34:20 -0600776 if ((s = bc_num_printNewline(nchars, line_len))) return s;
Gavin Howard2ed2bfb2018-03-14 02:42:32 -0600777
Gavin Howard63738202018-09-26 15:34:20 -0600778 div = num / pow;
779 num -= div * pow;
Gavin Howard9c4358c2018-03-22 20:11:28 -0600780
Gavin Howard63738202018-09-26 15:34:20 -0600781 if (putchar(((char) div) + '0') == EOF) return BC_STATUS_IO_ERR;
782 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700783
Gavin Howard63738202018-09-26 15:34:20 -0600784 return BC_STATUS_SUCCESS;
Gavin Howard6fbdb292018-02-27 15:44:48 -0700785}
Gavin Howardfe679f02018-02-14 15:50:09 -0700786
Gavin Howard902a16c2018-05-16 01:22:53 -0600787BcStatus bc_num_printHex(size_t num, size_t width, bool radix,
Gavin Howard0011a3a2018-03-29 23:11:32 -0600788 size_t *nchars, size_t line_len)
Gavin Howard2682a1f2018-03-03 09:09:13 -0700789{
Gavin Howard63738202018-09-26 15:34:20 -0600790 BcStatus s;
Gavin Howard80977b22018-09-06 14:29:13 -0600791
Gavin Howard63738202018-09-26 15:34:20 -0600792 assert(width == 1);
Gavin Howard80977b22018-09-06 14:29:13 -0600793
Gavin Howard63738202018-09-26 15:34:20 -0600794 if (radix) {
795 if ((s = bc_num_printNewline(nchars, line_len))) return s;
796 if (putchar('.') == EOF) return BC_STATUS_IO_ERR;
797 *nchars += 1;
798 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700799
Gavin Howard63738202018-09-26 15:34:20 -0600800 if ((s = bc_num_printNewline(nchars, line_len))) return s;
801 if (putchar(bc_num_hex_digits[num]) == EOF) return BC_STATUS_IO_ERR;
802 *nchars = *nchars + width;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700803
Gavin Howard63738202018-09-26 15:34:20 -0600804 return BC_STATUS_SUCCESS;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700805}
806
Gavin Howard5f46ee32018-09-25 13:02:43 -0600807BcStatus bc_num_printDecimal(BcNum *n, size_t *nchars, size_t len) {
Gavin Howard32f2beb2018-03-09 11:43:20 -0700808
Gavin Howard63738202018-09-26 15:34:20 -0600809 BcStatus s;
810 size_t i, rdx = n->rdx - 1;
Gavin Howard32f2beb2018-03-09 11:43:20 -0700811
Gavin Howard63738202018-09-26 15:34:20 -0600812 if (n->neg && putchar('-') == EOF) return BC_STATUS_IO_ERR;
813 (*nchars) += n->neg;
Gavin Howard32f2beb2018-03-09 11:43:20 -0700814
Gavin Howard63738202018-09-26 15:34:20 -0600815 s = BC_STATUS_SUCCESS;
Gavin Howard32f2beb2018-03-09 11:43:20 -0700816
Gavin Howard63738202018-09-26 15:34:20 -0600817 for (i = n->len - 1; !s && i < n->len; --i)
818 s = bc_num_printHex((size_t) n->num[i], 1, i == rdx, nchars, len);
Gavin Howard32f2beb2018-03-09 11:43:20 -0700819
Gavin Howard63738202018-09-26 15:34:20 -0600820 return s;
Gavin Howard32f2beb2018-03-09 11:43:20 -0700821}
822
Gavin Howard0011a3a2018-03-29 23:11:32 -0600823BcStatus bc_num_printBase(BcNum *n, BcNum *base, size_t base_t,
824 size_t *nchars, size_t line_len)
Gavin Howardbc7cae82018-03-14 13:43:04 -0600825{
Gavin Howard63738202018-09-26 15:34:20 -0600826 BcStatus s;
827 BcVec stack;
828 BcNum intp, fracp, digit, frac_len;
829 size_t width, i;
830 BcNumDigitOp print;
831 unsigned long dig, *ptr;
832 bool radix, neg = n->neg;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700833
Gavin Howard63738202018-09-26 15:34:20 -0600834 if (neg && putchar('-') == EOF) return BC_STATUS_IO_ERR;
835 (*nchars) += neg;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700836
Gavin Howard63738202018-09-26 15:34:20 -0600837 n->neg = false;
Gavin Howard7f45f1a2018-09-24 14:50:55 -0600838
Gavin Howard63738202018-09-26 15:34:20 -0600839 if (base_t <= BC_NUM_MAX_IBASE) {
840 width = 1;
841 print = bc_num_printHex;
842 }
843 else {
844 width = (size_t) floor(log10((double) (base_t - 1)) + 1.0);
845 print = bc_num_printDigits;
846 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700847
Gavin Howard63738202018-09-26 15:34:20 -0600848 if ((s = bc_vec_init(&stack, sizeof(long), NULL))) goto stack_err;
849 if ((s = bc_num_init(&intp, n->len))) goto int_err;
850 if ((s = bc_num_init(&fracp, n->rdx))) goto frac_err;
851 if ((s = bc_num_init(&digit, width))) goto digit_err;
852 if ((s = bc_num_init(&frac_len, BC_NUM_INT(n)))) goto frac_len_err;
853 if ((s = bc_num_copy(&intp, n))) goto err;
854 bc_num_one(&frac_len);
Gavin Howarda50fc542018-03-29 17:25:38 -0600855
Gavin Howard63738202018-09-26 15:34:20 -0600856 bc_num_truncate(&intp, intp.rdx);
857 if ((s = bc_num_sub(n, &intp, &fracp, 0))) goto err;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700858
Gavin Howard63738202018-09-26 15:34:20 -0600859 while (intp.len) {
860 if ((s = bc_num_mod(&intp, base, &digit, 0))) goto err;
861 if ((s = bc_num_ulong(&digit, &dig))) goto err;
862 if ((s = bc_vec_push(&stack, 1, &dig))) goto err;
863 if ((s = bc_num_div(&intp, base, &intp, 0))) goto err;
864 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700865
Gavin Howard63738202018-09-26 15:34:20 -0600866 for (i = 0; i < stack.len; ++i) {
867 ptr = bc_vec_item_rev(&stack, i);
868 assert(ptr);
869 if ((s = print(*ptr, width, false, nchars, line_len))) goto err;
870 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700871
Gavin Howard63738202018-09-26 15:34:20 -0600872 if (!n->rdx) goto err;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700873
Gavin Howard63738202018-09-26 15:34:20 -0600874 for (radix = true; frac_len.len <= n->rdx; radix = false) {
875 if ((s = bc_num_mul(&fracp, base, &fracp, n->rdx))) goto err;
876 if ((s = bc_num_ulong(&fracp, &dig))) goto err;
877 if ((s = bc_num_ulong2num(&intp, dig))) goto err;
878 if ((s = bc_num_sub(&fracp, &intp, &fracp, 0))) goto err;
879 if ((s = print(dig, width, radix, nchars, line_len))) goto err;
880 if ((s = bc_num_mul(&frac_len, base, &frac_len, 0))) goto err;
881 }
Gavin Howard2682a1f2018-03-03 09:09:13 -0700882
883err:
Gavin Howard63738202018-09-26 15:34:20 -0600884 bc_num_free(&frac_len);
Gavin Howard2682a1f2018-03-03 09:09:13 -0700885frac_len_err:
Gavin Howard63738202018-09-26 15:34:20 -0600886 bc_num_free(&digit);
Gavin Howard2682a1f2018-03-03 09:09:13 -0700887digit_err:
Gavin Howard63738202018-09-26 15:34:20 -0600888 bc_num_free(&fracp);
Gavin Howard2682a1f2018-03-03 09:09:13 -0700889frac_err:
Gavin Howard63738202018-09-26 15:34:20 -0600890 bc_num_free(&intp);
Gavin Howard2682a1f2018-03-03 09:09:13 -0700891int_err:
Gavin Howard63738202018-09-26 15:34:20 -0600892 bc_vec_free(&stack);
Gavin Howard7f45f1a2018-09-24 14:50:55 -0600893stack_err:
Gavin Howard63738202018-09-26 15:34:20 -0600894 n->neg = neg;
895 return s;
Gavin Howard2682a1f2018-03-03 09:09:13 -0700896}
897
Gavin Howard8d1f1db2018-02-23 11:29:41 -0700898BcStatus bc_num_init(BcNum *n, size_t request) {
Gavin Howardb5c77212018-02-14 17:12:34 -0700899
Gavin Howard63738202018-09-26 15:34:20 -0600900 assert(n);
Gavin Howardb5c77212018-02-14 17:12:34 -0700901
Gavin Howard63738202018-09-26 15:34:20 -0600902 memset(n, 0, sizeof(BcNum));
Gavin Howardb5c77212018-02-14 17:12:34 -0700903
Gavin Howard63738202018-09-26 15:34:20 -0600904 request = request >= BC_NUM_DEF_SIZE ? request : BC_NUM_DEF_SIZE;
905 if (!(n->num = malloc(request))) return BC_STATUS_ALLOC_ERR;
Gavin Howardb5c77212018-02-14 17:12:34 -0700906
Gavin Howard63738202018-09-26 15:34:20 -0600907 n->cap = request;
Gavin Howardb5c77212018-02-14 17:12:34 -0700908
Gavin Howard63738202018-09-26 15:34:20 -0600909 return BC_STATUS_SUCCESS;
Gavin Howardb5c77212018-02-14 17:12:34 -0700910}
911
Gavin Howard8d1f1db2018-02-23 11:29:41 -0700912BcStatus bc_num_expand(BcNum *n, size_t request) {
Gavin Howardb5c77212018-02-14 17:12:34 -0700913
Gavin Howard63738202018-09-26 15:34:20 -0600914 BcDigit *temp;
Gavin Howard9c4358c2018-03-22 20:11:28 -0600915
Gavin Howard63738202018-09-26 15:34:20 -0600916 assert(n && request);
Gavin Howardb5c77212018-02-14 17:12:34 -0700917
Gavin Howard63738202018-09-26 15:34:20 -0600918 if (request <= n->cap) return BC_STATUS_SUCCESS;
919 if (!(temp = realloc(n->num, request))) return BC_STATUS_ALLOC_ERR;
Gavin Howardb5c77212018-02-14 17:12:34 -0700920
Gavin Howard63738202018-09-26 15:34:20 -0600921 memset(temp + n->cap, 0, sizeof(char) * (request - n->cap));
922 n->num = temp;
923 n->cap = request;
Gavin Howardb5c77212018-02-14 17:12:34 -0700924
Gavin Howard63738202018-09-26 15:34:20 -0600925 return BC_STATUS_SUCCESS;
Gavin Howardb5c77212018-02-14 17:12:34 -0700926}
927
Gavin Howarded392aa2018-02-27 13:09:26 -0700928void bc_num_free(void *num) {
Gavin Howard63738202018-09-26 15:34:20 -0600929 BcNum *n = (BcNum*) num;
930 assert(n && n->num);
931 free(n->num);
Gavin Howardb5c77212018-02-14 17:12:34 -0700932}
933
Gavin Howard9c4358c2018-03-22 20:11:28 -0600934BcStatus bc_num_copy(BcNum *d, BcNum *s) {
Gavin Howard5a049c42018-02-15 11:24:11 -0700935
Gavin Howard63738202018-09-26 15:34:20 -0600936 BcStatus status;
Gavin Howard5a049c42018-02-15 11:24:11 -0700937
Gavin Howard63738202018-09-26 15:34:20 -0600938 assert(d && s);
Gavin Howardf23448c2018-02-27 20:36:24 -0700939
Gavin Howard63738202018-09-26 15:34:20 -0600940 if (d == s) return BC_STATUS_SUCCESS;
941 if ((status = bc_num_expand(d, s->cap))) return status;
Gavin Howard5a049c42018-02-15 11:24:11 -0700942
Gavin Howard63738202018-09-26 15:34:20 -0600943 d->len = s->len;
944 d->neg = s->neg;
945 d->rdx = s->rdx;
Gavin Howard5a049c42018-02-15 11:24:11 -0700946
Gavin Howard63738202018-09-26 15:34:20 -0600947 memcpy(d->num, s->num, sizeof(BcDigit) * d->len);
948 memset(d->num + d->len, 0, sizeof(BcDigit) * (d->cap - d->len));
Gavin Howard5a049c42018-02-15 11:24:11 -0700949
Gavin Howard63738202018-09-26 15:34:20 -0600950 return BC_STATUS_SUCCESS;
Gavin Howard5a049c42018-02-15 11:24:11 -0700951}
952
Gavin Howard5cea43b2018-03-02 11:32:11 -0700953BcStatus bc_num_parse(BcNum *n, const char *val, BcNum *base, size_t base_t) {
Gavin Howard025d04d2018-02-20 13:53:28 -0700954
Gavin Howard63738202018-09-26 15:34:20 -0600955 BcStatus s;
Gavin Howard3eb626f2018-02-14 13:54:35 -0700956
Gavin Howard63738202018-09-26 15:34:20 -0600957 assert(n && val && base);
958 assert(base_t >= BC_NUM_MIN_BASE && base_t <= BC_NUM_MAX_IBASE);
Gavin Howard3eb626f2018-02-14 13:54:35 -0700959
Gavin Howard63738202018-09-26 15:34:20 -0600960 if (!bc_num_strValid(val, base_t)) return BC_STATUS_MATH_BAD_STRING;
Gavin Howard3eb626f2018-02-14 13:54:35 -0700961
Gavin Howard63738202018-09-26 15:34:20 -0600962 if (base_t == 10) s = bc_num_parseDecimal(n, val);
963 else s = bc_num_parseBase(n, val, base);
Gavin Howard3eb626f2018-02-14 13:54:35 -0700964
Gavin Howard63738202018-09-26 15:34:20 -0600965 return s;
Gavin Howard3eb626f2018-02-14 13:54:35 -0700966}
967
Gavin Howard0011a3a2018-03-29 23:11:32 -0600968BcStatus bc_num_print(BcNum *n, BcNum *base, size_t base_t, bool newline,
969 size_t *nchars, size_t line_len)
Gavin Howard152f3e82018-03-07 12:33:15 -0700970{
Gavin Howard63738202018-09-26 15:34:20 -0600971 BcStatus s;
Gavin Howard3eb626f2018-02-14 13:54:35 -0700972
Gavin Howard63738202018-09-26 15:34:20 -0600973 assert(n && base && nchars);
974 assert(base_t >= BC_NUM_MIN_BASE && base_t <= BC_MAX_OBASE);
Gavin Howard3eb626f2018-02-14 13:54:35 -0700975
Gavin Howard63738202018-09-26 15:34:20 -0600976 if (*nchars >= line_len) {
977 if (putchar('\\') == EOF) return BC_STATUS_IO_ERR;
978 if (putchar('\n') == EOF) return BC_STATUS_IO_ERR;
979 *nchars = 0;
980 }
Gavin Howardbc7cae82018-03-14 13:43:04 -0600981
Gavin Howard63738202018-09-26 15:34:20 -0600982 if (!n->len) {
983 if (putchar('0') == EOF) return BC_STATUS_IO_ERR;
984 ++(*nchars);
985 s = BC_STATUS_SUCCESS;
986 }
987 else if (base_t == 10) s = bc_num_printDecimal(n, nchars, line_len);
988 else s = bc_num_printBase(n, base, base_t, nchars, line_len);
Gavin Howard3eb626f2018-02-14 13:54:35 -0700989
Gavin Howard63738202018-09-26 15:34:20 -0600990 if (s) return s;
Gavin Howard152f3e82018-03-07 12:33:15 -0700991
Gavin Howard63738202018-09-26 15:34:20 -0600992 if (newline) {
993 if (putchar('\n') == EOF) return BC_STATUS_IO_ERR;
994 *nchars = 0;
995 }
Gavin Howard152f3e82018-03-07 12:33:15 -0700996
Gavin Howard63738202018-09-26 15:34:20 -0600997 return s;
Gavin Howard3eb626f2018-02-14 13:54:35 -0700998}
999
Gavin Howard8d1f1db2018-02-23 11:29:41 -07001000BcStatus bc_num_ulong(BcNum *n, unsigned long *result) {
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001001
Gavin Howard63738202018-09-26 15:34:20 -06001002 size_t i;
1003 unsigned long pow;
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001004
Gavin Howard63738202018-09-26 15:34:20 -06001005 assert(n && result);
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001006
Gavin Howard63738202018-09-26 15:34:20 -06001007 if (n->neg) return BC_STATUS_MATH_NEGATIVE;
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001008
Gavin Howard63738202018-09-26 15:34:20 -06001009 for (*result = 0, pow = 1, i = n->rdx; i < n->len; ++i) {
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001010
Gavin Howard63738202018-09-26 15:34:20 -06001011 unsigned long prev = *result, powprev = pow;
1012 *result += ((unsigned long) n->num[i]) * pow;
1013 pow *= 10;
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001014
Gavin Howard63738202018-09-26 15:34:20 -06001015 if (*result < prev || pow < powprev) return BC_STATUS_MATH_OVERFLOW;
1016 }
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001017
Gavin Howard63738202018-09-26 15:34:20 -06001018 return BC_STATUS_SUCCESS;
Gavin Howard68b8a5c2018-02-15 11:41:24 -07001019}
1020
Gavin Howard8d1f1db2018-02-23 11:29:41 -07001021BcStatus bc_num_ulong2num(BcNum *n, unsigned long val) {
Gavin Howard8e2cc692018-02-15 17:39:14 -07001022
Gavin Howard63738202018-09-26 15:34:20 -06001023 BcStatus s;
1024 size_t len, i;
1025 BcDigit *ptr;
Gavin Howard8e2cc692018-02-15 17:39:14 -07001026
Gavin Howard63738202018-09-26 15:34:20 -06001027 assert(n);
Gavin Howard8e2cc692018-02-15 17:39:14 -07001028
Gavin Howard63738202018-09-26 15:34:20 -06001029 bc_num_zero(n);
Gavin Howard025d04d2018-02-20 13:53:28 -07001030
Gavin Howard63738202018-09-26 15:34:20 -06001031 if (!val) {
1032 memset(n->num, 0, sizeof(char) * n->cap);
1033 return BC_STATUS_SUCCESS;
1034 }
Gavin Howard8e2cc692018-02-15 17:39:14 -07001035
Gavin Howard63738202018-09-26 15:34:20 -06001036 len = (size_t) ceil(log10(((double) ULONG_MAX) + 1.0));
Gavin Howard8e2cc692018-02-15 17:39:14 -07001037
Gavin Howard63738202018-09-26 15:34:20 -06001038 if ((s = bc_num_expand(n, len))) return s;
Gavin Howard8e2cc692018-02-15 17:39:14 -07001039
Gavin Howard63738202018-09-26 15:34:20 -06001040 for (ptr = n->num, i = 0; val; ++i, ++n->len) {
1041 ptr[i] = (char) (val % 10);
1042 val /= 10;
1043 }
Gavin Howard8e2cc692018-02-15 17:39:14 -07001044
Gavin Howard63738202018-09-26 15:34:20 -06001045 return BC_STATUS_SUCCESS;
Gavin Howard025d04d2018-02-20 13:53:28 -07001046}
1047
Gavin Howard4538c902018-08-29 14:31:40 -06001048BcStatus bc_num_add(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001049 (void) scale;
1050 BcNumBinaryOp op = (!a->neg == !b->neg) ? bc_num_alg_a : bc_num_alg_s;
1051 return bc_num_binary(a, b, res, false, op, BC_NUM_AREQ(a, b));
Gavin Howard3eb626f2018-02-14 13:54:35 -07001052}
1053
Gavin Howard4538c902018-08-29 14:31:40 -06001054BcStatus bc_num_sub(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001055 (void) scale;
1056 BcNumBinaryOp op = (!a->neg == !b->neg) ? bc_num_alg_s : bc_num_alg_a;
1057 return bc_num_binary(a, b, res, true, op, BC_NUM_AREQ(a, b));
Gavin Howard3eb626f2018-02-14 13:54:35 -07001058}
1059
Gavin Howard4538c902018-08-29 14:31:40 -06001060BcStatus bc_num_mul(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001061 size_t req = BC_NUM_MREQ(a, b, scale);
1062 return bc_num_binary(a, b, res, scale, bc_num_alg_m, req);
Gavin Howard3eb626f2018-02-14 13:54:35 -07001063}
1064
Gavin Howard4538c902018-08-29 14:31:40 -06001065BcStatus bc_num_div(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001066 size_t req = BC_NUM_MREQ(a, b, scale);
1067 return bc_num_binary(a, b, res, scale, bc_num_alg_d, req);
Gavin Howard3eb626f2018-02-14 13:54:35 -07001068}
1069
Gavin Howard4538c902018-08-29 14:31:40 -06001070BcStatus bc_num_mod(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001071 size_t req = BC_NUM_MREQ(a, b, scale);
1072 return bc_num_binary(a, b, res, scale, bc_num_alg_mod, req);
Gavin Howard3eb626f2018-02-14 13:54:35 -07001073}
1074
Gavin Howard4538c902018-08-29 14:31:40 -06001075BcStatus bc_num_pow(BcNum *a, BcNum *b, BcNum *res, size_t scale) {
Gavin Howard63738202018-09-26 15:34:20 -06001076 return bc_num_binary(a, b, res, scale, bc_num_alg_p, a->len * b->len + 1);
Gavin Howard3eb626f2018-02-14 13:54:35 -07001077}
1078
Gavin Howard4538c902018-08-29 14:31:40 -06001079BcStatus bc_num_sqrt(BcNum *a, BcNum *res, size_t scale) {
Gavin Howard954276b2018-05-16 01:43:58 -06001080
Gavin Howard63738202018-09-26 15:34:20 -06001081 BcStatus s;
1082 BcNum a2, *ptr_a, num1, num2, half, f, fprime, *x0, *x1, *temp;
1083 size_t pow, len, digits, resrdx, req;
1084 ssize_t cmp;
Gavin Howard954276b2018-05-16 01:43:58 -06001085
Gavin Howard63738202018-09-26 15:34:20 -06001086 assert(a && res);
Gavin Howard954276b2018-05-16 01:43:58 -06001087
Gavin Howard63738202018-09-26 15:34:20 -06001088 req = BC_MAX(scale, a->rdx) + ((BC_NUM_INT(a)+ 1) >> 1) + 1;
Gavin Howard954276b2018-05-16 01:43:58 -06001089
Gavin Howard63738202018-09-26 15:34:20 -06001090 if (res == a) {
1091 memcpy(&a2, res, sizeof(BcNum));
1092 ptr_a = &a2;
1093 s = bc_num_init(res, req);
1094 }
1095 else {
1096 ptr_a = a;
1097 s = bc_num_expand(res, req);
1098 }
Gavin Howard954276b2018-05-16 01:43:58 -06001099
Gavin Howard63738202018-09-26 15:34:20 -06001100 if (s) goto init_err;
Gavin Howard954276b2018-05-16 01:43:58 -06001101
Gavin Howard63738202018-09-26 15:34:20 -06001102 if (!ptr_a->len) {
1103 bc_num_zero(res);
1104 goto init_err;
1105 }
1106 else if (ptr_a->neg) {
1107 s = BC_STATUS_MATH_NEG_SQRT;
1108 goto init_err;
1109 }
1110 else if (BC_NUM_ONE(a)) {
1111 bc_num_one(res);
1112 s = bc_num_extend(res, scale);
1113 goto init_err;
1114 }
Gavin Howard954276b2018-05-16 01:43:58 -06001115
Gavin Howard63738202018-09-26 15:34:20 -06001116 memset(res->num, 0, res->cap * sizeof(BcDigit));
1117 len = ptr_a->len;
Gavin Howard954276b2018-05-16 01:43:58 -06001118
Gavin Howard63738202018-09-26 15:34:20 -06001119 scale = BC_MAX(scale, ptr_a->rdx) + 1;
Gavin Howard954276b2018-05-16 01:43:58 -06001120
Gavin Howard63738202018-09-26 15:34:20 -06001121 if ((s = bc_num_init(&num1, len))) goto init_err;
1122 if ((s = bc_num_init(&num2, num1.len))) goto num2_err;
1123 if ((s = bc_num_init(&half, BC_NUM_DEF_SIZE))) goto two_err;
Gavin Howard954276b2018-05-16 01:43:58 -06001124
Gavin Howard63738202018-09-26 15:34:20 -06001125 bc_num_one(&half);
1126 half.num[0] = 5;
1127 half.rdx = 1;
Gavin Howard954276b2018-05-16 01:43:58 -06001128
Gavin Howard63738202018-09-26 15:34:20 -06001129 len += scale;
Gavin Howard954276b2018-05-16 01:43:58 -06001130
Gavin Howard63738202018-09-26 15:34:20 -06001131 if ((s = bc_num_init(&f, len))) goto f_err;
1132 if ((s = bc_num_init(&fprime, len + scale))) goto fprime_err;
Gavin Howard954276b2018-05-16 01:43:58 -06001133
Gavin Howard63738202018-09-26 15:34:20 -06001134 x0 = &num1;
1135 x1 = &num2;
Gavin Howard954276b2018-05-16 01:43:58 -06001136
Gavin Howard63738202018-09-26 15:34:20 -06001137 bc_num_one(x0);
Gavin Howard954276b2018-05-16 01:43:58 -06001138
Gavin Howard63738202018-09-26 15:34:20 -06001139 pow = BC_NUM_INT(ptr_a);
Gavin Howard954276b2018-05-16 01:43:58 -06001140
Gavin Howard63738202018-09-26 15:34:20 -06001141 if (pow) {
Gavin Howard954276b2018-05-16 01:43:58 -06001142
Gavin Howard63738202018-09-26 15:34:20 -06001143 if (pow & 1) {
1144 x0->num[0] = 2;
1145 pow -= 1;
1146 }
1147 else {
1148 x0->num[0] = 6;
1149 pow -= 2;
1150 }
Gavin Howard954276b2018-05-16 01:43:58 -06001151
Gavin Howard63738202018-09-26 15:34:20 -06001152 if ((s = bc_num_extend(x0, pow))) goto err;
Gavin Howarded742ce2018-08-29 14:23:12 -06001153
Gavin Howard63738202018-09-26 15:34:20 -06001154 // Make sure to move the radix back.
1155 x0->rdx -= pow;
1156 }
Gavin Howard954276b2018-05-16 01:43:58 -06001157
Gavin Howard63738202018-09-26 15:34:20 -06001158 cmp = 1;
1159 x0->rdx = digits = 0;
1160 resrdx = scale + 1;
1161 len = BC_NUM_INT(x0) + resrdx;
Gavin Howard954276b2018-05-16 01:43:58 -06001162
Gavin Howard63738202018-09-26 15:34:20 -06001163 while (!bcg.signe && cmp && digits <= len) {
Gavin Howard954276b2018-05-16 01:43:58 -06001164
Gavin Howard63738202018-09-26 15:34:20 -06001165 if ((s = bc_num_div(a, x0, &f, resrdx))) goto err;
1166 if ((s = bc_num_add(x0, &f, &fprime, resrdx))) goto err;
1167 if ((s = bc_num_mul(&fprime, &half, x1, resrdx))) goto err;
Gavin Howard954276b2018-05-16 01:43:58 -06001168
Gavin Howard63738202018-09-26 15:34:20 -06001169 cmp = bc_num_cmp(x1, x0);
1170 digits = x1->len - (unsigned long long) llabs(cmp);
Gavin Howard954276b2018-05-16 01:43:58 -06001171
Gavin Howard63738202018-09-26 15:34:20 -06001172 temp = x0;
1173 x0 = x1;
1174 x1 = temp;
1175 }
Gavin Howard954276b2018-05-16 01:43:58 -06001176
Gavin Howard63738202018-09-26 15:34:20 -06001177 if (bcg.signe) {
1178 s = BC_STATUS_EXEC_SIGNAL;
1179 goto err;
1180 }
Gavin Howard0dfe2922018-05-22 13:57:02 -06001181
Gavin Howard63738202018-09-26 15:34:20 -06001182 if ((s = bc_num_copy(res, x0))) goto err;
Gavin Howard954276b2018-05-16 01:43:58 -06001183
Gavin Howard63738202018-09-26 15:34:20 -06001184 if (res->rdx > --scale) bc_num_truncate(res, res->rdx - scale);
1185 else if (res->rdx < scale) s = bc_num_extend(res, scale - res->rdx);
Gavin Howard954276b2018-05-16 01:43:58 -06001186
1187err:
Gavin Howard63738202018-09-26 15:34:20 -06001188 bc_num_free(&fprime);
Gavin Howard954276b2018-05-16 01:43:58 -06001189fprime_err:
Gavin Howard63738202018-09-26 15:34:20 -06001190 bc_num_free(&f);
Gavin Howard954276b2018-05-16 01:43:58 -06001191f_err:
Gavin Howard63738202018-09-26 15:34:20 -06001192 bc_num_free(&half);
Gavin Howard954276b2018-05-16 01:43:58 -06001193two_err:
Gavin Howard63738202018-09-26 15:34:20 -06001194 bc_num_free(&num2);
Gavin Howard954276b2018-05-16 01:43:58 -06001195num2_err:
Gavin Howard63738202018-09-26 15:34:20 -06001196 bc_num_free(&num1);
Gavin Howard954276b2018-05-16 01:43:58 -06001197init_err:
Gavin Howard63738202018-09-26 15:34:20 -06001198 if (res == a) bc_num_free(&a2);
1199 return s;
Gavin Howard3eb626f2018-02-14 13:54:35 -07001200}