gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-02 06:16:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make window in play resizable (#190)

Browse Source
This commit is contained in:
Markus Krimmel
2022-12-09 15:54:34 +01:00
committed by GitHub
parent 3185dec7b3
commit a10bcd858e
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
gymnasium/utils/play.py
View File

@@ -15,7 +15,6 @@ try:
import pygame import pygame
from pygame import Surface from pygame import Surface
from pygame.event import Event from pygame.event import Event
from pygame.locals import VIDEORESIZE
except ImportError: except ImportError:
raise gym.error.DependencyNotInstalled( raise gym.error.DependencyNotInstalled(
"Pygame is not installed, run `pip install gymnasium[classic_control]`" "Pygame is not installed, run `pip install gymnasium[classic_control]`"
@@ -59,8 +58,10 @@ class PlayableGame:
self.env = env self.env = env
self.relevant_keys = self._get_relevant_keys(keys_to_action) self.relevant_keys = self._get_relevant_keys(keys_to_action)
# self.video_size is the size of the video that is being displayed.
# The window size may be larger, in that case we will add black bars
self.video_size = self._get_video_size(zoom) self.video_size = self._get_video_size(zoom)
self.screen = pygame.display.set_mode(self.video_size) self.screen = pygame.display.set_mode(self.video_size, pygame.RESIZABLE)
self.pressed_keys = [] self.pressed_keys = []
self.running = True self.running = True
@@ -113,9 +114,12 @@ class PlayableGame:
self.pressed_keys.remove(event.key) self.pressed_keys.remove(event.key)
elif event.type == pygame.QUIT: elif event.type == pygame.QUIT:
self.running = False self.running = False
elif event.type == VIDEORESIZE: elif event.type == pygame.WINDOWRESIZED:
self.video_size = event.size # Compute the maximum video size that fits into the new window
self.screen = pygame.display.set_mode(self.video_size) scale_width = event.x / self.video_size[0]
scale_height = event.y / self.video_size[1]
scale = min(scale_height, scale_width)
self.video_size = (scale * self.video_size[0], scale * self.video_size[1])
def display_arr( def display_arr(
@@ -133,7 +137,12 @@ def display_arr(
arr = 255.0 * (arr - arr_min) / (arr_max - arr_min) arr = 255.0 * (arr - arr_min) / (arr_max - arr_min)
pyg_img = pygame.surfarray.make_surface(arr.swapaxes(0, 1) if transpose else arr) pyg_img = pygame.surfarray.make_surface(arr.swapaxes(0, 1) if transpose else arr)
pyg_img = pygame.transform.scale(pyg_img, video_size) pyg_img = pygame.transform.scale(pyg_img, video_size)
screen.blit(pyg_img, (0, 0)) # We might have to add black bars if surface_size is larger than video_size
surface_size = screen.get_size()
width_offset = (surface_size[0] - video_size[0]) / 2
height_offset = (surface_size[1] - video_size[1]) / 2
screen.fill((0, 0, 0))
screen.blit(pyg_img, (width_offset, height_offset))
def play( def play(