billow 0.2.0

Simple Wave Function Collapse implementation in Rust
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
60

use rand::seq::SliceRandom;

use crate::module::Module;

/// A slot is a place in agrid where modules can be placed.
#[derive(Clone, Default)]
pub struct Slot {
    /// X coordinate of the slot.
    pub x: usize,
    /// Y coordinate of the slot.
    pub y: usize,
    /// Possible modules in the slot.
    pub superposition: Vec<Module>,
}

impl Slot {
    pub fn collapse(&mut self) {
        if let Some(module) = self.superposition.choose(&mut rand::thread_rng()) {
            self.superposition = vec![*module];
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::conector::ConnectorID;

    use super::*;

    #[test]
    fn test_collapse() {
        let mut slot = Slot {
            x: 0,
            y: 0,
            superposition: vec![
                Module {
                    connectors: [
                        ConnectorID::from("a"),
                        ConnectorID::from("b"),
                        ConnectorID::from("a"),
                        ConnectorID::from("b"),
                    ],
                    ..Default::default()
                },
                Module {
                    connectors: [
                        ConnectorID::from("c"),
                        ConnectorID::from("d"),
                        ConnectorID::from("e"),
                        ConnectorID::from("f"),
                    ],
                    ..Default::default()
                },
            ],
        };

        slot.collapse();
        assert_eq!(slot.superposition.len(), 1);
    }
}