Sztuczna_Inteligencja_2020/main.py

import sys
import secrets

from src.graphics import *
from src.waiter import *
from src.decisionTree import *

if __name__ == "__main__":
    # SETUP
    pygame.init()
    clock = pygame.time.Clock()
    graphics = Graphics()
    waiter = Waiter(graphics)
    tree = DecisionTree()

    # 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)