writing ThreeSegmentDudunsparceFinder

Author: Gin

SerialPrograms/Source/PokemonSV/Programs/Eggs/PokemonSV_EggHatcher.cpp

@@ -120,20 +120,20 @@ void EggHatcher::hatch_one_box(SingleSwitchProgramEnvironment& env, ProControlle
             if (num_non_egg_pokemon == 0){
                 // nothing in this column
                 env.log("Nothing in column " + std::to_string(column_index+1) + ".");
-                env.console.overlay().add_log("Empty column", COLOR_WHITE);
+                env.console.overlay().add_log("Empty column");
                 continue;
             }
 
             // we have only non-egg pokemon in the column
             // Move them back
             env.log("Only non-egg pokemon in column, move them back.");
-            env.console.overlay().add_log("No egg in column", COLOR_WHITE);
+            env.console.overlay().add_log("No egg in column");
             unload_one_column_from_party(env, env.console, context, NOTIFICATION_ERROR_RECOVERABLE, column_index, HAS_CLONE_RIDE_POKEMON);
             continue;
         }
 
         env.log("Loaded " + std::to_string(num_eggs) + " eggs to party.");
-        env.console.overlay().add_log("Load " + std::to_string(num_eggs) + " eggs", COLOR_WHITE);
+        env.console.overlay().add_log("Load " + std::to_string(num_eggs) + " eggs");
         leave_box_system_to_overworld(env.program_info(), env.console, context);
 
         auto hatched_callback = [&](uint8_t){  

SerialPrograms/Source/PokemonSV/Programs/PokemonSV_Navigation.cpp

@@ -999,10 +999,10 @@ void enter_menu_from_overworld(const ProgramInfo& info, VideoStream& stream, Pro
     bool success = false;
 
     while (true){
-        if (current_time() - start > std::chrono::minutes(3)){
+        if (current_time() - start > std::chrono::minutes(1)){
             OperationFailedException::fire(
                 ErrorReport::SEND_ERROR_REPORT,
-                "enter_menu_from_overworld(): Failed to enter specified menu after 3 minutes.",
+                "enter_menu_from_overworld(): Failed to enter specified menu after 1 minute.",
                 stream
             );
         }
@@ -1064,6 +1064,64 @@ void enter_menu_from_overworld(const ProgramInfo& info, VideoStream& stream, Pro
     }
 }
 
+void enter_menu_from_box_system(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context,
+    int menu_index,
+    MenuSide side
+){
+    WallClock start = current_time();
+    bool success = false;
+
+    while (true){
+        if (current_time() - start > std::chrono::seconds(20)){
+            OperationFailedException::fire(
+                ErrorReport::SEND_ERROR_REPORT,
+                "enter_menu_from_box_system(): Failed to enter specified menu after 20 seconds.",
+                stream
+            );
+        }
+
+        MainMenuWatcher main_menu(COLOR_RED);
+        context.wait_for_all_requests();
+
+        int ret = run_until<ProControllerContext>(
+            stream, context,
+            [](ProControllerContext& context){
+                // repeatedly pressing B and waiting for three seconds
+                for (int i = 0; i < 10; i++){
+                    pbf_press_button(context, BUTTON_B, 20, 3 * TICKS_PER_SECOND);
+                }
+            },
+            {main_menu}
+        );
+        context.wait_for(std::chrono::milliseconds(100));
+
+        const bool fast_mode = false;
+        switch (ret){
+        case 0:
+            stream.log("Detected main menu.");
+            if (menu_index == -1){
+                return;
+            }
+            success = main_menu.move_cursor(info, stream, context, side, menu_index, fast_mode);
+            if (success == false){
+                OperationFailedException::fire(
+                    ErrorReport::SEND_ERROR_REPORT,
+                    "enter_menu_from_overworld(): Cannot move menu cursor to specified menu.",
+                    stream
+                );
+            }
+            pbf_press_button(context, BUTTON_A, 20, 105);
+            return;     
+        default:
+            OperationFailedException::fire(
+                ErrorReport::SEND_ERROR_REPORT,
+                "enter_menu_from_overworld(): No recognized state after 30 seconds.",
+                stream
+            );
+        }
+    }
+}
+
 void press_button_until_gradient_arrow(
     const ProgramInfo& info, 
     VideoStream& stream,

SerialPrograms/Source/PokemonSV/Programs/PokemonSV_Navigation.h

@@ -26,7 +26,8 @@ void set_time_to_12am_from_home(const ProgramInfo& info, VideoStream& stream, Pr
 void day_skip_from_overworld(VideoStream& stream, ProControllerContext& context);
 
 // Press B to return to the overworld
-void press_Bs_to_back_to_overworld(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context, uint16_t seconds_between_b_presses = 3);
+void press_Bs_to_back_to_overworld(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context,
+    uint16_t seconds_between_b_presses = 3);
 
 //  From overworld, open map. Will change map view from rotated to fixed if not already fixed.
 void open_map_from_overworld(
@@ -164,14 +165,21 @@ void fly_to_overlapping_flypoint(const ProgramInfo& info, VideoStream& stream, P
 // throw exception if there is a fly point/pokecenter that overlaps with the player on the map
 void confirm_no_overlapping_flypoint(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context);
 
-// enter menu and move the cursor the given side, and index. then press the A button
+// enter menu and move the cursor to the given side and index, then press the A button (without wait for A button to press)
 // if menu_index is -1, return once the menu is detected.
 void enter_menu_from_overworld(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context,
     int menu_index,
     MenuSide side = MenuSide::RIGHT,
     bool has_minimap = true
 );
 
+// enter menu and move the cursor to the given side and index, then press the A button (without wait for A button to press)
+// if menu_index is -1, return once the menu is detected.
+void enter_menu_from_box_system(const ProgramInfo& info, VideoStream& stream, ProControllerContext& context,
+    int menu_index,
+    MenuSide side = MenuSide::RIGHT
+);
+
 // press given button until gradient arrow appears in given box_area_to_check.
 void press_button_until_gradient_arrow(
     const ProgramInfo& info, 

SerialPrograms/Source/PokemonSV/Programs/PokemonSV_ThreeSegmentDudunsparceFinder.cpp

@@ -4,24 +4,30 @@
  *
  */
 
+#include "Common/Cpp/Exceptions.h"
+#include "CommonFramework/Exceptions/FatalProgramException.h"
 #include "CommonFramework/Exceptions/OperationFailedException.h"
 #include "CommonFramework/Notifications/ProgramNotifications.h"
 #include "CommonFramework/ProgramStats/StatsTracking.h"
 #include "CommonTools/StartupChecks/VideoResolutionCheck.h"
 #include "NintendoSwitch/Commands/NintendoSwitch_Commands_PushButtons.h"
+#include "Pokemon/Pokemon_Strings.h"
+#include "PokemonSV/Programs/Boxes/PokemonSV_BoxRoutines.h"
+#include "PokemonSV/Programs/PokemonSV_Navigation.h"
 #include "PokemonSV_ThreeSegmentDudunsparceFinder.h"
 
 namespace PokemonAutomation{
 namespace NintendoSwitch{
 namespace PokemonSV{
 
+using namespace Pokemon;
 
 ThreeSegmentDudunsparceFinder_Descriptor::ThreeSegmentDudunsparceFinder_Descriptor()
     : SingleSwitchProgramDescriptor(
         "PokemonSV:ThreeSegmentDudunsparceFinder",
-        "Game Name", "Three-Segment Dudunsparce Finder",
+        STRING_POKEMON + " SV", "Three-Segment Dudunsparce Finder",
         "ComputerControl/blob/master/Wiki/Programs/PokemonSV/ThreeSegmentDudunsparceFinder.md",
-        "<Description of this program>.",
+        "Check whether a box of Dunsparce contain at least one that evolves into Three-Segment Dudunsparce.",
         FeedbackType::REQUIRED,
         AllowCommandsWhenRunning::DISABLE_COMMANDS,
         {ControllerFeature::NintendoSwitch_ProController},
@@ -47,6 +53,13 @@ std::unique_ptr<StatsTracker> ThreeSegmentDudunsparceFinder_Descriptor::make_sta
 
 ThreeSegmentDudunsparceFinder::ThreeSegmentDudunsparceFinder()
     : GO_HOME_WHEN_DONE(false)
+    , HAS_CLONE_RIDE_POKEMON(
+        "<b>Cloned Ride Legendary 2nd in Party:</b><br>"
+        "Ride legendary cannot be cloned after patch 1.0.1. To preserve the existing clone while hatching eggs, "
+        "place it as second in party before starting the program.</b>"
+        "The program will skip the first row of eggs in the box as a result.",
+        LockMode::LOCK_WHILE_RUNNING,
+        false)
     , NOTIFICATION_STATUS_UPDATE("Status Update", true, false, std::chrono::seconds(3600))
     , NOTIFICATIONS({
         &NOTIFICATION_STATUS_UPDATE,
@@ -56,9 +69,28 @@ ThreeSegmentDudunsparceFinder::ThreeSegmentDudunsparceFinder()
     })
 {
     PA_ADD_OPTION(GO_HOME_WHEN_DONE);
+    PA_ADD_OPTION(HAS_CLONE_RIDE_POKEMON);
     PA_ADD_OPTION(NOTIFICATIONS);
 }
 
+void ThreeSegmentDudunsparceFinder::check_one_column(SingleSwitchProgramEnvironment& env, ProControllerContext& context, int column_index){
+    enter_box_system_from_overworld(env.program_info(), env.console, context);
+    const uint8_t expected_empty_slots_in_party = HAS_CLONE_RIDE_POKEMON ? 4 : 5;
+    if (check_empty_slots_in_party(env.program_info(), env.console, context) != expected_empty_slots_in_party){
+        throw_and_log<FatalProgramException>(
+            env.console, ErrorReport::SEND_ERROR_REPORT,
+            "Your party should have " + std::to_string(expected_empty_slots_in_party) + " " + STRING_POKEMON + ".",
+            env.console
+        );
+    }
+
+    load_one_column_to_party(env, env.console, context, NOTIFICATION_ERROR_FATAL, column_index, HAS_CLONE_RIDE_POKEMON);
+    const int menu_index = 0;
+    // go to bag from box system
+    enter_menu_from_box_system(env.program_info(), env.console, context, menu_index, MenuSide::RIGHT);
+    context.wait_for_all_requests();
+    // BagWatcher bag_detector(BagWatcher::FinderType::PRESENT, COLOR_GREEN);
+}
 
 void ThreeSegmentDudunsparceFinder::program(SingleSwitchProgramEnvironment& env, ProControllerContext& context){
     assert_16_9_720p_min(env.logger(), env.console);
@@ -69,7 +101,10 @@ void ThreeSegmentDudunsparceFinder::program(SingleSwitchProgramEnvironment& env,
     pbf_press_button(context, BUTTON_L, 10, 100);
 
     try{
-        
+        for(int i = 0; i < 6; i++){
+            check_one_column(env, context, i);
+            break; // XXX
+        }
     } catch(OperationFailedException&){
         stats.m_errors++;
         env.update_stats();

SerialPrograms/Source/PokemonSV/Programs/PokemonSV_ThreeSegmentDudunsparceFinder.h

@@ -7,6 +7,7 @@
 #ifndef PokemonAutomation_PokemonSV_ThreeSegmentDudunsparceFinder_H
 #define PokemonAutomation_PokemonSV_ThreeSegmentDudunsparceFinder_H
 
+#include "Common/Cpp/Options/BooleanCheckBoxOption.h"
 #include "CommonFramework/Notifications/EventNotificationsTable.h"
 #include "NintendoSwitch/Options/NintendoSwitch_GoHomeWhenDoneOption.h"
 #include "NintendoSwitch/NintendoSwitch_SingleSwitchProgram.h"
@@ -33,11 +34,13 @@ class ThreeSegmentDudunsparceFinder : public SingleSwitchProgramInstance{
     virtual void program(SingleSwitchProgramEnvironment& env, ProControllerContext& context) override;
 
 private:
+    void check_one_column(SingleSwitchProgramEnvironment& env, ProControllerContext& context, int column_index);
 
 private:
-
     GoHomeWhenDoneOption GO_HOME_WHEN_DONE;
 
+    BooleanCheckBoxOption HAS_CLONE_RIDE_POKEMON;
+
 
     EventNotificationOption NOTIFICATION_STATUS_UPDATE;
     EventNotificationsOption NOTIFICATIONS;