diff --git a/garbageTruck.py b/garbageTruck.py new file mode 100644 index 0000000..2b0fad5 --- /dev/null +++ b/garbageTruck.py @@ -0,0 +1,34 @@ +from typing import List + +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) \ No newline at end of file