Experimental async packet processor

Author: VirxEC

rlbot/managers/bot.py

@@ -1,4 +1,5 @@
 import os
+from threading import Thread, Event
 from traceback import print_exc
 from typing import Optional
 
@@ -28,6 +29,12 @@ class Bot:
     _has_match_settings = False
     _has_field_info = False
 
+    _latest_packet = flat.GameTickPacket()
+    _lastest_prediction = flat.BallPrediction()
+    _packet_event = Event()
+    _packet_thread = None
+    _run_packet_thread = True
+
     def __init__(self):
         spawn_id = os.environ.get("RLBOT_SPAWN_IDS")
 
@@ -48,6 +55,9 @@ def __init__(self):
         )
         self._game_interface.packet_handlers.append(self._handle_packet)
 
+        self._packet_thread = Thread(target=self._packet_processor, daemon=True)
+        self._packet_thread.start()
+
         self.renderer = Renderer(self._game_interface)
 
     def _initialize_agent(self):
@@ -86,49 +96,67 @@ def _handle_field_info(self, field_info: flat.FieldInfo):
             self._initialize_agent()
 
     def _handle_ball_prediction(self, ball_prediction: flat.BallPrediction):
-        self.ball_prediction = ball_prediction
+        self._lastest_prediction = ball_prediction
 
     def _handle_packet(self, packet: flat.GameTickPacket):
-        if (
-            self.index == -1
-            or len(packet.players) <= self.index
-            or packet.players[self.index].spawn_id != self.spawn_id
-        ):
-            # spawn id should only be 0 if RLBOT_SPAWN_IDS was not set
-            if self.spawn_id == 0:
-                # in this case, if there's only one player, we can assume it's us
-                player_index = -1
-                for i, player in enumerate(packet.players):
-                    # skip human players/psyonix bots
-                    if not player.is_bot:
-                        continue
+        self._latest_packet = packet
+        self._packet_event.set()
 
-                    if player_index != -1:
-                        self.logger.error(
-                            "Multiple bots in the game, please set RLBOT_SPAWN_IDS"
-                        )
-                        return
+    def _packet_processor(self):
+        while self._run_packet_thread:
+            self._packet_event.wait()
 
-                    player_index = i
-                self.index = player_index
+            # if the thread was unblocked,
+            # but it we're not supposed to be running,
+            # then exit
+            if not self._run_packet_thread:
+                return
 
-            for i, player in enumerate(packet.players):
-                if player.spawn_id == self.spawn_id:
-                    self.index = i
-                    break
+            self.ball_prediction = self._lastest_prediction
+            packet: flat.GameTickPacket = self._latest_packet
+
+            self._packet_event.clear()
+
+            if (
+                self.index == -1
+                or len(packet.players) <= self.index
+                or packet.players[self.index].spawn_id != self.spawn_id
+            ):
+                # spawn id should only be 0 if RLBOT_SPAWN_IDS was not set
+                if self.spawn_id == 0:
+                    # in this case, if there's only one player, we can assume it's us
+                    player_index = -1
+                    for i, player in enumerate(packet.players):
+                        # skip human players/psyonix bots
+                        if not player.is_bot:
+                            continue
+
+                        if player_index != -1:
+                            self.logger.error(
+                                "Multiple bots in the game, please set RLBOT_SPAWN_IDS"
+                            )
+                            return
+
+                        player_index = i
+                    self.index = player_index
 
-            if self.index == -1:
+                for i, player in enumerate(packet.players):
+                    if player.spawn_id == self.spawn_id:
+                        self.index = i
+                        break
+
+                if self.index == -1:
+                    return
+
+            try:
+                controller = self.get_output(packet)
+            except Exception as e:
+                self.logger.error("Bot %s returned an error to RLBot: %s", self.name, e)
+                print_exc()
                 return
 
-        try:
-            controller = self.get_output(packet)
-        except Exception as e:
-            self.logger.error("Bot %s returned an error to RLBot: %s", self.name, e)
-            print_exc()
-            return
-
-        player_input = flat.PlayerInput(self.index, controller)
-        self._game_interface.send_player_input(player_input)
+            player_input = flat.PlayerInput(self.index, controller)
+            self._game_interface.send_player_input(player_input)
 
     def run(
         self,
@@ -144,6 +172,9 @@ def run(
                 rlbot_server_port=rlbot_server_port,
             )
         finally:
+            self._run_packet_thread = False
+            self._packet_event.set()
+
             self.retire()
             del self._game_interface
 

rlbot/version.py

@@ -1,4 +1,4 @@
-__version__ = "5.0.0-beta.1"
+__version__ = "5.0.0-beta.2"
 
 
 RESET_SEQ = "\033[0m"
@@ -15,9 +15,12 @@ def _get_color(color: int) -> str:
 )
 
 RELEASE_NOTES = {
+    "5.0.0-beta.2": """
+    Ensure bots don't fall behind the most recent GameTickPacket.
+    """,
     "5.0.0-beta.1": """
     Initial iteration of the Python interface for RLBot.
-    """
+    """,
 }
 
 RELEASE_BANNER = f"""