prepare for a wild-based detector

This one isn't implemented yet but might as well get the wrapper
functions out of the way.

1 files changed, 15 insertions, 0 deletions

diff --git a/pokemontools/vba/battle.py b/pokemontools/vba/battle.py
index 9fe4a84..6b52bce 100644
--- a/pokemontools/vba/battle.py
+++ b/pokemontools/vba/battle.py
@@ -64,6 +64,21 @@ class Battle(EmulatorController):
         """
         return self.emulator.is_trainer_switch_prompt()
 
+    def is_wild_switch_prompt(self):
+        """
+        Detects if the battle is waiting for the player to choose whether or
+        not to continue to fight the wild pokemon.
+        """
+        return self.emulator.is_wild_switch_prompt()
+
+    def is_switch_prompt(self):
+        """
+        Detects both trainer and wild switch prompts (for prompting whether to
+        switch pokemon). This is a yes/no box and not the actual pokemon
+        selection menu.
+        """
+        return self.is_trainer_switch_prompt() or self.is_wild_switch_prompt()
+
     def is_mandatory_switch(self):
         """
         Detects if the battle is waiting for the player to choose a next