working highlight RNG

Author: fyex

SerialPrograms/Source/Pokemon/Pokemon_Xoroshiro128Plus.cpp

@@ -77,6 +77,22 @@ std::vector<bool> Xoroshiro128Plus::generate_last_bit_sequence(size_t max_advanc
     return sequence;
 }
 
+
+std::pair<bool, uint64_t> Xoroshiro128Plus::advances_to_state(Xoroshiro128PlusState other_state, uint64_t max_advances) {
+    Xoroshiro128Plus temp_rng(get_state());
+    uint64_t advances = 0;
+
+    while (advances <= max_advances) {
+        Xoroshiro128PlusState temp_state = temp_rng.get_state();
+        if (temp_state.s0 == other_state.s0 && temp_state.s1 == other_state.s1) {
+            return { true, advances };
+        }
+        temp_rng.next();
+        advances++;
+    }
+    return { false, advances };
+}
+
 // The generic solution to the system of equations to calculate the initial state from the last bits of 128 consecutive Xoroshiro128+ results.
 uint64_t Xoroshiro128Plus::last_bits_reverse_matrix[128][2] = {
     /*s0 bit 0*/ {0b0101001100100001111011111110111001010011111110101011100011001101, 0b0111010111110111000101010100001111101001111001011111001011010111} ,

SerialPrograms/Source/Pokemon/Pokemon_Xoroshiro128Plus.h

@@ -31,6 +31,13 @@ class Xoroshiro128Plus{
     Xoroshiro128PlusState get_state();
     std::vector<bool> generate_last_bit_sequence(size_t max_advances);
 
+    // Calculates how many advances are required to reach the given state.
+    // The given state must be reachable within max_advances advances.
+    // Returns a pair:
+    // first: true if the state is reachable within max_advances, false otherwise
+    // second: the number of advances required (if first is true)
+    std::pair<bool, uint64_t> advances_to_state(Xoroshiro128PlusState other_state, uint64_t max_advances = 100000);
+
     static Xoroshiro128Plus xoroshiro128plus_from_last_bits(std::pair<uint64_t, uint64_t> last_bits);
 
 

SerialPrograms/Source/PokemonSwSh/Programs/RNG/PokemonSwSh_BasicRNG.cpp

@@ -66,7 +66,6 @@ Xoroshiro128PlusState find_rng_state(
     return rng.get_state();
 }
 
-
 Xoroshiro128PlusState refind_rng_state(
     VideoStream& stream,
     ProControllerContext& context,
@@ -75,6 +74,18 @@ Xoroshiro128PlusState refind_rng_state(
     size_t max_advances,
     bool save_screenshots,
     bool log_image_values
+) {
+    return refind_rng_state_and_animations(stream, context, last_known_state, min_advances, max_advances, save_screenshots, log_image_values).first;
+}
+
+std::pair<Xoroshiro128PlusState, uint64_t> refind_rng_state_and_animations(
+    VideoStream& stream,
+    ProControllerContext& context,
+    Xoroshiro128PlusState last_known_state,
+    size_t min_advances,
+    size_t max_advances,
+    bool save_screenshots,
+    bool log_image_values
 )
 {
     Xoroshiro128Plus rng(last_known_state.s0, last_known_state.s1);
@@ -140,7 +151,7 @@ Xoroshiro128PlusState refind_rng_state(
     stream.log("RNG: state[0] = " + tostr_hex(rng.get_state().s0));
     stream.log("RNG: state[1] = " + tostr_hex(rng.get_state().s1));
 
-    return rng.get_state();
+    return { rng.get_state(), sequence.size() };
 }
 
 

SerialPrograms/Source/PokemonSwSh/Programs/RNG/PokemonSwSh_BasicRNG.h

@@ -8,6 +8,8 @@
 #include "NintendoSwitch/Controllers/NintendoSwitch_ProController.h"
 #include "Pokemon/Pokemon_Xoroshiro128Plus.h"
 
+#include <optional>
+
 namespace PokemonAutomation{
 namespace NintendoSwitch{
 namespace PokemonSwSh{
@@ -33,6 +35,20 @@ Xoroshiro128PlusState refind_rng_state(
     bool log_values
 );
 
+// Performs Orbeetle attack animations until only one possible state is left.
+// Returns a pair:
+// first: current RNG state
+// second: the number of animations required to find the state
+std::pair<Xoroshiro128PlusState, uint64_t> refind_rng_state_and_animations(
+    VideoStream& stream,
+    ProControllerContext& context,
+    Xoroshiro128PlusState last_known_state,
+    size_t min_advances,
+    size_t max_advances,
+    bool save_screenshots,
+    bool log_values
+);
+
 void do_rng_advances(
     VideoStream& stream, ProControllerContext& context,
     Xoroshiro128Plus& rng,