fix bug when handling NMI interrupt and WIP for IRQ

snes-core/src/cpu/vectors.rs

@@ -1,4 +1,4 @@
-use super::{instructions::{phk::PHK, php::PHP, push_common, CPUInstruction}, interface::CPU, internal_registers::RDNMI};
+use super::{instructions::push_common, interface::CPU, internal_registers::RDNMI};
 use crate::cpu::bus::Bus;
 
 
@@ -38,23 +38,26 @@ impl CPU {
         self.registers.is_cpu_stopped = false;
     }
 
-    fn push_emulation_interrupt(&mut self, bus: &mut Bus) {
+    fn push_interrupt(&mut self, bus: &mut Bus) {
+        let pbr = self.registers.pbr;
         if !self.registers.emulation_mode {
-            PHK{}.execute(&mut self.registers, bus);
+            push_common::do_push(&mut self.registers, bus, &[pbr]);
         }
         let values = [
             (self.registers.pc >> 8) as u8,
             self.registers.pc as u8,
         ];
         push_common::do_push(&mut self.registers, bus, &values);
-        PHP{}.execute(&mut self.registers, bus);
+        let p = self.registers.p;
+        push_common::do_push(&mut self.registers, bus, &[p]);
     }
 
     fn handle_interrupt(&mut self, bus: &mut Bus, vector: Vector) {
-        self.push_emulation_interrupt(bus);
-        let effective_vector = vector.get_base_address();
-        self.registers.pc = effective_vector as u16;
-        self.registers.pbr = (effective_vector >> 16) as u8;
+        self.push_interrupt(bus);
+        let base_address = vector.get_base_address();
+        let effective_vector = CPU::get_vector(base_address, bus);
+        self.registers.pc = effective_vector;
+        self.registers.pbr = 0x00;
     }
 
     pub fn check_interrupts(&mut self, bus: &mut Bus) {
@@ -62,7 +65,10 @@ impl CPU {
         if rdnmi_byte >> 7 != 0 {
             self.registers.is_cpu_waiting_interrupt = false;
             self.handle_interrupt(bus, Vector::NMI);
-            return;
+        }
+        if !self.registers.get_irq_disable_flag() && bus.ppu.is_irq_set {
+            self.registers.is_cpu_waiting_interrupt = false;
+            self.handle_interrupt(bus, Vector::IRQ);
         }
     }
 

snes-core/src/ppu/interface.rs

@@ -4,6 +4,7 @@ pub struct PPU {
     framebuffer: Vec<u8>,
     pub registers: PPURegisters,
     was_vblank_nmi_set: bool,
+    pub is_irq_set: bool,
 }
 
 impl PPU {
@@ -12,6 +13,7 @@ impl PPU {
             framebuffer: vec![],
             registers: PPURegisters::new(),
             was_vblank_nmi_set: false,
+            is_irq_set: false,
         }
     }