import pygame
from pygame import locals


pygame.init()  # 初始化
# 创建一个窗口
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')      # 仙人掌
apple = 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')]

state='running'

index = 0 
y=400

t=30

while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            # 接收到退出事件后退出程序
            exit()
        if event.type==locals.KEYDOWN:
            if event.key==locals.K_SPACE:
                if state=="running":
                    state='up'
    if state=='up':
        if t >0:
            y-=t
            t-=2
        else:
            state='down'
    if state=='down':
        if t<=30:
            y+=t
            t+=2
        else:
            state='running'
            t=30

    wukong=hero[index]
    if state=="running":
        index+=1
        if index==5:
            index=0


    # 将背景图画上去
    screen.blit(background, (0, 0))
    screen.blit(road, (0, 500))
    screen.blit(wukong, (150, y))
    # 刷新画面
    pygame.display.update()
    FPS.tick(60)