Method render

1-grid-world/env.py:212–238

(self)

        return s

    def render(self):
        if self.render_mode is None: return
        if self._screen is None:
            self._screen = _open(self.title, (WIDTH * DYN_UNIT, HEIGHT * DYN_UNIT + self.HUD))
            self._hud_font = pygame.font.SysFont(None, 22)
            self._popup_font = pygame.font.SysFont(None, 28)
        _pump_events()
        s, hud = self._screen, self.HUD
        s.fill(WHITE)
        pygame.draw.rect(s, (30, 30, 30), pygame.Rect(0, 0, WIDTH * DYN_UNIT, hud))
        last = f"{self.last_score:+.1f}" if self.last_score is not None else "—"
        t = self._hud_font.render(
            f"Episode: {self.episode}   Score: {self.score:+.1f}   Last Score: {last}",
            True, (240, 240, 240))
        s.blit(t, (8, (hud - t.get_height()) // 2))
        _grid_lines(s, DYN_UNIT, y_off=hud)
        _circle(s, *self.goal, unit=DYN_UNIT, color=GOAL_COLOR, y_off=hud)
        for o in self.obstacles:
            _triangle(s, *o["state"], unit=DYN_UNIT, color=OBSTACLE_COLOR, y_off=hud)
        rect = _square(s, *self.agent, unit=DYN_UNIT,
                       color=OBSTACLE_COLOR if self._hit > 0 else AGENT_COLOR, y_off=hud)
        if self._hit > 0:
            p = self._popup_font.render("-1", True, OBSTACLE_COLOR)
            s.blit(p, p.get_rect(center=(rect.centerx, rect.top - 14)))
            self._hit -= 1
        pygame.display.flip()
        time.sleep(FPS_DELAY)


# ---------------------------------------------------------------------------