more readability

2 files changed, 38 insertions, 1 deletions

diff --git a/src/statistics/gamedata.rs b/src/statistics/gamedata.rs
index 3bd895e..80f269e 100644
--- a/src/statistics/gamedata.rs
+++ b/src/statistics/gamedata.rs
@@ -113,6 +113,23 @@ pub enum Trigger {
 #[derive(Debug, Clone, PartialEq, Eq, Hash)]
 pub struct Mechanic(pub u16, pub Trigger, pub &'static str);
 
+impl Mechanic {
+    #[inline]
+    pub fn boss_id(&self) -> u16 {
+        self.0
+    }
+
+    #[inline]
+    pub fn trigger(&self) -> &Trigger {
+        &self.1
+    }
+
+    #[inline]
+    pub fn name(&self) -> &'static str {
+        self.2
+    }
+}
+
 macro_rules! mechanics {
     ( $( $boss_id:expr => [ $($name:expr => $trigger:expr,)* ], )* ) => {
         &[
diff --git a/src/statistics/mechanics.rs b/src/statistics/mechanics.rs
index 5a16204..0cf6f24 100644
--- a/src/statistics/mechanics.rs
+++ b/src/statistics/mechanics.rs
@@ -3,8 +3,9 @@ use super::math::{Monoid, RecordFunc, Semigroup};
 
 use std::fmt;
 
+/// A simple wrapper for integers.
 #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
-pub struct Counter(u32);
+struct Counter(u32);
 
 impl Semigroup for Counter {
     #[inline]
@@ -20,19 +21,38 @@ impl Monoid for Counter {
     }
 }
 
+/// Provides access to the mechanic log.
 #[derive(Clone, Default)]
 pub struct MechanicLog {
     inner: RecordFunc<u64, (&'static Mechanic, u64), Counter>,
 }
 
 impl MechanicLog {
+    /// Increase the mechanic counter for the given mechanic and agent by one.
     pub fn increase(&mut self, time: u64, mechanic: &'static Mechanic, agent: u64) {
         self.inner.insert(time, (mechanic, agent), Counter(1));
     }
 
+    /// Return the count of mechanics.
+    ///
+    /// A function can be provided to filter entries by mechanic type and agent.
     pub fn count<F: FnMut(&'static Mechanic, u64) -> bool>(&self, mut filter: F) -> u32 {
         self.inner.tally_only(|(a, b)| filter(a, *b)).0
     }
+
+    /// Return the count of mechanics between the two given times.
+    ///
+    /// A function can be provided to filter entries by mechanic type and agent.
+    pub fn count_between<F: FnMut(&'static Mechanic, u64) -> bool>(
+        &self,
+        start: u64,
+        stop: u64,
+        mut filter: F,
+    ) -> u32 {
+        self.inner
+            .between_only(&start, &stop, |(a, b)| filter(a, *b))
+            .0
+    }
 }
 
 impl fmt::Debug for MechanicLog {