mdp_visualizer.py 文件源码

def _vis_init(screen, mdp, draw_state, cur_state, agent=None, value=False):
    # Pygame setup.
    pygame.init()
    screen.fill((255, 255, 255))
    pygame.display.update()
    done = False

    # Draw name of MDP:
    _draw_title_text(mdp, screen)
    if agent is not None:
        _draw_agent_text(agent, screen)
    if not value:
        # If we're not visualizing the value.
        _draw_state_text(cur_state, screen)
        agent_shape = draw_state(screen, mdp, cur_state, draw_statics=True)

        return agent_shape