Measure CPU usage, not frequency

1 files changed, 142 insertions, 109 deletions

diff --git a/src/main.rs b/src/main.rs
index 956285b..60c02e4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,127 +1,143 @@
 #[macro_use]
 extern crate clap;
-extern crate num;
+extern crate vec_map;
+extern crate libc;
 
-use std::path::{Path, PathBuf};
-use std::fs::File;
-use std::io::{self, Read};
-use std::{iter, thread, ops};
-use std::time::Duration;
-
-use clap::{Arg, App, AppSettings, SubCommand};
-
-use num::NumCast;
-
-const SYS_PRESENT: &'static str = "/sys/devices/system/cpu/present";
-static FORMAT: [char; 9] = [' ', '▁', '▂', '▃', '▄', '▅', '▆', '▇', '█'];
-
-const WINDOW_SIZE: usize = 32;
-
-#[derive(Debug, Default)]
-struct Window<T> {
-    data: Vec<T>,
-    idx: usize,
-    size: usize
-}
-
-impl<T> Window<T> where T: for<'a> iter::Sum<&'a T> + ops::Div<Output=T> + NumCast {
-    fn new(size: usize) -> Window<T> {
-        Window { data: Vec::with_capacity(size), idx: 0, size: size }
-    }
-
-    fn sample(&mut self, value: T) {
-        if self.data.len() < self.size { self.data.push(value) }
-        else { self.data[self.idx] = value; }
-        self.idx = (self.idx + 1) % self.size;
-    }
-
-    fn elements(&self) -> (&[T], &[T]) { (&self.data[self.idx..], &self.data[..self.idx]) }
-    fn average(&self) -> T { self.data.iter().sum::<T>() / T::from(self.data.len()).unwrap() }
-}
-
-#[derive(Debug, Default)]
-struct Core {
-    index: u32,
-    min_freq: u32,
-    max_freq: u32,
-    cur_freq: Window<u32>
-}
+#[macro_use]
+extern crate log;
+extern crate env_logger;
 
+use std::fs::File;
+use std::io::{self, BufRead, BufReader};
+use std::thread;
+use std::time::{Instant, Duration};
+
+use clap::{Arg, App};
+
+use vec_map::VecMap;
+
+const PROC_STAT: &'static str = "/proc/stat";
+static FORMAT: [char; 8] = ['▁', '▂', '▃', '▄', '▅', '▆', '▇', '█'];
+
+/// cpu  3357 0 4313 1362393
+///   The  amount  of  time, measured in units of USER_HZ (1/100ths of a
+///   second on most architectures, use sysconf(_SC_CLK_TCK)  to  obtain
+///   the right value), that the system spent in various states:
+///
+///   user   (1) Time spent in user mode.
+///
+///   nice   (2) Time spent in user mode with low priority (nice).
+///
+///   system (3) Time spent in system mode.
+///
+///   idle   (4) Time  spent  in  the  idle task.  This value should be
+///              USER_HZ times the second entry in the /proc/uptime  pseudo-
+///              file.
+///
+///   iowait (since Linux 2.5.41)
+///          (5) Time waiting for I/O to complete.
+///
+///   irq (since Linux 2.6.0-test4)
+///          (6) Time servicing interrupts.
+///
+///   softirq (since Linux 2.6.0-test4)
+///          (7) Time servicing softirqs.
+///
+///   steal (since Linux 2.6.11)
+///          (8) Stolen time, which is the time spent in other operating
+///              systems when running in a virtualized environment
+///
+///   guest (since Linux 2.6.24)
+///          (9) Time spent running a virtual CPU  for  guest  operating
+///              systems under the control of the Linux kernel.
+///
+///   guest_nice (since Linux 2.6.33)
+///        (10)  Time  spent  running  a  niced guest (virtual CPU for
+///              guest operating systems under the control of the Linux ker‐
+///              nel).
 #[derive(Debug)]
-struct System {
-    cores: Vec<Core>,
-    load: Window<f32>
+struct Stat {
+    time: Instant,
+    total: Option<CPU>,
+    cores: VecMap<CPU>
 }
 
-impl System {
-    fn new() -> System {
-        System {
-            cores: init_cores(),
-            load: Window::new(8)
+impl Stat {
+    pub fn read() -> io::Result<Stat> {
+        let file = try!(File::open(PROC_STAT));
+        let reader = BufReader::new(file);
+        let mut stat = Stat { time: Instant::now(), total: None, cores: VecMap::new() };
+
+        for line in reader.lines() {
+            let line = try!(line);
+            const OFFSET: usize = 3; // "cpu".len()
+            if line.starts_with("cpu ") {
+                stat.total = Some(CPU::from_line(&line[OFFSET..])); 
+            } else if line.starts_with("cpu") {
+                let num: u64 = line[OFFSET..].split_whitespace().next().and_then(|s| s.trim().parse().ok()).unwrap();
+                stat.cores.insert(num as usize, CPU::from_line(&line[OFFSET..]));
+            }
         }
-    }
 
-    // Direct printing to avoid allocation
-    fn print_cores(&self) {
-        for c in &self.cores {
-            let load = (c.cur_freq.average() - c.min_freq) as f32 / (c.max_freq - c.min_freq) as f32;
-            print!("{}", FORMAT[(FORMAT.len() as f32 * load) as usize]);
-        }
-        println!("");
+        Ok(stat)
     }
 
-    fn print_system(&self) {
-        let (first, last) = self.load.elements();
-        for load in first.iter().chain(last.iter()) {
-            print!("{}", FORMAT[(FORMAT.len() as f32 * load) as usize]);
+    pub fn load_since(&self, earlier: &Stat) -> Load {
+        Load {
+            duration: self.time.duration_since(earlier.time),
+            total: match (&self.total, &earlier.total) {
+                (&Some(ref now), &Some(ref old)) => Some(now.diff(old)),
+                _ => None
+            },
+            cores: self.cores.iter()
+                .flat_map(|(idx, core)| earlier.cores.get(idx).map(|ec| (idx, core.diff(ec))))
+                .collect()
         }
-        println!("");
     }
 }
 
-fn read_into_string<P: AsRef<Path>>(path: P) -> io::Result<String> {
-    let mut output = String::new();
-    try!(try!(File::open(path)).read_to_string(&mut output));
-    Ok(output)
+#[derive(Debug, Clone)]
+struct Load {
+    duration: Duration,
+    total: Option<CPU>,
+    cores: VecMap<CPU>
 }
 
-fn init_cores() -> Vec<Core> {
-    let present = read_into_string(SYS_PRESENT).expect("Can't read kernel interface to query present cores");
-    
-    // a-b
-    let mut parts = present.split('-').flat_map(|s| s.trim().parse::<u32>().ok());
-    let (a, b) = (parts.next().unwrap(), parts.next().unwrap());
-    (a..b + 1).map(|idx| {
-        let cpu_path: PathBuf = format!("/sys/devices/system/cpu/cpu{}/cpufreq", idx).into();
-        Core {
-            index: idx,
-            min_freq: read_into_string(cpu_path.join("scaling_min_freq")).ok()
-                .and_then(|s| s.trim().parse().ok()).expect("Can't read min freq"),
-            max_freq: read_into_string(cpu_path.join("scaling_max_freq")).ok()
-                .and_then(|s| s.trim().parse().ok()).expect("Can't read max frequency"),
-            cur_freq: Window::new(WINDOW_SIZE)
-        }
-    }).collect()
+#[derive(Debug, Clone)]
+struct CPU {
+    user: u64,
+    nice: u64,
+    system: u64,
+    idle: u64
 }
 
-fn update(system: &mut System) {
-    let mut frame_average = 0.;
-    for c in &mut system.cores {
-        let cpu_path: PathBuf = format!("/sys/devices/system/cpu/cpu{}/cpufreq/scaling_cur_freq", c.index).into();
-        let cur_freq = read_into_string(cpu_path).ok()
-                .and_then(|s| s.trim().parse().ok()).expect("Can't read current frequency");
-        c.cur_freq.sample(cur_freq);
-        frame_average += (cur_freq - c.min_freq) as f32 / (c.max_freq - c.min_freq) as f32;
+impl CPU {
+    pub fn from_line(line: &str) -> CPU {
+        fn parse(s: Option<&str>) -> u64 { s.and_then(|s| s.trim().parse().ok()).expect("Couldn't parse CPU stat") }
+        let mut tok = line.split_whitespace();
+        CPU {
+            user: parse(tok.next()),
+            nice: parse(tok.next()),
+            system: parse(tok.next()),
+            idle: parse(tok.next())
+        }
+    }
+
+    pub fn diff(&self, other: &CPU) -> CPU {
+        CPU {
+            user: self.user - other.user,
+            nice: self.nice - other.nice,
+            system: self.system - other.system,
+            idle: self.idle - other.idle
+        }
     }
-    system.load.sample(frame_average / system.cores.len() as f32);
 }
 
 fn main() {
+    env_logger::init().unwrap();
     let matches = App::new("cpuline")
         .version(crate_version!())
         .author(crate_authors!())
-        .about("Display CPU usage per-core or over time (Linux-only)")
-        .setting(AppSettings::SubcommandRequired)
         .arg(Arg::with_name("interval")
              .global(true)
              .short("i")
@@ -129,23 +145,40 @@ fn main() {
              .value_name("MS")
              .takes_value(true)
              .default_value("1000"))
-        .subcommand(SubCommand::with_name("cores"))
-        .subcommand(SubCommand::with_name("time"))
         .get_matches();
 
-    let action = match matches.subcommand_name() {
-        Some("cores") => System::print_cores,
-        Some("time") => System::print_system,
-        _ => unreachable!()
-    };
-
     let interval = value_t!(matches, "interval", u64).unwrap();
 
-    let mut system = System::new();
+    let mut stat = None;
 
+    let user_hz: u64 = unsafe { libc::sysconf(libc::_SC_CLK_TCK) as u64 };
+    debug!("user_hz({})", user_hz);
+    
     loop {
-        update(&mut system);
-        action(&system);
+        let old = stat;
+        stat = Stat::read().ok();
+
+        match (&stat, &old) {
+            (&Some(ref now), &Some(ref old)) => {
+                let load = now.load_since(&old);
+                let duration_ticks = load.duration.as_secs() * user_hz 
+                    + (load.duration.subsec_nanos() as f64 / 1E9 * user_hz as f64) as u64;
+
+                for (_, core) in load.cores.iter() {
+                    // How many ticks the core was in use
+                    let used = core.user + core.nice + core.system;
+                    // How long this core was used with 0 (not used) to 1 (fully used)
+                    let used_part = used as f32 / duration_ticks as f32;
+
+                    let output = FORMAT[((FORMAT.len() - 1) as f32 * used_part) as usize];
+                    debug!("used_part({}) = used({}) / duration_ticks({}) => '{}'", used_part, used, duration_ticks, output);
+                    print!("{}", output);
+                }
+                println!("");
+            },
+            _ => ()
+        }
+
         thread::sleep(Duration::from_millis(interval));
     }
 }