mltop 0.5.0

Resource monitor for ML engineers written in Rust
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::widgets::percentage_bar::percentage_bar;
use crate::{constants::BYTES_PER_GB, data::memory::MemorySnapshot};

use ratatui::{
    buffer::Buffer,
    layout::Rect,
    style::{Color, Style},
    text::{Line, Span},
    widgets::{Paragraph, Widget},
};

pub const MEMORY_WIDGET_HEIGHT: u16 = 1;

pub struct MemoryWidget {}

impl MemoryWidget {
    pub fn new() -> MemoryWidget {
        MemoryWidget {}
    }
}

impl MemoryWidget {
    pub fn render(&self, area: Rect, buf: &mut Buffer, data: &MemorySnapshot) {
        // memory
        let mut spans = vec![Span::styled(" Memory", Style::default().fg(Color::Yellow))];
        let percentage = data.used as f32 / data.total as f32 * 100.0;
        let text = format!(
            "{:.1}G/{:.1}G",
            data.used as f32 / BYTES_PER_GB as f32,
            data.total as f32 / BYTES_PER_GB as f32
        );
        let mem_bar_width = (area.width / 2) - 12;
        spans.extend(percentage_bar(mem_bar_width, percentage, &text));

        // swap
        spans.extend(vec![Span::styled(
            "    Swap",
            Style::default().fg(Color::Yellow),
        )]);
        let percentage = data.used_swap as f32 / data.total_swap as f32 * 100.0;
        let text = format!(
            "{:.1}G/{:.1}G",
            data.used_swap as f32 / BYTES_PER_GB as f32,
            data.total_swap as f32 / BYTES_PER_GB as f32
        );
        let swap_bar_width = area.width - mem_bar_width - 26;
        spans.extend(percentage_bar(swap_bar_width, percentage, &text));

        Paragraph::new(Line::from(spans))
            .left_aligned()
            .render(area, buf);
    }
}