import math

from dataclasses import dataclass

from rlbot.flat import GamePacket, PlayerInfo

from util.vec import Vec3

# When the ball is attached to a car's spikes, the distance will vary a bit depending on whether the ball is

# on the front bumper, the roof, etc. It tends to be most far away when the ball is on one of the front corners

# and that distance is a little under 200. We want to be sure that it's never over 200, otherwise bots will

# suffer from bad bugs when they don't think the ball is spiked to them but it actually is; they'll probably

# drive in circles. The opposite problem, where they think it's spiked before it really is, is not so bad because

# they usually spike it for real a split second later.

MAX_DISTANCE_WHEN_SPIKED = 200

@dataclass(init=False, slots=True)

class SpikeWatcher:

carrying_car: PlayerInfo | None = None

spike_moment: float = 0

carry_duration: float = 0

def read_packet(self, packet: GamePacket):

ball_location = Vec3(packet.balls[0].physics.location)

closest_candidate: PlayerInfo | None = None

closest_distance = math.inf

for car in packet.players:

car_location = Vec3(car.physics.location)

distance = car_location.dist(ball_location)

if distance < MAX_DISTANCE_WHEN_SPIKED:

if distance < closest_distance:

closest_candidate = car

closest_distance = distance

if closest_candidate != self.carrying_car and closest_candidate is not None:

self.spike_moment = packet.match_info.seconds_elapsed

self.carrying_car = closest_candidate

if self.carrying_car is not None:

self.carry_duration = packet.match_info.seconds_elapsed - self.spike_moment