sztuczna_inteligencja_2023_.../garbageTruck.py

33 lines
1011 B
Python

from typing import List
class garbageTruck:
position: List[int, int]
paper: []
plastic_and_metal: []
glass: []
bio: []
mixed: []
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)