drzewo_wiki #18

Merged
s473561 merged 5 commits from drzewo_wiki into master 2023-05-29 13:28:33 +02:00
8 changed files with 18 additions and 14 deletions
Showing only changes of commit 230b94064b - Show all commits

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@ -50,11 +50,10 @@ def zwroc_regaly_kategoria(kategoria):
lista_reg.append(reg) lista_reg.append(reg)
return lista_reg return lista_reg
def zwroc_regaly_wspolrzedne(x_reg, y_reg): # def zwroc_regaly_wspolrzedne(x_reg, y_reg):
for regal in lista_regalow: # for regal in lista_regalow:
if regal.wiersz == obliczPixeleNaPodstawieKratek(x_reg) and regal.kolumna == obliczPixeleNaPodstawieKratek(y_reg): # if regal.wiersz == obliczPixeleNaPodstawieKratek(x_reg) and regal.kolumna == obliczPixeleNaPodstawieKratek(y_reg):
return regal # return regal
def narysuj_siatke(): def narysuj_siatke():
blockSize = 70 # Set the size of the grid block blockSize = 70 # Set the size of the grid block

15
main.py
View File

@ -15,7 +15,7 @@ pygame.init()
def main(): def main():
wozek = Wozek() wozek = Wozek()
paczka = Paczka('duzy', 10, 'ogród', False, True, False, any, any, any, any, any) paczka = Paczka('duzy', 10, 'narzedzia', False, True, False, any, any, any, any, any)
ekran.lista_paczek.add(paczka) ekran.lista_paczek.add(paczka)
paczka.update_position(a_pix, b_pix) paczka.update_position(a_pix, b_pix)
@ -43,12 +43,11 @@ def main():
if wozek.ln == 0: if wozek.ln == 0:
wozek.load_package() wozek.load_package()
wozek.dynamic_wozek_picture wozek.dynamic_wozek_picture
print(wozek.storage)
## wozek jedzie odlozyc paczke na regal ## wozek jedzie odlozyc paczke na regal
paczka1 = wozek.storage.pop() paczka1 = wozek.storage[0]
array, reg = paczka1.tablica_do_drzewa('ogród') array, reg = paczka1.tablica_do_drzewa(paczka1.kategoria)
plikZPrzecinkami = open("DecisionTree/training_data.txt", 'w') plikZPrzecinkami = open("DecisionTree/training_data.txt", 'w')
@ -83,9 +82,11 @@ def main():
wozek.przemiesc_wozek_po_sciezce(sciezka) wozek.przemiesc_wozek_po_sciezce(sciezka)
# print(wozek.storage) # print(wozek.storage)
# if wozek.ln != 0: # drops package if wozek.ln != 0: # drops package
# wozek.drop_package() wozek.drop_package()
# wozek.dynamic_wozek_picture() reg.put_package_on_the_regal(paczka1, predictions)
wozek.dynamic_wozek_picture()
if event.type == pygame.MOUSEBUTTONDOWN: if event.type == pygame.MOUSEBUTTONDOWN:
# lewy przycisk myszy # lewy przycisk myszy

View File

@ -72,7 +72,6 @@ class Paczka(pygame.sprite.Sprite):
def tablica_do_drzewa(self, kategoria): def tablica_do_drzewa(self, kategoria):
tablica = [] tablica = []
# rozmiar # rozmiar
if self.rozmiar == 'maly': if self.rozmiar == 'maly':
tablica.append(0) tablica.append(0)
@ -103,7 +102,6 @@ class Paczka(pygame.sprite.Sprite):
else: tablica.append(1) else: tablica.append(1)
reg = ekran.zwroc_regaly_kategoria(kategoria) reg = ekran.zwroc_regaly_kategoria(kategoria)
# czy dolna wolna # czy dolna wolna
if reg[0].is_dolna_free() is True: if reg[0].is_dolna_free() is True:
tablica.append(0) tablica.append(0)

View File

@ -85,5 +85,11 @@ class Regal(pygame.sprite.Sprite):
return True return True
return False return False
def put_package_on_the_regal(self, package, where):
if(where == 0):
self.dolna.append(package)
else:
self.gorna.append(package)