blop

src/core/timer.rs

@@ -0,0 +1,84 @@
+use time::PreciseTime;
+
+pub struct OpenTimer<'a> {
+    name: &'static str,
+    timer_tree: &'a mut TimerTree,
+    start: PreciseTime,
+    depth: u32,
+}
+
+impl<'a> OpenTimer<'a> {
+    pub fn open(&mut self, name: &'static str) -> OpenTimer {
+        OpenTimer {
+            name: name,
+            timer_tree: self.timer_tree,
+            start: PreciseTime::now(),
+            depth: self.depth + 1,
+        }
+    }
+}
+
+impl<'a> Drop for OpenTimer<'a> {
+    fn drop(&mut self,) {
+        self.timer_tree.timings.push(Timing     {
+            name: self.name,
+            duration: self.start.to(PreciseTime::now()).num_microseconds().unwrap(),
+            depth: self.depth,
+        });
+    }
+}
+
+#[derive(Debug)]
+pub struct Timing {
+    name: &'static str,
+    duration: i64,
+    depth: u32,
+}
+
+#[derive(Debug)]
+pub struct TimerTree {
+    timings: Vec<Timing>,
+}
+
+impl TimerTree {
+    pub fn new() -> TimerTree {
+        TimerTree {
+            timings: Vec::new(),
+        }
+    }
+
+    pub fn open(&mut self, name: &'static str) -> OpenTimer {
+        OpenTimer {
+            name: name,
+            timer_tree: self,
+            start: PreciseTime::now(),
+            depth: 0,
+        }
+    }
+}
+
+
+
+#[cfg(test)]
+mod tests {
+
+    use super::*;
+
+    #[test]
+    fn test_timer() {
+        let mut timer_tree = TimerTree::new();
+        {
+            let mut a = timer_tree.open("a");
+            {
+                let mut ab = a.open("b");
+                {
+                    let abc = ab.open("c");
+                }
+                {
+                    let abd = ab.open("d");
+                }
+            }
+        }
+        assert_eq!(timer_tree.timings.len(), 4);
+    }
+}