about summary refs log tree commit diff
path: root/src/level_gen/util.rs
blob: 4f56fe6c955704ec8ed3531afa1735581a4dbab1 (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
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
}

/// Fill the outer edges of a generated level with walls
pub fn fill_outer_edges(level: &mut Vec<Vec<bool>>) {
    let xmax = level.len();
    if xmax == 0 {
        return;
    }
    let ymax = level[0].len();

    for row in level.iter_mut() {
        row[0] = true;
        row[ymax - 1] = true;
    }

    for y in 0..level[0].len() {
        level[0][y] = true;
        level[xmax - 1][y] = true;
    }
}