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
76
77
78
79
80
81
82
83
84
85
86
pub struct Grid<T>
where
T: Clone,
{
width: usize,
height: usize,
values: Vec<Vec<T>>,
}
pub struct GridItem<T>
where
T: Clone,
{
pub val: T,
pub x: usize,
pub y: usize,
}
impl<T> Grid<T>
where
T: Default,
T: Clone,
{
pub fn new(width: usize, height: usize) -> Self {
Grid {
width,
height,
values: vec![vec![T::default(); width]; height],
}
}
}
impl<T> Grid<T>
where
T: Clone,
{
pub fn from(grid: Vec<Vec<T>>) -> Self {
Grid {
width: grid[0].len(),
height: grid.len(),
values: grid,
}
}
pub fn get_near_4(&self, pos: (usize, usize)) -> Vec<GridItem<T>> {
let mut result = Vec::new();
if pos.0 > 0 {
result.push(GridItem {
x: pos.0 - 1,
y: pos.1,
val: self.values[pos.1][pos.0 - 1].clone(),
});
}
if pos.0 < self.width - 1 {
result.push(GridItem {
x: pos.0 + 1,
y: pos.1,
val: self.values[pos.1][pos.0 + 1].clone(),
});
}
if pos.1 > 0 {
result.push(GridItem {
x: pos.0,
y: pos.1 - 1,
val: self.values[pos.1 - 1][pos.0].clone(),
});
}
if pos.1 < self.height - 1 {
result.push(GridItem {
x: pos.0,
y: pos.1 + 1,
val: self.values[pos.1 + 1][pos.0].clone(),
});
}
result
}
pub fn is_near_4(&self, pos1: (usize, usize), pos2: (usize, usize)) -> bool {
let x = (pos1.0 as i32 - pos2.0 as i32).abs();
let y = (pos1.1 as i32 - pos2.1 as i32).abs();
x == 1 && y == 0 || x == 0 && y == 1
}
}