From ba49049cc3851cad40c77403c81b19f125ac2288 Mon Sep 17 00:00:00 2001 From: Pawel Felcyn Date: Mon, 3 Apr 2023 15:43:03 +0200 Subject: [PATCH] add street class --- street.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 street.py diff --git a/street.py b/street.py new file mode 100644 index 0000000..931b189 --- /dev/null +++ b/street.py @@ -0,0 +1,25 @@ +from enum import Enum +from typing import Tuple +from gameContext import GameContext + +class StreetType (Enum): + VERTICAL = 0 + HORIZONTAL = 0 + +class Street: + street_type: StreetType + start_cell: int + end_cell: int + row_or_column: int + + def __init__(self, start_cell: int, end_cell: int, row_or_column: int, street_type: StreetType = StreetType.VERTICAL) -> None: + self.start_cell = start_cell + self.end_cell = end_cell + self.street_type = street_type + self.row_or_column = row_or_column + + def render(self, game_context: GameContext) -> None: + for i in range(self.start_cell, self.end_cell): + img_str: str = 'imgs/street_vertical.png' if self.street_type == StreetType.VERTICAL else 'imgs/street_horizontal.png' + cell: Tuple[int, int] = (i, self.row_or_column) if self.street_type == StreetType.VERTICAL else (self.row_or_column, i) + game_context.render_in_cell(cell, img_str)