import pygame
from pygame import locals
import random

pygame.init()  # 初始化
class Block(pygame.sprite.Sprite):
    def __init__(self,image1,image2,image3):
        super().__init__()
        self.image=random.choice([image1,image2,image3])
        self.rect=self.image.get_rect()
        self.rect.x=1000
        self.rect.y=500-self.rect.height
# 创建一个窗口
screen = pygame.display.set_mode((1000, 600))
FPS = pygame.time.Clock()  # pygame时钟，控制游戏速度（帧数）
pygame.display.set_caption("悟空酷跑")
# 载入图片
background = pygame.image.load('bg.png')    # 背景
road = pygame.image.load('road.png')      # 路
stone = pygame.image.load('stone.png')      # 石头
cacti = pygame.image.load('cacti.png')      # 仙人掌
bush = pygame.image.load('bush.png')      # 灌木丛
hero = [pygame.image.load('hero1.png'),
        pygame.image.load('hero2.png'),
        pygame.image.load('hero3.png'),
        pygame.image.load('hero4.png'),
        pygame.image.load('hero5.png')]
index = 0   
y = 400
jumpState = "runing"
t = 30
# obstacle = random.choice([bush, stone, cacti])
# rect = obstacle.get_rect()
# rect.x = 1000
# rect.y = 500 - rect.height
roadx=0
bg_x=0
time=0
obstacle=Block(bush,cacti,stone)
while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            # 接收到退出事件后退出程序
            exit()
        if event.type == locals.KEYDOWN:
            if jumpState == "runing":
                if event.key == locals.K_SPACE:
                    jumpState = "up"

    if jumpState == "up":    # 起跳状态
        if t > 0:
            y -= t
            t -= 2
        else:
            jumpState = "down"  
    if jumpState == "down":    # 降落状态
        if t <= 30:
            y += t
            t += 2
        else:
            jumpState = "runing"
            t =30
    # 悟空造型
    wukong = hero[index]
    if jumpState == "runing":       # 跑步状态下
        index += 1
        if index >= 5:
            index = 0
    # 将背景图画上去
    screen.blit(background, (bg_x, 0))     # 远处背景
    screen.blit(road, (roadx, 500))     # 路
    screen.blit(wukong, (150, y))     # 悟空
    time+=1
    if time >=60:
        time=0
        num=random.randint(0,50)
        
    # if rect.x <= 0-rect.width:      # 障碍物消失
    #     # 创建障碍物对象
    #     obstacle = random.choice([bush,stone,cacti])
    #     rect = obstacle.get_rect()
    #     rect.x = 1000
    #     rect.y = 500 - rect.height
    # rect.x -= 25
    roadx-=25
    bg_x-=4
    if roadx<=-1000:
        roadx=0
    if bg_x<=-1000:
        bg_x=0
    if obstacle.rect.x<=0-obstacle.rect.width:
        obstacle=Block(bush,cacti,stone)
    obstacle.rect.x-=25
    screen.blit(obstacle.image, (obstacle.rect.x, obstacle.rect.y)) 
    # 刷新画面
    pygame.display.update()
    FPS.tick(60)