Inteligentna_smieciarka/garbage_truck.py

FIELDWIDTH = 50

class GarbageTank:
    def __init__(self, volume_capacity, mass_capacity):
        self.vcapacity = volume_capacity #m^3
        self.mcapacity = mass_capacity #kg

class Engine:
    def __init__(self, power):
        self.power = power #HP

class GarbageTruck:
    def __init__(self, dump_location, fuel_capacity, rect, orientation):
        self.dump_location = dump_location
        self.tank = GarbageTank(15, 18000)
        self.engine = Engine(400)
        self.fuel = fuel_capacity
        self.rect = rect
        self.orientation = orientation
        self.houses = [] #lista domów do odwiedzenia

    def turn_left(self):
        self.orientation = (self.orientation - 1) % 4

    def turn_right(self):
        self.orientation = (self.orientation + 1) % 4

    def forward(self):
        if self.orientation == 0:
            self.rect.x += FIELDWIDTH
        elif self.orientation == 1:
            self.rect.y += FIELDWIDTH
        elif self.orientation == 2:
            self.rect.x -= FIELDWIDTH
        else:
            self.rect.y -= FIELDWIDTH