list all seeds

Author: kirjavascript

tests/src/main.rs

@@ -2,6 +2,7 @@ mod block;
 mod input;
 mod labels;
 mod pushdown;
+mod rng;
 mod sps;
 mod util;
 mod video;
@@ -22,36 +23,14 @@ struct TestOptions {
     sps_seed: u32,
     #[options(help = "print SPS pieces")]
     sps_qty: u32,
+    #[options(help = "list RNG seeds")]
+    rng_seeds: bool,
     foo: bool,
 }
 
 fn main() {
     let options = TestOptions::parse_args_default_or_exit();
 
-    if options.foo {
-        let mut emu = util::emulator(None);
-        let mut view = video::Video::new();
-
-        // spend a few frames bootstrapping
-        for _ in 0..3 { emu.run_until_vblank(); }
-
-        emu.memory.iram_raw[labels::get("practiseType") as usize] = labels::get("MODE_TYPEB") as _;
-        emu.memory.iram_raw[labels::get("gameMode") as usize] = 4;
-        emu.memory.iram_raw[labels::get("levelNumber") as usize] = 18;
-        emu.memory.iram_raw[labels::get("typeBModifier") as usize] = 5;
-        let label = labels::get("mainLoop");
-        rusticnes_core::opcodes::push(&mut emu, (label >> 8) as u8);
-        rusticnes_core::opcodes::push(&mut emu, label as u8);
-
-        loop {
-            emu.run_until_vblank();
-            emu.ppu.render_ntsc(256);
-            view.update(&emu.ppu.filtered_screen);
-        }
-    }
-
-    // TODO: cycle counts for modes
-
     // run tests
     if options.test {
         sps::test();
@@ -75,4 +54,32 @@ fn main() {
         }
         println!("");
     }
+
+    // other stuff
+
+    if options.rng_seeds {
+        println!("{:?}", rng::seeds());
+    }
+
+    if options.foo {
+        let mut emu = util::emulator(None);
+        let mut view = video::Video::new();
+
+        // spend a few frames bootstrapping
+        for _ in 0..3 { emu.run_until_vblank(); }
+
+        emu.memory.iram_raw[labels::get("practiseType") as usize] = labels::get("MODE_TYPEB") as _;
+        emu.memory.iram_raw[labels::get("gameMode") as usize] = 4;
+        emu.memory.iram_raw[labels::get("levelNumber") as usize] = 18;
+        emu.memory.iram_raw[labels::get("typeBModifier") as usize] = 5;
+        let label = labels::get("mainLoop");
+        rusticnes_core::opcodes::push(&mut emu, (label >> 8) as u8);
+        rusticnes_core::opcodes::push(&mut emu, label as u8);
+
+        for _ in 0..10 {
+            emu.run_until_vblank();
+            emu.ppu.render_ntsc(256);
+            view.update(&emu.ppu.filtered_screen);
+        }
+    }
 }

tests/src/rng.rs

@@ -0,0 +1,35 @@
+use crate::{util, labels};
+use std::collections::HashSet;
+
+pub fn seeds() -> HashSet<u16> {
+    let mut emu = util::emulator(None);
+    let rng_seed = labels::get("rng_seed");
+    let next_rng = labels::get("generateNextPseudorandomNumber");
+
+    let mut seeds: HashSet<u16> = HashSet::new();
+
+    let mut seed = 0x8988;
+
+    loop {
+        seeds.insert(seed);
+
+        emu.memory.iram_raw[(rng_seed + 0) as usize] = (seed >> 8) as _;
+        emu.memory.iram_raw[(rng_seed + 1) as usize] = seed as u8;
+
+        emu.registers.x = rng_seed as u8;
+        emu.registers.y = 0x2;
+        emu.registers.pc = next_rng;
+
+        util::run_to_return(&mut emu, false);
+
+        seed =
+            ((emu.memory.iram_raw[rng_seed as usize] as u16) << 8)
+            + emu.memory.iram_raw[(rng_seed + 1) as usize] as u16;
+
+        if seed == 0x8988 {
+            break;
+        }
+    }
+
+    seeds
+}

tests/src/video.rs

@@ -1,4 +1,4 @@
-use minifb::{Key, Window, WindowOptions};
+use minifb::{Window, WindowOptions};
 
 const WIDTH: usize = 256;
 const HEIGHT: usize = 240;