forked from rust-random/rand
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
SmallRng: Replace PCG algorithm with xoshiro{128,256}++
Due to close correlations of PCG streams (rust-random#907) and lack of right-state propagation (rust-random#905), the `SmallRng` algorithm is switched to xoshiro{128,256}++. The implementation is taken from the `rand_xoshiro` crate and slightly simplified. Fixes rust-random#910.
- Loading branch information
Showing
5 changed files
with
222 additions
and
14 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,101 @@ | ||
// Copyright 2018 Developers of the Rand project. | ||
// | ||
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or | ||
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license | ||
// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your | ||
// option. This file may not be copied, modified, or distributed | ||
// except according to those terms. | ||
|
||
#[cfg(feature="serde1")] use serde::{Serialize, Deserialize}; | ||
use rand_core::impls::{next_u64_via_u32, fill_bytes_via_next}; | ||
use rand_core::le::read_u32_into; | ||
use rand_core::{SeedableRng, RngCore, Error}; | ||
|
||
/// A xoshiro128++ random number generator. | ||
/// | ||
/// The xoshiro128++ algorithm is not suitable for cryptographic purposes, but | ||
/// is very fast and has excellent statistical properties. | ||
/// | ||
/// The algorithm used here is translated from [the `xoshiro128plusplus.c` | ||
/// reference source code](http://xoshiro.di.unimi.it/xoshiro128plusplus.c) by | ||
/// David Blackman and Sebastiano Vigna. | ||
#[derive(Debug, Clone, PartialEq, Eq)] | ||
#[cfg_attr(feature="serde1", derive(Serialize, Deserialize))] | ||
pub struct Xoshiro128PlusPlus { | ||
s: [u32; 4], | ||
} | ||
|
||
impl SeedableRng for Xoshiro128PlusPlus { | ||
type Seed = [u8; 16]; | ||
|
||
/// Create a new `Xoshiro128PlusPlus`. If `seed` is entirely 0, it will be | ||
/// mapped to a different seed. | ||
#[inline] | ||
fn from_seed(seed: [u8; 16]) -> Xoshiro128PlusPlus { | ||
if seed.iter().all(|&x| x == 0) { | ||
return Self::seed_from_u64(0); | ||
} | ||
let mut state = [0; 4]; | ||
read_u32_into(&seed, &mut state); | ||
Xoshiro128PlusPlus { s: state } | ||
} | ||
} | ||
|
||
impl RngCore for Xoshiro128PlusPlus { | ||
#[inline] | ||
fn next_u32(&mut self) -> u32 { | ||
let result_starstar = self.s[0] | ||
.wrapping_add(self.s[3]) | ||
.rotate_left(7) | ||
.wrapping_add(self.s[0]); | ||
|
||
let t = self.s[1] << 9; | ||
|
||
self.s[2] ^= self.s[0]; | ||
self.s[3] ^= self.s[1]; | ||
self.s[1] ^= self.s[2]; | ||
self.s[0] ^= self.s[3]; | ||
|
||
self.s[2] ^= t; | ||
|
||
self.s[3] = self.s[3].rotate_left(11); | ||
|
||
result_starstar | ||
} | ||
|
||
#[inline] | ||
fn next_u64(&mut self) -> u64 { | ||
next_u64_via_u32(self) | ||
} | ||
|
||
#[inline] | ||
fn fill_bytes(&mut self, dest: &mut [u8]) { | ||
fill_bytes_via_next(self, dest); | ||
} | ||
|
||
#[inline] | ||
fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), Error> { | ||
self.fill_bytes(dest); | ||
Ok(()) | ||
} | ||
} | ||
|
||
#[cfg(test)] | ||
mod tests { | ||
use super::*; | ||
|
||
#[test] | ||
fn reference() { | ||
let mut rng = Xoshiro128PlusPlus::from_seed( | ||
[1, 0, 0, 0, 2, 0, 0, 0, 3, 0, 0, 0, 4, 0, 0, 0]); | ||
// These values were produced with the reference implementation: | ||
// http://xoshiro.di.unimi.it/xoshiro128plusplus.c | ||
let expected = [ | ||
641, 1573767, 3222811527, 3517856514, 836907274, 4247214768, | ||
3867114732, 1355841295, 495546011, 621204420, | ||
]; | ||
for &e in &expected { | ||
assert_eq!(rng.next_u32(), e); | ||
} | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,103 @@ | ||
// Copyright 2018 Developers of the Rand project. | ||
// | ||
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or | ||
// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license | ||
// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your | ||
// option. This file may not be copied, modified, or distributed | ||
// except according to those terms. | ||
|
||
#[cfg(feature="serde1")] use serde::{Serialize, Deserialize}; | ||
use rand_core::impls::fill_bytes_via_next; | ||
use rand_core::le::read_u64_into; | ||
use rand_core::{SeedableRng, RngCore, Error}; | ||
|
||
/// A xoshiro256** random number generator. | ||
/// | ||
/// The xoshiro256** algorithm is not suitable for cryptographic purposes, but | ||
/// is very fast and has excellent statistical properties. | ||
/// | ||
/// The algorithm used here is translated from [the `xoshiro256plusplus.c` | ||
/// reference source code](http://xoshiro.di.unimi.it/xoshiro256plusplus.c) by | ||
/// David Blackman and Sebastiano Vigna. | ||
#[derive(Debug, Clone, PartialEq, Eq)] | ||
#[cfg_attr(feature="serde1", derive(Serialize, Deserialize))] | ||
pub struct Xoshiro256PlusPlus { | ||
s: [u64; 4], | ||
} | ||
|
||
impl SeedableRng for Xoshiro256PlusPlus { | ||
type Seed = [u8; 32]; | ||
|
||
/// Create a new `Xoshiro256PlusPlus`. If `seed` is entirely 0, it will be | ||
/// mapped to a different seed. | ||
#[inline] | ||
fn from_seed(seed: [u8; 32]) -> Xoshiro256PlusPlus { | ||
if seed.iter().all(|&x| x == 0) { | ||
return Self::seed_from_u64(0); | ||
} | ||
let mut state = [0; 4]; | ||
read_u64_into(&seed, &mut state); | ||
Xoshiro256PlusPlus { s: state } | ||
} | ||
} | ||
|
||
impl RngCore for Xoshiro256PlusPlus { | ||
#[inline] | ||
fn next_u32(&mut self) -> u32 { | ||
self.next_u64() as u32 | ||
} | ||
|
||
#[inline] | ||
fn next_u64(&mut self) -> u64 { | ||
let result_plusplus = self.s[0] | ||
.wrapping_add(self.s[3]) | ||
.rotate_left(23) | ||
.wrapping_add(self.s[0]); | ||
|
||
let t = self.s[1] << 17; | ||
|
||
self.s[2] ^= self.s[0]; | ||
self.s[3] ^= self.s[1]; | ||
self.s[1] ^= self.s[2]; | ||
self.s[0] ^= self.s[3]; | ||
|
||
self.s[2] ^= t; | ||
|
||
self.s[3] = self.s[3].rotate_left(45); | ||
|
||
result_plusplus | ||
} | ||
|
||
#[inline] | ||
fn fill_bytes(&mut self, dest: &mut [u8]) { | ||
fill_bytes_via_next(self, dest); | ||
} | ||
|
||
#[inline] | ||
fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), Error> { | ||
self.fill_bytes(dest); | ||
Ok(()) | ||
} | ||
} | ||
|
||
#[cfg(test)] | ||
mod tests { | ||
use super::*; | ||
|
||
#[test] | ||
fn reference() { | ||
let mut rng = Xoshiro256PlusPlus::from_seed( | ||
[1, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, | ||
3, 0, 0, 0, 0, 0, 0, 0, 4, 0, 0, 0, 0, 0, 0, 0]); | ||
// These values were produced with the reference implementation: | ||
// http://xoshiro.di.unimi.it/xoshiro256plusplus.c | ||
let expected = [ | ||
41943041, 58720359, 3588806011781223, 3591011842654386, | ||
9228616714210784205, 9973669472204895162, 14011001112246962877, | ||
12406186145184390807, 15849039046786891736, 10450023813501588000, | ||
]; | ||
for &e in &expected { | ||
assert_eq!(rng.next_u64(), e); | ||
} | ||
} | ||
} |