96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
import pygame
|
|
|
|
MAX_STORAGE = 3
|
|
|
|
def obliczPixeleNaPodstawieKratek(wymiar): #Przeliczanie współrzędnych podanych w kratkach na pixele
|
|
i = 1
|
|
# pixele = 100
|
|
pixele = 73
|
|
while (i < wymiar):
|
|
pixele = pixele + 70
|
|
i = i + 1
|
|
return pixele
|
|
|
|
def obliczPixeleDlugosciRegalu(self): #Przeliczanie dlugości regału podanego w kratkach na pixele
|
|
i = 1
|
|
dlugoscRegalu = 40
|
|
while (i < 1) and (i <= 11 - self.numerKolumny): #Sprawdzenie, żeby regał nie wychodził poza plansze, jeżeli tak to jest ucinany tak, żeby nie wychodził
|
|
dlugoscRegalu = dlugoscRegalu + 80
|
|
i = i + 1
|
|
return dlugoscRegalu
|
|
|
|
class Regal(pygame.sprite.Sprite):
|
|
|
|
def __init__(self, nazwaRegalu, numerWiersza, numerKolumny):
|
|
super().__init__()
|
|
from ekran import screen
|
|
self.nazwaRegalu = nazwaRegalu
|
|
self.wysokoscRegalu = 64
|
|
self.numerKolumny = numerKolumny
|
|
self.numerWiersza = numerWiersza
|
|
|
|
self.wiersz = obliczPixeleNaPodstawieKratek(numerWiersza)
|
|
self.kolumna = obliczPixeleNaPodstawieKratek(numerKolumny)
|
|
self.dlugosc = obliczPixeleDlugosciRegalu(self)
|
|
|
|
package_storage = []
|
|
shelfs = { # here packages are stored
|
|
"dolna": package_storage,
|
|
"gorna": package_storage
|
|
}
|
|
|
|
|
|
storage_dolna = []
|
|
storage_gorna = []
|
|
|
|
self.dolna = storage_dolna
|
|
self.gorna = storage_gorna
|
|
|
|
if(self.nazwaRegalu == 'ogród'):
|
|
reg = pygame.Surface([self.dlugosc, self.wysokoscRegalu])
|
|
reg = pygame.image.load("images/regal.png")
|
|
self.rect = reg.get_rect()
|
|
screen.blit(reg, (self.wiersz, self.kolumna))
|
|
|
|
if(self.nazwaRegalu == 'narzedzia'):
|
|
reg = pygame.Surface([self.dlugosc, self.wysokoscRegalu])
|
|
reg = pygame.image.load("images/regal1.png")
|
|
self.rect = reg.get_rect()
|
|
screen.blit(reg, (self.wiersz, self.kolumna))
|
|
|
|
if(self.nazwaRegalu == 'kuchnia'):
|
|
reg = pygame.Surface([self.dlugosc, self.wysokoscRegalu])
|
|
reg = pygame.image.load("images/regal2.png")
|
|
self.rect = reg.get_rect()
|
|
screen.blit(reg, (self.wiersz, self.kolumna))
|
|
|
|
if(self.nazwaRegalu == 'motoryzacja'):
|
|
reg = pygame.Surface([self.dlugosc, self.wysokoscRegalu])
|
|
reg = pygame.image.load("images/regal3.png")
|
|
self.rect = reg.get_rect()
|
|
screen.blit(reg, (self.wiersz, self.kolumna))
|
|
|
|
def is_dolna_free(self):
|
|
if len(self.dolna) <= MAX_STORAGE:
|
|
return True
|
|
return False
|
|
|
|
def is_gorna_free(self):
|
|
if len(self.gorna) <= MAX_STORAGE:
|
|
return True
|
|
return False
|
|
|
|
def czy_na_gornej_wiecej_miejsca(self):
|
|
if len(self.gorna) > len(self.dolna):
|
|
return True
|
|
return False
|
|
|
|
def put_package_on_the_regal(self, package, where):
|
|
if(where == 0):
|
|
self.dolna.append(package)
|
|
else:
|
|
self.gorna.append(package)
|
|
|
|
|
|
|