import os

from agenci import *
from stale import *

OKNO = pygame.display.set_mode((SZEROKOSC_OKNA, WYSOKOSC_OKNA))
pygame.display.set_caption("Okno1")
Agenci1 = []

Krata = []
for rzad in range(LICZBA_POL_W_PIONIE):
    Krata.append([])
    for kolumna in range(LICZBA_POL_W_POZIOMIE):
        Krata[rzad].append(POLE.PUSTE)


def wyswietl_krate():
    OKNO.fill(SZARY1)
    for rzad in range(LICZBA_POL_W_PIONIE):
        for kolumna in range(LICZBA_POL_W_POZIOMIE):
            pygame.draw.rect(OKNO, BIALY, [(ODSTEP_MIEDZY_POLAMI + BOK_POLA) * kolumna + ODSTEP_MIEDZY_POLAMI,
                                           (ODSTEP_MIEDZY_POLAMI + BOK_POLA) * rzad + ODSTEP_MIEDZY_POLAMI,
                                           BOK_POLA, BOK_POLA])


def wyswietl_agentow():
    for a in Agenci1:
        OKNO.blit(a.tekstura, (a.hitbox.x, a.hitbox.y))
        if a.droga == 0:
            a.obierzNowyKierunek()
            a.okreslDlugoscDrogi()
        # a.ruszSie(Agenci1)
    pygame.display.update()


def czy_wylosowane_dla_agenta_pola_sa_puste(pole_lewe_gorne):
    wynik = True
    for wiersz in range(pole_lewe_gorne.wiersz, pole_lewe_gorne.wiersz + BOK_AGENTA1_W_POLACH):
        for kolumna in range(pole_lewe_gorne.kolumna, pole_lewe_gorne.kolumna + BOK_AGENTA1_W_POLACH):
            if Krata[wiersz][kolumna] != POLE.PUSTE:
                wynik = False
                break
        if wynik == False:
            break
    return wynik


def dodaj_agenta1():
    los = None
    if len(Agenci1) in range(0, 40):
        los = random.randint(1, 150)
    elif len(Agenci1) in range(40, 50):
        los = random.randint(1, 15)
    if los != None:
        pom = None
        if los in (1, 2) or len(Agenci1) == 0:
            pom = 'wozek.png'
        elif los in (3, 4):
            pom = 'wozek_ze_skrzynka.png'
        elif los == 5 and len(Agenci1) > 6:
            pom = 'traktor_ikona.png'
        if pom != None:
            pole_lewe_gorne = Pole(random.randint(0, LICZBA_POL_W_PIONIE - BOK_AGENTA1_W_POLACH),
                                   random.randint(0, LICZBA_POL_W_POZIOMIE - BOK_AGENTA1_W_POLACH))
            if len(Agenci1) == 0:
                pole_lewe_gorne = Pole(0, 0)
            if czy_wylosowane_dla_agenta_pola_sa_puste(pole_lewe_gorne):
                pom='test1_ikona.png'
                ikona = pygame.transform.scale(pygame.image.load(os.path.join('Ikony', pom)),
                                               (BOK_AGENTA1, BOK_AGENTA1))
                nowy_agent = Agent1(pole_lewe_gorne, ikona, KIERUNEK.GORA, 0)
                nowy_agent.zaznacz_zajmowane_pola_na_kracie(Krata)
                Agenci1.append(nowy_agent)
        # print(Krata)


def main():
    klatkaz = pygame.time.Clock()
    warunek_dzialania = True
    while warunek_dzialania:
        klatkaz.tick(FPS)
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                warunek_dzialania = False
                break
        wyswietl_krate()
        wyswietl_agentow()
        dodaj_agenta1()
    pygame.quit()


main()