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
use std::ops::{Index, IndexMut}; type Coordinate = (usize, usize); pub struct Matrix { width: usize, contents: Vec<f32>, } impl Matrix { pub fn new(width: usize, height: usize) -> Self { Matrix { width, contents: vec![0 as f32; width * height], } } } impl Index<Coordinate> for Matrix { type Output = f32; fn index(&self, (width, height): Coordinate) -> &Self::Output { &self.contents[height * self.width + width] } } impl IndexMut<Coordinate> for Matrix { fn index_mut(&mut self, (width, height): Coordinate) -> &mut Self::Output { &mut self.contents[height * self.width + width] } } #[cfg(test)] mod test { use super::*; #[test] fn test_matrix() { let mut matrix = Matrix::new(1024, 768); assert_eq!(matrix[(1023, 767)], 0.0); matrix[(12, 24)] = 123.456; assert_eq!(matrix[(12, 24)], 123.456); assert_eq!(matrix[(24, 12)], 0.0); } }