From 8c9567b610ac26316d5bb7db18ca012652fe4464 Mon Sep 17 00:00:00 2001
From: Rekai Musuka <rekai@musuka.dev>
Date: Sat, 14 Aug 2021 16:42:15 -0500
Subject: [PATCH] chore(cpu): rename discriminants of ImeState enum

---
 src/cpu.rs         | 19 ++++++-------------
 src/instruction.rs |  2 +-
 2 files changed, 7 insertions(+), 14 deletions(-)

diff --git a/src/cpu.rs b/src/cpu.rs
index 25c86fc..503eda4 100644
--- a/src/cpu.rs
+++ b/src/cpu.rs
@@ -118,7 +118,7 @@ impl Cpu {
                 let opcode = self.fetch();
                 let instr = self.decode(opcode);
                 let elapsed = self.execute(instr);
-                self.check_ime();
+                self.handle_ei();
                 elapsed
             }
         };
@@ -159,18 +159,11 @@ impl Cpu {
         &mut self.bus.joypad
     }
 
-    fn check_ime(&mut self) {
+    fn handle_ei(&mut self) {
         match self.ime {
-            ImeState::Pending => {
-                // This is within the context of the EI instruction, we need to not update EI until the end of the
-                // next executed Instruction
-                self.ime = ImeState::PendingEnd;
-            }
-            ImeState::PendingEnd => {
-                // The Instruction after EI has now been executed, so we want to enable the IME flag here
-                self.ime = ImeState::Enabled;
-            }
-            ImeState::Disabled | ImeState::Enabled => {} // Do Nothing
+            ImeState::EiExecuted => self.ime = ImeState::Pending,
+            ImeState::Pending => self.ime = ImeState::Enabled,
+            ImeState::Disabled | ImeState::Enabled => {}
         }
     }
 
@@ -514,8 +507,8 @@ pub(crate) enum HaltState {
 #[derive(Debug, Clone, Copy)]
 pub(crate) enum ImeState {
     Disabled,
+    EiExecuted,
     Pending,
-    PendingEnd,
     Enabled,
 }
 
diff --git a/src/instruction.rs b/src/instruction.rs
index 180bb70..b2393ae 100644
--- a/src/instruction.rs
+++ b/src/instruction.rs
@@ -971,7 +971,7 @@ impl Instruction {
             }
             Instruction::EI => {
                 // EI | Enable IME after the next instruction
-                cpu.set_ime(ImeState::Pending);
+                cpu.set_ime(ImeState::EiExecuted);
                 Cycle::new(4)
             }
             Instruction::CALL(cond) => {