Add place holder code for new program GeneratePokemonImageTrainingData

Author: Gin

SerialPrograms/CMakeLists.txt

@@ -77,8 +77,8 @@ file(GLOB MAIN_SOURCES
     ../Common/CRC32.h
     ../Common/Compiler.h
     ../Common/ControllerStates/HID_Keyboard_State.h
-    ../Common/ControllerStates/NintendoSwitch_WirelessController_State.h
     ../Common/ControllerStates/NintendoSwitch2_WiredController_State.h
+    ../Common/ControllerStates/NintendoSwitch_WirelessController_State.h
     ../Common/Cpp/AbstractLogger.h
     ../Common/Cpp/BitmapConversion.cpp
     ../Common/Cpp/BitmapConversion.h
@@ -728,25 +728,25 @@ file(GLOB MAIN_SOURCES
     Source/Controllers/SerialPABotBase/SerialPABotBase_StatusThread.h
     Source/Controllers/StandardHid/StandardHid_Keyboard.cpp
     Source/Controllers/StandardHid/StandardHid_Keyboard.h
+    Source/Controllers/StandardHid/StandardHid_KeyboardWithScheduler.cpp
+    Source/Controllers/StandardHid/StandardHid_KeyboardWithScheduler.h
     Source/Controllers/StandardHid/StandardHid_Keyboard_ControllerButtons.h
     Source/Controllers/StandardHid/StandardHid_Keyboard_SerialPABotBase.cpp
     Source/Controllers/StandardHid/StandardHid_Keyboard_SerialPABotBase.h
-    Source/Controllers/StandardHid/StandardHid_KeyboardWithScheduler.cpp
-    Source/Controllers/StandardHid/StandardHid_KeyboardWithScheduler.h
     Source/Integrations/DiscordIntegrationSettings.cpp
     Source/Integrations/DiscordIntegrationSettings.h
     Source/Integrations/DiscordIntegrationTable.cpp
     Source/Integrations/DiscordIntegrationTable.h
     Source/Integrations/DiscordSettingsOption.cpp
     Source/Integrations/DiscordSettingsOption.h
+    Source/Integrations/DiscordSocial/DiscordSocial.cpp
+    Source/Integrations/DiscordSocial/DiscordSocial.h
     Source/Integrations/DiscordWebhook.cpp
     Source/Integrations/DiscordWebhook.h
     Source/Integrations/DiscordWebhookSettings.cpp
     Source/Integrations/DiscordWebhookSettings.h
     Source/Integrations/DppIntegration/DppClient.cpp
     Source/Integrations/DppIntegration/DppClient.h
-    Source/Integrations/DiscordSocial/DiscordSocial.cpp
-    Source/Integrations/DiscordSocial/DiscordSocial.h
     Source/Integrations/DppIntegration/DppCommandHandler.cpp
     Source/Integrations/DppIntegration/DppCommandHandler.h
     Source/Integrations/DppIntegration/DppUtility.cpp
@@ -1492,6 +1492,8 @@ file(GLOB MAIN_SOURCES
     Source/PokemonLA/Programs/General/PokemonLA_RamanasIslandCombee.h
     Source/PokemonLA/Programs/General/PokemonLA_SkipToFullMoon.cpp
     Source/PokemonLA/Programs/General/PokemonLA_SkipToFullMoon.h
+    Source/PokemonLA/Programs/ML/PokemonLA_GeneratePokemonImageTrainingData.cpp
+    Source/PokemonLA/Programs/ML/PokemonLA_GeneratePokemonImageTrainingData.h
     Source/PokemonLA/Programs/PokemonLA_BattleRoutines.cpp
     Source/PokemonLA/Programs/PokemonLA_BattleRoutines.h
     Source/PokemonLA/Programs/PokemonLA_EscapeFromAttack.cpp
@@ -1723,7 +1725,7 @@ file(GLOB MAIN_SOURCES
     Source/PokemonSV/Inference/PokemonSV_WhiteButtonDetector.cpp
     Source/PokemonSV/Inference/PokemonSV_WhiteButtonDetector.h
     Source/PokemonSV/Inference/PokemonSV_WhiteTriangleDetector.cpp
-    Source/PokemonSV/Inference/PokemonSV_WhiteTriangleDetector.h    
+    Source/PokemonSV/Inference/PokemonSV_WhiteTriangleDetector.h
     Source/PokemonSV/Inference/PokemonSV_ZeroGateWarpPromptDetector.cpp
     Source/PokemonSV/Inference/PokemonSV_ZeroGateWarpPromptDetector.h
     Source/PokemonSV/Inference/Tera/PokemonSV_TeraCardDetector.cpp

SerialPrograms/Source/PokemonLA/PokemonLA_Panels.cpp

@@ -45,6 +45,8 @@
 #include "Programs/TestPrograms/PokemonLA_FlagNavigationTest.h"
 #include "Programs/TestPrograms/PokemonLA_SoundListener.h"
 
+#include "Programs/ML/PokemonLA_GeneratePokemonImageTrainingData.h"
+
 namespace PokemonAutomation{
 namespace NintendoSwitch{
 namespace PokemonLA{
@@ -99,6 +101,9 @@ std::vector<PanelEntry> PanelListFactory::make_panels() const{
         ret.emplace_back(make_single_switch_program<AutoMultiSpawn_Descriptor, AutoMultiSpawn>());
     }
     if (PreloadSettings::instance().DEVELOPER_MODE){
+        ret.emplace_back("---- Machine Learning ----");
+        ret.emplace_back(make_single_switch_program<GeneratePokemonImageTrainingData_Descriptor, GeneratePokemonImageTrainingData>());
+
         ret.emplace_back("---- Developer Tools ----");
         ret.emplace_back(make_single_switch_program<MountDetectionTest_Descriptor, MountDetectionTest>());
         ret.emplace_back(make_single_switch_program<OverworldWatcher_Descriptor, OverworldWatcher>());

SerialPrograms/Source/PokemonLA/Programs/ML/PokemonLA_GeneratePokemonImageTrainingData.cpp

@@ -0,0 +1,77 @@
+/*  Generate Pokemon Image Training Data
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *
+ */
+
+#include "Common/Cpp/Exceptions.h"
+#include "CommonFramework/Exceptions/ProgramFinishedException.h"
+#include "CommonFramework/ImageTools/ImageStats.h"
+#include "CommonFramework/Notifications/ProgramNotifications.h"
+#include "CommonFramework/VideoPipeline/VideoFeed.h"
+#include "NintendoSwitch/Commands/NintendoSwitch_Commands_PushButtons.h"
+#include "Pokemon/Pokemon_Strings.h"
+#include "PokemonLA/Programs/PokemonLA_GameEntry.h"
+#include "PokemonLA_GeneratePokemonImageTrainingData.h"
+
+namespace PokemonAutomation{
+namespace NintendoSwitch{
+namespace PokemonLA{
+
+using namespace Pokemon;
+
+
+GeneratePokemonImageTrainingData_Descriptor::GeneratePokemonImageTrainingData_Descriptor()
+    : SingleSwitchProgramDescriptor(
+        "PokemonLA:GeneratePokemonImageTrainingData",
+        STRING_POKEMON + " LA", "Generate " + STRING_POKEMON + " Image Training Data",
+        "ComputerControl/blob/master/Wiki/Programs/PokemonLA/GeneratePokemonImageTrainingData.md",
+        "Generate training images of " + STRING_POKEMON + " for machine learning purposes.",
+        ProgramControllerClass::StandardController_NoRestrictions,
+        FeedbackType::REQUIRED,
+        AllowCommandsWhenRunning::DISABLE_COMMANDS
+    )
+{}
+
+
+GeneratePokemonImageTrainingData::GeneratePokemonImageTrainingData()
+    : NUM_BOXES(
+        "<b>Number of Boxes to Process:</b><br>How many boxes to capture " + STRING_POKEMON + " from.",
+        LockMode::LOCK_WHILE_RUNNING,
+        5, 1, 30
+    )
+    , NOTIFICATION_STATUS_UPDATE("Status Update", true, false, std::chrono::seconds(3600))
+    , NOTIFICATIONS({
+        &NOTIFICATION_STATUS_UPDATE,
+        &NOTIFICATION_PROGRAM_FINISH,
+        &NOTIFICATION_ERROR_FATAL,
+    })
+{
+    PA_ADD_OPTION(NUM_BOXES);
+    PA_ADD_OPTION(NOTIFICATIONS);
+}
+
+void GeneratePokemonImageTrainingData::program(SingleSwitchProgramEnvironment& env, ProControllerContext& context){
+    // Send program starting notification
+    send_program_status_notification(env, NOTIFICATION_STATUS_UPDATE);
+
+    try{
+        // Send completion notification
+        send_program_finished_notification(env, NOTIFICATION_PROGRAM_FINISH);
+    }catch(ProgramFinishedException&){
+        // Program was stopped by user
+        send_program_finished_notification(env, NOTIFICATION_PROGRAM_FINISH);
+        throw;
+    }catch(std::exception& e){
+        // Handle any other errors
+        send_program_fatal_error_notification(env, NOTIFICATION_ERROR_FATAL, "Fatal error occurred: " + std::string(e.what()));
+        throw;
+    }
+}
+
+
+
+
+}
+}
+}

SerialPrograms/Source/PokemonLA/Programs/ML/PokemonLA_GeneratePokemonImageTrainingData.h

@@ -0,0 +1,50 @@
+/*  Generate Pokemon Image Training Data
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *  
+ *  Starting at Jubilife Village or any camp, talk to npc to get pokemon from box to party,
+ *  throw the pokemon out onto the field to take screenshots with it, move around it to take
+ *  screenshots from various angles. Repeat for rest of the pokemon in specified boxes.
+ */
+
+#ifndef PokemonAutomation_PokemonLA_GeneratePokemonImageTrainingData_H
+#define PokemonAutomation_PokemonLA_GeneratePokemonImageTrainingData_H
+
+#include "Common/Cpp/Options/BooleanCheckBoxOption.h"
+#include "Common/Cpp/Options/SimpleIntegerOption.h"
+#include "Common/Cpp/Options/TimeExpressionOption.h"
+#include "CommonFramework/Notifications/EventNotificationsTable.h"
+#include "NintendoSwitch/NintendoSwitch_SingleSwitchProgram.h"
+
+namespace PokemonAutomation{
+namespace NintendoSwitch{
+namespace PokemonLA{
+
+
+class GeneratePokemonImageTrainingData_Descriptor : public SingleSwitchProgramDescriptor{
+public:
+    GeneratePokemonImageTrainingData_Descriptor();
+};
+
+
+class GeneratePokemonImageTrainingData : public SingleSwitchProgramInstance{
+public:
+    GeneratePokemonImageTrainingData();
+
+    virtual void program(SingleSwitchProgramEnvironment& env, ProControllerContext& context) override;
+
+private:
+
+    SimpleIntegerOption<uint16_t> NUM_BOXES;
+
+    EventNotificationOption NOTIFICATION_STATUS_UPDATE;
+    EventNotificationsOption NOTIFICATIONS;
+};
+
+
+
+
+}
+}
+}
+#endif