Room Icons (#38)

* Add icons for the rooms

* Add pillow to the requirements

* Add helper function for getting the icon for a room type

* Draw the icons

* Add start room icon

* Add version for pillow

* Change pillow version to 11.0.0

Author: Joltras

requirements.txt

@@ -4,4 +4,5 @@ starlette~=0.41.2
 black~=24.10.0
 pylint~=3.3.1
 flake8~=7.1.1
-mypy~=1.8.0
+mypy~=1.8.0
+pillow~=11.0.0

resources/room_icons/boss_room.png

@@ -28,6 +28,7 @@ def __init__(
         super().__init__(x, y, room_id, room_type)
         self._width = width
         self._height = height
+        self._image = None
 
     def draw(self, canvas: tk.Canvas):
         """
@@ -40,6 +41,11 @@ def draw(self, canvas: tk.Canvas):
         color = room_type.room_colors[self._type].value
         hex_color = util_functions.rgb2hex(color[0], color[1], color[2])
         canvas.create_rectangle((x0, y0), (x1, y1), fill=hex_color)
+        center_x = x0 + self._width / 2
+        center_y = y0 + self._height / 2
+        self._image = util_functions.get_picture_for_room_type(self._type)
+        if self._image is not None:
+            canvas.create_image(center_x, center_y, image=self._image, anchor="center")
 
     @classmethod
     def from_room(cls, room: Room):

resources/room_icons/item_room.png

@@ -1,12 +1,39 @@
 """
 This module contains utility functions for the floor generator.
 """
+import os
 import random
 import math
 from typing import Tuple
 
 import utils.globals as my_globals
 from utils.direction import Direction
+from utils.room_type import RoomType
+from PIL import Image, ImageTk
+
+
+def get_picture_for_room_type(room_type: RoomType) -> ImageTk:
+    """
+    Gets the picture for the given room type.
+    @param room_type: room type to get the picture for
+    @return: picture for the given room type
+    """
+    # Get the directory of the current file
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    # Go up two levels to the project root directory
+    project_root = os.path.dirname(os.path.dirname(current_dir))
+    # Define the relative path to the image file
+    image_file = f"{room_type.name.lower()}.png"
+    # Join the project root directory with the relative path to get the absolute path
+    image_path = os.path.join(project_root, "resources", "room_icons", image_file)
+    try:
+        return ImageTk.PhotoImage(
+            Image.open(
+                image_path
+            ).resize((32, 32))
+        )
+    except FileNotFoundError:
+        return None
 
 
 def calculate_room_amount(stage_id: int):
@@ -36,7 +63,7 @@ def rgb2hex(r, g, b):
 
 
 def add_direction_to_coordinates(
-    direction: Direction, coordinates: Tuple[int, int]
+        direction: Direction, coordinates: Tuple[int, int]
 ) -> Tuple[int, int]:
     """
     Calculates the room according to the given room and direction
@@ -81,5 +108,5 @@ def calculate_distance(cord1, cord2) -> int:
     Calculates the distance between two points.
     """
     return (cord1[0] - cord2[0]) * (cord1[0] - cord2[0]) + (cord1[1] - cord2[1]) * (
-        cord1[1] - cord2[1]
+            cord1[1] - cord2[1]
     )