check floor 0 works

Author: kirjavascript

tests/src/floor.rs

@@ -0,0 +1,49 @@
+use crate::{util, labels, playfield};
+
+pub fn test() {
+    // check floor 0 doesnt burn lines
+
+    let mut emu = util::emulator(None);
+
+    for _ in 0..3 {
+        emu.run_until_vblank();
+    }
+
+    let practise_type = labels::get("practiseType") as usize;
+    let game_mode = labels::get("gameMode") as usize;
+    let main_loop = labels::get("mainLoop");
+    let level_number = labels::get("levelNumber") as usize;
+
+    // load floor 0
+
+    emu.memory.iram_raw[practise_type] = labels::get("MODE_FLOOR") as _;
+    emu.memory.iram_raw[level_number] = 18;
+    emu.memory.iram_raw[game_mode] = 4;
+
+    emu.registers.pc = main_loop;
+
+    for _ in 0..5 {
+        emu.run_until_vblank();
+    }
+
+    // setup a tetris
+
+    emu.memory.iram_raw[labels::get("currentPiece") as usize] = 0x11;
+    emu.memory.iram_raw[labels::get("tetriminoX") as usize] = 0x5;
+    emu.memory.iram_raw[labels::get("tetriminoY") as usize] = 0x11;
+    emu.memory.iram_raw[labels::get("autorepeatY") as usize] = 0;
+
+    playfield::set_str(&mut emu,r##"
+##### ####
+##### ####
+##### ####
+##### ####"##);
+
+    emu.memory.iram_raw[labels::get("vramRow") as usize] = 0;
+
+    for _ in 0..20 {
+        emu.run_until_vblank();
+    }
+
+    assert_ne!(playfield::get(&mut emu, 0, 19), 0xEF);
+}

tests/src/main.rs

@@ -1,14 +1,17 @@
 mod block;
 mod cycle_count;
-mod drought;
 mod input;
 mod labels;
+mod playfield;
+mod util;
+mod video;
+
+mod drought;
+mod floor;
 mod pushdown;
 mod rng;
 mod score;
 mod sps;
-mod util;
-mod video;
 
 use gumdrop::Options;
 
@@ -39,6 +42,8 @@ fn main() {
 
     // run tests
     if options.test {
+        floor::test();
+        println!("floor 0 works!");
         score::test();
         println!("score works!");
         score::test_render();

tests/src/playfield.rs

@@ -0,0 +1,49 @@
+use rusticnes_core::nes::NesState;
+use crate::{labels};
+
+pub fn set(emu: &mut NesState, x: u16, y: u16, value: u8) {
+    let index = ((y * 10) + x) + labels::get("playfield");
+    emu.memory.iram_raw[index as usize] = value;
+}
+
+pub fn get(emu: &mut NesState, x: u16, y: u16) -> u8 {
+    let index = ((y * 10) + x) + labels::get("playfield");
+    emu.memory.iram_raw[index as usize]
+}
+
+#[allow(dead_code)]
+pub fn clear(emu: &mut NesState) {
+    for line in 0..20 {
+        self::fill_line(emu, line);
+    }
+}
+
+#[allow(dead_code)]
+pub fn fill_line(emu: &mut NesState, y: u16) {
+    for x in 0..10 {
+        self::set(emu, x, y, labels::get("BLOCK_TILES") as _);
+    }
+}
+
+pub fn set_str(emu: &mut NesState, playfield: &str) {
+    let rows = playfield.trim_start_matches('\n').split("\n").collect::<Vec<_>>();
+    let offset = 20 - rows.len() as u16;
+    rows.iter().enumerate().for_each(|(y, line)| {
+        line.chars().enumerate().for_each(|(x, ch)| {
+            self::set(emu, x as _, offset + y as u16, if ch == '#' { 0x7b } else { 0xef });
+        });
+    });
+}
+
+#[allow(dead_code)]
+pub fn get_str(emu: &mut NesState) -> String {
+    let mut s = String::new();
+    for y in 0..20 {
+        for x in 0..10 {
+            s.push_str(if self::get(emu, x, y) == 0xEF { " " } else { "#" });
+        }
+        s.push_str("\n");
+    }
+
+    s
+}