149 lines
4.0 KiB
Python
149 lines
4.0 KiB
Python
import pygame
|
|
from classes.Household import *
|
|
from classes.Trashcan import *
|
|
|
|
|
|
class Garbagetruck:
|
|
|
|
def __init__(self, file="sprites/garbage_truck.png"):
|
|
self.capacity: int = 20
|
|
self.trash: list = []
|
|
self.trashweight: int = 0
|
|
self.image = pygame.image.load(file).convert_alpha()
|
|
self.image = pygame.transform.scale(self.image, (32, 32))
|
|
self.position = [3, 3]
|
|
self.houses: list = []
|
|
self.trashcans: list = []
|
|
self.state = None
|
|
self.segregation = {"Papier": "paper",
|
|
"MetalPlastik": "metals_and_plastics",
|
|
"Mixed": "mixed",
|
|
"Bio": "bio_waste",
|
|
"Szklo": "glass"}
|
|
self.route = None
|
|
self.scanner = None
|
|
self.planner = None
|
|
self.driver = None
|
|
self.runningtime = 0
|
|
|
|
def getRunningtime(self):
|
|
return self.runningtime
|
|
|
|
def setRunningtime(self, runningtime):
|
|
self.runningtime = runningtime
|
|
return self
|
|
|
|
def incrRunningtime(self):
|
|
self.runningtime += 1
|
|
return self
|
|
|
|
def getTrashcans(self):
|
|
return self.trashcans
|
|
|
|
def setTrashcans(self, others):
|
|
self.trashcans = others
|
|
return self
|
|
|
|
def setHouses(self, houses):
|
|
self.houses = houses
|
|
return self
|
|
|
|
def getHouses(self):
|
|
return self.houses
|
|
|
|
def getCapacity(self) -> int:
|
|
return self.capacity
|
|
|
|
def getTrash(self) -> list:
|
|
return self.trash
|
|
|
|
def setTrash(self, trash: list) -> None:
|
|
self.trash = trash
|
|
for item in trash:
|
|
self.addTrashweight(item.getWaga)
|
|
|
|
def addTrash(self, trash: Trash) -> None:
|
|
self.trash.append(trash)
|
|
self.addTrashweight(trash.getWaga())
|
|
|
|
def getTrashweight(self) -> int:
|
|
return self.trashweight
|
|
|
|
def setTrashweight(self, weight: int) -> None:
|
|
self.trashweight = weight
|
|
|
|
def addTrashweight(self, weight: int) -> None:
|
|
self.trashweight += weight
|
|
|
|
def getImage(self) -> object:
|
|
return self.image
|
|
|
|
def setImage(self, image: object) -> None:
|
|
self.image = image
|
|
|
|
def getPosition(self) -> list:
|
|
return self.position
|
|
|
|
def setPosition(self, position: list) -> object:
|
|
self.position = position
|
|
return self
|
|
|
|
def modPosiotion(self, modX, modY):
|
|
x = self.getPosition()[0] + modX
|
|
y = self.getPosition()[1] + modY
|
|
self.setPosition([x, y])
|
|
|
|
def getRoute(self):
|
|
return self.route
|
|
|
|
def setRoute(self, route) -> None:
|
|
self.route = route
|
|
|
|
def getScanner(self):
|
|
return self.scanner
|
|
|
|
def setScanner(self, scanner) -> None:
|
|
self.route = scanner
|
|
|
|
def getPlanner(self):
|
|
return self.planner
|
|
|
|
def setPlanner(self, planner) -> None:
|
|
self.route = planner
|
|
|
|
def getDriver(self):
|
|
return self.driver
|
|
|
|
def setDriver(self, driver) -> None:
|
|
self.route = driver
|
|
|
|
def distance(self, object1, object2) -> int:
|
|
lok1 = object1.getPosition()
|
|
lok2 = object2.getPosition()
|
|
|
|
dist = abs(lok1[0] - lok2[0]) + abs(lok1[1] - lok2[1])
|
|
return dist
|
|
|
|
def scanTile(self):
|
|
self.state = None
|
|
temp = self.houses[:]
|
|
temp.append(self.trashcans[:])
|
|
for loc in temp:
|
|
if tuple(self.position) == loc.getPosition():
|
|
self.state = loc
|
|
return
|
|
|
|
def printme(self):
|
|
x, y = self.getPosition()
|
|
return 32*x, 32*y
|
|
|
|
def throwGarbage(self, trash):
|
|
if self.segregation[trash.getTtype()] == self.state.getTrashtype():
|
|
self.addTrashweight(trash.getWeight * (-1))
|
|
self.trash.remove(trash)
|
|
|
|
def classifyTrash(self):
|
|
pass
|
|
# Tutaj jest plan żeby dopiero napisać funkcję jak już będzie klasyfikator
|
|
# ogólnie to myślałem żeby po prostu zklasyfikować śmieć i zmienić mu trashtype na rozpoznany, żeby śmieciarka go tak posegreowała
|