Added Some Values To Pokemon Game Wrapper

pyboy/plugins/game_wrapper_pokemon_gen1.py

@@ -15,6 +15,23 @@
 
 logger = pyboy.logging.get_logger(__name__)
 
+# Set of addresses for information on pokemon
+# Based on the following link: https://datacrystal.tcrf.net/wiki/Pok%C3%A9mon_Red_and_Blue/RAM_map
+ADDR_TITLE_SCREEN = 0x1FC3
+ADDR_NUMBER_OF_POKEMON = 0xD163 # In party
+
+#Item related values
+ADDR_TOTAL_ITEMS = 0xD53A
+ADDR_BADGES = 0xD356 # Currently requires conversion (each bit is an independent switch and represents a badge)
+
+# Game Time
+ADDR_HOURS = 0xDA40
+ADDR_MINUTES = 0xDA42
+ADDR_SECONDS = 0xDA44
+
+# Battle mode
+ADDR_BATTLE = 0xD057
+
 
 class GameWrapperPokemonGen1(PyBoyGameWrapper):
     """
@@ -28,6 +45,24 @@ class GameWrapperPokemonGen1(PyBoyGameWrapper):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, game_area_section=(0, 0, 20, 18), game_area_follow_scxy=True, **kwargs)
         self.sprite_offset = 0
+        if self.pyboy.memory[ADDR_TITLE_SCREEN] == 0:
+            pass
+        else:
+            self.number_of_pokemon = self.pyboy.memory[ADDR_NUMBER_OF_POKEMON]
+            self.total_items = self.pyboy.memory[ADDR_TOTAL_ITEMS]
+            # Extract the time in game:
+            _hours = self.pyboy.memory[ADDR_HOURS]
+            _minutes = self.pyboy.memory[ADDR_MINUTES]
+            _seconds = self.pyboy.memory[ADDR_SECONDS]
+            self.game_time = f'{_hours}:{_minutes}:{_seconds}'
+
+            # Check whether the player is in battle mode or out of battle
+            if self.pyboy.memory[ADDR_BATTLE] == 1:
+                self.battle_state = 'In Battle'
+            else:
+                self.battle_state = 'Overworld'
+
+            self.badges = self.pyboy.memory[ADDR_BADGES] #Research how this works.
 
     def enabled(self):
         return (self.pyboy.cartridge_title == "POKEMON RED") or (self.pyboy.cartridge_title == "POKEMON BLUE")
@@ -36,6 +71,22 @@ def post_tick(self):
         self._tile_cache_invalid = True
         self._sprite_cache_invalid = True
 
+        self.number_of_pokemon = self.pyboy.memory[ADDR_NUMBER_OF_POKEMON]
+        self.total_items = self.pyboy.memory[ADDR_TOTAL_ITEMS]
+        # Extract the time in game:
+        _hours = self.pyboy.memory[ADDR_HOURS]
+        _minutes = self.pyboy.memory[ADDR_MINUTES]
+        _seconds = self.pyboy.memory[ADDR_SECONDS]
+        self.game_time = f'{_hours}:{_minutes}:{_seconds}'
+
+        # Check whether the player is in battle mode or out of battle
+        if self.pyboy.memory[ADDR_BATTLE] == 1:
+            self.battle_state = 'In Battle'
+        else:
+            self.battle_state = 'Overworld'
+
+        self.badges = self.pyboy.memory[ADDR_BADGES] #Research how this works.
+
         scanline_parameters = self.pyboy.screen.tilemap_position_list
         # WX = scanline_parameters[0][2]
         WY = scanline_parameters[0][3]
@@ -78,4 +129,9 @@ def game_area_collision(self):
         return game_area
 
     def __repr__(self):
-        return "Pokemon Gen 1:\n" + super().__repr__()
+        return ("Pokemon Gen 1:\n"
+                + f"Pokemon in Party: {self.number_of_pokemon}\n"
+                + f"Battle State: {self.battle_state}\n"
+                + f"Game Time: {self.game_time}\n"
+                + f"Total items: {self.total_items}\n"
+                + super().__repr__())