s444383/AI-2020
3
3
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed

Browse Source
This commit is contained in:
Serhii Hromov 2020-06-28 12:58:19 +00:00
parent d9f60b27fc
commit b1f370a2a3
1 changed files with 22 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
main.py
View File

@ -62,6 +62,7 @@ for i in range(len(X)):
menu.append([X[i], y[i]]) menu.append([X[i], y[i]])
random.shuffle(menu) random.shuffle(menu)
'''
def image_recognition(): def image_recognition():
LOSSE += 1 LOSSE += 1
for _ in range(100): for _ in range(100):
@ -74,7 +75,7 @@ def image_recognition():
break break
waiter.order_list.pop() waiter.order_list.pop()
#print(WIN, LOSSE - WIN) #print(WIN, LOSSE - WIN)
'''
# ai settings # ai settings
#S_IDLE = ("kitchen", "middle", "inplace") #S_IDLE = ("kitchen", "middle", "inplace")
#S_FIRST = ("order", "food") #S_FIRST = ("order", "food")
@ -105,7 +106,6 @@ wall = pygame.image.load('wall.png')
EAT_TIME = 15 EAT_TIME = 15
tree = build_tree(training_data) tree = build_tree(training_data)
#order_len = len(tree_format)
#print_tree(tree) #print_tree(tree)
@ -135,6 +135,7 @@ class Client:
return (self.gender + " Age: " + str(self.age) +" "+ self.outfit+ return (self.gender + " Age: " + str(self.age) +" "+ self.outfit+
" $"+ str(self.cash)+ " Vege: "+ self.vege) " $"+ str(self.cash)+ " Vege: "+ self.vege)
'''
def order_drink(clt): def order_drink(clt):
frame = [] frame = []
if clt.gender == "Man": if clt.gender == "Man":
@ -166,7 +167,8 @@ def order_drink(clt):
drink_index = drink_predict[0] drink_index = drink_predict[0]
return objects[-1][drink_index] return objects[-1][drink_index]
'''
'''
def order_food(clt): def order_food(clt):
frame = [] frame = []
if clt.gender == "Man": if clt.gender == "Man":
@ -198,6 +200,8 @@ def order_food(clt):
food_index = food_predict[0] food_index = food_predict[0]
return objects[-2][food_index] return objects[-2][food_index]
'''
### ###
class Node: class Node:
@ -453,7 +457,7 @@ class Agent:
#jesli sie zatrzymal w kuchni z zamowieniem to oddaje zamowienie #jesli sie zatrzymal w kuchni z zamowieniem to oddaje zamowienie
if waiter.x == 1 and waiter.y == 1 and waiter.orders: if waiter.x == 1 and waiter.y == 1 and waiter.orders:
restaurant.kitchen.append([waiter.orders[0], waiter.orders[1], 50]) restaurant.kitchen.append([waiter.orders[0], waiter.orders[1], 50])
waiter.orders = False waiter.orders = False
#jesli sie zatrzymal w kuchni bez zamowienia to bierze jedzenie #jesli sie zatrzymal w kuchni bez zamowienia to bierze jedzenie
if waiter.x == 1 and waiter.y == 1 and not waiter.orders: if waiter.x == 1 and waiter.y == 1 and not waiter.orders:
for t in restaurant.kitchen: for t in restaurant.kitchen:
@ -468,6 +472,7 @@ class Agent:
def drawScreen(): def drawScreen():
pygame.draw.rect(display, (0, 0, 0), (0, 0, HEIGHT * 32, WIDTH * 32)) pygame.draw.rect(display, (0, 0, 0), (0, 0, HEIGHT * 32, WIDTH * 32))
for ih in range(HEIGHT): for ih in range(HEIGHT):
for iw in range(WIDTH): for iw in range(WIDTH):
tile = restaurant.tiles[ih][iw] tile = restaurant.tiles[ih][iw]
@ -476,35 +481,32 @@ def drawScreen():
display.blit(tileFoil, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tileFoil, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
if tile.cost == 5: if tile.cost == 5:
pygame.draw.circle(display, (128, 128, 255), (iw * 32 + 17, ih * 32 + 17), 8) pygame.draw.circle(display, (128, 128, 255), (iw * 32 + 17, ih * 32 + 17), 8)
if tile.table: if tile.table:
if tile.clientState: if tile.clientState:
if tile.clientState == "decide": if tile.clientState == "decide":
#pygame.draw.rect(display, (0, 128, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) # pygame.draw.rect(display, (0, 128, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
display.blit(tableDecide, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tableDecide, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
elif tile.clientState == "order": elif tile.clientState == "order":
#pygame.draw.rect(display, (0, 255, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) # pygame.draw.rect(display, (0, 255, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
display.blit(tableOrder, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tableOrder, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
elif tile.clientState == "wait": elif tile.clientState == "wait":
#pygame.draw.rect(display, (255, 128, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) # pygame.draw.rect(display, (255, 128, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
display.blit(tableWait, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tableWait, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
elif tile.clientState == "eat": elif tile.clientState == "eat":
#pygame.draw.rect(display, (128, 64, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) # pygame.draw.rect(display, (128, 64, 0), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
display.blit(tableEat, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tableEat, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
else: else:
pygame.draw.rect(display, (64, 64, 64), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) # pygame.draw.rect(display, (64, 64, 64), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
#display.blit(tableEmpty, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) display.blit(tableEmpty, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
if tile.kitchen: if tile.kitchen:
pygame.draw.rect(display, (255, 0, 255), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) pygame.draw.rect(display, (255, 0, 255), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
# WE CAN UNCOMMENT IT, DON'T WE?
# if tile.visited: # if tile.visited:
# pygame.draw.rect(display, (64,0,64), (iw * 32 + 1, ih * 32+1, 14, 14)) # pygame.draw.rect(display, (64,0,64), (iw * 32 + 1, ih * 32+1, 14, 14))
else: else:
pygame.draw.rect(display, (128, 0, 128), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (128, 0, 128), (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
display.blit(wall, (iw * 32 + 1, ih * 32 + 1, 32 - 1, 32 - 1))
#pygame.draw.circle(display, (255, 255, 255), (waiter.x * 32 + 16, waiter.y * 32 + 16), 16) #pygame.draw.circle(display, (255, 255, 255), (waiter.x * 32 + 16, waiter.y * 32 + 16), 16)
#Render actor's frames according to the direction
display.blit(waiterAct[waiter.dir - 1], (waiter.x * 32 + 8, waiter.y * 32 + 8)) display.blit(waiterAct[waiter.dir - 1], (waiter.x * 32 + 8, waiter.y * 32 + 8))
''' '''
@ -522,6 +524,7 @@ def drawScreen():
pygame.draw.circle(display, (255, 0, 0), (waiter.x * 32 + 16+xx, waiter.y * 32 + 16+yy), 8) pygame.draw.circle(display, (255, 0, 0), (waiter.x * 32 + 16+xx, waiter.y * 32 + 16+yy), 8)
''' '''
textsurface = font.render(str(restaurant.clients), False, (255, 255, 255)) textsurface = font.render(str(restaurant.clients), False, (255, 255, 255))
display.blit(textsurface, (WIDTH * 32 + 80, 300)) display.blit(textsurface, (WIDTH * 32 + 80, 300))
@ -540,49 +543,39 @@ ticks = 0
# draw info # draw info
help = True help = True
if help: if help:
font = pygame.font.SysFont('Arial', 18) font = pygame.font.SysFont('Arial', 18)
textsurface = font.render("kelner", False, (255, 255, 255)) textsurface = font.render("kelner", False, (255, 255, 255))
#pygame.draw.circle(display, (255, 255, 255), (WIDTH * 32 + 26, 16), 16) #pygame.draw.circle(display, (255, 255, 255), (WIDTH * 32 + 26, 16), 16)
display.blit(waiterAct3, (WIDTH * 32 + 18, 4)) display.blit(waiterAct3, (WIDTH * 32 + 18, 4))
display.blit(textsurface, (WIDTH * 32 + 50, 0)) display.blit(textsurface, (WIDTH * 32 + 50, 0))
textsurface = font.render("sciana", False, (255, 255, 255)) textsurface = font.render("sciana", False, (255, 255, 255))
#pygame.draw.rect(display, (128, 0, 128), (WIDTH * 32 + 10, 32, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (128, 0, 128), (WIDTH * 32 + 10, 32, 32 - 1, 32 - 1))
display.blit(wall, (WIDTH * 32 + 10, 32, 32 - 1, 32 - 1)) display.blit(wall, (WIDTH * 32 + 10, 32, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 32)) display.blit(textsurface, (WIDTH * 32 + 50, 32))
textsurface = font.render("stolik - pusty", False, (255, 255, 255)) textsurface = font.render("stolik - pusty", False, (255, 255, 255))
#pygame.draw.rect(display, (64, 64, 64), (WIDTH * 32 + 10, 64, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (64, 64, 64), (WIDTH * 32 + 10, 64, 32 - 1, 32 - 1))
display.blit(tableEmpty, (WIDTH * 32 + 10, 64, 32 - 1, 32 - 1)) display.blit(tableEmpty, (WIDTH * 32 + 10, 64, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 64)) display.blit(textsurface, (WIDTH * 32 + 50, 64))
textsurface = font.render("stolik - decyduje", False, (255, 255, 255)) textsurface = font.render("stolik - decyduje", False, (255, 255, 255))
#pygame.draw.rect(display, (0, 128, 0), (WIDTH * 32 + 10, 96, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (0, 128, 0), (WIDTH * 32 + 10, 96, 32 - 1, 32 - 1))
display.blit(tableDecide, (WIDTH * 32 + 10, 96, 32 - 1, 32 - 1)) display.blit(tableDecide, (WIDTH * 32 + 10, 96, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 96)) display.blit(textsurface, (WIDTH * 32 + 50, 96))
textsurface = font.render("stolik - zamawia", False, (255, 255, 255)) textsurface = font.render("stolik - zamawia", False, (255, 255, 255))
#pygame.draw.rect(display, (0, 255, 0), (WIDTH * 32 + 10, 128, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (0, 255, 0), (WIDTH * 32 + 10, 128, 32 - 1, 32 - 1))
display.blit(tableOrder, (WIDTH * 32 + 10, 128, 32 - 1, 32 - 1)) display.blit(tableOrder, (WIDTH * 32 + 10, 128, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 128)) display.blit(textsurface, (WIDTH * 32 + 50, 128))
textsurface = font.render("stolik - czeka", False, (255, 255, 255)) textsurface = font.render("stolik - czeka", False, (255, 255, 255))
#pygame.draw.rect(display, (255, 128, 0), (WIDTH * 32 + 10, 160, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (255, 128, 0), (WIDTH * 32 + 10, 160, 32 - 1, 32 - 1))
display.blit(tableWait, (WIDTH * 32 + 10, 160, 32 - 1, 32 - 1)) display.blit(tableWait, (WIDTH * 32 + 10, 160, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 160)) display.blit(textsurface, (WIDTH * 32 + 50, 160))
textsurface = font.render("stolik - je", False, (255, 255, 255)) textsurface = font.render("stolik - je", False, (255, 255, 255))
#pygame.draw.rect(display, (128, 64, 0), (WIDTH * 32 + 10, 192, 32 - 1, 32 - 1)) #pygame.draw.rect(display, (128, 64, 0), (WIDTH * 32 + 10, 192, 32 - 1, 32 - 1))
display.blit(tableWait, (WIDTH * 32 + 10, 192, 32 - 1, 32 - 1)) display.blit(tableEat, (WIDTH * 32 + 10, 192, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 192)) display.blit(textsurface, (WIDTH * 32 + 50, 192))
textsurface = font.render("kuchnia", False, (255, 255, 255)) textsurface = font.render("kuchnia", False, (255, 255, 255))
pygame.draw.rect(display, (255, 0, 255), (WIDTH * 32 + 10, 224, 32 - 1, 32 - 1)) pygame.draw.rect(display, (255, 0, 255), (WIDTH * 32 + 10, 224, 32 - 1, 32 - 1))
display.blit(textsurface, (WIDTH * 32 + 50, 224)) display.blit(textsurface, (WIDTH * 32 + 50, 224))
textsurface = font.render("kaluza", False, (255, 255, 255)) textsurface = font.render("kaluza", False, (255, 255, 255))
pygame.draw.circle(display, (128, 128, 255), (WIDTH * 32 + 26, 272), 8) pygame.draw.circle(display, (128, 128, 255), (WIDTH * 32 + 26, 272), 8)
display.blit(textsurface, (WIDTH * 32 + 50, 256)) display.blit(textsurface, (WIDTH * 32 + 50, 256))
@ -640,7 +633,7 @@ while True:
restaurant.tiles[table[1]][table[0]].clientState = False restaurant.tiles[table[1]][table[0]].clientState = False
totaltime = totaltime + ticks totaltime = totaltime + ticks
restaurant.left = restaurant.left - 1 restaurant.left = restaurant.left - 1
#update waiter #update waiter
if project == 0: if project == 0:
if waiter.idle: if waiter.idle:
waiter.getTask() waiter.getTask()