path: root/src/data_conversion.rs

//! This mainly concerns converting collected data into things that the canvas
//! can actually handle.

use crate::{
	app::{
		data_farmer,
		data_harvester::{self, processes::ProcessHarvest},
		App,
	},
	constants,
	utils::gen_util::{get_exact_byte_values, get_simple_byte_values},
};
use constants::*;
use std::collections::HashMap;

#[derive(Default, Debug)]
pub struct ConvertedNetworkData {
	pub rx: Vec<(f64, f64)>,
	pub tx: Vec<(f64, f64)>,
	pub rx_display: String,
	pub tx_display: String,
	pub total_rx_display: String,
	pub total_tx_display: String,
}

#[derive(Clone, Default, Debug)]
pub struct ConvertedProcessData {
	pub pid: u32,
	pub name: String,
	pub cpu_usage: f64,
	pub mem_usage: f64,
	pub group_pids: Vec<u32>,
}

#[derive(Clone, Default, Debug)]
pub struct ConvertedCpuData {
	pub cpu_name: String,
	pub cpu_data: Vec<(f64, f64)>,
}

pub fn convert_temp_row(app: &App) -> Vec<Vec<String>> {
	let mut sensor_vector: Vec<Vec<String>> = Vec::new();

	let current_data = &app.data_collection;
	let temp_type = &app.app_config_fields.temperature_type;

	if current_data.temp_harvest.is_empty() {
		sensor_vector.push(vec!["No Sensors Found".to_string(), "".to_string()])
	} else {
		for sensor in &current_data.temp_harvest {
			sensor_vector.push(vec![
				sensor.component_name.to_string(),
				(sensor.temperature.ceil() as u64).to_string()
					+ match temp_type {
						data_harvester::temperature::TemperatureType::Celsius => "C",
						data_harvester::temperature::TemperatureType::Kelvin => "K",
						data_harvester::temperature::TemperatureType::Fahrenheit => "F",
					},
			]);
		}
	}

	sensor_vector
}

pub fn convert_disk_row(current_data: &data_farmer::DataCollection) -> Vec<Vec<String>> {
	let mut disk_vector: Vec<Vec<String>> = Vec::new();
	for (itx, disk) in current_data.disk_harvest.iter().enumerate() {
		let io_activity = if current_data.io_labels.len() > itx {
			let converted_read = get_simple_byte_values(current_data.io_labels[itx].0, false);
			let converted_write = get_simple_byte_values(current_data.io_labels[itx].1, false);
			(
				format!("{:.*}{}/s", 0, converted_read.0, converted_read.1),
				format!("{:.*}{}/s", 0, converted_write.0, converted_write.1),
			)
		} else {
			("0B/s".to_string(), "0B/s".to_string())
		};

		let converted_free_space = get_simple_byte_values(disk.free_space, false);
		let converted_total_space = get_simple_byte_values(disk.total_space, false);
		disk_vector.push(vec![
			disk.name.to_string(),
			disk.mount_point.to_string(),
			format!(
				"{:.0}%",
				disk.used_space as f64 / disk.total_space as f64 * 100_f64
			),
			format!("{:.*}{}", 0, converted_free_space.0, converted_free_space.1),
			format!(
				"{:.*}{}",
				0, converted_total_space.0, converted_total_space.1
			),
			io_activity.0,
			io_activity.1,
		]);
	}

	disk_vector
}

pub fn convert_cpu_data_points(
	show_avg_cpu: bool, current_data: &data_farmer::DataCollection,
) -> Vec<ConvertedCpuData> {
	let mut cpu_data_vector: Vec<ConvertedCpuData> = Vec::new();
	let current_time = current_data.current_instant;
	let cpu_listing_offset = if show_avg_cpu { 0 } else { 1 };

	for (time, data) in &current_data.timed_data_vec {
		let time_from_start: f64 = (TIME_STARTS_FROM as f64
			- current_time.duration_since(*time).as_millis() as f64)
			.floor();

		for (itx, cpu) in data.cpu_data.iter().enumerate() {
			if !show_avg_cpu && itx == 0 {
				continue;
			}

			// Check if the vector exists yet
			let itx_offset = itx - cpu_listing_offset;
			if cpu_data_vector.len() <= itx_offset {
				cpu_data_vector.push(ConvertedCpuData::default());
				cpu_data_vector[itx_offset].cpu_name =
					current_data.cpu_harvest[itx].cpu_name.clone();
			}

			//Insert joiner points
			for &(joiner_offset, joiner_val) in &cpu.1 {
				let offset_time = time_from_start - joiner_offset as f64;
				cpu_data_vector[itx_offset]
					.cpu_data
					.push((offset_time, joiner_val));
			}

			cpu_data_vector[itx_offset]
				.cpu_data
				.push((time_from_start, cpu.0));
		}
	}

	cpu_data_vector
}

pub fn convert_mem_data_points(current_data: &data_farmer::DataCollection) -> Vec<(f64, f64)> {
	let mut result: Vec<(f64, f64)> = Vec::new();
	let current_time = current_data.current_instant;

	for (time, data) in &current_data.timed_data_vec {
		let time_from_start: f64 = (TIME_STARTS_FROM as f64
			- current_time.duration_since(*time).as_millis() as f64)
			.floor();

		//Insert joiner points
		for &(joiner_offset, joiner_val) in &data.mem_data.1 {
			let offset_time = time_from_start - joiner_offset as f64;
			result.push((offset_time, joiner_val));
		}

		result.push((time_from_start, data.mem_data.0));
	}

	result
}

pub fn convert_swap_data_points(current_data: &data_farmer::DataCollection) -> Vec<(f64, f64)> {
	let mut result: Vec<(f64, f64)> = Vec::new();
	let current_time = current_data.current_instant;

	for (time, data) in &current_data.timed_data_vec {
		let time_from_start: f64 = (TIME_STARTS_FROM as f64
			- current_time.duration_since(*time).as_millis() as f64)
			.floor();

		//Insert joiner points
		for &(joiner_offset, joiner_val) in &data.swap_data.1 {
			let offset_time = time_from_start - joiner_offset as f64;
			result.push((offset_time, joiner_val));
		}

		result.push((time_from_start, data.swap_data.0));
	}

	result
}

pub fn convert_mem_labels(current_data: &data_farmer::DataCollection) -> (String, String) {
	let mem_label = if current_data.memory_harvest.mem_total_in_mb == 0 {
		"".to_string()
	} else {
		"RAM:".to_string()
			+ &format!(
				"{:3.0}%",
				(current_data.memory_harvest.mem_used_in_mb as f64 * 100.0
					/ current_data.memory_harvest.mem_total_in_mb as f64)
					.round()
			) + &format!(
			"   {:.1}GB/{:.1}GB",
			current_data.memory_harvest.mem_used_in_mb as f64 / 1024.0,
			(current_data.memory_harvest.mem_total_in_mb as f64 / 1024.0).round()
		)
	};

	let swap_label = if current_data.swap_harvest.mem_total_in_mb == 0 {
		"".to_string()
	} else {
		"SWP:".to_string()
			+ &format!(
				"{:3.0}%",
				(current_data.swap_harvest.mem_used_in_mb as f64 * 100.0
					/ current_data.swap_harvest.mem_total_in_mb as f64)
					.round()
			) + &format!(
			"   {:.1}GB/{:.1}GB",
			current_data.swap_harvest.mem_used_in_mb as f64 / 1024.0,
			(current_data.swap_harvest.mem_total_in_mb as f64 / 1024.0).round()
		)
	};

	(mem_label, swap_label)
}

pub fn convert_network_data_points(
	current_data: &data_farmer::DataCollection,
) -> ConvertedNetworkData {
	let mut rx: Vec<(f64, f64)> = Vec::new<