#![no_std]
use core::marker::PhantomData;
use core::ptr;
pub trait Random: Sized {
fn random<G: ?Sized + RandomGen>(g: &mut G) -> Self;
}
pub trait RandomGen {
#[inline]
fn gen_u32(&mut self) -> u32 { self.gen_u64() as u32 }
#[inline]
fn gen_u64(&mut self) -> u64 { (self.gen_u32() as u64) << 32 | (self.gen_u32() as u64) }
#[inline]
fn gen_f32(&mut self) -> f32 { (self.gen_u32() >> 8) as f32 / (1u64 << 24) as f32 }
#[inline]
fn gen_f64(&mut self) -> f64 { (self.gen_u64() >> 11) as f64 / (1u64 << 53) as f64 }
#[inline]
fn fill_bytes(&mut self, bs: &mut [u8]) {
let mut p = bs.as_mut_ptr();
let mut l = bs.len();
loop { unsafe {
if l >= 8 {
let n = self.gen_u64();
ptr::copy_nonoverlapping(&n as *const _ as *const u8, p, 8);
p = p.offset(8);
l -= 8;
} else {
let n = self.gen_u32();
ptr::copy_nonoverlapping(&n as *const _ as *const u8, p, usize::min(4, l));
if 4 >= l { return; }
p = p.offset(4);
l -= 4;
}
} }
}
#[inline]
fn gen<A: Random>(&mut self) -> A { A::random(self) }
}
impl<'a, G: RandomGen> RandomGen for &'a mut G {
#[inline] fn gen_u32(&mut self) -> u32 { G::gen_u32(self) }
#[inline] fn gen_u64(&mut self) -> u64 { G::gen_u64(self) }
#[inline] fn gen_f32(&mut self) -> f32 { G::gen_f32(self) }
#[inline] fn gen_f64(&mut self) -> f64 { G::gen_f64(self) }
#[inline] fn fill_bytes(&mut self, bs: &mut [u8]) { G::fill_bytes(self, bs) }
}
#[derive(Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct Stream<G, A>(G, PhantomData<A>);
impl<G, A> Stream<G, A> {
#[inline]
pub fn new(g: G) -> Self { Stream(g, PhantomData) }
}
impl<G: RandomGen, A: Random> Iterator for Stream<G, A> {
type Item = A;
#[inline]
fn next(&mut self) -> Option<A> { Some(self.0.gen()) }
}
impl Random for u32 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> u32 { g.gen_u32() }
}
impl Random for u64 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> u64 { g.gen_u64() }
}
impl Random for i32 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> i32 { g.gen_u32() as _ }
}
impl Random for i64 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> i64 { g.gen_u64() as _ }
}
impl Random for f32 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> f32 { g.gen_f32() }
}
impl Random for f64 {
#[inline]
fn random<G: ?Sized + RandomGen>(g: &mut G) -> f64 { g.gen_f64() }
}
pub trait Seedable<Seed>: RandomGen {
fn reseed(&mut self, _: Seed);
fn from_seed(_: Seed) -> Self;
}