plotrs 0.1.3

CLI app for plotting data points from a csv and writing a png to disk
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

//! For reading a `.csv` file

use csv::StringRecord;
use tracing::{error, trace};

/// Reads a csv of data points and returns a vector of rows
pub fn load_data(data_path: &str, has_headers: bool, csv_delimiter: &str) -> Vec<StringRecord> {
	let mut data = match csv::ReaderBuilder::new()
		.has_headers(has_headers)
		.delimiter(csv_delimiter.as_bytes()[0])
		.from_path(data_path)
	{
		Ok(d) => d,
		Err(e) => {
			error!("Unable to read csv data {:?}", e);
			std::process::exit(1);
		}
	};
	let mut string_records: Vec<StringRecord> = Vec::new();
	for record in data.records() {
		match record {
			Ok(r) => {
				trace!("Csv data: {:?}", r);
				string_records.push(r)
			}
			Err(e) => {
				error!("Unable to read record in csv data: {}", e);
				std::process::exit(1);
			}
		}
	}
	string_records
}