import pygame
from pygame.locals import *
import sys
import random
import time

pygame.init()

vec = pygame.math.Vector2
HEIGHT = 450
WIDTH = 400
ACC = 0.5
FRIC = -0.12
FPS = 60

FramePerSec = pygame.time.Clock()
displaysurface = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Game")

# Groups
platforms = pygame.sprite.Group()
all_sprites = pygame.sprite.Group()

class Player(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()
        self.surf = pygame.Surface((30, 30))
        self.surf.fill((128, 255, 40))
        self.rect = self.surf.get_rect()
        self.pos = vec((WIDTH // 2, HEIGHT - 60))
        self.vel = vec(0, 0)
        self.acc = vec(0, 0)
        self.jumping = False
        self.score = 0

    def move(self):
        self.acc = vec(0, 0.5)
        pressed_keys = pygame.key.get_pressed()
        if pressed_keys[K_LEFT]:
            self.acc.x = -ACC
        if pressed_keys[K_RIGHT]:
            self.acc.x = ACC

        self.acc.x += self.vel.x * FRIC
        self.vel += self.acc
        self.pos += self.vel + 0.5 * self.acc

        if self.pos.x > WIDTH:
            self.pos.x = 0
        if self.pos.x < 0:
            self.pos.x = WIDTH

        self.rect.midbottom = self.pos

    def jump(self):
        hits = pygame.sprite.spritecollide(self, platforms, False)
        if hits and not self.jumping:
            self.jumping = True
            self.vel.y = -15

    def cancel_jump(self):
        if self.jumping and self.vel.y < -3:
            self.vel.y = -3

    def update(self):
        hits = pygame.sprite.spritecollide(self, platforms, False)
        if self.vel.y > 0 and hits:
            lowest = hits[0]
            if self.pos.y < lowest.rect.bottom:
                if lowest.point:
                    lowest.point = False
                    self.score += 1
                self.pos.y = lowest.rect.top + 1
                self.vel.y = 0
                self.jumping = False
        self.rect.midbottom = self.pos

class Platform(pygame.sprite.Sprite):
    def __init__(self):
        super().__init__()
        self.surf = pygame.Surface((random.randint(50, 100), 12))
        self.surf.fill((0, 255, 0))
        self.rect = self.surf.get_rect(center=(random.randint(0, WIDTH - 10), random.randint(0, HEIGHT - 30)))
        self.point = True
        self.speed = random.randint(-1, 1)
        self.moving = True

    def move(self):
        if self.moving:
            self.rect.move_ip(self.speed, 0)
            if self.rect.left > WIDTH:
                self.rect.right = 0
            if self.rect.right < 0:
                self.rect.left = WIDTH

def check(platform, groupies):
    if pygame.sprite.spritecollideany(platform, groupies):
        return True
    for entity in groupies:
        if entity == platform:
            continue
        if (abs(platform.rect.top - entity.rect.bottom) < 50) and (abs(platform.rect.bottom - entity.rect.top) < 50):
            return True
    return False

def plat_gen():
    while len(platforms) < 6:
        width = random.randrange(50, 100)
        p = Platform()
        C = True
        while C:
            p = Platform()
            p.rect.center = (random.randrange(0, WIDTH - width), random.randrange(-50, 0))
            C = check(p, platforms)
        platforms.add(p)
        all_sprites.add(p)

def restart_game():
    global all_sprites, platforms, player

    all_sprites.empty()
    platforms.empty()

    player = Player()
    all_sprites.add(player)

    ground = Platform()
    ground.surf = pygame.Surface((WIDTH, 20))
    ground.surf.fill((255, 0, 0))
    ground.rect = ground.surf.get_rect(center=(WIDTH / 2, HEIGHT - 10))
    ground.point = False
    ground.moving = False
    platforms.add(ground)
    all_sprites.add(ground)

    for _ in range(random.randint(4, 5)):
        C = True
        p = Platform()
        while C:
            p = Platform()
            C = check(p, platforms)
        platforms.add(p)
        all_sprites.add(p)

# --- Setup Game ---
player = Player()
all_sprites.add(player)

ground = Platform()
ground.surf = pygame.Surface((WIDTH, 20))
ground.surf.fill((255, 0, 0))
ground.rect = ground.surf.get_rect(center=(WIDTH / 2, HEIGHT - 10))
ground.point = False
ground.moving = False
platforms.add(ground)
all_sprites.add(ground)

for _ in range(random.randint(4, 5)):
    C = True
    p = Platform()
    while C:
        p = Platform()
        C = check(p, platforms)
    platforms.add(p)
    all_sprites.add(p)

font = pygame.font.SysFont("Verdana", 20)

# --- Game Loop ---
while True:
    player.update()

    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            sys.exit()
        if event.type == KEYDOWN:
            if event.key == K_SPACE:
                player.jump()
            if event.key == K_r:
                restart_game()
        if event.type == KEYUP:
            if event.key == K_SPACE:
                player.cancel_jump()

    if player.rect.top > HEIGHT:
        displaysurface.fill((255, 0, 0))
        pygame.display.update()
        time.sleep(1)
        restart_game()

    if player.rect.top <= HEIGHT / 3:
        player.pos.y += abs(player.vel.y)
        for plat in platforms:
            plat.rect.y += abs(player.vel.y)
            if plat.rect.top >= HEIGHT:
                plat.kill()

    plat_gen()

    displaysurface.fill((0, 0, 0))
    score_text = font.render(str(player.score), True, (123, 255, 0))
    displaysurface.blit(score_text, (WIDTH / 2, 10))

    for entity in all_sprites:
        displaysurface.blit(entity.surf, entity.rect)
        if hasattr(entity, 'move'):
            entity.move()

    pygame.display.update()
    FramePerSec.tick(FPS)