about summary refs log tree commit diff
path: root/src/level_gen/util.rs
blob: 629292c430fa19766f66e65163a6a58b29513061 (plain) (blame)
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
use crate::types::Dimensions;
use rand::{distributions, Rng};

pub fn falses(dims: &Dimensions) -> Vec<Vec<bool>> {
    let mut ret = Vec::with_capacity(dims.h as usize);
    for _ in 0..dims.h {
        let mut row = Vec::with_capacity(dims.w as usize);
        for _ in 0..dims.w {
            row.push(false);
        }
        ret.push(row);
    }
    ret
}

/// Randomly initialize a 2-dimensional boolean vector of the given
/// `Dimensions`, using the given random number generator and alive chance
pub fn rand_initialize<R: Rng + ?Sized>(
    dims: &Dimensions,
    rng: &mut R,
    alive_chance: f64,
) -> Vec<Vec<bool>> {
    let distrib = distributions::Bernoulli::new(alive_chance).unwrap();
    let mut ret = Vec::with_capacity(dims.h as usize);
    for _ in 0..dims.h {
        let mut row = Vec::with_capacity(dims.w as usize);
        for _ in 0..dims.w {
            row.push(rng.sample(distrib));
        }
        ret.push(row);
    }
    ret
}