drought probability generation

Author: kirjavascript

tests/src/drought.rs

@@ -0,0 +1,39 @@
+use crate::{labels, util, block};
+
+pub fn print_probabilities() {
+    let mut emu = util::emulator(None);
+    let rng_seed = labels::get("rng_seed");
+    let drought_modifier = labels::get("droughtModifier");
+    let next_rng = labels::get("pickRandomTetrimino");
+
+    let seed = 0x8988;
+    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.memory.iram_raw[labels::get("practiseType") as usize] = labels::get("MODE_DROUGHT") as u8;
+
+    for modifier in 0..19 {
+        emu.memory.iram_raw[drought_modifier as usize] = modifier;
+
+        let mut longbars = 0;
+        let mut total = 0;
+
+        for _ in 0..100000 {
+            emu.registers.pc = next_rng;
+
+            util::run_to_return(&mut emu, false);
+
+            let block: block::Block = emu.memory.iram_raw[labels::get("spawnID") as usize].into();
+
+            if block == block::Block::I {
+                longbars += 1;
+            }
+
+            total += 1;
+        }
+        println!(
+            "{} longbar%: {:.2}",
+            "0123456789ABCDEFGHI".as_bytes()[modifier as usize] as char,
+            (longbars as f64 / total as f64) * 100.
+        );
+    }
+}

tests/src/main.rs

@@ -1,4 +1,5 @@
 mod block;
+mod drought;
 mod input;
 mod labels;
 mod pushdown;
@@ -8,7 +9,6 @@ mod sps;
 mod util;
 mod video;
 
-
 use gumdrop::Options;
 
 fn parse_hex(s: &str) -> Result<u32, std::num::ParseIntError> {
@@ -26,6 +26,8 @@ struct TestOptions {
     sps_qty: u32,
     #[options(help = "list RNG seeds")]
     rng_seeds: bool,
+    #[options(help = "list drought mode probabilities")]
+    drought_probs: bool,
     foo: bool,
 }
 
@@ -64,6 +66,10 @@ fn main() {
         println!("{:?}", rng::seeds());
     }
 
+    if options.drought_probs {
+        drought::print_probabilities();
+    }
+
     // other stuff
 
     if options.foo {
@@ -79,11 +85,12 @@ fn main() {
         let mode_typeb = labels::get("MODE_TYPEB") as u8;
 
         rng::seeds().iter().for_each(|seed| {
-
             emu.reset();
 
             // spend a few frames bootstrapping
-            for _ in 0..3 { emu.run_until_vblank(); }
+            for _ in 0..3 {
+                emu.run_until_vblank();
+            }
 
             emu.memory.iram_raw[practise_type] = mode_typeb;
             emu.memory.iram_raw[game_mode] = 4;
@@ -103,7 +110,7 @@ fn main() {
             emu.ppu.render_ntsc(video::WIDTH);
             view.update(&emu.ppu.filtered_screen);
         });
-        loop {
-        }
+        loop {}
     }
+
 }