Madra_smieciarka/classes/Garbagetruck.py

import pygame
from classes.Household import *
from classes.Trashcan import *


class Garbagetruck:

    def __init__(self):
        self.capacity: int = 20
        self.trash: list = []
        self.trashweight: int = 0
        self.image = pygame.image.load("sprites/garbage_truck.png").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