cpumap 0.2.1

GUI/TUI to view and edit CPU affinities of processes and threads on Linux
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75

use std::sync::mpsc;
use crate::system::System;
use crate::uistate::UIBackend;
use crate::uifrontend::UIFrontend;
use crate::uifrontend::UIFrontendType;
use crate::topologycache::TopologyCache;

#[cfg(feature = "gui")]
use crate::eguifrontend::EguiFrontend;

/// Events that can happen outside the UI thread that the UI needs to handle.
pub enum Event {
    /// Process/system info update tick.
    Tick,
}

/// Main 'application'.
pub struct CPUMap {
    /// System information and configuration (cpu topology, getting/setting affinities, etc.)
    pub(crate) system:   System,

    /// Application state.
    pub(crate) backend:  UIBackend,

    /// Renders UI.
    pub(crate) frontend: UIFrontend,
}

impl CPUMap {
    /// Construct CPUMap with egui GUI frontend, see CPUMap member documentation.
    pub fn new(
        topo:          hwloc2::Topology,
        system:        sysinfo::System,
        event_rx:      mpsc::Receiver<Event>,
        frontend_type: UIFrontendType) -> Self
    {
        let cache  = TopologyCache::new(topo.object_at_root(), system.cpus())
                     .expect("Failed to init topology cache");
        // TODO maybe replace topo by cache completely here
        let system = System::new( system, cache, topo );
        let ui     = UIBackend::new(&system, event_rx);
        Self {
            system,
            backend: ui,
            frontend: match frontend_type {
                UIFrontendType::Egui => {
                    #[cfg(feature = "gui")] { Some(UIFrontend::egui()) }
                    #[cfg(not(feature = "gui"))] { None }
                }
                UIFrontendType::Ratatui => {
                    #[cfg(feature = "tui")] { Some(UIFrontend::ratatui())} 
                    #[cfg(not(feature = "tui"))] {None }
                }
            }.unwrap()
        }
    }

    /// Run and consume CPUMap, returning any fatal error.
    pub fn run(mut self) -> anyhow::Result<()> {
        #[cfg(any(feature = "gui", feature = "tui"))]
        match self.frontend {
            #[cfg(feature = "gui")]
            UIFrontend::Egui(_) => {
                EguiFrontend::run(Box::new(move|_cc| Box::new(self)))
            },
            #[cfg(feature = "tui")]
            UIFrontend::Ratatui(mut ratatui) => {
                ratatui.run(&mut self.backend, &mut self.system)
            }
        }
        #[cfg(not(any(feature = "gui", feature = "tui")))]
        Ok(())
    }
}