2020-01-02 04:39:47 +00:00
|
|
|
use crate::{app, constants, data_conversion::ConvertedProcessData, utils::error, utils::gen_util::*};
|
2019-12-06 05:57:04 +00:00
|
|
|
use tui::{
|
2019-09-17 03:53:20 +00:00
|
|
|
backend,
|
2020-01-01 22:55:15 +00:00
|
|
|
layout::{Alignment, Constraint, Direction, Layout, Rect},
|
2019-09-13 20:15:00 +00:00
|
|
|
style::{Color, Modifier, Style},
|
2019-12-28 03:39:25 +00:00
|
|
|
terminal::Frame,
|
2020-01-01 22:55:15 +00:00
|
|
|
widgets::{Axis, Block, Borders, Chart, Dataset, Marker, Paragraph, Row, Table, Text, Widget},
|
2019-09-11 03:37:20 +00:00
|
|
|
Terminal,
|
|
|
|
};
|
2019-09-12 02:10:49 +00:00
|
|
|
|
2019-12-06 05:57:04 +00:00
|
|
|
const TEXT_COLOUR: Color = Color::Gray;
|
|
|
|
const GRAPH_COLOUR: Color = Color::Gray;
|
|
|
|
const BORDER_STYLE_COLOUR: Color = Color::Gray;
|
|
|
|
const HIGHLIGHTED_BORDER_STYLE_COLOUR: Color = Color::LightBlue;
|
2019-12-27 23:19:57 +00:00
|
|
|
const GOLDEN_RATIO: f32 = 0.618_034; // Approx, good enough for use (also Clippy gets mad if it's too long)
|
2019-12-27 00:06:30 +00:00
|
|
|
|
|
|
|
lazy_static! {
|
|
|
|
static ref HELP_TEXT: [Text<'static>; 14] = [
|
|
|
|
Text::raw("\nGeneral Keybindings\n"),
|
|
|
|
Text::raw("q, Ctrl-c to quit.\n"),
|
|
|
|
Text::raw("Ctrl-r to reset all data.\n"),
|
|
|
|
Text::raw("f to toggle freezing and unfreezing the display.\n"),
|
|
|
|
Text::raw("Ctrl+Up/k, Ctrl+Down/j, Ctrl+Left/h, Ctrl+Right/l to navigate between panels.\n"),
|
|
|
|
Text::raw("Up and Down scrolls through a list.\n"),
|
|
|
|
Text::raw("Esc to close a dialog window (help or dd confirmation).\n"),
|
|
|
|
Text::raw("? to get this help screen.\n"),
|
|
|
|
Text::raw("\n Process Panel Keybindings\n"),
|
|
|
|
Text::raw("dd to kill the selected process.\n"),
|
|
|
|
Text::raw("c to sort by CPU usage.\n"),
|
|
|
|
Text::raw("m to sort by memory usage.\n"),
|
|
|
|
Text::raw("p to sort by PID.\n"),
|
|
|
|
Text::raw("n to sort by process name.\n"),
|
|
|
|
];
|
|
|
|
static ref COLOUR_LIST: Vec<Color> = gen_n_colours(constants::NUM_COLOURS);
|
2019-12-28 03:39:25 +00:00
|
|
|
static ref CANVAS_BORDER_STYLE: Style = Style::default().fg(BORDER_STYLE_COLOUR);
|
|
|
|
static ref CANVAS_HIGHLIGHTED_BORDER_STYLE: Style = Style::default().fg(HIGHLIGHTED_BORDER_STYLE_COLOUR);
|
2019-12-27 00:06:30 +00:00
|
|
|
}
|
2019-09-12 02:10:49 +00:00
|
|
|
|
|
|
|
#[derive(Default)]
|
|
|
|
pub struct CanvasData {
|
2019-12-06 05:57:04 +00:00
|
|
|
pub rx_display: String,
|
|
|
|
pub tx_display: String,
|
2019-12-27 23:30:35 +00:00
|
|
|
pub total_rx_display: String,
|
|
|
|
pub total_tx_display: String,
|
2019-12-06 05:57:04 +00:00
|
|
|
pub network_data_rx: Vec<(f64, f64)>,
|
|
|
|
pub network_data_tx: Vec<(f64, f64)>,
|
|
|
|
pub disk_data: Vec<Vec<String>>,
|
|
|
|
pub temp_sensor_data: Vec<Vec<String>>,
|
2020-01-02 04:39:47 +00:00
|
|
|
pub process_data: Vec<ConvertedProcessData>,
|
2019-12-28 03:39:25 +00:00
|
|
|
pub memory_labels: Vec<(u64, u64)>,
|
2019-12-06 05:57:04 +00:00
|
|
|
pub mem_data: Vec<(f64, f64)>,
|
|
|
|
pub swap_data: Vec<(f64, f64)>,
|
|
|
|
pub cpu_data: Vec<(String, Vec<(f64, f64)>)>,
|
2019-09-12 02:10:49 +00:00
|
|
|
}
|
|
|
|
|
2019-12-27 00:06:30 +00:00
|
|
|
/// Generates random colours.
|
|
|
|
/// Strategy found from https://martin.ankerl.com/2009/12/09/how-to-create-random-colors-programmatically/
|
|
|
|
fn gen_n_colours(num_to_gen: i32) -> Vec<Color> {
|
2019-12-27 21:47:04 +00:00
|
|
|
fn gen_hsv(h: f32) -> f32 {
|
|
|
|
let new_val = h + GOLDEN_RATIO;
|
|
|
|
if new_val > 1.0 {
|
|
|
|
new_val.fract()
|
|
|
|
} else {
|
|
|
|
new_val
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/// This takes in an h, s, and v value of range [0, 1]
|
|
|
|
/// For explanation of what this does, see
|
|
|
|
/// https://en.wikipedia.org/wiki/HSL_and_HSV#HSV_to_RGB_alternative
|
|
|
|
fn hsv_to_rgb(hue: f32, saturation: f32, value: f32) -> (u8, u8, u8) {
|
|
|
|
fn hsv_helper(num: u32, hu: f32, sat: f32, val: f32) -> f32 {
|
|
|
|
let k = (num as f32 + hu * 6.0) % 6.0;
|
|
|
|
val - val * sat * float_max(float_min(k, float_min(4.1 - k, 1.1)), 0.0)
|
|
|
|
}
|
|
|
|
|
|
|
|
(
|
|
|
|
(hsv_helper(5, hue, saturation, value) * 255.0) as u8,
|
|
|
|
(hsv_helper(3, hue, saturation, value) * 255.0) as u8,
|
|
|
|
(hsv_helper(1, hue, saturation, value) * 255.0) as u8,
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Generate colours
|
2020-01-01 22:55:15 +00:00
|
|
|
let mut colour_vec: Vec<Color> = vec![Color::LightCyan, Color::LightYellow, Color::Red, Color::Green, Color::LightMagenta];
|
2019-12-27 00:06:30 +00:00
|
|
|
|
|
|
|
let mut h: f32 = 0.4; // We don't need random colours... right?
|
|
|
|
for _i in 0..num_to_gen {
|
|
|
|
h = gen_hsv(h);
|
|
|
|
let result = hsv_to_rgb(h, 0.5, 0.95);
|
|
|
|
colour_vec.push(Color::Rgb(result.0, result.1, result.2));
|
|
|
|
}
|
|
|
|
|
|
|
|
colour_vec
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
pub fn draw_data<B: backend::Backend>(terminal: &mut Terminal<B>, app_state: &mut app::App) -> error::Result<()> {
|
2019-10-09 23:19:39 +00:00
|
|
|
terminal.autoresize()?;
|
2019-09-12 02:10:49 +00:00
|
|
|
terminal.draw(|mut f| {
|
2019-12-07 05:58:52 +00:00
|
|
|
if app_state.show_help {
|
2020-01-02 04:39:47 +00:00
|
|
|
// Only for the help
|
2019-12-07 05:58:52 +00:00
|
|
|
let vertical_dialog_chunk = Layout::default()
|
|
|
|
.direction(Direction::Vertical)
|
|
|
|
.margin(1)
|
|
|
|
.constraints([Constraint::Percentage(32), Constraint::Percentage(40), Constraint::Percentage(28)].as_ref())
|
|
|
|
.split(f.size());
|
2019-09-12 03:15:25 +00:00
|
|
|
|
2019-12-07 05:58:52 +00:00
|
|
|
let middle_dialog_chunk = Layout::default()
|
|
|
|
.direction(Direction::Horizontal)
|
|
|
|
.margin(0)
|
|
|
|
.constraints([Constraint::Percentage(30), Constraint::Percentage(40), Constraint::Percentage(30)].as_ref())
|
|
|
|
.split(vertical_dialog_chunk[1]);
|
2019-10-10 02:00:10 +00:00
|
|
|
|
2019-12-27 00:06:30 +00:00
|
|
|
Paragraph::new(HELP_TEXT.iter())
|
2019-10-10 02:50:36 +00:00
|
|
|
.block(Block::default().title("Help (Press Esc to close)").borders(Borders::ALL))
|
2019-10-10 02:00:10 +00:00
|
|
|
.style(Style::default().fg(Color::Gray))
|
|
|
|
.alignment(Alignment::Left)
|
|
|
|
.wrap(true)
|
|
|
|
.render(&mut f, middle_dialog_chunk[1]);
|
2020-01-02 04:39:47 +00:00
|
|
|
} else if app_state.show_dd {
|
|
|
|
let vertical_dialog_chunk = Layout::default()
|
|
|
|
.direction(Direction::Vertical)
|
|
|
|
.margin(1)
|
|
|
|
.constraints([Constraint::Percentage(40), Constraint::Percentage(20), Constraint::Percentage(40)].as_ref())
|
|
|
|
.split(f.size());
|
|
|
|
|
|
|
|
let middle_dialog_chunk = Layout::default()
|
|
|
|
.direction(Direction::Horizontal)
|
|
|
|
.margin(0)
|
|
|
|
.constraints([Constraint::Percentage(30), Constraint::Percentage(40), Constraint::Percentage(30)].as_ref())
|
|
|
|
.split(vertical_dialog_chunk[1]);
|
|
|
|
|
|
|
|
if let Some(dd_err) = app_state.dd_err.clone() {
|
|
|
|
let dd_text = [Text::raw(format!("\nFailure to properly kill the process - {}", dd_err))];
|
|
|
|
|
|
|
|
Paragraph::new(dd_text.iter())
|
|
|
|
.block(Block::default().title("Kill Process Error (Press Esc to close)").borders(Borders::ALL))
|
|
|
|
.style(Style::default().fg(Color::Gray))
|
|
|
|
.alignment(Alignment::Center)
|
|
|
|
.wrap(true)
|
|
|
|
.render(&mut f, middle_dialog_chunk[1]);
|
|
|
|
} else if let Some(process) = app_state.get_current_highlighted_process() {
|
|
|
|
let dd_text = [
|
|
|
|
Text::raw(format!(
|
|
|
|
"\nAre you sure you want to kill process {} with PID {}?",
|
|
|
|
process.name, process.pid
|
|
|
|
)),
|
|
|
|
Text::raw("\n\nPress ENTER to proceed, ESC to exit."),
|
|
|
|
Text::raw("\nNote that if bottom is frozen, it must be unfrozen for changes to be shown."),
|
|
|
|
];
|
|
|
|
|
|
|
|
Paragraph::new(dd_text.iter())
|
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.title("Kill Process Confirmation (Press Esc to close)")
|
|
|
|
.borders(Borders::ALL),
|
|
|
|
)
|
|
|
|
.style(Style::default().fg(Color::Gray))
|
|
|
|
.alignment(Alignment::Center)
|
|
|
|
.wrap(true)
|
|
|
|
.render(&mut f, middle_dialog_chunk[1]);
|
|
|
|
} else {
|
|
|
|
// This is a bit nasty, but it works well... I guess.
|
|
|
|
app_state.show_dd = false;
|
|
|
|
}
|
2019-12-06 05:57:04 +00:00
|
|
|
} else {
|
2019-10-10 02:00:10 +00:00
|
|
|
let vertical_chunks = Layout::default()
|
|
|
|
.direction(Direction::Vertical)
|
|
|
|
.margin(1)
|
2019-12-07 04:29:23 +00:00
|
|
|
.constraints([Constraint::Percentage(33), Constraint::Percentage(34), Constraint::Percentage(34)].as_ref())
|
2019-10-10 02:00:10 +00:00
|
|
|
.split(f.size());
|
|
|
|
|
|
|
|
let middle_chunks = Layout::default()
|
|
|
|
.direction(Direction::Horizontal)
|
|
|
|
.margin(0)
|
|
|
|
.constraints([Constraint::Percentage(60), Constraint::Percentage(40)].as_ref())
|
|
|
|
.split(vertical_chunks[1]);
|
|
|
|
|
|
|
|
let middle_divided_chunk_2 = Layout::default()
|
|
|
|
.direction(Direction::Vertical)
|
|
|
|
.margin(0)
|
|
|
|
.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
|
|
|
|
.split(middle_chunks[1]);
|
|
|
|
|
|
|
|
let bottom_chunks = Layout::default()
|
|
|
|
.direction(Direction::Horizontal)
|
|
|
|
.margin(0)
|
|
|
|
.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
|
|
|
|
.split(vertical_chunks[2]);
|
|
|
|
|
2019-12-11 06:01:24 +00:00
|
|
|
// Component specific chunks
|
2019-12-28 03:39:25 +00:00
|
|
|
let cpu_chunk = Layout::default()
|
2019-12-11 06:01:24 +00:00
|
|
|
.direction(Direction::Horizontal)
|
|
|
|
.margin(0)
|
2019-12-28 03:39:25 +00:00
|
|
|
.constraints(
|
|
|
|
if app_state.left_legend {
|
2020-01-01 22:56:28 +00:00
|
|
|
[Constraint::Percentage(15), Constraint::Percentage(85)]
|
2019-12-28 03:39:25 +00:00
|
|
|
} else {
|
2020-01-01 22:56:28 +00:00
|
|
|
[Constraint::Percentage(85), Constraint::Percentage(15)]
|
2019-12-28 03:39:25 +00:00
|
|
|
}
|
|
|
|
.as_ref(),
|
|
|
|
)
|
2019-12-11 06:01:24 +00:00
|
|
|
.split(vertical_chunks[0]);
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
let network_chunk = Layout::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.direction(Direction::Vertical)
|
2019-12-11 06:01:24 +00:00
|
|
|
.margin(0)
|
2019-12-28 03:39:25 +00:00
|
|
|
.constraints(
|
|
|
|
if app_state.left_legend {
|
|
|
|
[Constraint::Percentage(10), Constraint::Percentage(90)]
|
|
|
|
} else {
|
2019-12-28 06:20:05 +00:00
|
|
|
[Constraint::Percentage(75), Constraint::Percentage(10)]
|
2019-12-28 03:39:25 +00:00
|
|
|
}
|
|
|
|
.as_ref(),
|
|
|
|
)
|
2019-12-11 06:01:24 +00:00
|
|
|
.split(bottom_chunks[0]);
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
// Default chunk index based on left or right legend setting
|
|
|
|
let legend_index = if app_state.left_legend { 0 } else { 1 };
|
|
|
|
let graph_index = if app_state.left_legend { 1 } else { 0 };
|
2019-10-10 02:00:10 +00:00
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
// Set up blocks and their components
|
|
|
|
// CPU graph
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_cpu_graph(&mut f, &app_state, cpu_chunk[graph_index]);
|
2019-09-15 05:29:49 +00:00
|
|
|
|
2020-01-01 22:55:15 +00:00
|
|
|
// CPU legend
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_cpu_legend(&mut f, app_state, cpu_chunk[legend_index]);
|
2019-09-15 05:29:49 +00:00
|
|
|
|
2019-10-10 02:00:10 +00:00
|
|
|
//Memory usage graph
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_memory_graph(&mut f, &app_state, middle_chunks[0]);
|
2019-10-10 02:00:10 +00:00
|
|
|
|
|
|
|
// Network graph
|
2020-01-02 23:34:09 +00:00
|
|
|
draw_network_graph(&mut f, &app_state, network_chunk[0]);
|
2019-10-10 02:00:10 +00:00
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_network_labels(&mut f, app_state, network_chunk[1]);
|
2019-12-28 06:20:05 +00:00
|
|
|
|
|
|
|
// Temperature table
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_temp_table(&mut f, app_state, middle_divided_chunk_2[0]);
|
2019-12-28 06:20:05 +00:00
|
|
|
|
|
|
|
// Disk usage table
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_disk_table(&mut f, app_state, middle_divided_chunk_2[1]);
|
2019-10-10 02:00:10 +00:00
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
// Processes table
|
2020-01-02 04:39:47 +00:00
|
|
|
draw_processes_table(&mut f, app_state, bottom_chunks[1]);
|
2019-09-14 20:46:14 +00:00
|
|
|
}
|
2019-09-12 02:10:49 +00:00
|
|
|
})?;
|
|
|
|
|
|
|
|
Ok(())
|
|
|
|
}
|
2019-09-25 20:43:13 +00:00
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_cpu_graph<B: backend::Backend>(f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect) {
|
|
|
|
let cpu_data: &[(String, Vec<(f64, f64)>)] = &app_state.canvas_data.cpu_data;
|
2019-12-28 03:39:25 +00:00
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
// CPU usage graph
|
2019-12-28 03:39:25 +00:00
|
|
|
let x_axis: Axis<String> = Axis::default()
|
|
|
|
.style(Style::default().fg(GRAPH_COLOUR))
|
|
|
|
.bounds([0.0, constants::TIME_STARTS_FROM as f64 * 10.0]);
|
|
|
|
let y_axis = Axis::default()
|
|
|
|
.style(Style::default().fg(GRAPH_COLOUR))
|
|
|
|
.bounds([-0.5, 100.5])
|
|
|
|
.labels(&["0%", "100%"]);
|
|
|
|
|
|
|
|
let mut dataset_vector: Vec<Dataset> = Vec::new();
|
|
|
|
|
|
|
|
for (i, cpu) in cpu_data.iter().enumerate() {
|
|
|
|
let mut avg_cpu_exist_offset = 0;
|
|
|
|
if app_state.show_average_cpu {
|
|
|
|
if i == 0 {
|
|
|
|
// Skip, we want to render the average cpu last!
|
|
|
|
continue;
|
|
|
|
} else {
|
|
|
|
avg_cpu_exist_offset = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dataset_vector.push(
|
|
|
|
Dataset::default()
|
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[(i - avg_cpu_exist_offset) % COLOUR_LIST.len()]))
|
|
|
|
.data(&(cpu.1)),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
if !cpu_data.is_empty() && app_state.show_average_cpu {
|
|
|
|
// Unwrap should be safe here, this assumes that the cpu_data vector is populated...
|
|
|
|
dataset_vector.push(
|
|
|
|
Dataset::default()
|
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[(cpu_data.len() - 1) % COLOUR_LIST.len()]))
|
|
|
|
.data(&(cpu_data.first().unwrap().1)),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
Chart::default()
|
|
|
|
.block(
|
|
|
|
Block::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.title("CPU")
|
2019-12-28 03:39:25 +00:00
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Cpu => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.x_axis(x_axis)
|
|
|
|
.y_axis(y_axis)
|
|
|
|
.datasets(&dataset_vector)
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_cpu_legend<B: backend::Backend>(f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect) {
|
|
|
|
let cpu_data: &[(String, Vec<(f64, f64)>)] = &(app_state.canvas_data.cpu_data);
|
|
|
|
|
2020-01-01 22:55:15 +00:00
|
|
|
let num_rows = i64::from(draw_loc.height) - 4;
|
|
|
|
let start_position = get_start_position(
|
|
|
|
num_rows,
|
|
|
|
&(app_state.scroll_direction),
|
|
|
|
&mut app_state.previous_cpu_table_position,
|
|
|
|
&mut app_state.currently_selected_cpu_table_position,
|
|
|
|
);
|
|
|
|
|
|
|
|
let sliced_cpu_data = (&cpu_data[start_position as usize..]).to_vec();
|
|
|
|
let mut stringified_cpu_data: Vec<Vec<String>> = Vec::new();
|
|
|
|
|
|
|
|
for cpu in sliced_cpu_data {
|
|
|
|
if let Some(cpu_data) = cpu.1.last() {
|
|
|
|
stringified_cpu_data.push(vec![cpu.0.clone(), format!("{:.0}%", cpu_data.1.round())]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
let mut cpu_row_counter = 0;
|
|
|
|
|
|
|
|
let cpu_rows = stringified_cpu_data.iter().enumerate().map(|(itx, cpu_string_row)| {
|
|
|
|
Row::StyledData(
|
|
|
|
cpu_string_row.iter(),
|
|
|
|
if cpu_row_counter == app_state.currently_selected_cpu_table_position - start_position {
|
|
|
|
cpu_row_counter = -1;
|
|
|
|
Style::default().fg(Color::Black).bg(Color::Cyan)
|
|
|
|
} else {
|
|
|
|
if cpu_row_counter >= 0 {
|
|
|
|
cpu_row_counter += 1;
|
|
|
|
}
|
|
|
|
Style::default().fg(COLOUR_LIST[itx % COLOUR_LIST.len()])
|
|
|
|
},
|
|
|
|
)
|
2019-12-28 06:20:05 +00:00
|
|
|
});
|
2019-12-28 03:39:25 +00:00
|
|
|
|
2020-01-01 22:55:15 +00:00
|
|
|
Table::new(["CPU", "Use%"].iter(), cpu_rows)
|
2019-12-28 06:20:05 +00:00
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
|
|
|
app::ApplicationPosition::Cpu => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
2020-01-01 22:55:15 +00:00
|
|
|
.header_style(Style::default().fg(Color::LightBlue))
|
|
|
|
.widths(&[Constraint::Percentage(50), Constraint::Percentage(50)])
|
2019-12-28 06:20:05 +00:00
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_memory_graph<B: backend::Backend>(f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect) {
|
|
|
|
let mem_data: &[(f64, f64)] = &(app_state.canvas_data.mem_data);
|
|
|
|
let swap_data: &[(f64, f64)] = &(app_state.canvas_data.swap_data);
|
|
|
|
let memory_labels: &[(u64, u64)] = &(app_state.canvas_data.memory_labels);
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
let x_axis: Axis<String> = Axis::default()
|
|
|
|
.style(Style::default().fg(GRAPH_COLOUR))
|
|
|
|
.bounds([0.0, constants::TIME_STARTS_FROM as f64 * 10.0]);
|
|
|
|
let y_axis = Axis::default()
|
|
|
|
.style(Style::default().fg(GRAPH_COLOUR))
|
|
|
|
.bounds([-0.5, 100.5]) // Offset as the zero value isn't drawn otherwise...
|
|
|
|
.labels(&["0%", "100%"]);
|
|
|
|
|
2019-12-28 06:20:05 +00:00
|
|
|
let mem_name = "RAM:".to_string()
|
|
|
|
+ &format!("{:3}%", (mem_data.last().unwrap_or(&(0_f64, 0_f64)).1.round() as u64))
|
|
|
|
+ &format!(
|
|
|
|
" {:.1}GB/{:.1}GB",
|
|
|
|
memory_labels.first().unwrap_or(&(0, 0)).0 as f64 / 1024.0,
|
|
|
|
memory_labels.first().unwrap_or(&(0, 0)).1 as f64 / 1024.0
|
|
|
|
);
|
|
|
|
let swap_name: String;
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
let mut mem_canvas_vec: Vec<Dataset> = vec![Dataset::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.name(&mem_name)
|
2019-12-28 03:39:25 +00:00
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[0]))
|
|
|
|
.data(&mem_data)];
|
|
|
|
|
|
|
|
if !(&swap_data).is_empty() {
|
|
|
|
if let Some(last_canvas_result) = (&swap_data).last() {
|
|
|
|
if last_canvas_result.1 >= 0.0 {
|
2019-12-28 06:20:05 +00:00
|
|
|
swap_name = "SWP:".to_string()
|
|
|
|
+ &format!("{:3}%", (swap_data.last().unwrap_or(&(0_f64, 0_f64)).1.round() as u64))
|
|
|
|
+ &format!(
|
|
|
|
" {:.1}GB/{:.1}GB",
|
|
|
|
memory_labels[1].0 as f64 / 1024.0,
|
|
|
|
memory_labels[1].1 as f64 / 1024.0
|
|
|
|
);
|
2019-12-28 03:39:25 +00:00
|
|
|
mem_canvas_vec.push(
|
|
|
|
Dataset::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.name(&swap_name)
|
2019-12-28 03:39:25 +00:00
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[1]))
|
|
|
|
.data(&swap_data),
|
|
|
|
);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
Chart::default()
|
|
|
|
.block(
|
|
|
|
Block::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.title("Memory")
|
2019-12-28 03:39:25 +00:00
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Mem => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.x_axis(x_axis)
|
|
|
|
.y_axis(y_axis)
|
|
|
|
.datasets(&mem_canvas_vec)
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_network_graph<B: backend::Backend>(f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect) {
|
|
|
|
let network_data_rx: &[(f64, f64)] = &(app_state.canvas_data.network_data_rx);
|
|
|
|
let network_data_tx: &[(f64, f64)] = &(app_state.canvas_data.network_data_tx);
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
let x_axis: Axis<String> = Axis::default().style(Style::default().fg(GRAPH_COLOUR)).bounds([0.0, 600_000.0]);
|
|
|
|
let y_axis = Axis::default()
|
|
|
|
.style(Style::default().fg(GRAPH_COLOUR))
|
|
|
|
.bounds([-0.5, 30_f64])
|
|
|
|
.labels(&["0B", "1KiB", "1MiB", "1GiB"]);
|
|
|
|
Chart::default()
|
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.title("Network")
|
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Network => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.x_axis(x_axis)
|
|
|
|
.y_axis(y_axis)
|
|
|
|
.datasets(&[
|
|
|
|
Dataset::default()
|
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[0]))
|
|
|
|
.data(&network_data_rx),
|
|
|
|
Dataset::default()
|
|
|
|
.marker(if app_state.use_dot { Marker::Dot } else { Marker::Braille })
|
|
|
|
.style(Style::default().fg(COLOUR_LIST[1]))
|
|
|
|
.data(&network_data_tx),
|
|
|
|
])
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_network_labels<B: backend::Backend>(f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect) {
|
|
|
|
let rx_display: String = app_state.canvas_data.rx_display.clone();
|
|
|
|
let tx_display: String = app_state.canvas_data.tx_display.clone();
|
|
|
|
let total_rx_display: String = app_state.canvas_data.total_rx_display.clone();
|
|
|
|
let total_tx_display: String = app_state.canvas_data.total_tx_display.clone();
|
|
|
|
|
2019-12-28 06:20:05 +00:00
|
|
|
// Gross but I need it to work...
|
2020-01-02 23:34:09 +00:00
|
|
|
let total_network = if cfg!(not(target_os = "windows")) {
|
|
|
|
vec![vec![rx_display, tx_display, total_rx_display, total_tx_display]]
|
|
|
|
} else {
|
|
|
|
vec![vec![rx_display, tx_display]]
|
|
|
|
};
|
2019-12-28 06:20:05 +00:00
|
|
|
let mapped_network = total_network.iter().map(|val| Row::Data(val.iter()));
|
|
|
|
|
2020-01-02 23:34:09 +00:00
|
|
|
Table::new(
|
|
|
|
if cfg!(not(target_os = "windows")) {
|
|
|
|
vec!["RX", "TX", "Total RX", "Total TX"]
|
|
|
|
} else {
|
|
|
|
vec!["RX", "TX"]
|
|
|
|
}
|
|
|
|
.iter(),
|
|
|
|
mapped_network,
|
|
|
|
)
|
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
|
|
|
app::ApplicationPosition::Network => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.header_style(Style::default().fg(Color::LightBlue))
|
|
|
|
.widths(&if cfg!(not(target_os = "windows")) {
|
|
|
|
vec![
|
2020-01-01 21:32:20 +00:00
|
|
|
Constraint::Percentage(25),
|
|
|
|
Constraint::Percentage(25),
|
|
|
|
Constraint::Percentage(25),
|
|
|
|
Constraint::Percentage(25),
|
2020-01-02 23:34:09 +00:00
|
|
|
]
|
|
|
|
} else {
|
|
|
|
vec![Constraint::Percentage(50), Constraint::Percentage(50)]
|
|
|
|
})
|
|
|
|
.render(f, draw_loc);
|
2019-12-28 06:20:05 +00:00
|
|
|
}
|
2019-12-28 03:39:25 +00:00
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_temp_table<B: backend::Backend>(f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect) {
|
|
|
|
let temp_sensor_data: &[Vec<String>] = &(app_state.canvas_data.temp_sensor_data);
|
|
|
|
|
2019-12-28 03:39:25 +00:00
|
|
|
let num_rows = i64::from(draw_loc.height) - 4;
|
|
|
|
let start_position = get_start_position(
|
|
|
|
num_rows,
|
|
|
|
&(app_state.scroll_direction),
|
|
|
|
&mut app_state.previous_temp_position,
|
|
|
|
&mut app_state.currently_selected_temperature_position,
|
|
|
|
);
|
|
|
|
|
|
|
|
let sliced_vec: Vec<Vec<String>> = (&temp_sensor_data[start_position as usize..]).to_vec();
|
2020-01-01 22:55:15 +00:00
|
|
|
let mut temp_row_counter = 0;
|
2019-12-28 03:39:25 +00:00
|
|
|
|
2020-01-01 22:55:15 +00:00
|
|
|
let temperature_rows = sliced_vec.iter().map(|temp_row| {
|
2019-12-28 03:39:25 +00:00
|
|
|
Row::StyledData(
|
2020-01-01 22:55:15 +00:00
|
|
|
temp_row.iter(),
|
|
|
|
if temp_row_counter == app_state.currently_selected_temperature_position - start_position {
|
|
|
|
temp_row_counter = -1;
|
2019-12-28 03:39:25 +00:00
|
|
|
Style::default().fg(Color::Black).bg(Color::Cyan)
|
|
|
|
} else {
|
2020-01-01 22:55:15 +00:00
|
|
|
if temp_row_counter >= 0 {
|
|
|
|
temp_row_counter += 1;
|
2019-12-28 03:39:25 +00:00
|
|
|
}
|
|
|
|
Style::default().fg(TEXT_COLOUR)
|
|
|
|
},
|
|
|
|
)
|
|
|
|
});
|
|
|
|
|
|
|
|
let width = f64::from(draw_loc.width);
|
|
|
|
Table::new(["Sensor", "Temp"].iter(), temperature_rows)
|
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.title("Temperatures")
|
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Temp => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.header_style(Style::default().fg(Color::LightBlue))
|
|
|
|
.widths(&[Constraint::Length((width * 0.45) as u16), Constraint::Length((width * 0.4) as u16)])
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_disk_table<B: backend::Backend>(f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect) {
|
|
|
|
let disk_data: &[Vec<String>] = &(app_state.canvas_data.disk_data);
|
2019-12-28 03:39:25 +00:00
|
|
|
let num_rows = i64::from(draw_loc.height) - 4;
|
|
|
|
let start_position = get_start_position(
|
|
|
|
num_rows,
|
|
|
|
&(app_state.scroll_direction),
|
|
|
|
&mut app_state.previous_disk_position,
|
|
|
|
&mut app_state.currently_selected_disk_position,
|
|
|
|
);
|
|
|
|
|
|
|
|
let sliced_vec: Vec<Vec<String>> = (&disk_data[start_position as usize..]).to_vec();
|
|
|
|
let mut disk_counter = 0;
|
|
|
|
|
|
|
|
let disk_rows = sliced_vec.iter().map(|disk| {
|
|
|
|
Row::StyledData(
|
|
|
|
disk.iter(),
|
|
|
|
if disk_counter == app_state.currently_selected_disk_position - start_position {
|
|
|
|
disk_counter = -1;
|
|
|
|
Style::default().fg(Color::Black).bg(Color::Cyan)
|
|
|
|
} else {
|
|
|
|
if disk_counter >= 0 {
|
|
|
|
disk_counter += 1;
|
|
|
|
}
|
|
|
|
Style::default().fg(TEXT_COLOUR)
|
|
|
|
},
|
|
|
|
)
|
|
|
|
});
|
|
|
|
|
|
|
|
// TODO: We may have to dynamically remove some of these table elements based on size...
|
|
|
|
let width = f64::from(draw_loc.width);
|
|
|
|
Table::new(["Disk", "Mount", "Used", "Total", "Free", "R/s", "W/s"].iter(), disk_rows)
|
|
|
|
.block(
|
|
|
|
Block::default()
|
2019-12-28 06:20:05 +00:00
|
|
|
.title("Disk")
|
2019-12-28 03:39:25 +00:00
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Disk => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.header_style(Style::default().fg(Color::LightBlue).modifier(Modifier::BOLD))
|
|
|
|
.widths(&[
|
|
|
|
Constraint::Length((width * 0.18).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.14).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.11).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.11).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.11).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.11).floor() as u16),
|
|
|
|
Constraint::Length((width * 0.11).floor() as u16),
|
|
|
|
])
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
fn draw_processes_table<B: backend::Backend>(f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect) {
|
|
|
|
let process_data: &[ConvertedProcessData] = &(app_state.canvas_data.process_data);
|
2019-12-28 03:39:25 +00:00
|
|
|
let width = f64::from(draw_loc.width);
|
|
|
|
|
|
|
|
// Admittedly this is kinda a hack... but we need to:
|
|
|
|
// * Scroll
|
|
|
|
// * Show/hide elements based on scroll position
|
|
|
|
// As such, we use a process_counter to know when we've hit the process we've currently scrolled to. We also need to move the list - we can
|
|
|
|
// do so by hiding some elements!
|
|
|
|
let num_rows = i64::from(draw_loc.height) - 4;
|
|
|
|
|
|
|
|
let start_position = get_start_position(
|
|
|
|
num_rows,
|
|
|
|
&(app_state.scroll_direction),
|
|
|
|
&mut app_state.previous_process_position,
|
|
|
|
&mut app_state.currently_selected_process_position,
|
|
|
|
);
|
|
|
|
|
2020-01-02 04:39:47 +00:00
|
|
|
let sliced_vec: Vec<ConvertedProcessData> = (&process_data[start_position as usize..]).to_vec();
|
2019-12-28 03:39:25 +00:00
|
|
|
let mut process_counter = 0;
|
|
|
|
|
|
|
|
let process_rows = sliced_vec.iter().map(|process| {
|
2020-01-02 04:39:47 +00:00
|
|
|
let stringified_process_vec: Vec<String> = vec![
|
|
|
|
process.pid.to_string(),
|
|
|
|
process.name.clone(),
|
|
|
|
process.cpu_usage.clone(),
|
|
|
|
process.mem_usage.clone(),
|
|
|
|
];
|
2019-12-28 03:39:25 +00:00
|
|
|
Row::StyledData(
|
2020-01-02 04:39:47 +00:00
|
|
|
stringified_process_vec.into_iter(),
|
2019-12-28 03:39:25 +00:00
|
|
|
if process_counter == app_state.currently_selected_process_position - start_position {
|
|
|
|
process_counter = -1;
|
|
|
|
Style::default().fg(Color::Black).bg(Color::Cyan)
|
|
|
|
} else {
|
|
|
|
if process_counter >= 0 {
|
|
|
|
process_counter += 1;
|
|
|
|
}
|
|
|
|
Style::default().fg(TEXT_COLOUR)
|
|
|
|
},
|
|
|
|
)
|
|
|
|
});
|
|
|
|
|
|
|
|
{
|
|
|
|
use app::data_collection::processes::ProcessSorting;
|
|
|
|
let mut pid = "PID(p)".to_string();
|
|
|
|
let mut name = "Name(n)".to_string();
|
|
|
|
let mut cpu = "CPU%(c)".to_string();
|
|
|
|
let mut mem = "Mem%(m)".to_string();
|
|
|
|
|
|
|
|
let direction_val = if app_state.process_sorting_reverse {
|
|
|
|
"⯆".to_string()
|
|
|
|
} else {
|
|
|
|
"⯅".to_string()
|
|
|
|
};
|
|
|
|
|
|
|
|
match app_state.process_sorting_type {
|
|
|
|
ProcessSorting::CPU => cpu += &direction_val,
|
|
|
|
ProcessSorting::MEM => mem += &direction_val,
|
|
|
|
ProcessSorting::PID => pid += &direction_val,
|
|
|
|
ProcessSorting::NAME => name += &direction_val,
|
|
|
|
};
|
|
|
|
|
|
|
|
Table::new([pid, name, cpu, mem].iter(), process_rows)
|
|
|
|
.block(
|
|
|
|
Block::default()
|
|
|
|
.title("Processes")
|
|
|
|
.borders(Borders::ALL)
|
|
|
|
.border_style(match app_state.current_application_position {
|
2019-12-28 06:20:05 +00:00
|
|
|
app::ApplicationPosition::Process => *CANVAS_HIGHLIGHTED_BORDER_STYLE,
|
2019-12-28 03:39:25 +00:00
|
|
|
_ => *CANVAS_BORDER_STYLE,
|
|
|
|
}),
|
|
|
|
)
|
|
|
|
.header_style(Style::default().fg(Color::LightBlue))
|
|
|
|
.widths(&[
|
|
|
|
Constraint::Length((width * 0.2) as u16),
|
|
|
|
Constraint::Length((width * 0.35) as u16),
|
|
|
|
Constraint::Length((width * 0.2) as u16),
|
|
|
|
Constraint::Length((width * 0.2) as u16),
|
|
|
|
])
|
|
|
|
.render(f, draw_loc);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-25 20:43:13 +00:00
|
|
|
fn get_start_position(
|
2019-12-06 05:57:04 +00:00
|
|
|
num_rows: i64, scroll_direction: &app::ScrollDirection, previous_position: &mut i64, currently_selected_position: &mut i64,
|
2019-09-25 20:43:13 +00:00
|
|
|
) -> i64 {
|
|
|
|
match scroll_direction {
|
|
|
|
app::ScrollDirection::DOWN => {
|
|
|
|
if *currently_selected_position < num_rows {
|
|
|
|
0
|
2019-12-06 05:57:04 +00:00
|
|
|
} else if *currently_selected_position - num_rows < *previous_position {
|
2019-09-25 20:43:13 +00:00
|
|
|
*previous_position
|
2019-12-06 05:57:04 +00:00
|
|
|
} else {
|
2019-09-25 20:43:13 +00:00
|
|
|
*previous_position = *currently_selected_position - num_rows + 1;
|
|
|
|
*previous_position
|
|
|
|
}
|
|
|
|
}
|
|
|
|
app::ScrollDirection::UP => {
|
2019-12-28 06:20:05 +00:00
|
|
|
if *currently_selected_position == 0 {
|
|
|
|
*previous_position = 0;
|
|
|
|
return *previous_position;
|
|
|
|
}
|
|
|
|
|
2019-09-25 20:43:13 +00:00
|
|
|
if *currently_selected_position == *previous_position - 1 {
|
|
|
|
*previous_position = if *previous_position > 0 { *previous_position - 1 } else { 0 };
|
|
|
|
*previous_position
|
2019-12-06 05:57:04 +00:00
|
|
|
} else {
|
2019-09-25 20:43:13 +00:00
|
|
|
*previous_position
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|