add SPS tests

Author: kirjavascript

tests/src/main.rs

@@ -2,14 +2,23 @@ mod labels;
 mod pushdown;
 mod util;
 mod block;
+mod sps;
 
 use gumdrop::Options;
 
+fn parse_hex(s: &str) -> Result<u32, std::num::ParseIntError> {
+    u32::from_str_radix(s, 16)
+}
+
 #[derive(Debug, Options)]
 struct TestOptions {
     help: bool,
     #[options(help = "run tests")]
     test: bool,
+    #[options(help = "set SPS seed", parse(try_from_str = "parse_hex"))]
+    sps_seed: u32,
+    #[options(help = "print SPS pieces")]
+    sps_qty: u32,
 }
 
 fn main() {
@@ -20,26 +29,19 @@ fn main() {
         println!("pushdown works!");
     }
 
-    let mut emu = util::emulator(None);
-
-    let seed_0 = 0x88;
-    let seed_1 = 0x88;
-    let seed_2 = 0x88;
-
-    emu.memory.iram_raw[(labels::get("set_seed_input") + 0) as usize] = seed_0;
-    emu.memory.iram_raw[(labels::get("set_seed") + 0) as usize] = seed_0;
-    emu.memory.iram_raw[(labels::get("set_seed_input") + 1) as usize] = seed_1;
-    emu.memory.iram_raw[(labels::get("set_seed") + 1) as usize] = seed_1;
-    emu.memory.iram_raw[(labels::get("set_seed_input") + 2) as usize] = seed_2;
-    emu.memory.iram_raw[(labels::get("set_seed") + 2) as usize] = seed_2;
+    if options.sps_qty > 0 {
+        let mut blocks = sps::SPS::new();
 
+        blocks.set_input((
+            ((options.sps_seed >> 16) & 0xFF) as u8,
+            ((options.sps_seed >> 8) & 0xFF) as u8,
+            (options.sps_seed& 0xFF) as u8,
+        ));
 
-    for i in 0..12 {
-        emu.registers.pc = labels::get("pickTetriminoSeed");
-        util::run_to_return(&mut emu, false);
-
-        let block: block::Block = emu.memory.iram_raw[labels::get("spawnID") as usize].into();
-
-        println!("{:#?}", block);
+        for i in 0..options.sps_qty {
+            print!("{:?}", blocks.next());
+        }
     }
+
+    // TODO: cycle counts for modes
 }

tests/src/sps.rs

@@ -0,0 +1,31 @@
+use rusticnes_core::nes::NesState;
+use crate::{labels, util, block};
+
+pub struct SPS {
+    emu: NesState,
+}
+
+impl SPS {
+    pub fn new() -> Self {
+        Self {
+            emu: util::emulator(None),
+        }
+    }
+
+    pub fn set_input(&mut self, seed: (u8, u8, u8)) {
+        self.emu.memory.iram_raw[(labels::get("set_seed_input") + 0) as usize] = seed.0;
+        self.emu.memory.iram_raw[(labels::get("set_seed") + 0) as usize] = seed.0;
+        self.emu.memory.iram_raw[(labels::get("set_seed_input") + 1) as usize] = seed.1;
+        self.emu.memory.iram_raw[(labels::get("set_seed") + 1) as usize] = seed.1;
+        self.emu.memory.iram_raw[(labels::get("set_seed_input") + 2) as usize] = seed.2;
+        self.emu.memory.iram_raw[(labels::get("set_seed") + 2) as usize] = seed.2;
+    }
+
+    pub fn next(&mut self) -> block::Block {
+        self.emu.registers.pc = labels::get("pickTetriminoSeed");
+
+        util::run_to_return(&mut self.emu, false);
+
+        self.emu.memory.iram_raw[labels::get("spawnID") as usize].into()
+    }
+}