terrain_forge/
rng.rs

1//! Seeded random number generator for deterministic generation
2
3use rand::{Rng as RandRng, SeedableRng};
4use rand_chacha::ChaCha8Rng;
5
6/// Seeded RNG wrapper for deterministic generation
7pub 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}