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'),]
index = 0 
jumpstate='runing'
y=400
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:
                jumpstate='up'
    if jumpstate=='up':
        if y>150:
            y-=5
        else:
            jumpstate='down'
    if jumpstate=='down':
        if y<400:
            y+=5
        else:
            jumpstate='runing'    
    wukong=hero[index]        
    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)