import pygame

from plansza import x1, y1, x2, y2, x3, y3, x4, y4, a_pix, b_pix
import packageList
import regal
import wozek
from packageList import *
from paczka import Paczka

EKRAN_SZEROKOSC = 770
EKRAN_WYSOKOSC = 770
screen = pygame.display.set_mode((EKRAN_SZEROKOSC, EKRAN_WYSOKOSC))
miejsce = pygame.image.load('images/miejsce_paczek.png')
miejsce = pygame.transform.scale(miejsce, (140, 140))
pygame.display.set_caption("Inteligentny wozek")
icon = pygame.image.load('images/icon.png')
pygame.display.set_icon(icon)

lista_paczek = packageList.zainicjuj_liste_paczek(a_pix, b_pix)
wozek_group = pygame.sprite.Group()
regals_collision_group = pygame.sprite.Group()

def narysuj_regaly():
    regalList = []
    allRegals = {
        "aCategory": regalList,
        "bCategory": regalList,
        "cCategory": regalList,
        "dCategory": regalList,
    }
    allRegals["aCategory"].append(regal.Regal(1, 1, x1, y1))
    allRegals["aCategory"].append(regal.Regal(2, 1, x1, y1+1))
    allRegals["aCategory"].append(regal.Regal(3, 1, x1+1, y1))
    allRegals["aCategory"].append(regal.Regal(4, 1, x1+1, y1+1))

    allRegals["bCategory"].append(regal.Regal(5, 1, x2, y2))
    allRegals["bCategory"].append(regal.Regal(6, 1, x2, y2+1))
    allRegals["bCategory"].append(regal.Regal(7, 1, x2+1, y2))
    allRegals["bCategory"].append(regal.Regal(8, 1, x2+1, y2+1))

    allRegals["cCategory"].append(regal.Regal(9, 1, x3, y3))
    allRegals["cCategory"].append(regal.Regal(10, 1, x3, y3+1))
    allRegals["cCategory"].append(regal.Regal(11, 1, x3+1, y3))
    allRegals["cCategory"].append(regal.Regal(12, 1, x3+1, y3+1))

    allRegals["dCategory"].append(regal.Regal(13, 1, x4, y4))
    allRegals["dCategory"].append(regal.Regal(14, 1, x4, y4+1))
    allRegals["dCategory"].append(regal.Regal(15, 1, x4+1, y4))
    allRegals["dCategory"].append(regal.Regal(16, 1, x4+1, y4+1))

    for category in allRegals:
        for regal_obj in allRegals[category]:
            regals_collision_group.add(regal_obj)


def narysuj_siatke():
    blockSize = 70  # Set the size of the grid block
    WHITE = (200, 200, 200)
    for x in range(0, EKRAN_SZEROKOSC, blockSize):
        for y in range(0, EKRAN_WYSOKOSC, blockSize):
            rect = pygame.Rect(x, y, blockSize, blockSize)
            pygame.draw.rect(screen, WHITE, rect, 1)


def odswiez_ekran(wozek):
    screen.fill((51, 51, 51)) # removes object trail
    screen.blit(miejsce,(a_pix, b_pix))

    narysuj_siatke()
    narysuj_paczki(wozek)
    narysuj_regaly()
    wozek.draw()
    pygame.display.flip()


def sprawdz_ktory_wiersz(x):
    nr_wiersza = 0
    for i in range(70, EKRAN_WYSOKOSC + 70, 70):
        if x < i:
            return nr_wiersza
        nr_wiersza = nr_wiersza + 1


def sprawdz_ktora_kolumna(y):
    nr_kolumny = 0
    for i in range(70, EKRAN_SZEROKOSC + 70, 70):
        if y < i:
            return nr_kolumny
        nr_kolumny = nr_kolumny + 1


def narysuj_paczki(wozek):
    if wozek.ln == 0:
        for paczka in lista_paczek.list:
            paczka.narysuj(paczka.x, paczka.y, screen)