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
70
71
72
73
74
75
|
use std::collections::HashSet;
use specs::{
Entities,
ReadStorage,
WriteStorage,
Read,
System,
Join,
Write
};
use crate::{
Pos,
components::{
Controller,
Position,
Flags,
Flag,
Moved,
Entered,
Movable,
ControlCooldown
},
controls::{
Control
},
resources::{
Size,
Ground
},
};
pub struct Move;
impl <'a> System<'a> for Move {
type SystemData = (
Entities<'a>,
ReadStorage<'a, Controller>,
WriteStorage<'a, Position>,
Read<'a, Size>,
ReadStorage<'a, Flags>,
Write<'a, Ground>,
WriteStorage<'a, Moved>,
WriteStorage<'a, Entered>,
ReadStorage<'a, Movable>,
WriteStorage<'a, ControlCooldown>
);
fn run(&mut self, (entities, controllers, mut positions, size, flags, mut ground, mut moved, mut entered, movables, mut cooldowns): Self::SystemData) {
moved.clear();
entered.clear();
for (ent, controller, mut pos, movable) in (&entities, &controllers, &mut positions.restrict_mut(), &movables).join(){
match &controller.control {
Control::Move(direction) => {
let newpos = (pos.get_unchecked().pos + direction.to_position()).clamp(Pos::new(0, 0), Pos::new(size.width - 1, size.height - 1));
let ground_flags = ground.flags_on(newpos, &flags);
if !ground_flags.contains(&Flag::Blocking) && ground_flags.contains(&Flag::Floor) {
let mut pos_mut = pos.get_mut_unchecked();
moved.insert(ent, Moved{from: pos_mut.pos}).expect("can't insert Moved");
ground.cells.get_mut(&pos_mut.pos).unwrap().remove(&ent);
pos_mut.pos = newpos;
ground.cells.entry(newpos).or_insert_with(HashSet::new).insert(ent);
for ent in ground.cells.get(&newpos).unwrap() {
let _ = entered.insert(*ent, Entered);
}
cooldowns.insert(ent, ControlCooldown{amount: movable.cooldown}).unwrap();
}
}
_ => {}
}
}
}
}
|
