Add numpy utilities and game data pre-processing (#36)

ViliamVadocz · web-flow · commit 935acd891b22 · 2020-02-13T20:33:15.000-08:00
diff --git a/src/bot.py b/src/bot.py
@@ -11,6 +11,7 @@
 from util.spikes import SpikeWatcher
 from util.vec import Vec3
 
+# Would you like to use numpy utilities? Check out the np_util folder!
 
 class MyBot(BaseAgent):
 
@@ -97,7 +98,7 @@ def find_correction(current: Vec3, ideal: Vec3) -> float:
 
 def draw_debug(renderer, text_lines: List[str]):
     renderer.begin_rendering()
-    y = 200
+    y = 250
     for line in text_lines:
         renderer.draw_string_2d(50, y, 1, 1, line, renderer.yellow())
         y += 20
diff --git a/src/np_util/data.py b/src/np_util/data.py
@@ -0,0 +1,141 @@
+'''Rocket League data pre-processing.'''
+
+from .utils import Car, Ball, BoostPad, arr_from_list, arr_from_rot, arr_from_vec, orient_matrix
+
+import numpy as np
+
+
+def setup(self, packet, field_info):
+    """Sets up the variables and classes for the agent.
+
+    Arguments:
+        self {BaseAgent} -- The agent.
+        packet {GameTickPacket} -- Information about the game.
+        field_info {FieldInfoPacket} -- Information about the game field.
+    """
+
+    # Game info
+    self.game_time = packet.game_info.seconds_elapsed
+    self.dt = 1.0 / 120.0
+    self.last_time = 0.0
+    self.r_active = packet.game_info.is_round_active
+    self.ko_pause = packet.game_info.is_kickoff_pause
+    self.m_ended = packet.game_info.is_match_ended
+    self.gravity = packet.game_info.world_gravity_z
+
+    # Creates Car objects for each car.
+    self.teammates = []
+    self.opponents = []
+    for index in range(packet.num_cars):
+        car = packet.game_cars[index]
+        if index == self.index:
+            self.player = Car(self.index, self.team, self.name)
+        elif car.team == self.team:
+            self.teammates.append(Car(index, car.team, car.name))
+        else:
+            self.opponents.append(Car(index, car.team, car.name))
+
+    # Creates a Ball object.
+    self.ball = Ball()
+
+    # Creates Boostpad objects.
+    self.l_pads = []
+    self.s_pads = []
+    for i in range(field_info.num_boosts):
+        pad = field_info.boost_pads[i]
+        pad_type = self.l_pads if pad.is_full_boost else self.s_pads
+        pad_obj = BoostPad(i, arr_from_vec(pad.location))
+        pad_type.append(pad_obj)
+
+
+def process(self, packet):
+    """Processes the gametick packet.
+    Arguments:
+        self {BaseAgent} -- The agent.
+        packet {GameTickPacket} -- The game packet being processed.
+    """
+
+    # Processing game info.
+    self.game_time = packet.game_info.seconds_elapsed
+    self.dt = self.game_time - self.last_time
+    self.last_time = self.game_time
+    self.r_active = packet.game_info.is_round_active
+    self.ko_pause = packet.game_info.is_kickoff_pause
+    self.m_ended = packet.game_info.is_match_ended
+    self.gravity = packet.game_info.world_gravity_z
+
+    # Processing player data.
+    # From packet:
+    self.player.pos = arr_from_vec(packet.game_cars[self.player.index].physics.location)
+    self.player.rot = arr_from_rot(packet.game_cars[self.player.index].physics.rotation)
+    self.player.vel = arr_from_vec(packet.game_cars[self.player.index].physics.velocity)
+    self.player.ang_vel = arr_from_vec(packet.game_cars[self.player.index].physics.angular_velocity)
+    self.player.dead = packet.game_cars[self.player.index].is_demolished
+    self.player.wheel_c = packet.game_cars[self.player.index].has_wheel_contact
+    self.player.sonic = packet.game_cars[self.player.index].is_super_sonic
+    self.player.jumped = packet.game_cars[self.player.index].jumped
+    self.player.d_jumped = packet.game_cars[self.player.index].double_jumped
+    self.player.boost = packet.game_cars[self.player.index].boost
+    # Calculated:
+    self.player.orient_m = orient_matrix(self.player.rot, self.player.orient_m)
+
+    # Processing teammates.
+    for teammate in self.teammates:
+        # From packet:
+        teammate.pos = arr_from_vec(packet.game_cars[teammate.index].physics.location)
+        teammate.rot        = arr_from_rot(packet.game_cars[teammate.index].physics.rotation)
+        teammate.vel        = arr_from_vec(packet.game_cars[teammate.index].physics.velocity)
+        teammate.ang_vel    = arr_from_vec(packet.game_cars[teammate.index].physics.angular_velocity)
+        teammate.dead       = packet.game_cars[teammate.index].is_demolished
+        teammate.wheel_c    = packet.game_cars[teammate.index].has_wheel_contact
+        teammate.sonic      = packet.game_cars[teammate.index].is_super_sonic
+        teammate.jumped     = packet.game_cars[teammate.index].jumped
+        teammate.d_jumped   = packet.game_cars[teammate.index].double_jumped
+        teammate.boost      = packet.game_cars[teammate.index].boost
+
+    # Processing opponents.
+    for opponent in self.opponents:
+        # From packet:
+        opponent.pos = arr_from_vec(packet.game_cars[opponent.index].physics.location)
+        opponent.rot = arr_from_rot(packet.game_cars[opponent.index].physics.rotation)
+        opponent.vel = arr_from_vec(packet.game_cars[opponent.index].physics.velocity)
+        opponent.ang_vel = arr_from_vec(packet.game_cars[opponent.index].physics.angular_velocity)
+        opponent.dead = packet.game_cars[opponent.index].is_demolished
+        opponent.wheel_c = packet.game_cars[opponent.index].has_wheel_contact
+        opponent.sonic = packet.game_cars[opponent.index].is_super_sonic
+        opponent.jumped = packet.game_cars[opponent.index].jumped
+        opponent.d_jumped = packet.game_cars[opponent.index].double_jumped
+        opponent.boost = packet.game_cars[opponent.index].boost
+
+    # Processing Ball data.
+    self.ball.pos = arr_from_vec(packet.game_ball.physics.location)
+    self.ball.rot = arr_from_rot(packet.game_ball.physics.rotation)
+    self.ball.vel = arr_from_vec(packet.game_ball.physics.velocity)
+    self.ball.ang_vel = arr_from_vec(packet.game_ball.physics.angular_velocity)
+    self.ball.last_touch = packet.game_ball.latest_touch
+
+    # Processing ball prediction.
+    ball_prediction_struct = self.get_ball_prediction_struct()
+    dtype = [
+        ('physics', [
+            ('pos', '<f4', 3),
+            ('rot', [
+                ('pitch', '<f4'),
+                ('yaw', '<f4'),
+                ('roll', '<f4')
+            ]),
+            ('vel', '<f4', 3),
+            ('ang_vel', '<f4', 3)
+        ]),
+        ('time', '<f4')
+    ]
+    self.ball_prediction = np.ctypeslib.as_array(
+        ball_prediction_struct.slices).view(dtype)[:ball_prediction_struct.num_slices]
+
+    # Processing Boostpads.
+    self.active_pads = []
+    for pad in self.l_pads + self.s_pads:
+        pad.active = packet.game_boosts[pad.index].is_active
+        pad.timer = packet.game_boosts[pad.index].timer
+        if pad.active == True:
+            self.active_pads.append(pad)
diff --git a/src/np_util/utils.py b/src/np_util/utils.py
