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 rand::{Rng, seq::SliceRandom};
use specs::{
ReadStorage,
System,
Join,
Write,
Read
};
use crate::{
components::{
Position,
Loot,
Trigger,
TriggerBox,
Flags,
Flag
},
resources::{NewEntities, Ground},
Pos
};
pub struct DropLoot;
impl <'a> System<'a> for DropLoot{
type SystemData = (
ReadStorage<'a, Position>,
Write<'a, NewEntities>,
ReadStorage<'a, TriggerBox>,
ReadStorage<'a, Loot>,
Read<'a, Ground>,
ReadStorage<'a, Flags>
);
fn run(&mut self, (positions, mut new, triggerboxes, loots, ground, flags): Self::SystemData) {
for (position, triggerbox, loot) in (&positions, &triggerboxes, &loots).join(){
if triggerbox.has_message(&[Trigger::Die, Trigger::Loot]) {
for (template, chance) in &loot.loot {
if *chance > rand::thread_rng().gen_range(0.0, 1.0) {
let pos = if loot.spread {
pick_position(position.pos, &ground, &flags)
} else {position.pos};
// todo: better error handling
new.create(pos, &template).unwrap();
}
}
}
}
}
}
fn pick_position<'a>(pos: Pos, ground: &Read<'a, Ground>, flags: &ReadStorage<'a, Flags>) -> Pos {
let ground_flags = ground.flags_on(pos, &flags);
if !ground_flags.contains(&Flag::Blocking) && ground_flags.contains(&Flag::Floor) {
return pos
}
let mut neighbours = vec![(1,0), (-1,0), (0,1), (0,-1)];
neighbours.shuffle(&mut rand::thread_rng());
for t in neighbours {
let p = pos + Pos::from_tuple(t);
let ground_flags = ground.flags_on(p, &flags);
if !ground_flags.contains(&Flag::Blocking) && ground_flags.contains(&Flag::Floor) {
return p
}
}
return pos;
}
|
