lgpe battle menu detector

Author: kichithewolf

SerialPrograms/CMakeLists.txt

@@ -1375,6 +1375,8 @@ file(GLOB MAIN_SOURCES
     Source/PokemonLA/Resources/PokemonLA_WeatherAndTimeIcons.h
     Source/PokemonLGPE/Commands/PokemonLGPE_DateSpam.cpp
     Source/PokemonLGPE/Commands/PokemonLGPE_DateSpam.h
+    Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.cpp
+    Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.h
     Source/PokemonLGPE/Inference/Sounds/PokemonLGPE_ShinySoundDetector.cpp
     Source/PokemonLGPE/Inference/Sounds/PokemonLGPE_ShinySoundDetector.h
     Source/PokemonLGPE/Inference/PokemonLGPE_ShinySymbolDetector.cpp

SerialPrograms/SerialPrograms.pro

@@ -674,6 +674,7 @@ SOURCES += \
     Source/PokemonLA/Resources/PokemonLA_PokemonSprites.cpp \
     Source/PokemonLA/Resources/PokemonLA_WeatherAndTimeIcons.cpp \
     Source/PokemonLGPE/Commands/PokemonLGPE_DateSpam.cpp \
+    Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.cpp \
     Source/PokemonLGPE/Inference/Sounds/PokemonLGPE_ShinySoundDetector.cpp \
     Source/PokemonLGPE/Inference/PokemonLGPE_ShinySymbolDetector.cpp \
     Source/PokemonLGPE/Programs/Farming/PokemonLGPE_DailyItemFarmer.cpp \
@@ -1871,6 +1872,7 @@ HEADERS += \
     Source/PokemonLA/Resources/PokemonLA_PokemonSprites.h \
     Source/PokemonLA/Resources/PokemonLA_WeatherAndTimeIcons.h \
     Source/PokemonLGPE/Commands/PokemonLGPE_DateSpam.h \
+    Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.h \
     Source/PokemonLGPE/Inference/Sounds/PokemonLGPE_ShinySoundDetector.h \
     Source/PokemonLGPE/Inference/PokemonLGPE_ShinySymbolDetector.h \
     Source/PokemonLGPE/Programs/Farming/PokemonLGPE_DailyItemFarmer.h \

SerialPrograms/Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.cpp

@@ -0,0 +1,44 @@
+/*  Battle Menu Detector
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *
+ */
+
+#include "CommonTools/Images/SolidColorTest.h"
+#include "CommonTools/Images/ImageFilter.h"
+#include "CommonFramework/ImageTools/ImageBoxes.h"
+#include "CommonFramework/ImageTypes/ImageRGB32.h"
+#include "CommonFramework/ImageTools/ImageStats.h"
+#include "CommonFramework/ImageTypes/ImageViewRGB32.h"
+#include "CommonFramework/VideoPipeline/VideoOverlayScopes.h"
+#include "PokemonLGPE_BattleMenuDetector.h"
+
+namespace PokemonAutomation{
+namespace NintendoSwitch{
+namespace PokemonLGPE{
+
+BattleMenuDetector::BattleMenuDetector(Color color)
+    : m_your_box(0.114, 0.916, 0.016, 0.027)
+    , m_opponent_box(0.904, 0.048, 0.022, 0.031)
+{}
+void BattleMenuDetector::make_overlays(VideoOverlaySet& items) const{
+    items.add(COLOR_RED, m_your_box);
+    items.add(COLOR_RED, m_opponent_box);
+}
+bool BattleMenuDetector::detect(const ImageViewRGB32& screen) const{
+    ImageViewRGB32 left_image = extract_box_reference(screen, m_your_box);
+    ImageViewRGB32 right_image = extract_box_reference(screen, m_opponent_box);
+
+    //Has a bit of a gradient in the box
+    //TODO: Do more boxes, horizonal so less gradient variation.
+    if (is_solid(left_image, { 0.040, 0.467, 0.493 }) && is_solid(right_image, { 0.074, 0.458, 0.468 })){
+        return true;
+    }
+    return false;
+}
+
+
+
+}
+}
+}

SerialPrograms/Source/PokemonLGPE/Inference/Battles/PokemonLGPE_BattleMenuDetector.h

@@ -0,0 +1,60 @@
+/*  Battle Menu Detector
+ *
+ *  From: https://github.com/PokemonAutomation/
+ *
+ */
+
+#ifndef PokemonAutomation_PokemonLGPE_BattleMenuDetector_H
+#define PokemonAutomation_PokemonLGPE_BattleMenuDetector_H
+
+#include <chrono>
+#include <atomic>
+#include "CommonFramework/VideoPipeline/VideoOverlayScopes.h"
+#include "Common/Cpp/Color.h"
+#include "CommonFramework/ImageTools/ImageBoxes.h"
+#include "CommonTools/VisualDetector.h"
+#include "CommonTools/InferenceCallbacks/VisualInferenceCallback.h"
+
+namespace PokemonAutomation{
+    class CancellableScope;
+    class VideoFeed;
+namespace NintendoSwitch{
+namespace PokemonLGPE{
+
+//Detect the light blue of your HP box and your opponent's HP box
+class BattleMenuDetector : public StaticScreenDetector{
+public:
+    BattleMenuDetector(Color color);
+
+    virtual void make_overlays(VideoOverlaySet& items) const override;
+    virtual bool detect(const ImageViewRGB32& screen) const override;
+
+private:
+    ImageFloatBox m_your_box;
+    ImageFloatBox m_opponent_box;
+};
+class BattleMenuWatcher : public DetectorToFinder<BattleMenuDetector>{
+public:
+    BattleMenuWatcher(Color color)
+        : DetectorToFinder("BattleMenuWatcher", std::chrono::milliseconds(250), color)
+    {}
+};
+
+//Battle dialog detector - white of the box
+
+/*
+Possible future improvements:
+Detect arrow and position
+|Fight|Pokemon|Bag|Run| for wild battle
+|---|Fight|Pokemon|Bag| for trainer battle
+(align right)
+
+Detect Y Info button located next to menu.
+Don't know what changes for a double battle.
+*/
+
+}
+}
+}
+
+#endif