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
61
62
63
64
65
66
67
68
69

use crate::{conector::ConnectorID, direction::Direction, slot::Slot};

/// Module represents a tile that can inhabit a slot.
#[derive(Debug, Clone, Copy, Default)]
pub struct Module {
    pub index: usize,
    /// Valid connectors for each direction.
    pub connectors: [ConnectorID; 4],
    // TODO: add Image field. (for now, we just use the index)
    //pub image: Box<dyn Image>,
}

impl Module {
    pub fn new(index: usize) -> Self {
        Self {
            index,
            connectors: [ConnectorID::default(); 4],
        }
    }

    pub fn is_possible(&self, from: &Slot, dir: Direction) -> bool {
        let back = dir.reverse();

        for slot in from.superposition.iter() {
            if self.connectors[back as usize] == slot.connectors[dir as usize] {
                return true;
            }
        }
        false
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_is_possible() {
        let module_a = Module {
            connectors: [
                ConnectorID::from("a"),
                ConnectorID::from("b"),
                ConnectorID::from("a"),
                ConnectorID::from("b"),
            ],
            ..Default::default()
        };

        let module_b = Module {
            connectors: [
                ConnectorID::from("c"),
                ConnectorID::from("d"),
                ConnectorID::from("e"),
                ConnectorID::from("f"),
            ],
            ..Default::default()
        };

        let slot = Slot {
            superposition: vec![module_a.clone()],
            x: 0,
            y: 0,
        };

        assert!(module_a.is_possible(&slot, Direction::Up));
        assert!(module_a.is_possible(&slot, Direction::Left));
        assert!(!module_b.is_possible(&slot, Direction::Up));
    }
}