From d794a94b68fb8d5a2f8f17889b062cb86d56ae6a Mon Sep 17 00:00:00 2001
From: Rekai Musuka <rekai@musuka.dev>
Date: Fri, 20 Aug 2021 00:17:28 -0500
Subject: [PATCH] fix(timer): increase accuracy of timer

---
 src/timer.rs | 52 ++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 38 insertions(+), 14 deletions(-)

diff --git a/src/timer.rs b/src/timer.rs
index 264c44c..3ef7080 100644
--- a/src/timer.rs
+++ b/src/timer.rs
@@ -21,17 +21,28 @@ impl Timer {
         use State::*;
         use TimerSpeed::*;
 
-        if let TIMAOverflow(step) | AbortedTIMAOverflow(step) = self.state {
-            if step < 4 {
-                self.state = TIMAOverflow(step + 1);
-                return;
-            }
-
-            if self.state == TIMAOverflow(step) {
+        match self.state {
+            TIMAOverflow(_) | AbortedTIMAOverflow(_) => self.state = self.state.next(),
+            LoadTMA => {
                 self.counter = self.modulo;
                 self.interrupt = true;
+
+                self.state.next();
+            }
+            Normal => {}
+        }
+
+        if let TIMAOverflow(step) | AbortedTIMAOverflow(step) = self.state {
+            if step < 3 {
+                self.state = self.state.next();
+            } else {
+                if self.state == TIMAOverflow(step) {
+                    self.counter = self.modulo;
+                    self.interrupt = true;
+                }
+
+                self.state = Normal;
             }
-            self.state = Normal;
         }
 
         self.divider = self.divider.wrapping_add(1);
@@ -67,14 +78,12 @@ impl Timer {
         use State::*;
 
         match self.state {
-            Normal => self.counter = byte,
+            Normal | AbortedTIMAOverflow(_) => self.counter = byte,
             TIMAOverflow(step) => {
-                if step < 4 {
-                    self.counter = byte;
-                    self.state = AbortedTIMAOverflow(step);
-                }
+                self.counter = byte;
+                self.state = AbortedTIMAOverflow(step);
             }
-            AbortedTIMAOverflow(_) => self.counter = byte,
+            LoadTMA => {}
         }
     }
 
@@ -173,4 +182,19 @@ enum State {
     TIMAOverflow(u8),
     AbortedTIMAOverflow(u8),
     Normal,
+    LoadTMA,
+}
+
+impl State {
+    fn next(&self) -> Self {
+        use State::*;
+
+        match self {
+            Normal | LoadTMA => Normal,
+            TIMAOverflow(3) => LoadTMA,
+            AbortedTIMAOverflow(3) => Normal,
+            TIMAOverflow(step) => TIMAOverflow(step + 1),
+            AbortedTIMAOverflow(step) => AbortedTIMAOverflow(step + 1),
+        }
+    }
 }