Added blackjack toytext rendering support (#2550)

gym/envs/toy_text/blackjack.py

@@ -1,4 +1,8 @@
 from typing import Optional
+import os
+
+import numpy as np
+import pygame
 
 import gym
 from gym import spaces
@@ -75,6 +79,8 @@ class BlackjackEnv(gym.Env):
     http://incompleteideas.net/book/the-book-2nd.html
     """
 
+    metadata = {"render.modes": ["human", "rgb_array"]}
+
     def __init__(self, natural=False, sab=False):
         self.action_space = spaces.Discrete(2)
         self.observation_space = spaces.Tuple(
@@ -124,3 +130,107 @@ class BlackjackEnv(gym.Env):
         self.dealer = draw_hand(self.np_random)
         self.player = draw_hand(self.np_random)
         return self._get_obs()
+
+    def render(self, mode="human"):
+        player_sum, dealer_card_value, usable_ace = self._get_obs()
+        screen_width, screen_height = 600, 500
+        card_img_height = screen_height // 3
+        card_img_width = int(card_img_height * 142 / 197)
+        spacing = screen_height // 20
+
+        bg_color = (7, 99, 36)
+        white = (255, 255, 255)
+
+        if not hasattr(self, "screen"):
+            if mode == "human":
+                pygame.init()
+                self.screen = pygame.display.set_mode((screen_width, screen_height))
+            else:
+                pygame.font.init()
+                self.screen = pygame.Surface((screen_width, screen_height))
+
+        self.screen.fill(bg_color)
+
+        def get_image(path):
+            cwd = os.path.dirname(__file__)
+            image = pygame.image.load(cwd + "/" + path)
+            return image
+
+        def get_font(path, size):
+            cwd = os.path.dirname(__file__)
+            font = pygame.font.Font((cwd + "/" + path), size)
+            return font
+
+        small_font = get_font(
+            os.path.join("font", "Minecraft.ttf"), screen_height // 15
+        )
+        dealer_text = small_font.render(
+            "Dealer: " + str(dealer_card_value), True, white
+        )
+        dealer_text_rect = self.screen.blit(dealer_text, (spacing, spacing))
+
+        suits = ["C", "D", "H", "S"]
+        dealer_card_suit = self.np_random.choice(suits)
+
+        if dealer_card_value == 1:
+            dealer_card_value_str = "A"
+        elif dealer_card_value == 10:
+            dealer_card_value_str = self.np_random.choice(["J", "Q", "K"])
+        else:
+            dealer_card_value_str = str(dealer_card_value)
+
+        def scale_card_img(card_img):
+            return pygame.transform.scale(card_img, (card_img_width, card_img_height))
+
+        dealer_card_img = scale_card_img(
+            get_image(
+                os.path.join("img", dealer_card_suit + dealer_card_value_str + ".png")
+            )
+        )
+        dealer_card_rect = self.screen.blit(
+            dealer_card_img,
+            (
+                screen_width // 2 - card_img_width - spacing // 2,
+                dealer_text_rect.bottom + spacing,
+            ),
+        )
+
+        hidden_card_img = scale_card_img(get_image(os.path.join("img", "Card.png")))
+        self.screen.blit(
+            hidden_card_img,
+            (
+                screen_width // 2 + spacing // 2,
+                dealer_text_rect.bottom + spacing,
+            ),
+        )
+
+        player_text = small_font.render("Player", True, white)
+        player_text_rect = self.screen.blit(
+            player_text, (spacing, dealer_card_rect.bottom + 1.5 * spacing)
+        )
+
+        large_font = get_font(os.path.join("font", "Minecraft.ttf"), screen_height // 6)
+        player_sum_text = large_font.render(str(player_sum), True, white)
+        player_sum_text_rect = self.screen.blit(
+            player_sum_text,
+            (
+                screen_width // 2 - player_sum_text.get_width() // 2,
+                player_text_rect.bottom + spacing,
+            ),
+        )
+
+        if usable_ace:
+            usable_ace_text = small_font.render("usable ace", True, white)
+            self.screen.blit(
+                usable_ace_text,
+                (
+                    screen_width // 2 - usable_ace_text.get_width() // 2,
+                    player_sum_text_rect.bottom + spacing // 2,
+                ),
+            )
+        if mode == "human":
+            pygame.display.update()
+        else:
+            return np.transpose(
+                np.array(pygame.surfarray.pixels3d(self.screen)), axes=(1, 0, 2)
+            )

requirements.txt

@@ -5,5 +5,6 @@ mujoco_py>=1.50, <2.0
 scipy>=1.4.1
 numpy>=1.18.0
 pyglet>=1.4.0
+pygame==2.1.0
 cloudpickle>=1.2.0
 lz4>=3.1.0