pub use libcsv::{Reader, Writer, Error};
use rustc_serialize::{Decodable, Encodable};
use std::io::{Read, Write};
use super::super::matrix::{Matrix, BaseMatrix};
impl<T> Matrix<T> where T: Decodable {
pub fn read_csv<'a, R: Read>(mut reader: Reader<R>)
-> Result<Matrix<T>, Error> {
let header: Vec<String> = try!(reader.headers());
let mut nrows = 0;
let ncols = header.len();
let mut records: Vec<T> = vec![];
for record in reader.decode() {
let values: Vec<T> = try!(record);
records.extend(values);
nrows += 1;
}
Ok(Matrix::new(nrows, ncols, records))
}
}
impl<T> Matrix<T> where T: Encodable {
pub fn write_csv<W: Write>(&self, writer: &mut Writer<W>)
-> Result<(), Error> {
for row in self.row_iter() {
try!(writer.encode(row.raw_slice()));
}
Ok(())
}
}