AI_PROJECT/App.py

import pygame
import Colors
import Tractor
import Pole
import time
import displayControler as dCon
import Image
import Osprzet


pygame.init()
screen = pygame.display.set_mode((dCon.getScreenWidth(), dCon.getScreenHeihgt()))

image_loader=Image.Image()
image_loader.load_images()
pole=Pole.Pole(screen,image_loader)
pole.draw_grid() #musi byc tutaj wywołane ponieważ inicjalizuje sloty do slownika

#Tractor creation
traktor_slot = pole.get_slot_from_cord((0, 0))
traktor = Tractor.Tractor(traktor_slot, screen, Osprzet.plug)


def init_demo(): #Demo purpose
    old_info=""
    traktor.draw_tractor()
    time.sleep(2)
    pole.randomize_colors()
    traktor.draw_tractor()
    while True:
        time.sleep(0.5)
        demo_move()
        old_info=get_info(old_info)
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                quit()

def init(demo):
    pygame.display.update()
    if(demo==True):
        init_demo()
    #TODO: Implement


def demo_move():
    current_slot = traktor.slot
    if current_slot:
        current_slot.redraw_image()  # Przerysowanie obrazu dla aktualnego slotu
    traktor.random_move(pole)

def get_info(old_info):
    (x,y)=pygame.mouse.get_pos()
    new_info=pole.check_collision(x,y)
    if(old_info!=new_info):
        print(new_info)
    return new_info