v/vlib/rand/splitmix64.v

54 lines
1.0 KiB
V
Raw Normal View History

module rand
// Ported from http://xoshiro.di.unimi.it/splitmix64.c
struct Splitmix64 {
mut:
state u64
}
2019-12-19 22:29:37 +01:00
/**
* new_splitmix64 - a Splitmix64 PRNG generator
2019-12-07 13:51:00 +01:00
* @param seed the initial seed of the PRNG.
* @return a new Splitmix64 PRNG instance
*/
2019-12-19 22:29:37 +01:00
pub fn new_splitmix64(seed u64) Splitmix64 {
2019-12-19 22:29:37 +01:00
return Splitmix64{
seed}
}
2019-12-19 22:29:37 +01:00
/**
* Splitmix64.next - update the PRNG state and get back the next random number
* @return the generated pseudo random number
*/
2019-12-19 22:29:37 +01:00
[inline]
pub fn (rng mut Splitmix64) next() u64 {
2019-12-07 13:51:00 +01:00
rng.state += (0x9e3779b97f4a7c15)
mut z := rng.state
2019-12-19 22:29:37 +01:00
z = (z ^ ((z>>u64(30)))) * (0xbf58476d1ce4e5b9)
z = (z ^ ((z>>u64(27)))) * (0x94d049bb133111eb)
return z ^ (z>>(31))
}
2019-12-19 22:29:37 +01:00
/**
* Splitmix64.bounded_next - Get the next random number < bound
* @param bound - the returned random number will be < bound
* @return the generated pseudo random number
*/
2019-12-19 22:29:37 +01:00
[inline]
pub fn (rng mut Splitmix64) bounded_next(bound u64) u64 {
2019-12-07 13:51:00 +01:00
threshold := -bound % bound
for {
r := rng.next()
if r >= threshold {
2019-12-07 13:51:00 +01:00
return r % bound
}
}
return u64(0)
}
2019-12-19 22:29:37 +01:00