use std::mem::{size_of};
use crate::quantify::Quantifiable;
#[derive(Debug)]
pub struct Matrix<T>
{
data: Vec<T>,
num_columns: usize,
}
impl<T> Matrix<T>
{
pub fn get(&self,row:usize,column:usize) -> &T
{
&self.data[row*self.num_columns+column]
}
pub fn get_mut(&mut self,row:usize,column:usize) -> &mut T
{
&mut self.data[row*self.num_columns+column]
}
pub fn get_rows(&self) -> usize
{
self.data.len()/self.num_columns
}
pub fn get_columns(&self) -> usize
{
self.num_columns
}
pub fn constant(value:T,num_rows:usize,num_columns:usize) -> Matrix<T> where T:Clone
{
Matrix{
data: vec![value;num_rows*num_columns],
num_columns,
}
}
pub fn map<U,F:FnMut(&T)->U>(&self, f:F) -> Matrix<U>
{
Matrix{
data: self.data.iter().map(f).collect(),
num_columns: self.num_columns,
}
}
pub fn outside_diagonal(&self) -> OutsideDiagonal<T>
{
OutsideDiagonal{ matrix: self, row:0, column:0 }
}
}
impl<T:Quantifiable> Quantifiable for Matrix<T>
{
fn total_memory(&self) -> usize
{
size_of::<Matrix<T>>() + self.data.total_memory()
}
fn print_memory_breakdown(&self)
{
unimplemented!();
}
fn forecast_total_memory(&self) -> usize
{
unimplemented!();
}
}
impl<T> IntoIterator for Matrix<T>
{
type Item = T;
type IntoIter = <Vec<T> as IntoIterator>::IntoIter;
fn into_iter(self) -> <Self as IntoIterator>::IntoIter
{
self.data.into_iter()
}
}
pub struct OutsideDiagonal<'a, T>
{
matrix: &'a Matrix<T>,
row: usize,
column: usize,
}
impl<'a,T> Iterator for OutsideDiagonal<'a,T>
{
type Item = &'a T;
fn next(&mut self) -> Option<<Self as Iterator>::Item>
{
self.column+=1;
while self.column==self.row
{
self.column+=1;
}
if self.column>=self.matrix.get_columns()
{
self.column = 0;
self.row += 1;
}
if self.row < self.matrix.get_rows() {
Some(self.matrix.get(self.row,self.column))
} else {
None
}
}
}