import pygame
from pygame import locals
import random

pygame.init()   # 初始化
score = 0
grid_size = 20  # 格子大小
grid_num_width = 15     # 横向格子数量
grid_num_height = 25      # 纵向格子数量
FPS = 30
count = 0
states = False

# 创建窗口
screen = pygame.display.set_mode((460, 500))
pygame.display.set_caption("俄罗斯方块")
clock = pygame.time.Clock()  # pygame时钟，控制游戏速度（帧数）
# 载入素材
background = pygame.image.load('bg.png')
font = pygame.font.Font('STKAITI.TTF', 60)  # 字体

# 俄罗斯方块所有形状
O = [[(0, 0), (0, 1), (1, 0), (1, 1)]]
I = [[(0, -1), (0, 0), (0, 1), (0, 2)],
     [(-1, 0), (0, 0), (1, 0), (2, 0)]]
Z = [[(0, -1), (0, 0), (1, 0), (1, 1)],
     [(-1, 0), (0, 0), (0, -1), (1, -1)]]
S = [[(-1, 0), (0, 0), (0, 1), (1, 1)],
     [(1, -1), (1, 0), (0, 0), (0, 1)]]
T = [[(0, -1), (0, 0), (0, 1), (-1, 0)],
     [(-1, 0), (0, 0), (1, 0), (0, 1)],
     [(0, -1), (0, 0), (0, 1), (1, 0)],
     [(-1, 0), (0, 0), (1, 0), (0, -1)]]
J = [[(-1, 0), (0, 0), (1, 0), (1, -1)],
     [(0, -1), (0, 0), (0, 1), (-1, -1)],
     [(-1, 0), (0, 0), (1, 0), (-1, 1)],
     [(0, -1), (0, 0), (0, 1), (1, 1)]]
L = [[(-1, 0), (0, 0), (1, 0), (1, 1)],
     [(0, -1), (0, 0), (0, 1), (1, -1)],
     [(-1, 0), (0, 0), (1, 0), (-1, -1)],
     [(0, -1), (0, 0), (0, 1), (-1, 1)]]
shape_list = [I, J, L, O, S, T, Z]  # 7种类型俄罗斯方块

# 一些RGB颜色
cube_colors = [
    (204, 153, 153), (102, 102, 153),(153, 0, 102), 
    (255, 204, 0), (204, 0, 51),(255, 0, 51), (0, 102, 153),
    (153, 0, 51), (204, 255, 102), (255, 153, 0)]



def check(center):
    for i in current_shape:
        i = (i[0] + center[0], i[1] + center[1] )
        if i[0] <1 or i[0] > grid_num_height or i[1] <1 or i[1] > grid_num_width:
            return False

while True:
    for event in pygame.event.get():
        if event.type == locals.QUIT:
            exit()
        if event.type == locals.KEYDOWN:
            if event.key == locals.K_RIGHT:
                center[1] += 1
                if check(center) == False:
                    center[1] -= 1
            elif event.key == locals.K_LEFT:    # 向左
                center[1] -= 1
                if check(center) == False:
                    center[1] -= 1
            elif event.key == locals.K_DOWN:    # 向下
                center[0] += 1
                if check(center) == False:
                    center[0] -= 1
            elif event.key == locals.K_UP:
                old_index = index
                index += 1
                if index >= len(shape):
                    index = 0    
                current_shape = shape[index]
                if check(center) == False:
                    index = old_index
                    current_shape = shape[index]
    if states == False:
        states = True
        center = [2, 8]     # 第2行第8列
        shape = random.choice(shape_list)
        index = random.randint(0,len(shape) - 1)
        current_shape = shape[index]
        color = random.choice(cube_colors)
    count += 1
    if count % FPS == 0 :
        center[0] += 1
        if check(center) == False:
            center[0] -= 1
            states = False
    # 将背景图画上去
    screen.blit(background, (0, 0))
    # 计算出所有小方块的行、列位置
    current_pos = []
    for cube in current_shape:
        pos = (cube[0] + center[0], cube[1] + center[1])
        current_pos.append(pos)
    # 取出所有小方块的行、列位置，计算坐标，绘制俄罗斯方块
    for cube in current_pos:
        pygame.draw.rect(screen, color,
                    (cube[1] * 20-20, cube[0] * 20-20, 20, 20), 0)
        pygame.draw.rect(screen, (255, 255, 255), 
                        (cube[1] * 20-20, cube[0] * 20-20, 20, 20), 1)
    # 得分
    text_surface = font.render(str(score), True, (0, 0, 0))
    screen.blit(text_surface, (350,70))
    # 刷新画面
    pygame.display.update()
    clock.tick(FPS)