Method mainloop

1-grid-world/env.py:318–337

(self)

                              max(0, min(HEIGHT - 1, self.agent_pos[1] + dy))]

    def mainloop(self):
        self._screen = _open(self.title, (WIDTH * UNIT, HEIGHT * UNIT + self.BAR))
        self._font = pygame.font.SysFont(None, 22)
        self._small = pygame.font.SysFont(None, 16)
        while True:
            for e in pygame.event.get():
                if e.type == pygame.QUIT:
                    pygame.quit(); return
                if e.type == pygame.MOUSEBUTTONDOWN and e.button == 1:
                    for rect, btn in zip(self._btn_rects(), self.buttons):
                        if rect.collidepoint(e.pos) and self._btn_enabled(btn):
                            label = btn[0]
                            self._press_label = label
                            self._press_frames = 6
                            self._render()  # draw the pressed state immediately
                            btn[1]()
                            self.clicks[label] = self.clicks.get(label, 0) + 1
                            break
            self._render()
            time.sleep(0.03)

    @staticmethod
    def _btn_enabled(btn):