sztuczna_inteligencja_2023_smieciarka/startup.py

from gameContext import GameContext
from city import City
from PIL import Image
import pygame
from typing import Tuple, List
from street import Street, StreetType
from garbageTruck import GarbageTruck

def startup(game_context: GameContext):
    render_background(game_context)
    game_context.city = create_city()
    game_context.city.render_city(game_context)
    car = create_dust_car(game_context)
    car.render(game_context)
    game_context.dust_car = car

def create_dust_car(game_context: GameContext) -> GarbageTruck:
    return GarbageTruck((3, 3))

def render_background(game_context: GameContext):
    bg_img = Image.open("imgs/background.jpg")
    pygame_bg_image = pygame.image.frombuffer(bg_img.tobytes(), bg_img.size, 'RGB')
    game_context.canvas.blit(pygame_bg_image, (0, 0))

def create_city() -> City:
    city: City = City()
    streets = create_streets()
    for s in streets:
        city.add_street(s)
    return city

def create_streets() -> List[Street]:
    streets = []
    streets.append(Street(3, 30, 3, StreetType.HORIZONTAL))
    streets.append(Street(4, 15, 10, StreetType.VERTICAL))
    streets.append(Street(4, 10, 25, StreetType.VERTICAL))
    streets.append(Street(11, 24, 10, StreetType.HORIZONTAL))
    streets.append(Street(4, 9, 18, StreetType.VERTICAL))
    streets.append(Street(1, 30, 16, StreetType.HORIZONTAL))
    streets.append(Street(17, 30, 2, StreetType.VERTICAL))
    streets.append(Street(3, 25, 23, StreetType.HORIZONTAL))
    streets.append(Street(17, 30, 13, StreetType.VERTICAL))
    streets.append(Street(17, 23, 25, StreetType.VERTICAL))
    return streets