You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.1 KiB
50 lines
1.1 KiB
3 years ago
|
use std::ops::{Add, Index, IndexMut};
|
||
|
|
|
||
|
|
use ndarray::Array2;
|
||
|
|
|
||
|
|
#[derive(Copy, Clone)]
|
||
|
|
pub struct CellDiff {
|
||
|
|
x: i8,
|
||
|
|
y: i8,
|
||
|
|
}
|
||
|
|
|
||
|
|
pub const MINICUBE_DIFF_ARRAY: [CellDiff; 6] = [
|
||
|
|
CellDiff { x: -1, y: 0 },
|
||
|
|
CellDiff { x: 0, y: 1 },
|
||
|
|
CellDiff { x: 1, y: 1 },
|
||
|
|
CellDiff { x: 1, y: 0 },
|
||
|
|
CellDiff { x: 0, y: -1 },
|
||
|
|
CellDiff { x: -1, y: -1 },
|
||
|
|
];
|
||
|
|
|
||
|
|
#[derive(Copy, Clone)]
|
||
|
|
pub struct CellLocation {
|
||
|
|
pub x: usize,
|
||
|
|
pub y: usize,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Add<CellDiff> for CellLocation {
|
||
|
|
type Output = Self;
|
||
|
|
|
||
|
|
fn add(self, other: CellDiff) -> Self {
|
||
|
|
CellLocation {
|
||
|
|
x: (self.x as isize + other.x as isize) as usize,
|
||
|
|
y: (self.y as isize + other.y as isize) as usize,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<T> Index<CellLocation> for Array2<T> {
|
||
|
|
type Output = T;
|
||
|
|
fn index<'a>(&'a self, cell_location: CellLocation) -> &'a T {
|
||
|
|
&self[[cell_location.x, cell_location.y]]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<T> IndexMut<CellLocation> for Array2<T> {
|
||
|
|
fn index_mut<'a>(&'a mut self, cell_location: CellLocation) -> &'a mut T {
|
||
|
|
&mut self[[cell_location.x, cell_location.y]]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|