rktl_autonomy.rocket_league_interface

This the main environment intended for use. It interfaces the agent with the mess of other nodes that exist in this repository. Look at higher level documentation to learn more about the inputs and outputs from this node and how they are consumed or created by others.

License:

BSD 3-Clause License Copyright (c) 2023, Autonomous Robotics Club of Purdue (Purdue ARC) All rights reserved.

Module Contents

Classes

CarActions

Possible actions for car.

RocketLeagueInterface

ROS interface for the Rocket League.

class rktl_autonomy.rocket_league_interface.CarActions

Bases: enum.IntEnum

Possible actions for car.

STOP = 0
FWD_LEFT
FWD_RIGHT
FWD
REV_LEFT
REV_RIGHT
REV
SIZE
class rktl_autonomy.rocket_league_interface.RocketLeagueInterface(eval=False, launch_file=('rktl_autonomy', 'rocket_league_train.launch'), launch_args=[], run_id=None, env_number=0)

Bases: rktl_autonomy.ROSInterface

ROS interface for the Rocket League.

property action_space

The Space object corresponding to valid actions.

property observation_space

The Space object corresponding to valid observations.

_reset_env()

Reset environment for a new training episode.

_reset_self()

Reset internally for a new episode.

_has_state()

Determine if the new state is ready.

_clear_state()

Clear state variables / flags in preparation for new ones.

_get_state()

Get state tuple (observation, reward, done, info).

_publish_action(action)

Publish an action to the ROS network.

_car_odom_cb(odom_msg)

Callback for odometry of car.

_ball_odom_cb(odom_msg)

Callback for odometry of ball.

_score_cb(score_msg)

Callback for score of game.