pomdp_py.problems.tag.env package

Submodules

pomdp_py.problems.tag.env.env module

class pomdp_py.problems.tag.env.env.TagEnvironment(init_state, grid_map, pr_stay=0.2, small=1, big=10)

Bases: Environment

property width

property length

property grid_map

classmethod from_str(worldstr, **kwargs)

pomdp_py.problems.tag.env.visual module

Largely based on MosViz, except this is not an OO-POMDP

class pomdp_py.problems.tag.env.visual.TagViz(env, res=30, fps=30, controllable=False, observation_model=None)

Bases: object

property img_width

property img_height

property last_observation

update(action, observation, belief)

Update the visualization after there is new real action and observation and updated belief.

static draw_robot(img, x, y, th, size, color=(255, 12, 12))

static draw_observation(img, z, rx, ry, rth, r, size, color=(12, 12, 255))

static draw_belief(img, belief, r, size, target_color)

belief (OOBelief)

on_init()

pygame init

on_event(event)

on_loop()

on_render()

on_cleanup()

on_execute()

render_env(display_surf)

pomdp_py.problems.tag.env.visual.unittest()

Module contents