36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from typing import List, Tuple
|
|
|
|
from garbage import RecognizedGarbage
|
|
|
|
class GarbageTruck:
|
|
position: Tuple[int, int]
|
|
paper: List[RecognizedGarbage]
|
|
plastic_and_metal: List[RecognizedGarbage]
|
|
glass: List[RecognizedGarbage]
|
|
bio: List[RecognizedGarbage]
|
|
mixed: List[RecognizedGarbage]
|
|
truckSpritePath = 'imgs/dust_car.png'
|
|
def __init__(self, position: List[int, int]) -> None:
|
|
self.position = position
|
|
self.paper = []
|
|
self.plastic_and_metal = []
|
|
self.glass = []
|
|
self.bio = []
|
|
self.mixed = []
|
|
|
|
|
|
def sort_garbage(self, RecognizedGarbage) -> None:
|
|
if RecognizedGarbage.garbage_type == 0:
|
|
self.paper.append(RecognizedGarbage)
|
|
|
|
elif RecognizedGarbage.garbage_type == 1:
|
|
self.plastic_and_metal.append(RecognizedGarbage)
|
|
|
|
elif RecognizedGarbage.garbage_type == 3:
|
|
self.glass.append(RecognizedGarbage)
|
|
|
|
elif RecognizedGarbage.garbage_type == 4:
|
|
self.bio.append(RecognizedGarbage)
|
|
|
|
elif RecognizedGarbage.garbage_type == 5:
|
|
self.mixed.append(RecognizedGarbage) |