import pygame
import sys

pygame.init()

SIRKA = 800
VYSKA = 600
okno = pygame.display.set_mode((SIRKA, VYSKA))
pygame.display.set_caption("Pong - Plná Hra")

BILA = (255, 255, 255)
CERNA = (0, 0, 0)
ZELENA = (0, 255, 0)

hodiny = pygame.time.Clock()
font_velky = pygame.font.SysFont("Arial", 50)
font_maly = pygame.font.SysFont("Arial", 30)

STAV_MENU = 0
STAV_HRA = 1
STAV_KONEC = 2
stav = STAV_MENU

palka_sirka = 15
palka_vyska = 100
rychlost_palky = 7
micek_velikost = 15

# Funkce pro reset hry
def reset_hry():
    global hrac1_y, hrac2_y, skore1, skore2, micek_x, micek_y, micek_rychlost_x, micek_rychlost_y
    hrac1_y = VYSKA // 2 - palka_vyska // 2
    hrac2_y = VYSKA // 2 - palka_vyska // 2
    skore1 = 0
    skore2 = 0
    micek_x = SIRKA // 2 - micek_velikost // 2
    micek_y = VYSKA // 2 - micek_velikost // 2
    micek_rychlost_x = 5
    micek_rychlost_y = 5

reset_hry()

bezime = True
while bezime:
    for udalost in pygame.event.get():
        if udalost.type == pygame.QUIT:
            bezime = False
        if udalost.type == pygame.KEYDOWN:
            if stav == STAV_MENU and udalost.key == pygame.K_SPACE:
                stav = STAV_HRA
            elif stav == STAV_KONEC and udalost.key == pygame.K_SPACE:
                reset_hry()
                stav = STAV_HRA

    klavesy = pygame.key.get_pressed()

    if stav == STAV_HRA:
        if klavesy[pygame.K_w] and hrac1_y > 0:
            hrac1_y -= rychlost_palky
        if klavesy[pygame.K_s] and hrac1_y < VYSKA - palka_vyska:
            hrac1_y += rychlost_palky

        if klavesy[pygame.K_UP] and hrac2_y > 0:
            hrac2_y -= rychlost_palky
        if klavesy[pygame.K_DOWN] and hrac2_y < VYSKA - palka_vyska:
            hrac2_y += rychlost_palky

        micek_x += micek_rychlost_x
        micek_y += micek_rychlost_y

        if micek_y <= 0 or micek_y >= VYSKA - micek_velikost:
            micek_rychlost_y *= -1

        rect_micek = pygame.Rect(micek_x, micek_y, micek_velikost, micek_velikost)
        rect_hrac1 = pygame.Rect(30, hrac1_y, palka_sirka, palka_vyska)
        rect_hrac2 = pygame.Rect(SIRKA - 30 - palka_sirka, hrac2_y, palka_sirka, palka_vyska)

        if rect_micek.colliderect(rect_hrac1) or rect_micek.colliderect(rect_hrac2):
            micek_rychlost_x *= -1.1 # Zrychlení po odrazu

        if micek_x < 0:
            skore2 += 1
            micek_x, micek_y = SIRKA // 2, VYSKA // 2
            micek_rychlost_x = 5
        elif micek_x > SIRKA:
            skore1 += 1
            micek_x, micek_y = SIRKA // 2, VYSKA // 2
            micek_rychlost_x = -5

        # Podmínka výhry
        if skore1 >= 5 or skore2 >= 5:
            stav = STAV_KONEC

    # Vykreslování
    okno.fill(CERNA)
    
    if stav == STAV_MENU:
        nadpis = font_velky.render("PONG", True, BILA)
        navod = font_maly.render("Stiskni MEZERNÍK pro start", True, ZELENA)
        okno.blit(nadpis, (SIRKA//2 - nadpis.get_width()//2, 200))
        okno.blit(navod, (SIRKA//2 - navod.get_width()//2, 300))

    elif stav == STAV_HRA:
        pygame.draw.aaline(okno, BILA, (SIRKA // 2, 0), (SIRKA // 2, VYSKA))
        pygame.draw.rect(okno, BILA, rect_hrac1)
        pygame.draw.rect(okno, BILA, rect_hrac2)
        pygame.draw.ellipse(okno, BILA, rect_micek)
        text_skore = font_velky.render(f"{skore1}   {skore2}", True, BILA)
        okno.blit(text_skore, (SIRKA // 2 - text_skore.get_width() // 2, 20))

    elif stav == STAV_KONEC:
        vitez = "Hráč 1" if skore1 >= 5 else "Hráč 2"
        text_vitez = font_velky.render(f"{vitez} vyhrál!", True, ZELENA)
        text_restart = font_maly.render("Stiskni MEZERNÍK pro novou hru", True, BILA)
        okno.blit(text_vitez, (SIRKA//2 - text_vitez.get_width()//2, 200))
        okno.blit(text_restart, (SIRKA//2 - text_restart.get_width()//2, 300))

    pygame.display.flip()
    hodiny.tick(60)

pygame.quit()
sys.exit()