JSBGym provides reinforcement learning environments for the control of fixed-wing aircraft using the JSBSim flight dynamics model. The package's environments implement the Farama-Foundation's Gymnasium interface allowing environments to be created and interacted with.
Open a terminal and install jsbgym via pip:
pip install jsbgym
To render the environment with FlightGear, download and install it from here. Make sure the FlightGear bin directory is in PATH (Usually C:\Program Files\FlightGear 2020.3\bin
) and if not already existant, add a system variable called FG_ROOT
with the FG data folder as it's value (Usually C:\Program Files\FlightGear 2020.3\data
).
If there are aircraft installed in a different location, add the folder to the FG_AIRCRAFT
system variable.
3D visualisation requires installation of the FlightGear simulator. Confirm FlightGear is runnable from terminal with:
fgfs --version
Open a terminal and install jsbgym via pip:
pip install jsbgym
Rendering with some modes will require additional packages:
sudo apt-get install python3-tk
To render the environment with FlightGear, download the AppImage from here.
and rename the AppImage file to fgfs
and place it in /usr/local/bin
:
sudo mv fgfs /usr/local/bin
The data files must also be downloaded from here (approximately 2 GB) then in terminal, enter
export FG_ROOT=/path/to/datafolder
If there are aircraft installed in a different location, add the folder to the FG_AIRCRAFT
system variable.
3D visualisation requires installation of the FlightGear simulator. Confirm FlightGear is runnable from terminal with:
fgfs --version
import jsbgym
import gymnasium as gym
env = gym.make(ENV_ID)
env.reset()
observation, reward, terminated, truncated, info = env.step(action)
Environment ID strings are constructed as follows:
f"{aircraft}-{task}-{shaping}-{flightgear}-v0"
The environment can be configured to use one of 14 aircraft:
- C172 Cessna 172P Skyhawk (Default FlightGear Aircraft)
- PA28 Piper PA-28-161 Warrior II
- J3 Piper J-3 Cub
- F15 McDonnell Douglas F-15C Eagle (F-15C in FlightGear)
- F16 General Dynamics F-16CJ Block 52
- OV10 North American OV-10A USAFE Bronco
- PC7 Pilatus PC-7
- A320 Airbus A320 (A320 Familiy in Flightgear)
- B747 Boeing 747-400
- MD11 McDonnell Douglas MD-11
- DHC6 de Havilland Canada DHC-6-300 Twin Otter
- C130 Lockheed C-130 Hercules
- WF Wright Flyer II 1903
- SS Royal Naval Air Service Submarine Scout Zero Airship
All aircraft except the Cessna 172P requires the aircraft to be downloaded via the launcher using the default FlightGear Hangar if using flightgear.
JSBGym implements two tasks for controlling the altitude and heading of aircraft:
- HeadingControlTask aircraft must fly in a straight line, maintaining its initial altitude and direction of travel (heading)
- TurnHeadingControlTask aircraft must turn to face a random target heading while maintaining their initial altitude
The environment can use three different shaping types:
- Shaping.STANDARD
- Shaping.EXTRA
- Shaping.EXTRA_SEQUENTIAL
If using FlightGear as a render mode, use FG
, if not, use NoFG
To fly a Cessna on the Heading Control task withoug using FlightGear,
env = gym.make("C172-HeadingControlTask-Shaping.STANDARD-NoFG-v0")
A basic plot of agent actions and current state information can be using human
render mode by calling env.render()
after specifying the render mode in gym.make()
.
env = gym.make("C172-HeadingControlTask-Shaping.STANDARD-NoFG-v0", render_mode="human")
env.reset()
env.render()
Visualising with FlightGear requires the Gymnasium environment to be created with a FlightGear-enabled environment ID by specifying the render_mode in gym.make()
and changing the value after {shaping}
to FG
. Using this render mode while training is strongly discouraged due to an error occuring midway through the training (Could not connect to socket for output!
).
env = gym.make("C172-HeadingControlTask-Shaping.STANDARD-FG-v0", render_mode="flightgear")
env.reset()
env.render()
JSBGym's environments have a continuous state and action space. The state is a 11-tuple:
(name='position/h-sl-ft', description='altitude above mean sea level [ft]', min=-1400, max=85000)
(name='attitude/pitch-rad', description='pitch [rad]', min=-1.5707963267948966, max=1.5707963267948966)
(name='attitude/roll-rad', description='roll [rad]', min=-3.141592653589793, max=3.141592653589793)
(name='velocities/u-fps', description='body frame x-axis velocity [ft/s]', min=-2200, max=2200)
(name='velocities/v-fps', description='body frame y-axis velocity [ft/s]', min=-2200, max=2200)
(name='velocities/w-fps', description='body frame z-axis velocity [ft/s]', min=-2200, max=2200)
(name='velocities/p-rad_sec', description='roll rate [rad/s]', min=-6.283185307179586, max=6.283185307179586)
(name='velocities/q-rad_sec', description='pitch rate [rad/s]', min=-6.283185307179586, max=6.283185307179586)
(name='velocities/r-rad_sec', description='yaw rate [rad/s]', min=-6.283185307179586, max=6.283185307179586)
(name='error/altitude-error-ft', description='error to desired altitude [ft]', min=-1400, max=85000)
(name='error/track-error-deg', description='error to desired track [deg]', min=-180, max=180)
Actions are 3-tuples of floats in the range [-1,+1] describing commands to move the aircraft's control surfaces (ailerons, elevator, rudder):
(name='fcs/aileron-cmd-norm', description='aileron commanded position, normalised', min=-1.0, max=1.0)
(name='fcs/elevator-cmd-norm', description='elevator commanded position, normalised', min=-1.0, max=1.0)
(name='fcs/rudder-cmd-norm', description='rudder commanded position, normalised', min=-1.0, max=1.0)