za fly spot num of resets option

Author: kichithewolf

SerialPrograms/Source/PokemonLZA/Programs/ShinyHunting/PokemonLZA_ShinyHunt_FlySpotReset.cpp

@@ -13,6 +13,7 @@
 #include "CommonTools/StartupChecks/VideoResolutionCheck.h"
 #include "NintendoSwitch/Commands/NintendoSwitch_Commands_PushButtons.h"
 #include "NintendoSwitch/Commands/NintendoSwitch_Commands_Superscalar.h"
+#include "NintendoSwitch/Programs/NintendoSwitch_GameEntry.h"
 #include "Pokemon/Pokemon_Strings.h"
 #include "PokemonLA/Inference/Sounds/PokemonLA_ShinySoundDetector.h"
 #include "PokemonLZA/Programs/PokemonLZA_BasicNavigation.h"
@@ -71,6 +72,11 @@ ShinyHunt_FlySpotReset::ShinyHunt_FlySpotReset()
         LockMode::LOCK_WHILE_RUNNING,
         Route::NO_MOVEMENT
     )
+    , NUM_RESETS(
+        "<b>Resets:</b><br>Number of resets, for use when performing fly resets in Hyperspace Wild Zones. Each fly takes about 1 Cal. of time. Make sure to leave enough time to catch found shinies. Zero disables this option.",
+        LockMode::UNLOCK_WHILE_RUNNING,
+        0, 0
+    )
     , NOTIFICATION_STATUS("Status Update", true, false, std::chrono::seconds(3600))
     , NOTIFICATIONS({
         &NOTIFICATION_STATUS,
@@ -82,6 +88,7 @@ ShinyHunt_FlySpotReset::ShinyHunt_FlySpotReset()
 {
     PA_ADD_STATIC(SHINY_REQUIRES_AUDIO);
     PA_ADD_OPTION(ROUTE);
+    PA_ADD_OPTION(NUM_RESETS);
     PA_ADD_OPTION(SHINY_DETECTED);
     PA_ADD_OPTION(NOTIFICATIONS);
 }
@@ -221,6 +228,7 @@ void ShinyHunt_FlySpotReset::program(SingleSwitchProgramEnvironment& env, ProCon
         );
     }
 
+    uint64_t num_resets = 0;
     bool to_zoom_to_max = true;
     run_until<ProControllerContext>(
         env.console, context,
@@ -230,11 +238,19 @@ void ShinyHunt_FlySpotReset::program(SingleSwitchProgramEnvironment& env, ProCon
                 shiny_sound_handler.process_pending(context);
                 route(env, context, stats, to_zoom_to_max);
                 to_zoom_to_max = false;
+                num_resets++;
                 stats.resets++;
                 env.update_stats();
                 if (stats.resets.load(std::memory_order_relaxed) % 10 == 0){
                     send_program_status_notification(env, NOTIFICATION_STATUS);
                 }
+
+                uint64_t num_resets_temp = NUM_RESETS;
+                if (num_resets_temp != 0 && num_resets >= num_resets_temp){
+                    env.log("Number of resets hit. Going to home to pause the game.");
+                    go_home(env.console, context);
+                    break;
+                }
             } // end while
         },
         {{shiny_detector}}

SerialPrograms/Source/PokemonLZA/Programs/ShinyHunting/PokemonLZA_ShinyHunt_FlySpotReset.h

@@ -9,6 +9,7 @@
 #ifndef PokemonAutomation_PokemonLZA_ShinyHunt_FlySpotReset_H
 #define PokemonAutomation_PokemonLZA_ShinyHunt_FlySpotReset_H
 
+#include "Common/Cpp/Options/SimpleIntegerOption.h"
 #include "CommonFramework/Notifications/EventNotificationsTable.h"
 #include "NintendoSwitch/NintendoSwitch_SingleSwitchProgram.h"
 #include "PokemonLA/Options/PokemonLA_ShinyDetectedAction.h"
@@ -46,6 +47,7 @@ class ShinyHunt_FlySpotReset : public SingleSwitchProgramInstance{
     PokemonLA::ShinyRequiresAudioText SHINY_REQUIRES_AUDIO;
     ShinySoundDetectedActionOption SHINY_DETECTED;
     EnumDropdownOption<Route> ROUTE;
+    SimpleIntegerOption<uint64_t> NUM_RESETS;
 
     EventNotificationOption NOTIFICATION_STATUS;
     EventNotificationsOption NOTIFICATIONS;