sztuczna_inteligencja_2023_smieciarka/city.py

from typing import List
from garbageCan import GarbageCan
from speedBump import SpeedBump
from street import Street
from gameContext import GameContext

class Node:
    garbageCan: GarbageCan
    id: int

    def __init__(self, id: int, can: GarbageCan) -> None:
        self.id 
        self.can = can
        
class Bump:
    speedBump: SpeedBump
    id: int
    
    def __init__(self, id: int, bump: SpeedBump) -> None:
        self.id 
        self.bump = bump
        
        
class City:
    nodes: List[GarbageCan]
    bumps: List[SpeedBump]
    streets: List[Street]

    def __init__(self) -> None:
        self.nodes = []
        self.streets = []
        self.bumps = []

    def add_node(self, node: GarbageCan) -> None:
        self.nodes.append(node)
    
    def add_street(self, street: Street) -> None:
        self.streets.append(street)
        
    def add_bump(self, bump: SpeedBump) -> None:
        self.streets.append(bump)

    def render_city(self, game_context: GameContext) -> None:
        self._render_streets(game_context)
        self._render_nodes(game_context)
        self._render_bumps(game_context)

    def _render_streets(self, game_context: GameContext) -> None:
        for street in self.streets:
            street.render(game_context)
            
    def _render_nodes(self, game_context: GameContext) -> None:
        for node in self.nodes:
            node.render(game_context)
            
    def _render_bumps(self, game_context: GameContext) -> None:
        for bump in self.bumps:
            bump.render(game_context)