Fixed #13: Properly buffer screen transitions to prevent memory corru…

…ption

main.mfk

@@ -54,12 +54,15 @@ void main() {
 
   while(true) {
     // Wait for NMI to complete rendering frame
-    while (new_frame) {}
-
-    sprite_tick()
-    input_and_screen_tick()
-    clear_remaining_oam()
-
+    while (new_frame && not(buffer_screen_draw)) {}
+
+    if (buffer_screen_draw) {
+      screen_draw_tick()
+    } else {
+      sprite_tick()
+      input_and_screen_tick()
+      clear_remaining_oam()
+    }
     // wait_for_sprite0_clear()
 
     new_frame = true
@@ -68,17 +71,6 @@ void main() {
 }
 
 void nmi() {
-  if (buffer_screen_draw) {
-    screen_transfer_tick()
-    nmi_always_run()
-    return
-  }
-
-  if (not(drawing_enabled)) {
-    nmi_always_run()
-    return
-  }
-
   if in_nmi != 0 {
     // Overran NMI
     nmi_fail = 1
@@ -89,15 +81,31 @@ void nmi() {
 
   in_nmi = 1
 
+  nmi_work()
+
+  in_nmi = 0
+}
+
+inline void nmi_work() {
+  if (buffer_screen_blank) {
+    screen_transfer_tick()
+    nmi_always_run()
+    new_frame = false
+    return
+  }
+
+  if (not(drawing_enabled)) {
+    nmi_always_run()
+    return
+  }
+
   if (new_frame) {
     nmi_frame()
 
     new_frame = false
   }
 
   nmi_always_run()
-
-  in_nmi = 0
 }
 
 void nmi_frame() {

ui/screen.mfk

@@ -19,6 +19,7 @@ enum Screen {
 }
 
 bool buffer_screen_draw = false
+bool buffer_screen_blank = false
 Screen current_screen
 
 bool screen_tick() {
@@ -47,7 +48,7 @@ void draw_screen() {
 
 void set_screen_forcibly(Screen new_screen) {
   current_screen = new_screen
-  buffer_screen_draw = true
+  buffer_screen_blank = true
 }
 
 void set_screen(Screen new_screen) {
@@ -56,18 +57,24 @@ void set_screen(Screen new_screen) {
   }
 
   current_screen = new_screen
-  buffer_screen_draw = true
+  buffer_screen_blank = true
 }
 
 void screen_transfer_tick() {
   // Wait for NMI to disable rendering and NMIs
-  // Draw new screen
   drawing_enabled = false
-  buffer_screen_draw = false
+  buffer_screen_blank = false
   read_ppu_status()
   ppu_ctrl = standard_ppu_ctrl
   ppu_mask = 0
 
+  buffer_screen_draw = true
+}
+
+void screen_draw_tick() {
+  // NMI has cleared screen, now write it
+  // Draw new screen
+  buffer_screen_draw = false
   ppu_wait_vblank()
 
   clear_buffer()