AI/wheel.py

import pygame
from floor import Floor
from boxOnTheFloor import BoxOnTheFloor
from unboxOnTheFloor import UnboxOnTheFloor
from shelf import Shelf

class Wheel:
    def __init__(self, screen, cell):
        self.cell = cell
        self.ns = 1
        self.we = 2
        self.direction = 3
        self.m_ns = 0
        self.m_we = 0
        self.occupied = False
        self.box = False
        # 1 - w gore
        # 2 - w prawo
        # 3 - w dol
        # 4 - w lewo
        self.screen = screen
        self.imageorg = pygame.image.load(r'images/wo.png')
        self.imageorg = pygame.transform.scale(self.imageorg, (cell, cell))
        self.image = pygame.transform.scale(self.imageorg, (cell, cell))
    def draw(self):
        self.screen.blit(self.image, ((self.cell*self.we)+self.m_we, (self.cell*self.ns)+self.m_ns))
        self.m_ns = self.m_ns*(0.9)
        self.m_we = self.m_we*(0.9)
    def isOccupied(self):
        return self.occupied
    def putBox(self, boxik):
        self.box = boxik
        self.occupied = True
        self.imageorg = pygame.image.load(r'images/wop.png')
        self.imageorg = pygame.transform.scale(self.imageorg, (self.cell, self.cell))
        self.image = pygame.transform.scale(self.imageorg, (self.cell, self.cell))
    def getBox(self):
        self.occupied = False
        self.imageorg = pygame.image.load(r'images/wo.png')
        self.imageorg = pygame.transform.scale(self.imageorg, (self.cell, self.cell))
        self.image = pygame.transform.scale(self.imageorg, (self.cell, self.cell))
        return self.box
    def move(self, move, krata):
        if(move.key==pygame.K_DOWN):
            if(type(krata[self.ns][self.we])==UnboxOnTheFloor):
                pass
            elif(type(krata[self.ns+1][self.we])==Floor):
                self.ns+=1
                self.m_ns=self.cell*(-1)
                self.direction = 3
            elif(type(krata[self.ns+1][self.we])==Shelf):
                if(self.occupied==True and krata[self.ns+1][self.we].isOccupied()==False and krata[self.ns+1][self.we].kierunek==1):
                    krata[self.ns+1][self.we].put(self.getBox())
                    self.direction = 3
                elif(self.occupied==False and krata[self.ns+1][self.we].isOccupied()==True and krata[self.ns+1][self.we].kierunek==1):
                    self.putBox(krata[self.ns+1][self.we].get())
                    self.direction = 3
            elif(type(krata[self.ns+1][self.we])==UnboxOnTheFloor):
                if(self.ns+1==len(krata)-1):
                    self.ns+=1
                    self.m_ns=self.cell*(-1)
                    self.direction = 1
                    self.getBox()
        elif(move.key==pygame.K_UP):
            if(type(krata[self.ns-1][self.we])==Floor):
                self.ns-=1
                self.m_ns=self.cell
                self.direction = 1
            elif(type(krata[self.ns-1][self.we])==BoxOnTheFloor):
                if(self.occupied==False):
                    self.putBox(krata[self.ns - 1][self.we].get())
                    krata[self.ns - 1][self.we] = Floor(self.screen, self.cell, self.ns - 1, self.we)
                    self.ns -= 1
                    self.m_ns = self.cell
                    self.direction = 1
            elif(type(krata[self.ns-1][self.we])==Shelf):
                if(self.occupied==True and krata[self.ns-1][self.we].isOccupied()==False and krata[self.ns-1][self.we].kierunek==0):
                    krata[self.ns-1][self.we].put(self.getBox())
                    self.direction = 1
                elif(self.occupied==False and krata[self.ns-1][self.we].isOccupied()==True and krata[self.ns-1][self.we].kierunek==0):
                    self.putBox(krata[self.ns-1][self.we].get())
                    self.direction = 1
        elif(move.key==pygame.K_LEFT):
            if(type(krata[self.ns][self.we-1])==Floor):
                self.we-=1
                self.m_we=self.cell
                self.direction = 4
            elif(type(krata[self.ns][self.we-1])==Shelf):
                if(self.occupied==True and krata[self.ns][self.we-1].isOccupied()==False and krata[self.ns][self.we-1].kierunek==3):
                    krata[self.ns][self.we-1].put(self.getBox())
                    self.direction = 4
                elif(self.occupied==False and krata[self.ns][self.we-1].isOccupied()==True and krata[self.ns][self.we-1].kierunek==3):
                    self.putBox(krata[self.ns][self.we-1].get())
                    self.direction = 4
        elif(move.key==pygame.K_RIGHT):
            if(type(krata[self.ns][self.we+1])==Floor):
                self.we+=1
                self.m_we=self.cell*(-1)
                self.direction = 2
            elif(type(krata[self.ns][self.we+1])==Shelf):
                if(self.occupied==True and krata[self.ns][self.we+1].isOccupied()==False and krata[self.ns][self.we+1].kierunek==2):
                    krata[self.ns][self.we+1].put(self.getBox())
                    self.direction = 2
                elif(self.occupied==False and krata[self.ns][self.we+1].isOccupied()==True and krata[self.ns][self.we+1].kierunek==2):
                    self.putBox(krata[self.ns][self.we+1].get())
                    self.direction = 2
        if (self.direction==1):
            self.image = pygame.transform.rotate(self.imageorg, 180)
        elif (self.direction==2):
            self.image = pygame.transform.rotate(self.imageorg, 90)
        elif (self.direction==3):
            self.image = pygame.transform.rotate(self.imageorg, 0)
        elif (self.direction==4):
            self.image = pygame.transform.rotate(self.imageorg, 270)