1use rand::{Rng as RandRng, SeedableRng};
4use rand_chacha::ChaCha8Rng;
5
6pub struct Rng {
8 inner: ChaCha8Rng,
9}
10
11impl Rng {
12 pub fn new(seed: u64) -> Self {
13 Self {
14 inner: ChaCha8Rng::seed_from_u64(seed),
15 }
16 }
17
18 pub fn range(&mut self, min: i32, max: i32) -> i32 {
19 self.inner.gen_range(min..max)
20 }
21
22 pub fn range_usize(&mut self, min: usize, max: usize) -> usize {
23 self.inner.gen_range(min..max)
24 }
25
26 pub fn random(&mut self) -> f64 {
27 self.inner.gen()
28 }
29
30 pub fn chance(&mut self, probability: f64) -> bool {
31 self.random() < probability
32 }
33
34 pub fn pick<'a, T>(&mut self, slice: &'a [T]) -> Option<&'a T> {
35 if slice.is_empty() {
36 None
37 } else {
38 Some(&slice[self.range_usize(0, slice.len())])
39 }
40 }
41
42 pub fn shuffle<T>(&mut self, slice: &mut [T]) {
43 for i in (1..slice.len()).rev() {
44 let j = self.range_usize(0, i + 1);
45 slice.swap(i, j);
46 }
47 }
48}