Integrated bigint. Mostly works, but getting "cast" error for make Test.

1 files changed, 424 insertions, 0 deletions

diff --git a/src/main/stanza/bigint.stanza b/src/main/stanza/bigint.stanza
new file mode 100644
index 00000000..f5122323
--- /dev/null
+++ b/src/main/stanza/bigint.stanza
@@ -0,0 +1,424 @@
+defpackage bigint :
+   import core
+   import verse
+
+
+public defn to-hex (num: Int) -> String :
+  val buf = StringBuffer()
+  print(buf, "0x")
+  var rems = List<Int>()
+  var x : Int = num
+  defn* loop () :
+    rems = List(x % 16, rems)
+    x = x / 16
+    if x > 0: loop()
+  loop()
+  for r in rems do :
+    print(buf, to-char((to-int('0') + r) when r < 10 else (to-int('a') + r - 10)))
+  to-string(buf)
+
+
+val val-n-bytes = 4
+val val-n-bits = val-n-bytes * 8
+val val-all-ones = -1 ;; TODO
+val val-n-half-bits = 32 / 2
+defn val-n-words (nbits: Int) -> Int : 1 + (nbits - 1) / 32 ;val-n-bits
+defn val-n-half-words (nbits: Int) -> Int : (1 + (nbits - 1)) / val-n-half-bits
+defn val-top-bit (v: Int) -> Int : v >> (32 - 1)
+defn val-n-full-words (nbits: Int) -> Int : nbits / 32
+defn val-n-word-bits (nbits: Int) -> Int : nbits % 32
+defn mask-val (n: Int) : val-all-ones >> (32 - n)
+
+public defclass BigInt <: Gettable & Settable & Lengthable
+public defmulti to-bin (b: BigInt) -> String
+public defmulti num-words (b: BigInt) -> Int
+public defmulti dat (b: BigInt) -> Array<Int>
+public defn BigInt (len: Int) :
+  val d = Array<Int>(val-n-words(len))
+  new BigInt :
+     defmethod get (this, i: Int) -> Int : d[i]
+     defmethod set (this, i: Int, x: Int) : d[i] = x; & (-1) ;val-all-ones
+     defmethod dat (this) -> Array<Int> : d
+     defmethod length (this) -> Int : len
+     defmethod num-words (this) -> Int : length(d)
+     defmethod to-bin (this) -> String :
+       string-join $
+       generate<Char> :
+         defn* loop (pos:Int) :
+           if (pos >= 0) :
+             yield(if (d[pos / 32] >> (pos % 32))&1 == 1: '1' else: '0')
+             loop(pos - 1)
+         loop(len - 1)
+     defmethod to-string (this) : 
+       var s = ""
+       for x in d do : s = string-join([x s])
+       s
+     ;defmethod to-string (this) : string-join(["BigInt<" len ">(" to-bin(this) ")"])
+
+defmethod print (o:OutputStream, b:BigInt) :
+  print(o, to-string(b))
+
+defn assert (cond:True|False, msg:String) : if not cond: error(msg)
+
+defn assert (cond:True|False) : assert(cond, "failure")
+
+;; loop macro with starting values
+
+;; defn map-range (s:Int, e:Int, f: (Int) -> False) :
+;;   for i in s to e do : f(i)
+
+defn map! (f: (Int, Int) -> Int, d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+  ;; assert(length(d) == length(s0) and length(s0) == length(s1),
+  ;;        string-join(["LENS != " num-words(d) " " num-words(s0) " " num-words(s1)]))
+  for i in 0 to num-words(d) do :
+    d[i] = f(s0[i], s1[i])
+  d
+
+defn mask! (d: BigInt) -> BigInt :
+  val n-full-words = val-n-full-words(length(d))
+  val n-word-bits  = val-n-word-bits(length(d))
+  for i in 0 to n-full-words do :
+    d[i] = val-all-ones
+  for i in n-full-words to num-words(d) do :
+    d[i] = 0
+  if n-word-bits > 0 :
+    d[n-full-words] = mask-val(n-word-bits)
+  d
+
+defn trim! (d:BigInt) -> BigInt :
+  val n-full-words = val-n-full-words(length(d))
+  val n-word-bits  = val-n-word-bits(length(d))
+  if n-word-bits > 0 :
+    d[n-full-words] = d[n-full-words] & mask-val(n-word-bits)
+  d
+
+defn extract! (d:BigInt, s0:BigInt, e:Int, s:Int) -> BigInt : trim!(rsh!(d, s0, s))
+
+public defn bits (s0:BigInt, e:Int, s:Int) -> BigInt : extract!(BigInt(e - s + 1), s0, e, s)
+
+public defn bit (s0:BigInt, s:Int) -> Int :
+  val wi = s / 32 ;val-n-bits
+  val bi = s % 32 ;val-n-bits
+  s0[wi] >> bi
+
+defn inject! (d:BigInt, s0:BigInt, f:BigInt, start:Int) -> BigInt :
+  val bw          = length(f)
+  val msk         = mask!(BigInt(bw))
+  val msk-lsh     = msk << start
+  val inv-msk-lsh = bit-invert!(msk-lsh, msk-lsh)
+  val f-lsh   = f << start
+  (s0 & inv-msk-lsh) | f-lsh
+
+defn map! (f: (Int) -> Int, d: BigInt, s0: BigInt) -> BigInt :
+  ;; assert(length(d) == length(s0), string-join(["LENS != " num-words(d) " " num-words(s0)]))
+  for i in 0 to num-words(d) do :
+    d[i] = f(s0[i])
+  d
+
+defn fill! (f: (Int) -> Int, b: BigInt) :
+  for i in 0 to num-words(b) do :
+    b[i] = f(i)
+  b
+
+public defmethod equal? (x:BigInt, y:BigInt) -> True|False :
+  if num-words(x) != num-words(y) :
+    false
+  else :
+    var eq? = true
+    for i in 0 to num-words(x) do :
+      val e = (x[i] == y[i])
+      if not e : println-all(["NOT-EQUAL " x " AND " y " i=" i " X[i] " x[i] " Y[i] " y[i]])
+      eq? = eq? and e
+    eq?
+
+public defn less?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False :
+  sub!(diff, x, y)
+  val-top-bit(diff[num-words(diff) - 1]) == 1
+
+public defn less? (x:BigInt, y:BigInt) -> True|False : op(less?!, x, y)
+
+public defn less-eq?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False :
+  sub!(diff, y, x)
+  val-top-bit(diff[num-words(diff) - 1]) == 0
+
+public defn less-eq? (x:BigInt, y:BigInt) -> True|False : op(less-eq?!, x, y)
+
+public defn greater?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False : not less-eq?!(diff, x, y)
+
+public defn greater? (x:BigInt, y:BigInt) -> True|False : op(greater?!, x, y)
+
+public defn greater-eq?! (diff:BigInt, x:BigInt, y:BigInt) -> True|False : not less?!(diff, x, y)
+
+public defn greater-eq? (x:BigInt, y:BigInt) -> True|False : op(greater-eq?!, x, y)
+
+public defn BigIntLit (f: (Int) -> Int, len: Int) :
+  fill!(f, BigInt(len))
+
+;val hex-nibbles = "0123456789abcdef"
+
+defn as-digit (c: Char) -> Int :
+  var index = 0
+  for (x in "0123456789abcdef", i in 0 to false) do :
+     if x == c: index = i
+  index
+  ;val x = index-of(hex-nibbles, c)
+  ;x as Int
+
+public defn BigIntLit (s: String) : BigIntLit(s, -1)
+
+;public defn BigIntLit (i: Int) : BigIntLit(string-join(["x" to-hex(to-string(i))]))
+
+public defn BigIntLit (s: String, w:Int) :
+  val hd = s[0]
+  val base = if hd == 'b': 1 else if hd == 'x': 4 else: 'd'
+  val shamt   = if base == 'b': 1 else if base == 'x': 4 else: 4
+  val digits  = if base == 'b': substring(s, 1) else if base == 'x': substring(s, 1) else: s
+  val len     = if w == -1: length(digits) * shamt else: w
+  println-all(["STRING " s])
+  println-all(["BASE " base " SHAMT " shamt " DIGITS " digits " LEN " len])
+  var lit     = BigInt(len)
+  for i in 0 to num-words(lit) do :
+    lit[i] = 0
+
+
+  switch {base == _} :
+    'd' :
+      for i in 0 to length(digits) do :
+        val y = shift-left(lit,3) + shift-left(lit,1)
+        val x = to-int(substring(digits,i,i + 1))
+        val z = BigInt(length(y))
+        z[0] = x
+        lit = y + z
+        println-all(["DIGIT = " x])
+        println-all(["Before LIT = " lit])
+
+        ;println-all(["After LIT = " lit])
+      ;println-all(["RES = " to-bin(lit)])
+      ;println-all(["RES = " lit])
+      lit
+    else : 
+      for i in 0 to length(digits) do :
+        val off   = (length(digits) - 1 - i) * shamt
+        val wi    = off / 32;val-n-bits
+        val bi    = off % 32;val-n-bits
+        ;println-all(["OFF " off " wi " wi " bi " bi " digits " digits])
+        lit[wi] = lit[wi] | ((as-digit(digits[i]) & 1) << bi)
+        ;println-all([" lit[wi] " lit[wi] " => " lit])
+      ;println-all(["RES = " to-bin(lit)])
+      ;println-all(["RES = " lit])
+      lit
+
+public defn sizeof (in: Int) -> Int :
+  max(1, ceil-log2(in + 1))
+
+public defn BigIntLit (x: Int, w: Int) :
+  fill!({_ + x}, BigInt(w))
+
+public defn BigIntLit (x: Int) : BigIntLit(x, sizeof(x))
+
+defn op (f:(BigInt, BigInt, BigInt) -> True|False, x:BigInt, y:BigInt) -> True|False :
+  f(BigInt(max(length(x), length(y))), x, y)
+
+defn op (f:(BigInt, BigInt, BigInt) -> BigInt, x:BigInt, y:BigInt) -> BigInt :
+  f(BigInt(max(length(x), length(y))), x, y)
+
+defn op (f:(BigInt, BigInt) -> BigInt, x:BigInt) -> BigInt :
+  f(BigInt(length(x)), x)
+
+defn int (x: True|False) -> Int : if x : 1 else : 0
+
+public defn plus! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+  var carry = false
+  for i in 0 to num-words(d) do :
+    d[i] = s0[i] + s1[i] + int(carry)
+    carry = ((s0[i] + s1[i]) < s0[i]) or (d[i] < int(carry))
+  d
+
+public defn plus (x:BigInt, y:BigInt) -> BigInt : op(plus!, x, y)
+
+public defn cat! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+  lsh!(d, s0, length(s1))
+  ;; println-all(["  LSH! " s0 " " length(s1) " => " d])
+  for i in 0 to num-words(s1) do :
+    d[i] = d[i] | s1[i]
+  d
+
+public defn cat (x:BigInt, y:BigInt) -> BigInt :
+  ;; println-all(["> CAT " x " " y])
+  val res = cat!(BigInt(length(x) + length(y)), x, y)
+  ;; println-all(["< CAT " x " " y " => " res])
+  res
+
+public defn cat (args:Streamable<BigInt>) -> BigInt :
+  reduce(cat, args)
+
+public defn sub! (d: BigInt, s0: BigInt, s1: BigInt) -> BigInt :
+  var borrow = false
+  for i in 0 to num-words(d) do :
+    d[i] = s0[i] - s1[i] + int(borrow)
+    borrow = (s0[i] < (s0[i] - s1[i])) or (s0[i] - s1[i]) < d[i]
+  d
+
+public defn sub (x:BigInt, y:BigInt) -> BigInt : op(sub!, x, y)
+
+public defn neg! (d: BigInt, s0: BigInt) -> BigInt :
+  var borrow = false
+  for i in 0 to num-words(d) do :
+    d[i] = neg(s0[i]) - int(borrow)
+    borrow = (s0[i] > 0) or (d[i] > 0)
+  d
+
+public defn neg (x:BigInt) -> BigInt : op(neg!, x)
+
+public defn rsha! (d:BigInt, s0:BigInt, amount:Int) -> BigInt :
+  val w = length(s0)
+  val nw = num-words(d)
+  val n-shift-bits     = amount % 32;val-n-bits
+  val n-shift-words    = amount / 32;val-n-bits
+  val n-rev-shift-bits = 32 - n-shift-bits;val-n-bits
+  val is-zero-carry    = n-shift-bits == 0
+  val msb              = s0[nw - 1] >> (w - nw * 32 - 1);val-n-bits
+  var carry = 0;
+
+  if msb == 0 :
+    for i in 0 to n-shift-words do :
+      d[nw - i - 1] = 0
+
+  defn* loopy (i:Int) :
+    if i >= n-shift-words :
+      val x = s0[i]
+      d[i - n-shift-words] = (x >> n-shift-bits) | carry
+      carry                = if is-zero-carry: 0 else: x << n-rev-shift-bits
+      loopy(i - 1)
+  loopy(nw - 1)
+
+  if msb != 0 :
+    val boundary = (w - amount)
+
+    defn* loop (i:Int) :
+      if i >= 0 :
+        val idx = i * val-n-bits
+        if idx  > boundary :
+          d[i] = val-all-ones
+          loop(i - 1)
+        else :
+          d[i] = d[i] | (val-all-ones << (boundary - idx))
+          d[nw - 1] = d[nw - 1] & (val-all-ones >> ((nw - 1) * val-n-bits - w))
+    loop(nw - 1)
+  d
+
+public defn signed-shift-right (b:BigInt, n:Int) -> BigInt : rsha!(BigInt(length(b)), b, n)
+
+public defn rsh! (d:BigInt, s0:BigInt, amount:Int) -> BigInt :
+  val nw               = num-words(d)
+  var carry            = 0
+  val n-shift-bits     = amount % val-n-bits
+  val n-shift-words    = amount / val-n-bits
+  val n-rev-shift-bits = val-n-bits - n-shift-bits
+  val is-zero-carry    = n-shift-bits == 0
+  for i in 0 to n-shift-words do :
+    d[nw - i - 1] = 0
+  defn* loop (i:Int) :
+    if i >= n-shift-words :
+      val x = s0[i]
+      d[i - n-shift-words] = (x >> n-shift-bits) | carry
+      carry                = if is-zero-carry: 0 else: x << n-rev-shift-bits
+      loop(i - 1)
+  loop(nw - 1)
+  d
+
+public defn shift-right (b:BigInt, n:Int) -> BigInt : rsh!(BigInt(length(b)), b, n)
+
+public defn lsh! (d:BigInt, s0:BigInt, amount:Int) :
+  ;; println-all(["LSH " s0 " AMOUNT " amount " INTO BIGINT<" length(d) ">"])
+  val n-shift-bits     = amount % val-n-bits
+  val n-shift-words    = amount / val-n-bits
+  val n-rev-shift-bits = val-n-bits - n-shift-bits
+  val is-zero-carry    = n-shift-bits == 0
+  for i in 0 to num-words(d) do :
+    d[i] = 0
+  var carry = 0;
+  ;; println-all(["LSH AMOUNT " amount " VNB " val-n-bits " NSB " n-shift-bits " NSW " n-shift-words " NRSB " n-rev-shift-bits])
+  for i in 0 to (num-words(d) - n-shift-words) do :
+    val x = if i >= num-words(s0) : 0 else: s0[i]
+    ;; println-all(["  SHIFTING " (i + n-shift-words) " VAL " (x << n-shift-bits)])
+    d[i + n-shift-words] = (x << n-shift-bits) | carry
+    carry                = if is-zero-carry: 0 else: x >> n-rev-shift-bits
+  d
+
+public defn shift-left (b:BigInt, n:Int) -> BigInt : lsh!(BigInt(length(b) + n), b, n)
+
+;; defn mul! (d: BigInt, s0: BigInt, s1: BigInt, nb0: Int, nb1: Int) -> BigInt :
+;;   ;; Adapted from Hacker's Delight, from Knuth
+;;   var nbd = nb0 + nb1
+;;   for i in 0 to val-n-words(nbd) do :
+;;     d[i] = 0
+;; 
+;;   half-val-t* w = reinterpret-cast<half-val-t*>(d);
+;;   half-val-t* u = reinterpret-cast<half-val-t*>(s0);
+;;   half-val-t* v = reinterpret-cast<half-val-t*>(s1);
+;;   val m = val-n-half-words(nb0)
+;;   val n = val-n-half-words(nb1)
+;;   val p = val-n-half-words(nbd)
+;; 
+;;   for j in 0 to n do :
+;;     var k = 0
+;;     for i in 0 to min(m, p - j) do :
+;;       val t = u[i] * v[j] + w[i + j] + k
+;;       w[i + j] = t
+;;       k = t >> val-n-half-bits()
+;;     if ((j + m) < p)
+;;       w[j + m] = k
+;;   d
+
+public defn bit-xor! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-xor, d, s0, s1)
+
+public defn bit-xor (b0:BigInt, b1:BigInt) : op(bit-xor!, b0, b1)
+
+public defn bit-or! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-or, d, s0, s1)
+
+public defn bit-or (b0:BigInt, b1:BigInt) : op(bit-or!, b0, b1)
+
+public defn bit-and! (d: BigInt, s0: BigInt, s1: BigInt) : map!(bit-and, d, s0, s1)
+
+public defn bit-and (b0:BigInt, b1:BigInt) : op(bit-and!, b0, b1)
+
+public defn bit-invert! (d: BigInt, s0: BigInt) : map!(bit-xor, d, s0, mask!(d))
+
+public defn bit-invert (b0:BigInt) : op(bit-invert!, b0)
+
+defn check (msg:String, x:BigInt, e:BigInt) :
+  println-all([msg " " x " E " e " ? " x == e])
+
+defn check (msg:String, x:BigInt) :
+  println-all([msg " " x])
+
+;; check("Ba ", BigIntLit({ _ },   1))
+;; check("Bb ", BigIntLit({ _ },  16))
+;; check("Bc ", BigIntLit({ _ },  32))
+;; check("Bd ", BigIntLit({ _ },  48))
+;; check("Be ", BigIntLit({ _ },  64))
+;; check("Bf ", BigIntLit({ _ },  65))
+;; check("B1 ", BigIntLit(1, 8))
+;; check("B2 ", BigIntLit(2, 8))
+;; check("B+ ", BigIntLit(3, 8) + BigIntLit(5, 8), BigIntLit(3 + 5, 8))
+;; check("B- ", BigIntLit(5, 8) + BigIntLit(3, 8), BigIntLit(5 + 3, 8))
+;; check("B| ", BigIntLit(5, 8) | BigIntLit(9, 8), BigIntLit(5 | 9, 8))
+;; check("B& ", BigIntLit(5, 8) & BigIntLit(9, 8), BigIntLit(5 & 9, 8))
+;; check("B^ ", BigIntLit(5, 8) ^ BigIntLit(9, 8), BigIntLit(5 ^ 9, 8))
+;; check("B< ", BigIntLit(5, 8) << 1, BigIntLit(5 << 1, 9))
+;; check("B< ", BigIntLit(5, 3) << 10, BigIntLit(5 << 1, 13))
+;; check("B< ", BigIntLit(5, 3) << 32, BigIntLit(5 << 1, 38))
+;; check("B< ", BigIntLit("b1010") << 1, BigIntLit(10 << 1, 5))
+;check("S1 ", BigIntLit("xfafa") << 16, BigIntLit("xfafa0000", 32))
+;;; check("B< ", BigIntLit(5, 3) << 64, BigIntLit(5 << 1, 67))
+;;; check("BN ", neg(BigIntLit(2, 8)), BigIntLit(-2, 8))
+;check("S2 ", BigIntLit("b11111010") << 8, BigIntLit("b1111101000000000", 16))
+;check("C1 ", cat(BigIntLit("b11111010", 8), BigIntLit("b10111100", 8)), BigIntLit("b1111101010111100", 16))
+;check("C3 ", cat(cat(BigIntLit("b1111"), BigIntLit("b1010")), cat(BigIntLit("b1011"), BigIntLit("b1100"))), BigIntLit("b1111101010111100", 16))
+;check("C4 ", cat([BigIntLit("b1111"), BigIntLit("b1010"), BigIntLit("b1011"), BigIntLit("b1100")]), BigIntLit("b1111101010111100", 16))
+;check("C5 ", BigIntLit("b101111001"), BigIntLit("b101111001"))
+;check("C6 ", cat(BigIntLit("b1"), BigIntLit("b01111001")), BigIntLit("b101111001"))
+;check("C7 ", cat(BigIntLit("b11101"), BigIntLit("b101111001")), BigIntLit("b11101101111001"))
+;check("C8 ", cat([BigIntLit("b11"), BigIntLit("b101"), BigIntLit("b1011"), BigIntLit("b11001")]), BigIntLit("b11101101111001"))
+;check("C0 ", bits(BigIntLit("b11101101111001"), 10, 1), BigIntLit("b0110111100"))