pong_env.py 文件源码

def main():
    pygame.init()
    pygame.display.set_caption('Pong')
    pygame.mouse.set_visible(0)  # make cursor invisible
    surface = pygame.display.set_mode((400, 300))
    fps_clock = pygame.time.Clock()

    game = PongGame(window_size=(400, 300), speed=4)

    while True:  #main game loop
        for event in pygame.event.get():
            if event.type == QUIT:
                pygame.quit()
                sys.exit()
            # mouse movement commands
            #elif event.type == MOUSEMOTION:
            #game.bats['user'].move(event.pos)

        _, done = game.step(1, -1)
        if done:
            game.reset()
        game.draw(surface)
        pygame.display.update()
        fps_clock.tick(120)