// Copyright 2009 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. /* * algorithm by * DP Mitchell and JA Reeds */ package rand // rand, rand31, Int63 - return non-negative random int, int32, int64 // urand32 - return random uint32 // nrand, nrand31, Int63n - return 0 <= random < n // frand, frand64, frand32 - return 0 <= random float, float64, float32 < 1 // perm gives a random permutation []int const ( _LEN = 607; _TAP = 273; _MASK = (1<<63)-1; _A = 48271; _M = 2147483647; _Q = 44488; _R = 3399; ) var ( rng_cooked [_LEN]int64; // cooked random numbers rng_vec [_LEN]int64; // current feedback register rng_tap int; // index into vector rng_feed int; // index into vector ) func seedrand(x int32) int32 { // seed rng x[n+1] = 48271 * x[n] mod (2**31 - 1) hi := x / _Q; lo := x % _Q; x = _A*lo - _R*hi; if x < 0 { x += _M; } return x; } func Seed(seed int32) { rng_tap = 0; rng_feed = _LEN-_TAP; seed = seed%_M; if seed < 0 { seed += _M; } if seed == 0 { seed = 89482311; } x := seed; for i := -20; i < _LEN; i++ { x = seedrand(x); if i >= 0 { var u int64; u = int64(x) << 20; x = seedrand(x); u ^= int64(x) << 10; x = seedrand(x); u ^= int64(x); u ^= rng_cooked[i]; rng_vec[i] = u & _MASK; } } } func Int63() int64 { rng_tap--; if rng_tap < 0 { rng_tap += _LEN; } rng_feed--; if rng_feed < 0 { rng_feed += _LEN; } x := (rng_vec[rng_feed] + rng_vec[rng_tap]) & _MASK; rng_vec[rng_feed] = x; return x; } func Uint32() uint32 { return uint32(Int63() >> 31); } func Int31() int32 { return int32(Int63() >> 32); } func Int() int { u := uint(Int63()); return int(u << 1 >> 1); // clear sign bit if int == int32 } func Int63n(n int64) int64 { if n <= 0 { return 0 } max := int64((1<<63)-1 - (1<<63) % uint64(n)); v := Int63(); for v > max { v = Int63() } return v % n } func Int31n(n int32) int32 { return int32(Int63n(int64(n))) } func Intn(n int) int { return int(Int63n(int64(n))) } func Float64() float64 { x := float64(Int63()) / float64(_MASK); for x >= 1 { x = float64(Int63()) / float64(_MASK); } return x; } func Float32() float32 { return float32(Float64()) } func Float() float { return float(Float64()) } func Perm(n int) []int { m := make([]int, n); for i:=0; i