import sys
import pygame

screen = []
objectArray = []


class Position:
    def __init__(self, x, y):
        self.x = x
        self.y = y


class Object:
    def __init__(self, name, pos):
        self.name = name
        self.pos = pos

    def draw(self, square):
        pass


class Agent(Object):
    def __init__(self, name, pos):
        super().__init__(name, pos)

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


class House(Object):
    def __init__(self, name, pos):
        super().__init__(name, pos)

    def draw(self, square):
        leftTopX, leftTopY = 50 + self.pos.x * square, 10 + self.pos.y * square
        pygame.draw.rect(screen, (0, 255, 0), (leftTopX, leftTopY, square, square))


class Junkyard(Object):
    def __init__(self, name, pos):
        super().__init__(name, pos)

    def draw(self, square):
        leftTopX, leftTopY = 50 + self.pos.x * square, 10 + self.pos.y * square
        pygame.draw.rect(screen, (255, 0, 0), (leftTopX, leftTopY, square, square))

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 o in objectArr:
        o.draw(square)

    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


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].pos.x > 0:
                objectArray[0].pos.x = objectArray[0].pos.x - 1
            if event.key == pygame.K_RIGHT and objectArray[0].pos.x < gridLength - 1:
                objectArray[0].pos.x = objectArray[0].pos.x + 1
            if event.key == pygame.K_UP and objectArray[0].pos.y > 1:
                objectArray[0].pos.y = objectArray[0].pos.y - 1
            if event.key == pygame.K_DOWN and objectArray[0].pos.y < gridLength:
                objectArray[0].pos.y = objectArray[0].pos.y + 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", Position(5, 7))
    junkyard = Junkyard("wysypisko", Position(10, 10))
    housePositions = [Position(x, y) for x, y in [
        (7, 4), (3, 10), (8, 10), (4, 5), (1, 2), (10, 4), (14, 14), (6, 9)
    ]]
    houses = [House(f'dom-{i}', pos) for i, pos in enumerate(housePositions)]
    objectArray.append(agent)
    objectArray.append(junkyard)
    objectArray += houses

    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