flatland.envs.graph.graph_rendering module

flatland.envs.graph.graph_rendering module#

flatland.envs.graph.graph_rendering.add_flatland_styling(env: RailEnv, ax: Axes)[source]#

Adds Flatland 3 background image, sets ticks and grid.

Parameters#

env: RailEnv

The underlying rail env

ax: plt.Axes

The ax to style

flatland.envs.graph.graph_rendering.get_positions(g: DiGraph, delta=0.2)[source]#

Get positions for Flatland 3 DiGraph rendering.

Parameters#

g: The DiGraph of the Graph Transition Map delta: offset to separate the two directions

Returns#

Dict[GridNode, Tuple[int,int]]

The position dict