Sztuczna_Inteligencja_2020/main.py

67 lines
1.9 KiB
Python

import sys
import secrets
from src.graphics import *
from src.rabbit import Rabbit
from src.waiter import *
from src.decisionTree import *
if __name__ == "__main__":
# SETUP
pygame.init()
clock = pygame.time.Clock()
graphics = Graphics()
waiter = Waiter(graphics)
rabbit = Rabbit()
# init functions
graphics.drawBackground(waiter.matrix)
graphics.update(waiter)
goal = None
path = ''
while True:
for event in pygame.event.get():
# rabbit.check(waiter.matrix, waiter.X, waiter.Y)
if event.type == pygame.QUIT:
pygame.quit()
sys.exit()
break
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
pygame.quit()
sys.exit()
break
if event.key == pygame.K_s:
tree.TasksList('check', [10,3])
tree.TasksList('eat', [5,12])
tree.TasksList('order', [12,4])
tree.TasksList('goToBar', [10,10])
tree.TasksList('check', [3,4])
tree.TasksList('eat', [10,5])
tree.print()
if event.key == pygame.K_r:
temp = False
while not temp:
x = secrets.randbelow(graphics.width)
y = secrets.randbelow(graphics.height)
if waiter.matrix.matrix[x][y].walk_through == 1:
temp = True
goal = (x, y)
path = waiter.findPath(goal)
path = waiter.translatePath(path)
if path != '':
nextStep = path[0]
path = path[1:]
waiter.travel(nextStep, graphics)
pygame.display.flip()
clock.tick(graphics.fps)