squares 0.1.1

A Counter-RNG faster than Philox
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

#![allow(unused)]

use rand_core::{impls::fill_bytes_via_next, RngCore};
use crate::Key;

/// An RNG compatible with `rand`.
/// Increments counter internally.
#[derive(Clone, Copy, Debug)]
struct Squares {
  key: Key,
  index: u64
}

impl Squares {
  #[must_use] #[inline(always)]
  pub const fn with_key(key: Key) -> Self {
    Self { key, index: 0 }
  }

  /// Set the location in the RNG sequence
  #[must_use] #[inline(always)]
  pub const fn with_index(self, index: u64) -> Self {
    Self { key: self.key, index }
  }

  /// Get the current location in the RNG sequence
  #[must_use] #[inline(always)]
  pub const fn index(&self) -> u64 { self.index }

  /// Skip ahead in the sequence
  #[must_use] #[inline(always)]
  pub const fn skip(mut self, n: u64) -> Self {
    self.index += n;
    self
  }
}

impl RngCore for Squares {
  fn next_u32(&mut self) -> u32 {
    let r = super::u32(self.key, self.index);
    self.index += 1;
    r
  }

  fn next_u64(&mut self) -> u64 {
    let r = super::u64(self.key, self.index);
    self.index += 1;
    r
  }

  fn fill_bytes(&mut self, dest: &mut [u8]) {
    fill_bytes_via_next(self, dest);
  }

  fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), rand_core::Error> {
    fill_bytes_via_next(self, dest);
    Ok(())
  }
}