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

# 创建窗口
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)]

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)

def check(center):
    for cube in current_shape:
        cube = (cube[0] + center[0],cube[1] + center[1])
        if cube[0] < 1 or cube[1] > 1 or cube[0]
        > grid_num_height or cube[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:     # 向右
                if center[1] < 15:
                    center[1] += 1
            elif event.key == locals.K_LEFT:    # 向左
                if center[1] > 1:
                    center[1] -= 1
            elif event.key == locals.K_DOWN:    # 向下
                if center[0] < 25:
                    center[0] += 1

    # 将背景图画上去
    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)