102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
import time
|
|
from typing import Tuple
|
|
import pygame
|
|
|
|
from wyszukiwanie import Stan, SearchSpace
|
|
|
|
# from main import screen, miejsce
|
|
|
|
# screen nie działa
|
|
|
|
class Wozek():
|
|
def __init__(self, y, x):
|
|
# self.x = 55
|
|
# self.y = 55
|
|
self.x_change = 0
|
|
self.y_change = 0
|
|
self.height = 64
|
|
self.width = 64
|
|
self.image = pygame.image.load("images/pusty_wozek.png")
|
|
# self.screen = pygame.display.set_mode((980, 980))
|
|
# self.miejsce = pygame.image.load('images/miejsce_paczek.png')
|
|
# Credit: Forklift icons created by Smashicons - Flaticon
|
|
# https://www.flaticon.com/free-icons/forklift
|
|
#self.__zainicjuj_stan_poczatkowy()
|
|
self.state = Stan(y, x, 0)
|
|
|
|
def draw(self):
|
|
from main import screen
|
|
# screen.blit(self.image, (self.x, self.y))
|
|
#screen.blit(self.image, (self.obecnyStan.x, self.obecnyStan.y))
|
|
screen.blit(self.image, (self.state.x, self.state.y))
|
|
def draw_bfs(self):
|
|
from main import screen
|
|
# screen.blit(self.image, (self.x, self.y))
|
|
#screen.blit(self.image, (self.obecnyStan.x, self.obecnyStan.y))
|
|
screen.blit(self.image, (self.state.x*98, self.state.y*98))
|
|
|
|
# storage = ["none"] * 10
|
|
storage = []
|
|
max_size = 10
|
|
|
|
def add_element(element, storage, max_size):
|
|
if len(storage) < max_size:
|
|
storage.append(element)
|
|
else:
|
|
print("I'm full!")
|
|
|
|
def remove_element(storage):
|
|
if len(storage) > 0:
|
|
place = storage.pop()
|
|
return place
|
|
else:
|
|
print("I'm empty!")
|
|
|
|
def dynamic_wozek_picture(wozek, storage):
|
|
if len(storage) == 0:
|
|
wozek.image = pygame.image.load("images/pusty_wozek.png")
|
|
elif ((len(storage) > 0) and (len(storage) < 4)):
|
|
wozek.image = pygame.image.load("images/pelny_wozek_1_crate.png")
|
|
elif ((len(storage) > 3) and (len(storage) < 10)):
|
|
wozek.image = pygame.image.load("images/pelny_wozek_2_crates.png")
|
|
elif (len(storage) == 10):
|
|
wozek.image = pygame.image.load("images/pelny_wozek_full_3_crates.png")
|
|
|
|
# def __zainicjuj_stan_poczatkowy(self):
|
|
# from wyszukiwanie import Stan
|
|
# # self.obecnyStan = Stan(55, 55, 3)
|
|
# self.obecnyStan = Stan(1, 1, 3)
|
|
|
|
# def _drive(self, actions):
|
|
# # for action in actions.reverse():
|
|
# for action in actions:
|
|
# self._move(action)
|
|
# pygame.event.pump()
|
|
# time.sleep(0.01)
|
|
# # self.screen.fill((51,51,51)) # removes object trail
|
|
# # self.screen.blit(self.miejsce, (430, 400))
|
|
# self.draw()
|
|
# pygame.display.flip()
|
|
|
|
def _move(self, action):
|
|
x = self.state.x
|
|
y = self.state.y
|
|
kierunek = self.state.kierunek
|
|
|
|
if action == "Lewo":
|
|
self.state.kierunek = 3 if kierunek == 0 else kierunek - 1
|
|
|
|
elif action == "Prawo":
|
|
self.state.kierunek = (kierunek + 1) % 4
|
|
|
|
elif action == "Do przodu":
|
|
new_x = x + 1 if kierunek == 1 else (x - 1 if kierunek == 3 else x)
|
|
new_y = y - 1 if kierunek == 0 else (y + 1 if kierunek == 2 else y)
|
|
|
|
self.state.x = new_x
|
|
self.state.y = new_y
|
|
|
|
# def ustaw_wozek_w_kierunku(self, kierunek):
|
|
# TODO
|
|
|