From: Evan Shaw Date: Thu, 22 Apr 2010 23:57:29 +0000 (-0700) Subject: big: Create type nat X-Git-Tag: weekly.2010-04-27~48 X-Git-Url: http://www.git.cypherpunks.su/?a=commitdiff_plain;h=841a32dd5ec836e171d04604710d88a6b8e59467;p=gostls13.git big: Create type nat Changed most of the functions in nat.go to methods on nat. R=gri CC=golang-dev https://golang.org/cl/976041 --- diff --git a/src/pkg/big/arith_test.go b/src/pkg/big/arith_test.go index b0f6bf6f1f..49908e342d 100644 --- a/src/pkg/big/arith_test.go +++ b/src/pkg/big/arith_test.go @@ -52,7 +52,7 @@ func TestFunWW(t *testing.T) { } -func addr(x []Word) *Word { +func addr(x nat) *Word { if len(x) == 0 { return nil } @@ -62,26 +62,26 @@ func addr(x []Word) *Word { type funVV func(z, x, y *Word, n int) (c Word) type argVV struct { - z, x, y []Word + z, x, y nat c Word } var sumVV = []argVV{ argVV{}, - argVV{[]Word{0}, []Word{0}, []Word{0}, 0}, - argVV{[]Word{1}, []Word{1}, []Word{0}, 0}, - argVV{[]Word{0}, []Word{_M}, []Word{1}, 1}, - argVV{[]Word{80235}, []Word{12345}, []Word{67890}, 0}, - argVV{[]Word{_M - 1}, []Word{_M}, []Word{_M}, 1}, - argVV{[]Word{0, 0, 0, 0}, []Word{_M, _M, _M, _M}, []Word{1, 0, 0, 0}, 1}, - argVV{[]Word{0, 0, 0, _M}, []Word{_M, _M, _M, _M - 1}, []Word{1, 0, 0, 0}, 0}, - argVV{[]Word{0, 0, 0, 0}, []Word{_M, 0, _M, 0}, []Word{1, _M, 0, _M}, 1}, + argVV{nat{0}, nat{0}, nat{0}, 0}, + argVV{nat{1}, nat{1}, nat{0}, 0}, + argVV{nat{0}, nat{_M}, nat{1}, 1}, + argVV{nat{80235}, nat{12345}, nat{67890}, 0}, + argVV{nat{_M - 1}, nat{_M}, nat{_M}, 1}, + argVV{nat{0, 0, 0, 0}, nat{_M, _M, _M, _M}, nat{1, 0, 0, 0}, 1}, + argVV{nat{0, 0, 0, _M}, nat{_M, _M, _M, _M - 1}, nat{1, 0, 0, 0}, 0}, + argVV{nat{0, 0, 0, 0}, nat{_M, 0, _M, 0}, nat{1, _M, 0, _M}, 1}, } func testFunVV(t *testing.T, msg string, f funVV, a argVV) { n := len(a.z) - z := make([]Word, n) + z := make(nat, n) c := f(addr(z), addr(a.x), addr(a.y), n) for i, zi := range z { if zi != a.z[i] { @@ -118,39 +118,39 @@ func TestFunVV(t *testing.T) { type funVW func(z, x *Word, y Word, n int) (c Word) type argVW struct { - z, x []Word + z, x nat y Word c Word } var sumVW = []argVW{ argVW{}, - argVW{[]Word{0}, []Word{0}, 0, 0}, - argVW{[]Word{1}, []Word{0}, 1, 0}, - argVW{[]Word{1}, []Word{1}, 0, 0}, - argVW{[]Word{0}, []Word{_M}, 1, 1}, - argVW{[]Word{0, 0, 0, 0}, []Word{_M, _M, _M, _M}, 1, 1}, + argVW{nat{0}, nat{0}, 0, 0}, + argVW{nat{1}, nat{0}, 1, 0}, + argVW{nat{1}, nat{1}, 0, 0}, + argVW{nat{0}, nat{_M}, 1, 1}, + argVW{nat{0, 0, 0, 0}, nat{_M, _M, _M, _M}, 1, 1}, } var prodVW = []argVW{ argVW{}, - argVW{[]Word{0}, []Word{0}, 0, 0}, - argVW{[]Word{0}, []Word{_M}, 0, 0}, - argVW{[]Word{0}, []Word{0}, _M, 0}, - argVW{[]Word{1}, []Word{1}, 1, 0}, - argVW{[]Word{22793}, []Word{991}, 23, 0}, - argVW{[]Word{0, 0, 0, 22793}, []Word{0, 0, 0, 991}, 23, 0}, - argVW{[]Word{0, 0, 0, 0}, []Word{7893475, 7395495, 798547395, 68943}, 0, 0}, - argVW{[]Word{0, 0, 0, 0}, []Word{0, 0, 0, 0}, 894375984, 0}, - argVW{[]Word{_M << 1 & _M}, []Word{_M}, 1 << 1, _M >> (_W - 1)}, - argVW{[]Word{_M << 7 & _M}, []Word{_M}, 1 << 7, _M >> (_W - 7)}, - argVW{[]Word{_M << 7 & _M, _M, _M, _M}, []Word{_M, _M, _M, _M}, 1 << 7, _M >> (_W - 7)}, + argVW{nat{0}, nat{0}, 0, 0}, + argVW{nat{0}, nat{_M}, 0, 0}, + argVW{nat{0}, nat{0}, _M, 0}, + argVW{nat{1}, nat{1}, 1, 0}, + argVW{nat{22793}, nat{991}, 23, 0}, + argVW{nat{0, 0, 0, 22793}, nat{0, 0, 0, 991}, 23, 0}, + argVW{nat{0, 0, 0, 0}, nat{7893475, 7395495, 798547395, 68943}, 0, 0}, + argVW{nat{0, 0, 0, 0}, nat{0, 0, 0, 0}, 894375984, 0}, + argVW{nat{_M << 1 & _M}, nat{_M}, 1 << 1, _M >> (_W - 1)}, + argVW{nat{_M << 7 & _M}, nat{_M}, 1 << 7, _M >> (_W - 7)}, + argVW{nat{_M << 7 & _M, _M, _M, _M}, nat{_M, _M, _M, _M}, 1 << 7, _M >> (_W - 7)}, } func testFunVW(t *testing.T, msg string, f funVW, a argVW) { n := len(a.z) - z := make([]Word, n) + z := make(nat, n) c := f(addr(z), addr(a.x), a.y, n) for i, zi := range z { if zi != a.z[i] { @@ -179,41 +179,41 @@ func TestFunVW(t *testing.T) { type funVWW func(z, x *Word, y, r Word, n int) (c Word) type argVWW struct { - z, x []Word + z, x nat y, r Word c Word } var prodVWW = []argVWW{ argVWW{}, - argVWW{[]Word{0}, []Word{0}, 0, 0, 0}, - argVWW{[]Word{991}, []Word{0}, 0, 991, 0}, - argVWW{[]Word{0}, []Word{_M}, 0, 0, 0}, - argVWW{[]Word{991}, []Word{_M}, 0, 991, 0}, - argVWW{[]Word{0}, []Word{0}, _M, 0, 0}, - argVWW{[]Word{991}, []Word{0}, _M, 991, 0}, - argVWW{[]Word{1}, []Word{1}, 1, 0, 0}, - argVWW{[]Word{992}, []Word{1}, 1, 991, 0}, - argVWW{[]Word{22793}, []Word{991}, 23, 0, 0}, - argVWW{[]Word{22800}, []Word{991}, 23, 7, 0}, - argVWW{[]Word{0, 0, 0, 22793}, []Word{0, 0, 0, 991}, 23, 0, 0}, - argVWW{[]Word{7, 0, 0, 22793}, []Word{0, 0, 0, 991}, 23, 7, 0}, - argVWW{[]Word{0, 0, 0, 0}, []Word{7893475, 7395495, 798547395, 68943}, 0, 0, 0}, - argVWW{[]Word{991, 0, 0, 0}, []Word{7893475, 7395495, 798547395, 68943}, 0, 991, 0}, - argVWW{[]Word{0, 0, 0, 0}, []Word{0, 0, 0, 0}, 894375984, 0, 0}, - argVWW{[]Word{991, 0, 0, 0}, []Word{0, 0, 0, 0}, 894375984, 991, 0}, - argVWW{[]Word{_M << 1 & _M}, []Word{_M}, 1 << 1, 0, _M >> (_W - 1)}, - argVWW{[]Word{_M<<1&_M + 1}, []Word{_M}, 1 << 1, 1, _M >> (_W - 1)}, - argVWW{[]Word{_M << 7 & _M}, []Word{_M}, 1 << 7, 0, _M >> (_W - 7)}, - argVWW{[]Word{_M<<7&_M + 1<<6}, []Word{_M}, 1 << 7, 1 << 6, _M >> (_W - 7)}, - argVWW{[]Word{_M << 7 & _M, _M, _M, _M}, []Word{_M, _M, _M, _M}, 1 << 7, 0, _M >> (_W - 7)}, - argVWW{[]Word{_M<<7&_M + 1<<6, _M, _M, _M}, []Word{_M, _M, _M, _M}, 1 << 7, 1 << 6, _M >> (_W - 7)}, + argVWW{nat{0}, nat{0}, 0, 0, 0}, + argVWW{nat{991}, nat{0}, 0, 991, 0}, + argVWW{nat{0}, nat{_M}, 0, 0, 0}, + argVWW{nat{991}, nat{_M}, 0, 991, 0}, + argVWW{nat{0}, nat{0}, _M, 0, 0}, + argVWW{nat{991}, nat{0}, _M, 991, 0}, + argVWW{nat{1}, nat{1}, 1, 0, 0}, + argVWW{nat{992}, nat{1}, 1, 991, 0}, + argVWW{nat{22793}, nat{991}, 23, 0, 0}, + argVWW{nat{22800}, nat{991}, 23, 7, 0}, + argVWW{nat{0, 0, 0, 22793}, nat{0, 0, 0, 991}, 23, 0, 0}, + argVWW{nat{7, 0, 0, 22793}, nat{0, 0, 0, 991}, 23, 7, 0}, + argVWW{nat{0, 0, 0, 0}, nat{7893475, 7395495, 798547395, 68943}, 0, 0, 0}, + argVWW{nat{991, 0, 0, 0}, nat{7893475, 7395495, 798547395, 68943}, 0, 991, 0}, + argVWW{nat{0, 0, 0, 0}, nat{0, 0, 0, 0}, 894375984, 0, 0}, + argVWW{nat{991, 0, 0, 0}, nat{0, 0, 0, 0}, 894375984, 991, 0}, + argVWW{nat{_M << 1 & _M}, nat{_M}, 1 << 1, 0, _M >> (_W - 1)}, + argVWW{nat{_M<<1&_M + 1}, nat{_M}, 1 << 1, 1, _M >> (_W - 1)}, + argVWW{nat{_M << 7 & _M}, nat{_M}, 1 << 7, 0, _M >> (_W - 7)}, + argVWW{nat{_M<<7&_M + 1<<6}, nat{_M}, 1 << 7, 1 << 6, _M >> (_W - 7)}, + argVWW{nat{_M << 7 & _M, _M, _M, _M}, nat{_M, _M, _M, _M}, 1 << 7, 0, _M >> (_W - 7)}, + argVWW{nat{_M<<7&_M + 1<<6, _M, _M, _M}, nat{_M, _M, _M, _M}, 1 << 7, 1 << 6, _M >> (_W - 7)}, } func testFunVWW(t *testing.T, msg string, f funVWW, a argVWW) { n := len(a.z) - z := make([]Word, n) + z := make(nat, n) c := f(addr(z), addr(a.x), a.y, a.r, n) for i, zi := range z { if zi != a.z[i] { @@ -232,16 +232,16 @@ func testFunVWW(t *testing.T, msg string, f funVWW, a argVWW) { type funWVW func(z *Word, xn Word, x *Word, y Word, n int) (r Word) type argWVW struct { - z []Word + z nat xn Word - x []Word + x nat y Word r Word } func testFunWVW(t *testing.T, msg string, f funWVW, a argWVW) { n := len(a.z) - z := make([]Word, n) + z := make(nat, n) r := f(addr(z), a.xn, addr(a.x), a.y, n) for i, zi := range z { if zi != a.z[i] { diff --git a/src/pkg/big/int.go b/src/pkg/big/int.go index ca94c5a427..6b570a07d6 100644 --- a/src/pkg/big/int.go +++ b/src/pkg/big/int.go @@ -9,8 +9,8 @@ package big // An Int represents a signed multi-precision integer. // The zero value for an Int represents the value 0. type Int struct { - neg bool // sign - abs []Word // absolute value of the integer + neg bool // sign + abs nat // absolute value of the integer } @@ -21,7 +21,7 @@ func (z *Int) New(x int64) *Int { z.neg = true x = -x } - z.abs = newN(z.abs, uint64(x)) + z.abs = z.abs.new(uint64(x)) return z } @@ -33,7 +33,7 @@ func NewInt(x int64) *Int { return new(Int).New(x) } // Set sets z to x. func (z *Int) Set(x *Int) *Int { z.neg = x.neg - z.abs = setN(z.abs, x.abs) + z.abs = z.abs.set(x.abs) return z } @@ -44,16 +44,16 @@ func (z *Int) Add(x, y *Int) *Int { // x + y == x + y // (-x) + (-y) == -(x + y) z.neg = x.neg - z.abs = addNN(z.abs, x.abs, y.abs) + z.abs = z.abs.add(x.abs, y.abs) } else { // x + (-y) == x - y == -(y - x) // (-x) + y == y - x == -(x - y) - if cmpNN(x.abs, y.abs) >= 0 { + if x.abs.cmp(y.abs) >= 0 { z.neg = x.neg - z.abs = subNN(z.abs, x.abs, y.abs) + z.abs = z.abs.sub(x.abs, y.abs) } else { z.neg = !x.neg - z.abs = subNN(z.abs, y.abs, x.abs) + z.abs = z.abs.sub(y.abs, x.abs) } } if len(z.abs) == 0 { @@ -69,16 +69,16 @@ func (z *Int) Sub(x, y *Int) *Int { // x - (-y) == x + y // (-x) - y == -(x + y) z.neg = x.neg - z.abs = addNN(z.abs, x.abs, y.abs) + z.abs = z.abs.add(x.abs, y.abs) } else { // x - y == x - y == -(y - x) // (-x) - (-y) == y - x == -(x - y) - if cmpNN(x.abs, y.abs) >= 0 { + if x.abs.cmp(y.abs) >= 0 { z.neg = x.neg - z.abs = subNN(z.abs, x.abs, y.abs) + z.abs = z.abs.sub(x.abs, y.abs) } else { z.neg = !x.neg - z.abs = subNN(z.abs, y.abs, x.abs) + z.abs = z.abs.sub(y.abs, x.abs) } } if len(z.abs) == 0 { @@ -94,7 +94,7 @@ func (z *Int) Mul(x, y *Int) *Int { // x * (-y) == -(x * y) // (-x) * y == -(x * y) // (-x) * (-y) == x * y - z.abs = mulNN(z.abs, x.abs, y.abs) + z.abs = z.abs.mul(x.abs, y.abs) z.neg = len(z.abs) > 0 && x.neg != y.neg // 0 has no sign return z } @@ -126,14 +126,14 @@ func (z *Int) DivMod(x, y, r *Int) (*Int, *Int) { func div(q, r, x, y *Int) { q.neg = x.neg != y.neg r.neg = x.neg - q.abs, r.abs = divNN(q.abs, r.abs, x.abs, y.abs) + q.abs, r.abs = q.abs.div(r.abs, x.abs, y.abs) return } // Neg computes z = -x. func (z *Int) Neg(x *Int) *Int { - z.abs = setN(z.abs, x.abs) + z.abs = z.abs.set(x.abs) z.neg = len(z.abs) > 0 && !x.neg // 0 has no sign return z } @@ -152,7 +152,7 @@ func (x *Int) Cmp(y *Int) (r int) { // (-x) cmp (-y) == -(x cmp y) switch { case x.neg == y.neg: - r = cmpNN(x.abs, y.abs) + r = x.abs.cmp(y.abs) if x.neg { r = -r } @@ -170,7 +170,7 @@ func (z *Int) String() string { if z.neg { s = "-" } - return s + stringN(z.abs, 10) + return s + z.abs.string(10) } @@ -212,7 +212,7 @@ func (z *Int) SetString(s string, base int) (*Int, bool) { z.neg = false } - z.abs, _, scanned = scanN(z.abs, s, base) + z.abs, _, scanned = z.abs.scan(s, base) if scanned != len(s) { goto Error } @@ -230,7 +230,7 @@ Error: // sets z to that value. func (z *Int) SetBytes(b []byte) *Int { s := int(_S) - z.abs = makeN(z.abs, (len(b)+s-1)/s, false) + z.abs = z.abs.make((len(b)+s-1)/s, false) z.neg = false j := 0 @@ -258,7 +258,7 @@ func (z *Int) SetBytes(b []byte) *Int { z.abs[j] = w } - z.abs = normN(z.abs) + z.abs = z.abs.norm() return z } @@ -306,12 +306,12 @@ func (z *Int) Exp(x, y, m *Int) *Int { return z } - var mWords []Word + var mWords nat if m != nil { mWords = m.abs } - z.abs = expNNN(z.abs, x.abs, y.abs, mWords) + z.abs = z.abs.expNN(x.abs, y.abs, mWords) z.neg = x.neg && y.abs[0]&1 == 1 return z } @@ -379,20 +379,20 @@ func GcdInt(d, x, y, a, b *Int) { // ProbablyPrime performs n Miller-Rabin tests to check whether z is prime. // If it returns true, z is prime with probability 1 - 1/4^n. // If it returns false, z is not prime. -func ProbablyPrime(z *Int, n int) bool { return !z.neg && probablyPrime(z.abs, n) } +func ProbablyPrime(z *Int, n int) bool { return !z.neg && z.abs.probablyPrime(n) } // Lsh sets z = x << n and returns z. func (z *Int) Lsh(x *Int, n uint) *Int { addedWords := int(n) / _W // Don't assign z.abs yet, in case z == x - znew := makeN(z.abs, len(x.abs)+addedWords+1, false) + znew := z.abs.make(len(x.abs)+addedWords+1, false) z.neg = x.neg - shiftLeft(znew[addedWords:], x.abs, n%_W) + znew[addedWords:].shiftLeft(x.abs, n%_W) for i := range znew[0:addedWords] { znew[i] = 0 } - z.abs = normN(znew) + z.abs = znew.norm() return z } @@ -401,9 +401,9 @@ func (z *Int) Lsh(x *Int, n uint) *Int { func (z *Int) Rsh(x *Int, n uint) *Int { removedWords := int(n) / _W // Don't assign z.abs yet, in case z == x - znew := makeN(z.abs, len(x.abs)-removedWords, false) + znew := z.abs.make(len(x.abs)-removedWords, false) z.neg = x.neg - shiftRight(znew, x.abs[removedWords:], n%_W) - z.abs = normN(znew) + znew.shiftRight(x.abs[removedWords:], n%_W) + z.abs = znew.norm() return z } diff --git a/src/pkg/big/int_test.go b/src/pkg/big/int_test.go index 914a631e51..bb42f81856 100644 --- a/src/pkg/big/int_test.go +++ b/src/pkg/big/int_test.go @@ -327,8 +327,8 @@ func TestDivStepD6(t *testing.T) { // See Knuth, Volume 2, section 4.3.1, exercise 21. This code exercises // a code path which only triggers 1 in 10^{-19} cases. - u := &Int{false, []Word{0, 0, 1 + 1<<(_W-1), _M ^ (1 << (_W - 1))}} - v := &Int{false, []Word{5, 2 + 1<<(_W-1), 1 << (_W - 1)}} + u := &Int{false, nat{0, 0, 1 + 1<<(_W-1), _M ^ (1 << (_W - 1))}} + v := &Int{false, nat{5, 2 + 1<<(_W-1), 1 << (_W - 1)}} r := new(Int) q, r := new(Int).DivMod(u, v, r) diff --git a/src/pkg/big/nat.go b/src/pkg/big/nat.go index 8843d43549..2c8f837de6 100644 --- a/src/pkg/big/nat.go +++ b/src/pkg/big/nat.go @@ -34,10 +34,9 @@ import "rand" // always normalized before returning the final result. The normalized // representation of 0 is the empty or nil slice (length = 0). -// TODO(gri) - convert these routines into methods for type 'nat' -// - decide if type 'nat' should be exported +type nat []Word -func normN(z []Word) []Word { +func (z nat) norm() nat { i := len(z) for i > 0 && z[i-1] == 0 { i-- @@ -47,7 +46,7 @@ func normN(z []Word) []Word { } -func makeN(z []Word, m int, clear bool) []Word { +func (z nat) make(m int, clear bool) nat { if cap(z) > m { z = z[0:m] // reuse z - has at least one extra word for a carry, if any if clear { @@ -62,18 +61,18 @@ func makeN(z []Word, m int, clear bool) []Word { if m > c { c = m } - return make([]Word, m, c+1) // +1: extra word for a carry, if any + return make(nat, m, c+1) // +1: extra word for a carry, if any } -func newN(z []Word, x uint64) []Word { +func (z nat) new(x uint64) nat { if x == 0 { - return makeN(z, 0, false) + return z.make(0, false) } // single-digit values if x == uint64(Word(x)) { - z = makeN(z, 1, false) + z = z.make(1, false) z[0] = Word(x) return z } @@ -85,7 +84,7 @@ func newN(z []Word, x uint64) []Word { } // split x into n words - z = makeN(z, n, false) + z = z.make(n, false) for i := 0; i < n; i++ { z[i] = Word(x & _M) x >>= _W @@ -95,8 +94,8 @@ func newN(z []Word, x uint64) []Word { } -func setN(z, x []Word) []Word { - z = makeN(z, len(x), false) +func (z nat) set(x nat) nat { + z = z.make(len(x), false) for i, d := range x { z[i] = d } @@ -104,23 +103,23 @@ func setN(z, x []Word) []Word { } -func addNN(z, x, y []Word) []Word { +func (z nat) add(x, y nat) nat { m := len(x) n := len(y) switch { case m < n: - return addNN(z, y, x) + return z.add(y, x) case m == 0: // n == 0 because m >= n; result is 0 - return makeN(z, 0, false) + return z.make(0, false) case n == 0: // result is x - return setN(z, x) + return z.set(x) } // m > 0 - z = makeN(z, m, false) + z = z.make(m, false) c := addVV(&z[0], &x[0], &y[0], n) if m > n { c = addVW(&z[n], &x[n], c, m-n) @@ -134,7 +133,7 @@ func addNN(z, x, y []Word) []Word { } -func subNN(z, x, y []Word) []Word { +func (z nat) sub(x, y nat) nat { m := len(x) n := len(y) @@ -143,14 +142,14 @@ func subNN(z, x, y []Word) []Word { panic("underflow") case m == 0: // n == 0 because m >= n; result is 0 - return makeN(z, 0, false) + return z.make(0, false) case n == 0: // result is x - return setN(z, x) + return z.set(x) } // m > 0 - z = makeN(z, m, false) + z = z.make(m, false) c := subVV(&z[0], &x[0], &y[0], n) if m > n { c = subVW(&z[n], &x[n], c, m-n) @@ -158,13 +157,13 @@ func subNN(z, x, y []Word) []Word { if c != 0 { panic("underflow") } - z = normN(z) + z = z.norm() return z } -func cmpNN(x, y []Word) (r int) { +func (x nat) cmp(y nat) (r int) { m := len(x) n := len(y) if m != n || m == 0 { @@ -192,14 +191,14 @@ func cmpNN(x, y []Word) (r int) { } -func mulAddNWW(z, x []Word, y, r Word) []Word { +func (z nat) mulAddWW(x nat, y, r Word) nat { m := len(x) if m == 0 || y == 0 { - return newN(z, uint64(r)) // result is r + return z.new(uint64(r)) // result is r } // m > 0 - z = makeN(z, m, false) + z = z.make(m, false) c := mulAddVWW(&z[0], &x[0], y, r, m) if c > 0 { z = z[0 : m+1] @@ -210,81 +209,81 @@ func mulAddNWW(z, x []Word, y, r Word) []Word { } -func mulNN(z, x, y []Word) []Word { +func (z nat) mul(x, y nat) nat { m := len(x) n := len(y) switch { case m < n: - return mulNN(z, y, x) + return z.mul(y, x) case m == 0 || n == 0: - return makeN(z, 0, false) + return z.make(0, false) case n == 1: - return mulAddNWW(z, x, y[0], 0) + return z.mulAddWW(x, y[0], 0) } // m >= n && m > 1 && n > 1 if z == nil || &z[0] == &x[0] || &z[0] == &y[0] { - z = makeN(nil, m+n, true) // z is an alias for x or y - cannot reuse + z = nat(nil).make(m+n, true) // z is an alias for x or y - cannot reuse } else { - z = makeN(z, m+n, true) + z = z.make(m+n, true) } for i := 0; i < n; i++ { if f := y[i]; f != 0 { z[m+i] = addMulVVW(&z[i], &x[0], f, m) } } - z = normN(z) + z = z.norm() return z } // q = (x-r)/y, with 0 <= r < y -func divNW(z, x []Word, y Word) (q []Word, r Word) { +func (z nat) divW(x nat, y Word) (q nat, r Word) { m := len(x) switch { case y == 0: panic("division by zero") case y == 1: - q = setN(z, x) // result is x + q = z.set(x) // result is x return case m == 0: - q = setN(z, nil) // result is 0 + q = z.set(nil) // result is 0 return } // m > 0 - z = makeN(z, m, false) + z = z.make(m, false) r = divWVW(&z[0], 0, &x[0], y, m) - q = normN(z) + q = z.norm() return } -func divNN(z, z2, u, v []Word) (q, r []Word) { +func (z nat) div(z2, u, v nat) (q, r nat) { if len(v) == 0 { - panic("Divide by zero undefined") + panic("division by zero") } - if cmpNN(u, v) < 0 { - q = makeN(z, 0, false) - r = setN(z2, u) + if u.cmp(v) < 0 { + q = z.make(0, false) + r = z2.set(u) return } if len(v) == 1 { var rprime Word - q, rprime = divNW(z, u, v[0]) + q, rprime = z.divW(u, v[0]) if rprime > 0 { - r = makeN(z2, 1, false) + r = z2.make(1, false) r[0] = rprime } else { - r = makeN(z2, 0, false) + r = z2.make(0, false) } return } - q, r = divLargeNN(z, z2, u, v) + q, r = z.divLarge(z2, u, v) return } @@ -294,23 +293,23 @@ func divNN(z, z2, u, v []Word) (q, r []Word) { // Preconditions: // len(v) >= 2 // len(uIn) >= len(v) -func divLargeNN(z, z2, uIn, v []Word) (q, r []Word) { +func (z nat) divLarge(z2, uIn, v nat) (q, r nat) { n := len(v) m := len(uIn) - len(v) - var u []Word + var u nat if z2 == nil || &z2[0] == &uIn[0] { - u = makeN(nil, len(uIn)+1, true) // uIn is an alias for z2 + u = u.make(len(uIn)+1, true) // uIn is an alias for z2 } else { - u = makeN(z2, len(uIn)+1, true) + u = z2.make(len(uIn)+1, true) } - qhatv := make([]Word, len(v)+1) - q = makeN(z, m+1, false) + qhatv := make(nat, len(v)+1) + q = z.make(m+1, false) // D1. shift := uint(leadingZeroBits(v[n-1])) - shiftLeft(v, v, shift) - shiftLeft(u, uIn, shift) + v.shiftLeft(v, shift) + u.shiftLeft(uIn, shift) u[len(uIn)] = uIn[len(uIn)-1] >> (_W - uint(shift)) // D2. @@ -351,10 +350,10 @@ func divLargeNN(z, z2, uIn, v []Word) (q, r []Word) { q[j] = qhat } - q = normN(q) - shiftRight(u, u, shift) - shiftRight(v, v, shift) - r = normN(u) + q = q.norm() + u.shiftRight(u, shift) + v.shiftRight(v, shift) + r = u.norm() return q, r } @@ -372,10 +371,10 @@ func log2(x Word) int { } -// log2N computes the integer binary logarithm of x. +// log2 computes the integer binary logarithm of x. // The result is the integer n for which 2^n <= x < 2^(n+1). // If x == 0, the result is -1. -func log2N(x []Word) int { +func (x nat) log2() int { m := len(x) if m > 0 { return (m-1)*_W + log2(x[m-1]) @@ -410,7 +409,7 @@ func hexValue(ch byte) int { // conversion base. A prefix of ``0x'' or ``0X'' selects base 16; the // ``0'' prefix selects base 8. Otherwise the selected base is 10. // -func scanN(z []Word, s string, base int) ([]Word, int, int) { +func (z nat) scan(s string, base int) (nat, int, int) { // determine base if necessary i, n := 0, len(s) if base == 0 { @@ -436,7 +435,7 @@ func scanN(z []Word, s string, base int) ([]Word, int, int) { for ; i < n; i++ { d := hexValue(s[i]) if 0 <= d && d < base { - z = mulAddNWW(z, z, Word(base), Word(d)) + z = z.mulAddWW(z, Word(base), Word(d)) } else { break } @@ -449,7 +448,7 @@ func scanN(z []Word, s string, base int) ([]Word, int, int) { // string converts x to a string for a given base, with 2 <= base <= 16. // TODO(gri) in the style of the other routines, perhaps this should take // a []byte buffer and return it -func stringN(x []Word, base int) string { +func (x nat) string(base int) string { if base < 2 || 16 < base { panic("illegal base") } @@ -459,17 +458,17 @@ func stringN(x []Word, base int) string { } // allocate buffer for conversion - i := (log2N(x)+1)/log2(Word(base)) + 1 // +1: round up + i := (x.log2()+1)/log2(Word(base)) + 1 // +1: round up s := make([]byte, i) // don't destroy x - q := setN(nil, x) + q := nat(nil).set(x) // convert for len(q) > 0 { i-- var r Word - q, r = divNW(q, q, Word(base)) + q, r = q.divW(q, Word(base)) s[i] = "0123456789abcdef"[r] } @@ -536,40 +535,42 @@ func trailingZeroBits(x Word) int { } -// To avoid losing the top n bits, dst should be sized so that -// len(dst) == len(src) + 1. -func shiftLeft(dst, src []Word, n uint) { - if len(src) == 0 { - return +// To avoid losing the top n bits, z should be sized so that +// len(z) == len(x) + 1. +func (z nat) shiftLeft(x nat, n uint) nat { + if len(x) == 0 { + return x } ñ := _W - n - x := src[len(src)-1] - if len(dst) > len(src) { - dst[len(src)] = x >> ñ + m := x[len(x)-1] + if len(z) > len(x) { + z[len(x)] = m >> ñ } - for i := len(src) - 1; i >= 1; i-- { - y := src[i-1] - dst[i] = x<>ñ - x = y + for i := len(x) - 1; i >= 1; i-- { + y := x[i-1] + z[i] = m<>ñ + m = y } - dst[0] = x << n + z[0] = m << n + return z } -func shiftRight(dst, src []Word, n uint) { - if len(src) == 0 { - return +func (z nat) shiftRight(x nat, n uint) nat { + if len(x) == 0 { + return x } ñ := _W - n - x := src[0] - for i := 0; i < len(src)-1; i++ { - y := src[i+1] - dst[i] = x>>n | y<<ñ - x = y + m := x[0] + for i := 0; i < len(x)-1; i++ { + y := x[i+1] + z[i] = m>>n | y<<ñ + m = y } - dst[len(src)-1] = x >> n + z[len(x)-1] = m >> n + return z } @@ -577,16 +578,17 @@ func shiftRight(dst, src []Word, n uint) { func greaterThan(x1, x2, y1, y2 Word) bool { return x1 > y1 || x1 == y1 && x2 > y2 } -// modNW returns x % d. -func modNW(x []Word, d Word) (r Word) { +// modW returns x % d. +func (x nat) modW(d Word) (r Word) { // TODO(agl): we don't actually need to store the q value. - q := makeN(nil, len(x), false) + var q nat + q = q.make(len(x), false) return divWVW(&q[0], 0, &x[0], d, len(x)) } // powersOfTwoDecompose finds q and k such that q * 1<= 32 { - runModNWTests(t, modNWTests32) + runModWTests(t, modWTests32) } if _W >= 64 { - runModNWTests(t, modNWTests32) + runModWTests(t, modWTests64) } } @@ -269,19 +269,19 @@ func TestTrailingZeroBits(t *testing.T) { } -type expNNNTest struct { +type expNNTest struct { x, y, m string out string } -var expNNNTests = []expNNNTest{ - expNNNTest{"0x8000000000000000", "2", "", "0x40000000000000000000000000000000"}, - expNNNTest{"0x8000000000000000", "2", "6719", "4944"}, - expNNNTest{"0x8000000000000000", "3", "6719", "5447"}, - expNNNTest{"0x8000000000000000", "1000", "6719", "1603"}, - expNNNTest{"0x8000000000000000", "1000000", "6719", "3199"}, - expNNNTest{ +var expNNTests = []expNNTest{ + expNNTest{"0x8000000000000000", "2", "", "0x40000000000000000000000000000000"}, + expNNTest{"0x8000000000000000", "2", "6719", "4944"}, + expNNTest{"0x8000000000000000", "3", "6719", "5447"}, + expNNTest{"0x8000000000000000", "1000", "6719", "1603"}, + expNNTest{"0x8000000000000000", "1000000", "6719", "3199"}, + expNNTest{ "2938462938472983472983659726349017249287491026512746239764525612965293865296239471239874193284792387498274256129746192347", "298472983472983471903246121093472394872319615612417471234712061", "29834729834729834729347290846729561262544958723956495615629569234729836259263598127342374289365912465901365498236492183464", @@ -290,20 +290,20 @@ var expNNNTests = []expNNNTest{ } -func TestExpNNN(t *testing.T) { - for i, test := range expNNNTests { - x, _, _ := scanN(nil, test.x, 0) - y, _, _ := scanN(nil, test.y, 0) - out, _, _ := scanN(nil, test.out, 0) +func TestExpNN(t *testing.T) { + for i, test := range expNNTests { + x, _, _ := nat(nil).scan(test.x, 0) + y, _, _ := nat(nil).scan(test.y, 0) + out, _, _ := nat(nil).scan(test.out, 0) - var m []Word + var m nat if len(test.m) > 0 { - m, _, _ = scanN(nil, test.m, 0) + m, _, _ = nat(nil).scan(test.m, 0) } - z := expNNN(nil, x, y, m) - if cmpNN(z, out) != 0 { + z := nat(nil).expNN(x, y, m) + if z.cmp(out) != 0 { t.Errorf("#%d got %v want %v", i, z, out) } }