import sys
import pygame

screen = []
objectArray = []

class Agent:
    def __init__(self, name, xPos, yPos):
        self.name = name
        self.xPos = xPos
        self.yPos = yPos

def draw(square_num, objectArr):
    #następne dwie linijki do odkomentowania, jak będzie wgrane zdjęcie do tła
    #background = pygame.image.load("ścieżka do pliku").convert() #tu ścieżka do zdjęcia w tle
    #screen.blit(background, (0, 0))
    grid_color = (0, 0, 0) #kolor czarny

    grid_size = 500  #rozmiar kraty
    square = grid_size/square_num #rozmiar pojedyńczego kwadracika

    a = 50
    b = 10  #odległości kraty od krawędzi okna

    for i in range(square_num):
        pygame.draw.line(screen, grid_color, (a + i*square, b), (a + i*square, b + grid_size), 2)

        pygame.draw.line(screen, grid_color, (a, b + i*square), (a + grid_size, b + i*square), 2)

    pygame.draw.line(screen, grid_color, (a, b + grid_size), (a + grid_size,
                     b + grid_size), 2)
    pygame.draw.line(screen, grid_color, (a + grid_size, b),
                     (a + grid_size, b + grid_size), 2)

    ## tutaj rysujemy agenta i inne obiekty juz na gotowej mapie

    ## RYSUJEMY AGENTA
    #agent_color = (255, 0, 0)
    circleX = objectArr[0].xPos * square + square + square/ 2 #dodane jedno +square, by śmieciara nie wychodziła poza kratę
    circleY = objectArr[0].yPos * square - square / 2
    #radius = 10
    #pygame.draw.circle(screen, agent_color, (a + circleX, b + circleY), radius)
    truck = pygame.image.load("C:/Users/gabi9/Downloads/car.png.png").convert_alpha() #tu ścieżka do zdjęcia w tle
    truck = pygame.transform.scale(truck, (square, square))
    screen.blit(truck, (circleX, circleY))

def kb_listen(objectArray, gridLength):
    for event in pygame.event.get():
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT and objectArray[0].xPos > 0:
                objectArray[0].xPos = objectArray[0].xPos - 1
            if event.key == pygame.K_RIGHT and objectArray[0].xPos < gridLength - 1:
                objectArray[0].xPos = objectArray[0].xPos + 1
            if event.key == pygame.K_UP and objectArray[0].yPos > 1:
                objectArray[0].yPos = objectArray[0].yPos - 1
            if event.key == pygame.K_DOWN and objectArray[0].yPos < gridLength:
                objectArray[0].yPos = objectArray[0].yPos + 1
        if event.type == pygame.QUIT:
            sys.exit()


if __name__ == '__main__':
    pygame.init()  #inicjalizacja modułów, na razie niepotrzebna

    #Tworzymy nowego playera, czy tam agenta
    agent = Agent("smieciarka", 5, 7)
    objectArray.append(agent)

    width = 600
    height = 530
    screen = pygame.display.set_mode((width, height))  #ustalanie rozmiarów okna

    while 1:
        c = (255, 255, 255)  #tymczasowy kolor tła - do usunięcia, jak już będzie zdjęcie
        screen.fill(c)
        draw(15, objectArray)
        kb_listen(objectArray, 15)
        pygame.display.update()  #by krata pojawiła się w okienku - update powierzchni