from AI.base import *
class TeamAI(BaseAI):
def __init__(self, helper):
self.helper = helper
self.equipments = [0, 0, 0, 0, 0] # Set the number of your equipments.
self.color = (255, 255, 255) # Set the color you like.
def decide(self):
my_pos = self.helper.get_player_position()
radius = self.helper.player_radius
carry = self.helper.get_player_value()
nearest_oil_pos = self.helper.get_nearest_oil()
home = self.helper.get_base_center()
destination = nearest_oil_pos if carry < 5000 else home
if destination is None:
return AI_DIR_STOP
else:
direction = (destination[0] - my_pos[0], destination[1] - my_pos[1])
return self.helper.get_direction(direction)