flatland.utils.graphics_pil module

class flatland.utils.graphics_pil.PILGL(width, height, jupyter=False, screen_width=800, screen_height=600)

Bases: flatland.utils.graphics_layer.GraphicsLayer

AGENT_LAYER = 3

PREDICTION_PATH_LAYER = 1

RAIL_LAYER = 0

SELECTED_AGENT_LAYER = 4

SELECTED_TARGET_LAYER = 5

TARGET_LAYER = 2

alpha_composite_layers()

begin_frame()

build_background_map(dTargets)

clear_layer(iLayer=0, opacity=None)

close_window()

create_image(opacity=255)

create_layer(iLayer=0, clear=True)

create_layers(clear=True)

draw_image_row_col(pil_img, rcTopLeft, layer=0)

draw_image_xy(pil_img, xyPixLeftTop, layer=0)

get_agent_color(iAgent)

get_image()

return a blended / alpha composited image composed of all the layers, with layer 0 at the “back”.

idle(seconds=1e-05)

process any display events eg redraw, resize. Return only after the given number of seconds, ie idle / loop until that number.

open_window()

pause(seconds=1e-05)

deprecated

plot(gX, gY, color=None, linewidth=3, layer=0, opacity=255, **kwargs)

Draw a line joining the points in gX, GY - each an

prettify(*args, **kwargs)

prettify2(width, height, cell_size)

rgb_s2i(sRGB)

convert a hex RGB string like 0091ea to 3-tuple of ints

save_image(filename)

Renders the current scene into a image file :param filename: filename where to store the rendering output_generator (supported image format *.bmp , .. , *.png)

scatter(gX, gY, color=None, marker='o', s=50, layer=0, opacity=255, *args, **kwargs)

show(block=False)

text(xPx, yPx, strText, layer=0)

text_rowcol(rcTopLeft, strText, layer=3)

class flatland.utils.graphics_pil.PILSVG(width, height, jupyter=False, screen_width=800, screen_height=600)

Bases: flatland.utils.graphics_pil.PILGL

Note : This class should now ideally be called as PILPNG, but for backward compatibility, and to not introduce any breaking changes at this point we are sticking to the legacy name of PILSVG (when in practice we are not using SVG anymore)

clear_agents()

clear_rails()

draw_malfunction(agent_idx, rcTopLeft)

load_agent()

load_buildings()

load_pngs(file_directory, rotate=False, agent_colors=False, background_image=None, whitefilter=None)

load_rail()

Load the rail SVG images, apply rotations, and store as PIL images.

load_scenery()

pil_from_png_file(package, resource)

process_events()

recolor_image(pil, a3BaseColor, ltColors, invert=False)

set_agent_at(agent_idx, row, col, in_direction, out_direction, is_selected, rail_grid=None, show_debug=False, clear_debug_text=True, malfunction=False)

set_cell_occupied(agent_idx, row, col)

set_predicion_path_at(row, col, binary_trans, agent_rail_color)

set_rail_at(row, col, binary_trans, target=None, is_selected=False, rail_grid=None, num_agents=None, show_debug=True)

Set the rail at cell (row, col) to have transitions binTrans. The target argument can contain the index of the agent to indicate that agent’s target is at that cell, so that a station can be rendered in the static rail layer.

flatland.utils.graphics_pil.main()

flatland.utils.graphics_pil.main2()