Skip to content

Updated to use simple controller state #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tarehart merged 4 commits into from
Jun 4, 2018
Merged

Updated to use simple controller state #1

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
afcb6f5
Updated to use simple controller state
dtracers Jun 4, 2018
6cdb017
changed to get_output
dtracers Jun 4, 2018
bc332c7
made changes cleaner
dtracers Jun 4, 2018
c809724
made changes even simpler
dtracers Jun 4, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 8 additions & 11 deletions python_example/python_example.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,23 +1,20 @@
import math

from rlbot.agents.base_flatbuffer_agent import BaseFlatbufferAgent, SimpleControllerState
from rlbot.messages.flat import GameTickPacket
from rlbot.agents.base_agent import BaseAgent, SimpleControllerState
from rlbot.utils.structures.game_data_struct import GameTickPacket

URotationToRadians = math.pi / float(32768)


class PythonExample(BaseFlatbufferAgent):
class PythonExample(BaseAgent):

def get_output(self, packet: GameTickPacket) -> SimpleControllerState:
controller_state = SimpleControllerState()

if packet.Ball() is None: # This happens during replays
return controller_state
ball_location = Vector2(packet.game_ball.physics.location.x, packet.game_ball.physics.location.y)

ball_location = Vector2(packet.Ball().Physics().Location().X(), packet.Ball().Physics().Location().Y())

my_car = packet.Players(self.index)
car_location = Vector2(my_car.Physics().Location().X(), my_car.Physics().Location().Y())
my_car = packet.game_cars[self.index]
car_location = Vector2(my_car.physics.location.x, my_car.physics.location.y)
car_direction = get_car_facing_vector(my_car)
car_to_ball = ball_location - car_location

Expand Down Expand Up @@ -64,8 +61,8 @@ def correction_to(self, ideal):


def get_car_facing_vector(car):
pitch = car.Physics().Rotation().Pitch()
yaw = car.Physics().Rotation().Yaw()
pitch = float(car.physics.rotation.pitch)
yaw = float(car.physics.rotation.yaw)

facing_x = math.cos(pitch) * math.cos(yaw)
facing_y = math.cos(pitch) * math.sin(yaw)
Expand Down