use crate::block::Block;
use std::fmt;
use std::fmt::Debug;
use std::ops::{Index, IndexMut};
pub struct Memory {
rows: usize,
cols: usize,
blocks: Box<[Block]>,
}
impl Memory {
pub fn new(lanes: u32, lane_length: u32) -> Memory {
let rows = lanes as usize;
let cols = lane_length as usize;
let total = rows * cols;
let blocks = vec![Block::zero(); total].into_boxed_slice();
Memory { rows, cols, blocks }
}
}
impl Debug for Memory {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Memory {{ rows: {}, cols: {} }}", self.rows, self.cols)
}
}
impl Index<u32> for Memory {
type Output = Block;
fn index(&self, index: u32) -> &Block {
&self.blocks[index as usize]
}
}
impl Index<u64> for Memory {
type Output = Block;
fn index(&self, index: u64) -> &Block {
&self.blocks[index as usize]
}
}
impl Index<(u32, u32)> for Memory {
type Output = Block;
fn index(&self, index: (u32, u32)) -> &Block {
let pos = ((index.0 as usize) * self.cols) + (index.1 as usize);
&self.blocks[pos]
}
}
impl IndexMut<u32> for Memory {
fn index_mut(&mut self, index: u32) -> &mut Block {
&mut self.blocks[index as usize]
}
}
impl IndexMut<u64> for Memory {
fn index_mut(&mut self, index: u64) -> &mut Block {
&mut self.blocks[index as usize]
}
}
impl IndexMut<(u32, u32)> for Memory {
fn index_mut(&mut self, index: (u32, u32)) -> &mut Block {
let pos = ((index.0 as usize) * self.cols) + (index.1 as usize);
&mut self.blocks[pos]
}
}
#[cfg(test)]
mod tests {
use crate::memory::Memory;
#[test]
fn new_returns_correct_instance() {
let lanes = 4;
let lane_length = 128;
let memory = Memory::new(lanes, lane_length);
assert_eq!(memory.rows, lanes as usize);
assert_eq!(memory.cols, lane_length as usize);
assert_eq!(memory.blocks.len(), 512);
}
}