rust-sokoban/src/systems/gameplay_state_sytem.rs

55 lines
1.7 KiB
Rust
Raw Permalink Normal View History

use specs::{Join, ReadStorage, System, Write};
use std::collections::HashMap;
use crate::{
components::{Box, BoxSpot, Position},
resources::{Gameplay, GameplayState},
};
pub struct GameplayStateSystem {}
impl<'a> System<'a> for GameplayStateSystem {
// Data
type SystemData = (
Write<'a, Gameplay>,
ReadStorage<'a, Position>,
ReadStorage<'a, Box>,
ReadStorage<'a, BoxSpot>,
);
fn run(&mut self, data: Self::SystemData) {
let (mut gameplay_state, positions, boxes, box_spots) = data;
// get all boxes indexed by position
let boxes_by_position: HashMap<(u8, u8), &Box> = (&positions, &boxes)
.join()
2020-07-23 21:24:36 -04:00
.map(|t| {
let (pos, b) = t;
((pos.x, pos.y), b)
})
.collect();
// loop through all box spots and check if there is a corresponding
2020-07-24 18:00:23 -04:00
// box at that position. since we now have different types of boxes
// we need to make sure the right type of box is on the right
// type of spot.
for (box_spot, position) in (&box_spots, &positions).join() {
if let Some(the_box) = boxes_by_position.get(&(position.x, position.y)) {
if the_box.color == box_spot.color {
// continue
} else {
// return, haven't won yet
return;
}
} else {
gameplay_state.state = GameplayState::Playing;
return;
}
}
// If we made it this far, then all box spots have boxes on them, and the
// game has been won
gameplay_state.state = GameplayState::Won;
}
}