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
|
use std::collections::{HashMap, HashSet};
use specs::{
ReadStorage,
Component,
Entity
};
use crate::{
components::{Visible, Removed},
Pos
};
#[derive(Default)]
pub struct Ground {
pub cells: HashMap<Pos, HashSet<Entity>>
}
impl Ground {
pub fn components_on<'a, C: Component>(&self, pos: Pos, component_type: &'a ReadStorage<C>, removals: &'a ReadStorage<Removed>) -> Vec<&'a C> {
self.cells
.get(&pos)
.unwrap_or(&HashSet::new())
.iter()
.filter(|e| !removals.contains(**e))
.filter_map(|e| component_type.get(*e))
.collect()
}
pub fn by_height(&self, pos: &Pos, visibles: &ReadStorage<Visible>, ignore: &Entity) -> Vec<Entity> {
let mut entities: Vec<Entity> = self.cells
.get(&pos).unwrap_or(&HashSet::new())
.iter()
.cloned()
.filter(|e| e != ignore && visibles.contains(*e))
.collect();
entities.sort_by(|a, b|
visibles.get(*b).unwrap().height.partial_cmp(&visibles.get(*a).unwrap().height).unwrap()
);
entities
}
}
|
