Show initial tinymt state while tools move

Author: zaksabeast

reader_core/src/gen6/draw.rs

@@ -13,10 +13,23 @@ pub use crate::draw::{draw_header, draw_pkx};
 pub fn draw_tinymt(reader: &Gen6Reader, rng: &Gen6Rng) {
     let tinymt = rng.tinymt();
     let tinymt_state = tinymt.current_state();
+    let tinymt_init_state = tinymt.rng().initial_state();
     let tinymt_seed = reader.tinymt_seed();
 
-    pnp::println!("TinyMT seed: {:08X}", tinymt_seed);
+    pnp::println!("Tiny u32 seed: {:08X}", tinymt_seed);
     pnp::println!("Advances: {}", tinymt.advances());
+    pnp::println!("");
+    pnp::println!("TinyMT seed:");
+    pnp::println!(
+        "[3]{:08X} [2]{:08X}",
+        tinymt_init_state[3],
+        tinymt_init_state[2]
+    );
+    pnp::println!(
+        "[1]{:08X} [0]{:08X}",
+        tinymt_init_state[1],
+        tinymt_init_state[0]
+    );
     pnp::println!("TinyMT state:");
     pnp::println!("[3]{:08X} [2]{:08X}", tinymt_state[3], tinymt_state[2]);
     pnp::println!("[1]{:08X} [0]{:08X}", tinymt_state[1], tinymt_state[0]);

reader_core/src/rng/rng_wrapper.rs

@@ -55,6 +55,10 @@ impl<T: rng::Rng + Copy> RngWrapper<T> {
             }
         }
     }
+
+    pub fn rng(&self) -> &T {
+        &self.rng
+    }
 }
 
 #[cfg(test)]

reader_core/src/rng/tiny_mt.rs

@@ -2,6 +2,7 @@ use crate::rng::Rng;
 
 #[derive(Clone, Copy, Debug, PartialEq, Eq, Default)]
 pub struct TinyMT {
+    initial_state: [u32; 4],
     state: [u32; 4],
 }
 
@@ -15,7 +16,10 @@ impl TinyMT {
                 .wrapping_add(i as u32);
         }
 
-        let mut rng = Self { state };
+        let mut rng = Self {
+            state,
+            initial_state: state,
+        };
 
         for _ in 0..8 {
             rng.next_state();
@@ -40,6 +44,10 @@ impl TinyMT {
     fn get_state(&self) -> [u32; 4] {
         self.state
     }
+
+    pub fn initial_state(&self) -> &[u32; 4] {
+        &self.initial_state
+    }
 }
 
 impl Rng for TinyMT {