djkato/grid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Speedup idx function

Browse source
This commit is contained in:
becheran 2022-08-13 11:07:39 +02:00
parent 7979a4fbe0
commit f28b3a3281
2 changed files with 19 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
benches/benches.rs
View file

@ -83,7 +83,7 @@ fn criterion_benchmark(c: &mut Criterion) {
b.iter_batched( b.iter_batched(
|| (rand(), rand()), || (rand(), rand()),
|(x, y)| { |(x, y)| {
let _v = vec_vec[criterion::black_box(x)][criterion::black_box(y)]; let _v = vec_vec[x][y];
}, },
criterion::BatchSize::SmallInput, criterion::BatchSize::SmallInput,
) )
@ -92,7 +92,7 @@ fn criterion_benchmark(c: &mut Criterion) {
let grid = init_grid(); let grid = init_grid();
b.iter_batched( b.iter_batched(
|| (rand(), rand()), || (rand(), rand()),
|(x, y)| grid[criterion::black_box(x)][criterion::black_box(y)], |(x, y)| grid[x][y],
criterion::BatchSize::SmallInput, criterion::BatchSize::SmallInput,
) )
}); });
@ -102,9 +102,9 @@ fn criterion_benchmark(c: &mut Criterion) {
|| (rand(), rand()), || (rand(), rand()),
|(x, y)| { |(x, y)| {
let _v = vec_vec let _v = vec_vec
.get(criterion::black_box(x)) .get(x)
.unwrap() .unwrap()
.get(criterion::black_box(y)) .get(y)
.unwrap(); .unwrap();
}, },
criterion::BatchSize::SmallInput, criterion::BatchSize::SmallInput,
@ -116,7 +116,7 @@ fn criterion_benchmark(c: &mut Criterion) {
|| (rand(), rand()), || (rand(), rand()),
|(x, y)| { |(x, y)| {
let _v = grid let _v = grid
.get(criterion::black_box(x), criterion::black_box(y)) .get(x, y)
.unwrap(); .unwrap();
}, },
criterion::BatchSize::SmallInput, criterion::BatchSize::SmallInput,

28
src/lib.rs
View file

@ -723,22 +723,19 @@ impl<T: Clone> Clone for Grid<T> {
impl<T> Index<usize> for Grid<T> { impl<T> Index<usize> for Grid<T> {
type Output = [T]; type Output = [T];
fn index(&self, idx: usize) -> &Self::Output { #[inline]
if idx < self.rows { fn index(&self, idx: usize) -> &[T] {
let start_idx = idx * self.cols; let start_idx = idx * self.cols;
&self.data[start_idx..start_idx + self.cols] &self.data[start_idx..start_idx + self.cols]
} else {
panic!(
"index {:?} out of bounds. Grid has {:?} rows.",
self.rows, idx
);
}
} }
} }
impl<T> IndexMut<usize> for Grid<T> { impl<T> IndexMut<usize> for Grid<T> {
fn index_mut(&mut self, idx: usize) -> &mut Self::Output {
&mut self.data[(idx * self.cols)..] #[inline]
fn index_mut(&mut self, idx: usize) -> &mut [T] {
let start_idx = idx * self.cols;
&mut self.data[start_idx..start_idx + self.cols]
} }
} }
@ -1217,8 +1214,11 @@ mod test {
#[test] #[test]
fn idx() { fn idx() {
let grid = Grid::init(1, 2, 3); let grid: Grid<u8> = Grid::from_vec(vec![1, 2, 3, 4], 2);
assert_eq!(grid[0][0], 3); assert_eq!(grid[0][0], 1);
assert_eq!(grid[0][1], 2);
assert_eq!(grid[1][0], 3);
assert_eq!(grid[1][1], 4);
} }
#[test] #[test]