vecgrid.rs

pub struct VecGrid<T> {
    data: Vec<T>,
    width: usize,
    height: usize,
}

impl<T> VecGrid<T> {
    pub fn new(width: usize, height: usize) -> VecGrid<T> {
        VecGrid {
            data: Vec::with_capacity(width * height),
            width,
            height,
        }
    }

    pub fn get(&self, x: usize, y: usize) -> Option<&T> {
        self.data.get(x + (y * self.width))
    }

    pub fn get_mut(&mut self, x: usize, y: usize) -> Option<&mut T> {
        self.data.get_mut(x + (y * self.width))
    }
}

impl<T: Clone> VecGrid<T> {
    pub fn with(width: usize, height: usize, item: T) -> VecGrid<T> {
        VecGrid {
            data: vec![item; width * height],
            width,
            height,
        }
    }
}