import pygame
import random

WIDTH = 640
HEIGHT = 480
FPS = 30
GROUND = HEIGHT - 30

# RGB = (red, green, blue)
WHITE = (255, 255, 255)
BLACK = (0, 0, 0)
RED = (255, 0, 0)
GREEN = (0, 255, 0)
BLUE = (0, 0, 255)


pygame.init()
pygame.mixer.init()
screen = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("My best game")
clock = pygame.time.Clock()

class Dino(pygame.sprite.Sprite):
    __jump = 0
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.Surface((50, 50))
        self.image.fill(GREEN)
        self.rect = self.image.get_rect()
        self.rect.left = WIDTH/3
        self.rect.bottom = GROUND


    def update(self):
        self.rect.y -= self.__jump
        if self.rect.bottom >= GROUND:
            self.__jump = 0
        else:
            self.__jump -= 1

    def jump (self):
        if self.rect.bottom == GROUND:
            self.__jump = 20

all_sprites = pygame.sprite.Group()
dinosuar = Dino()
all_sprites.add(dinosuar)


class Cactus(pygame.sprite.Sprite):
    def __init__(self):
        pygame.sprite.Sprite.__init__(self)
        self.image = pygame.Surface((50, 50))
        self.image.fill(RED)
        self.rect = self.image.get_rect()
        self.rect.center = (WIDTH/2, HEIGHT/2)
        self.rect.bottom = HEIGHT - 20


    def update(self):
        self.rect.x -= 4
        if (self.rect.left <= 0):
            self.rect.left = WIDTH

all_sprites = pygame.sprite.Group()
cactus = Cactus()
all_sprites.add(cactus)



runnning = True
currentColor = WHITE
while runnning:
    clock.tick(FPS)
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            runnning = False
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_SPACE:
                dinosuar.jump()


    all_sprites.update()


    screen.fill(currentColor)
    all_sprites.draw(screen)
    pygame.display.flip()

pygame.quit()