add render method in city class
This commit is contained in:
parent
ba49049cc3
commit
b8280a3854
32
city.py
32
city.py
@ -1,17 +1,19 @@
|
|||||||
from typing import List, Tuple
|
from typing import List
|
||||||
from garbageCan import GarbageCan
|
from garbageCan import GarbageCan
|
||||||
|
from street import Street
|
||||||
|
from gameContext import GameContext
|
||||||
|
|
||||||
class Node:
|
class Node:
|
||||||
garbageCan: GarbageCan
|
garbageCan: GarbageCan
|
||||||
id: int
|
id: int
|
||||||
|
|
||||||
def __init__(self, id: int, can: GarbageCan) -> None:
|
def __init__(self, id: int, can: GarbageCan) -> None:
|
||||||
self.id = id
|
self.id
|
||||||
self.can = can
|
self.can = can
|
||||||
|
|
||||||
class City:
|
class City:
|
||||||
nodes: List[Node]
|
nodes: List[Node]
|
||||||
streets: List[Tuple[int, int]]
|
streets: List[Street]
|
||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
self.nodes = []
|
self.nodes = []
|
||||||
@ -20,20 +22,12 @@ class City:
|
|||||||
def add_node(self, node: Node) -> None:
|
def add_node(self, node: Node) -> None:
|
||||||
self.nodes.append(node)
|
self.nodes.append(node)
|
||||||
|
|
||||||
def add_street(self, street: Tuple[int, int]) -> None:
|
def add_street(self, street: Street) -> None:
|
||||||
firstFound: bool = False
|
|
||||||
secondFound: bool = False
|
|
||||||
|
|
||||||
for node in self.nodes:
|
|
||||||
if firstFound and secondFound:
|
|
||||||
break
|
|
||||||
if node.id == street.__getitem__(0):
|
|
||||||
firstFound = True
|
|
||||||
continue
|
|
||||||
if node.id == street.__getitem__(1):
|
|
||||||
secondFound = True
|
|
||||||
|
|
||||||
if not firstFound or not secondFound:
|
|
||||||
return
|
|
||||||
|
|
||||||
self.streets.append(street)
|
self.streets.append(street)
|
||||||
|
|
||||||
|
def render_city(self, game_context: GameContext) -> None:
|
||||||
|
self._render_streets(game_context)
|
||||||
|
|
||||||
|
def _render_streets(self, game_context: GameContext) -> None:
|
||||||
|
for street in self.streets:
|
||||||
|
street.render(game_context)
|
Loading…
Reference in New Issue
Block a user