donut options test program

Author: kichithewolf

SerialPrograms/Source/PokemonLZA/PokemonLZA_Panels.cpp

@@ -48,6 +48,7 @@
 #include "Programs/TestPrograms/PokemonLZA_MoveBoxArrow.h"
 #include "Programs/TestPrograms/PokemonLZA_TestBoxCellInfo.h"
 #include "InferenceTraining/PokemonLZA_GenerateLocationNameOCR.h"
+#include "Programs/TestPrograms/PokemonLZA_DonutOptionsTest.h"
 
 namespace PokemonAutomation{
 namespace NintendoSwitch{
@@ -114,6 +115,7 @@ std::vector<PanelEntry> PanelListFactory::make_panels() const{
         ret.emplace_back(make_single_switch_program<MoveBoxArrow_Descriptor, MoveBoxArrow>());
         ret.emplace_back(make_single_switch_program<TestBoxCellInfo_Descriptor, TestBoxCellInfo>());
         ret.emplace_back(make_single_switch_program<GenerateLocationNameOCR_Descriptor, GenerateLocationNameOCR>());
+        ret.emplace_back(make_single_switch_program<DonutOptionsTest_Descriptor, DonutOptionsTest>());
     }
     return ret;
 }

SerialPrograms/Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_DonutOptionsTest.cpp

@@ -0,0 +1,93 @@
+/*  Donut Options Test
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *
+ */
+
+#include "CommonFramework/Notifications/ProgramNotifications.h"
+#include "CommonFramework/Exceptions/OperationFailedException.h"
+#include "CommonTools/Async/InferenceRoutines.h"
+#include "CommonTools/StartupChecks/VideoResolutionCheck.h"
+#include "NintendoSwitch/Commands/NintendoSwitch_Commands_PushButtons.h"
+#include "Pokemon/Pokemon_Strings.h"
+#include "PokemonLZA/Inference/PokemonLZA_SelectionArrowDetector.h"
+#include "PokemonLZA/Inference/PokemonLZA_DialogDetector.h"
+#include "PokemonSwSh/Inference/PokemonSwSh_IvJudgeReader.h" //TODO: change/remove later
+#include "PokemonLZA_DonutOptionsTest.h"
+
+namespace PokemonAutomation{
+namespace NintendoSwitch{
+namespace PokemonLZA{
+
+using namespace Pokemon;
+
+DonutOptionsTest_Descriptor::DonutOptionsTest_Descriptor()
+    : SingleSwitchProgramDescriptor(
+        "PokemonLZA:DonutOptionsTest",
+        STRING_POKEMON + " LZA", "Donut Options Test",
+        "Programs/PokemonLZA/DonutOptionsTest.html",
+        "Testing options layout for donut making.",
+        ProgramControllerClass::StandardController_NoRestrictions,
+        FeedbackType::REQUIRED,
+        AllowCommandsWhenRunning::DISABLE_COMMANDS
+    )
+{}
+
+DonutOptionsTest::DonutOptionsTest()
+    : LANGUAGE(
+        "<b>Game Language:</b>",
+        PokemonSwSh::IV_READER().languages(), //TODO: replace?
+        LockMode::LOCK_WHILE_RUNNING,
+        true
+    )
+    , BERRIES("<b>Berries:</b><br>The berries used to make the donut. Minimum 3 berries, maximum 8 berries.")
+    , NUM_DONUTS(
+        "<b>Number of Donuts:</b><br>The number of donuts to make.",
+        LockMode::LOCK_WHILE_RUNNING,
+        1, 1
+    )
+    , GO_HOME_WHEN_DONE(false)
+    , NOTIFICATION_STATUS_UPDATE("Status Update", true, false, std::chrono::seconds(3600))
+    , NOTIFICATIONS({
+        &NOTIFICATION_STATUS_UPDATE,
+        &NOTIFICATION_PROGRAM_FINISH,
+        &NOTIFICATION_ERROR_FATAL,
+    })
+{
+    PA_ADD_OPTION(LANGUAGE);
+    PA_ADD_OPTION(BERRIES);
+    PA_ADD_OPTION(NUM_DONUTS);
+    PA_ADD_OPTION(GO_HOME_WHEN_DONE);
+    PA_ADD_OPTION(NOTIFICATIONS);
+}
+
+void DonutOptionsTest::program(SingleSwitchProgramEnvironment& env, ProControllerContext& context){
+    assert_16_9_720p_min(env.logger(), env.console);
+
+    const Language language = LANGUAGE;
+    if (language == Language::None){
+        throw UserSetupError(env.console, "Must set game language option to read ingredient lists.");
+    }
+
+    //Validate number of selected berries
+    env.log("Checking berry count.");
+
+    size_t num_berries = 0;
+    std::vector<std::unique_ptr<DonutBerriesTableRow>> berries_table = BERRIES.copy_snapshot();
+    num_berries = berries_table.size();
+
+    if (num_berries < 3 || num_berries > 8) {
+        throw UserSetupError(env.console, "Must have at least 3 berries and no more than 8 berries.");
+    }
+    env.log("Number of berries validated.", COLOR_BLACK);
+
+
+    GO_HOME_WHEN_DONE.run_end_of_program(context);
+    send_program_finished_notification(env, NOTIFICATION_PROGRAM_FINISH);
+}
+
+
+}
+}
+}
+

SerialPrograms/Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_DonutOptionsTest.h

@@ -0,0 +1,49 @@
+/*  Donut Options Test
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *
+ */
+
+#ifndef PokemonAutomation_PokemonLZA_DonutOptionsTest_H
+#define PokemonAutomation_PokemonLZA_DonutOptionsTest_H
+
+#include "NintendoSwitch/NintendoSwitch_SingleSwitchProgram.h"
+#include "CommonFramework/Notifications/EventNotificationsTable.h"
+#include "NintendoSwitch/Options/NintendoSwitch_GoHomeWhenDoneOption.h"
+#include "Common/Cpp/Options/SimpleIntegerOption.h"
+#include "CommonTools/Options/LanguageOCROption.h"
+#include "PokemonLZA/Options/PokemonLZA_DonutBerriesOption.h"
+
+namespace PokemonAutomation{
+namespace NintendoSwitch{
+namespace PokemonLZA{
+
+class DonutOptionsTest_Descriptor : public SingleSwitchProgramDescriptor{
+public:
+    DonutOptionsTest_Descriptor();
+};
+
+class DonutOptionsTest : public SingleSwitchProgramInstance{
+public:
+    DonutOptionsTest();
+
+    virtual void program(SingleSwitchProgramEnvironment& env, ProControllerContext& context) override;
+
+private:
+    OCR::LanguageOCROption LANGUAGE;
+    DonutBerriesTable BERRIES;
+    SimpleIntegerOption<uint8_t> NUM_DONUTS;
+
+    GoHomeWhenDoneOption GO_HOME_WHEN_DONE;
+    EventNotificationOption NOTIFICATION_STATUS_UPDATE;
+    EventNotificationsOption NOTIFICATIONS;
+};
+
+
+
+
+
+}
+}
+}
+#endif

SerialPrograms/SourceFiles.cmake

@@ -1692,6 +1692,8 @@ file(GLOB LIBRARY_SOURCES
     Source/PokemonLZA/Programs/ShinyHunting/PokemonLZA_WildZoneCafe.h
     Source/PokemonLZA/Programs/ShinyHunting/PokemonLZA_WildZoneEntrance.cpp
     Source/PokemonLZA/Programs/ShinyHunting/PokemonLZA_WildZoneEntrance.h
+    Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_DonutOptionsTest.cpp
+    Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_DonutOptionsTest.h
     Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_MoveBoxArrow.cpp
     Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_MoveBoxArrow.h
     Source/PokemonLZA/Programs/TestPrograms/PokemonLZA_OverworldWatcher.cpp