bigint: Add prototypes for immediate add+sub.

This commit is contained in:
Jeroen van Rijn
2021-08-11 20:59:50 +02:00
parent d57e1be89f
commit c5cbd3260a
2 changed files with 40 additions and 14 deletions
+98
View File
@@ -0,0 +1,98 @@
package bigint
/*
Copyright 2021 Jeroen van Rijn <nom@duclavier.com>.
Made available under Odin's BSD-2 license.
A BigInt implementation in Odin.
For the theoretical underpinnings, see Knuth's The Art of Computer Programming, Volume 2, section 4.3.
The code started out as an idiomatic source port of libTomMath, which is in the public domain, with thanks.
*/
import "core:intrinsics"
is_initialized :: proc(a: ^Int) -> bool {
return a != rawptr(uintptr(0));
}
is_zero :: proc(a: ^Int) -> bool {
return is_initialized(a) && a.used == 0;
}
is_positive :: proc(a: ^Int) -> bool {
return is_initialized(a) && a.sign == .Zero_or_Positive;
}
is_pos :: is_positive;;
is_negative :: proc(a: ^Int) -> bool {
return is_initialized(a) && a.sign == .Negative;
}
is_neg :: is_negative;
/*
Compare two `Int`s, signed.
*/
compare :: proc(a, b: ^Int) -> Comparison_Flag {
if !is_initialized(a) { return .Uninitialized; }
if !is_initialized(b) { return .Uninitialized; }
/* Compare based on sign */
if a.sign != b.sign {
return .Less_Than if is_negative(a) else .Greater_Than;
}
x, y := a, b;
/* If negative, compare in the opposite direction */
if is_neg(a) {
x, y = b, a;
}
return cmp_mag(x, y);
}
cmp :: compare;
/*
Compare the magnitude of two `Int`s, unsigned.
*/
compare_magnitude :: proc(a, b: ^Int) -> Comparison_Flag {
if !is_initialized(a) { return .Uninitialized; }
if !is_initialized(b) { return .Uninitialized; }
/* Compare based on used digits */
if a.used != b.used {
return .Greater_Than if a.used > b.used else .Less_Than;
}
/* Same number of used digits, compare based on their value */
for n := a.used - 1; n >= 0; n -= 1 {
if a.digit[n] != b.digit[n] {
return .Greater_Than if a.digit[n] > b.digit[n] else .Less_Than;
}
}
return .Equal;
}
cmp_mag :: compare_magnitude;
/*
Compare an `Int` to an unsigned number upto the size of the backing type.
*/
compare_digit :: proc(a: ^Int, u: DIGIT) -> Comparison_Flag {
if !is_initialized(a) { return .Uninitialized; }
/* Compare based on sign */
if is_neg(a) {
return .Less_Than;
}
/* Compare based on magnitude */
if a.used > 1 {
return .Greater_Than;
}
/* Compare the only digit in `a` to `u`. */
if a.digit[0] != u {
return .Greater_Than if a.digit[0] > u else .Less_Than;
}
return .Equal;
}