pip install pygame -i https://mirrors.aliyun.com/pypi/simple/

"""
作者：it項目實例網
更多項目實例，請訪問：www.itprojects.cn
"""

import random
import sys
import time

import pygame

from blocks import block_s, block_i, block_j, block_l, block_o, block_t, block_z

SCREEN_WIDTH, SCREEN_HEIGHT = 450, 750
BG_COLOR = (40, 40, 60) # 背景色
BLOCK_COL_NUM = 10 # 每行的方格數
SIZE = 30 # 每個小方格大小
BLOCK_ROW_NUM = 25 # 每列的方個數
BORDER_WIDTH = 4 # 游戲區邊框寬度
RED = (200, 30, 30) # 紅色，GAME OVER 的字體顏色


def judge_game_over(stop_all_block_list):
 """
 判斷游戲是否結束
 """
 if "O" in stop_all_block_list[0]:
  return True


def change_speed(score):
 speed_level = [("1", 0.5, 0, 20), ("2", 0.4, 21, 50), ("3", 0.3, 51, 100), ("4", 0.2, 101, 200), ("5", 0.1, 201, None)]
 for speed_info, speed, score_start, score_stop in speed_level:
  if score_stop and score_start = score = score_stop:
   return speed_info, speed
  elif score_stop is None and score >= score_start:
   return speed_info, speed


def judge_lines(stop_all_block_list):
 """
 判斷是否有同一行的方格，如果有則消除
 """
 # 記錄剛剛消除的行數
 move_row_list = list()
 # 消除滿格的行
 for row, line in enumerate(stop_all_block_list):
  if "." not in line:
   # 如果這一行沒有. 那么就意味著全部是O，則消除這一行
   stop_all_block_list[row] = ['.' for _ in range(len(line))]
   move_row_list.append(row)

 # 如果沒有滿格的行，則結束此函數
 if not move_row_list:
  return 0

 # 移動剩余的行到下一行
 for row in move_row_list:
  stop_all_block_list.pop(row)
  stop_all_block_list.insert(0, ['.' for _ in range(len(line))])

 return len(move_row_list) * 10


def add_to_stop_all_block_list(stop_all_block_list, current_block, current_block_start_row, current_block_start_col):
 """
 將當前已經停止移動的block添加到列表中
 """
 for row, line in enumerate(current_block):
  for col, block in enumerate(line):
   if block != '.':
    stop_all_block_list[current_block_start_row + row][current_block_start_col + col] = "O"


def change_current_block_style(current_block):
 """
 改變圖形的樣式
 """
 # 計算出，當前圖形樣式屬于哪個圖形
 current_block_style_list = None
 for block_style_list in [block_s, block_i, block_j, block_l, block_o, block_t, block_z]:
  if current_block in block_style_list:
   current_block_style_list = block_style_list

 # 得到當前正在用的圖形的索引（下標）
 index = current_block_style_list.index(current_block)
 # 它的下一個圖形的索引
 index += 1
 # 防止越界
 index = index % len(current_block_style_list)
 # 返回下一個圖形
 return current_block_style_list[index]


def judge_move_right(current_block, current_block_start_col):
 """
 判斷是否可以向右移動
 """
 # 先判斷列的方式是從右到左
 for col in range(len(current_block[0]) - 1, -1, -1):
  # 得到1列的所有元素
  col_list = [line[col] for line in current_block]
  # 判斷是否碰到右邊界
  if 'O' in col_list and current_block_start_col + col >= BLOCK_COL_NUM:
   return False
 return True


def judge_move_left(current_block, current_block_start_col):
 """
 判斷是否可以向左移動
 """
 # 先判斷列的方式是從左到右
 for col in range(len(current_block[0])):
  # 得到1列的所有元素
  col_list = [line[col] for line in current_block]
  # 判斷是否碰到右邊界
  if 'O' in col_list and current_block_start_col + col  0:
   return False
 return True


def judge_move_down(current_block, current_block_start_row, current_block_start_col, stop_all_block_list):
 """
 判斷是否碰撞到其它圖形或者底邊界
 """
 # 得到其它圖形所有的坐標
 stop_all_block_position = list()
 for row, line in enumerate(stop_all_block_list):
  for col, block in enumerate(line):
   if block != ".":
    stop_all_block_position.append((row, col))
 # print(stop_all_block_position)

 # 判斷碰撞
 for row, line in enumerate(current_block):
  if 'O' in line and current_block_start_row + row >= BLOCK_ROW_NUM:
   # 如果當前行有0，且從起始行開始算+當前顯示的行，超過了總行數，那么就認為碰到了底部
   return False
  for col, block in enumerate(line):
   if block != "." and (current_block_start_row + row, current_block_start_col + col) in stop_all_block_position:
    return False

 return True


def get_block():
 """
 創建一個圖形
 """
 block_style_list = random.choice([block_s, block_i, block_j, block_l, block_o, block_t, block_z])
 return random.choice(block_style_list)


def main():
 pygame.init()
 screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
 pygame.display.set_caption('俄羅斯方塊')

 current_block = get_block() # 當前圖形
 current_block_start_row = -2 # 當前圖片從哪一行開始顯示圖形
 current_block_start_col = 4 # 當前圖形從哪一列開始顯示
 next_block = get_block() # 下一個圖形
 last_time = time.time()
 speed = 0.5 # 降落的速度
 speed_info = '1' # 顯示的速度等級

 # 定義一個列表，用來存儲所有的已經停止移動的形狀
 stop_all_block_list = [['.' for i in range(BLOCK_COL_NUM)] for j in range(BLOCK_ROW_NUM)]

 # 字體
 font = pygame.font.Font('yh.ttf', 24) # 黑體24
 game_over_font = pygame.font.Font("yh.ttf", 72)
 game_over_font_width, game_over_font_height = game_over_font.size('GAME OVER')
 game_again_font_width, game_again_font_height = font.size('鼠標點擊任意位置，再來一局')

 # 得分
 score = 0

 # 標記游戲是否結束
 game_over = False

 # 創建計時器（防止while循環過快，占用太多CPU的問題）
 clock = pygame.time.Clock()
 while True:
  for event in pygame.event.get():
   if event.type == pygame.QUIT:
    sys.exit()
   elif event.type == pygame.KEYDOWN:
    if event.key == pygame.K_LEFT:
     if judge_move_left(current_block, current_block_start_col - 1):
      current_block_start_col -= 1
    elif event.key == pygame.K_RIGHT:
     if judge_move_right(current_block, current_block_start_col + 1):
      current_block_start_col += 1
    elif event.key == pygame.K_UP:
     current_block_next_style = change_current_block_style(current_block)
     if judge_move_left(current_block_next_style, current_block_start_col) and \

       judge_move_right(current_block_next_style, current_block_start_col) and \

       judge_move_down(current_block, current_block_start_row, current_block_start_col, stop_all_block_list):
      # 判斷新的樣式沒有越界
      current_block = current_block_next_style
    elif event.key == pygame.K_DOWN:
     # 判斷是否可以向下移動，如果碰到底部或者其它的圖形就不能移動了
     if judge_move_down(current_block, current_block_start_row + 1, current_block_start_col, stop_all_block_list):
      current_block_start_row += 1
   elif event.type == pygame.MOUSEBUTTONDOWN and event.button:
    if game_over:
     # 重置游戲用到的變量
     current_block = get_block() # 當前圖形
     current_block_start_row = -2 # 當前圖片從哪一行開始顯示圖形
     current_block_start_col = 4 # 當前圖形從哪一列開始顯示
     next_block = get_block() # 下一個圖形
     stop_all_block_list = [['.' for i in range(BLOCK_COL_NUM)] for j in range(BLOCK_ROW_NUM)]
     score = 0
     game_over = False

  # 判斷是否修改當前圖形顯示的起始行
  if not game_over and time.time() - last_time > speed:
   last_time = time.time()
   # 判斷是否可以向下移動，如果碰到底部或者其它的圖形就不能移動了
   if judge_move_down(current_block, current_block_start_row + 1, current_block_start_col, stop_all_block_list):
    current_block_start_row += 1
   else:
    # 將這個圖形存儲到統一的列表中，這樣便于判斷是否成為一行
    add_to_stop_all_block_list(stop_all_block_list, current_block, current_block_start_row, current_block_start_col)
    # 判斷是否有同一行的，如果有就消除，且加上分數
    score += judge_lines(stop_all_block_list)
    # 判斷游戲是否結束（如果第一行中間有O那么就表示游戲結束）
    game_over = judge_game_over(stop_all_block_list)
    # 調整速度
    speed_info, speed = change_speed(score)
    # 創建新的圖形
    current_block = next_block
    next_block = get_block()
    # 重置數據
    current_block_start_col = 4
    current_block_start_row = -2

  # 畫背景（填充背景色）
  screen.fill(BG_COLOR)

  # 畫游戲區域分隔線
  pygame.draw.line(screen, (100, 40, 200), (SIZE * BLOCK_COL_NUM, 0), (SIZE * BLOCK_COL_NUM, SCREEN_HEIGHT), BORDER_WIDTH)

  # 顯示當前圖形
  for row, line in enumerate(current_block):
   for col, block in enumerate(line):
    if block != '.':
     pygame.draw.rect(screen, (20, 128, 200), ((current_block_start_col + col) * SIZE, (current_block_start_row + row) * SIZE, SIZE, SIZE), 0)

  # 顯示所有停止移動的圖形
  for row, line in enumerate(stop_all_block_list):
   for col, block in enumerate(line):
    if block != '.':
     pygame.draw.rect(screen, (20, 128, 200), (col * SIZE, row * SIZE, SIZE, SIZE), 0)

  # 畫網格線 豎線
  for x in range(BLOCK_COL_NUM):
   pygame.draw.line(screen, (0, 0, 0), (x * SIZE, 0), (x * SIZE, SCREEN_HEIGHT), 1)
  # 畫網格線 橫線
  for y in range(BLOCK_ROW_NUM):
   pygame.draw.line(screen, (0, 0, 0), (0, y * SIZE), (BLOCK_COL_NUM * SIZE, y * SIZE), 1)

  # 顯示右側（得分、速度、下一行圖形）
  # 得分
  score_show_msg = font.render('得分: ', True, (255, 255, 255))
  screen.blit(score_show_msg, (BLOCK_COL_NUM * SIZE + 10, 10))
  score_show_msg = font.render(str(score), True, (255, 255, 255))
  screen.blit(score_show_msg, (BLOCK_COL_NUM * SIZE + 10, 50))
  # 速度
  speed_show_msg = font.render('速度: ', True, (255, 255, 255))
  screen.blit(speed_show_msg, (BLOCK_COL_NUM * SIZE + 10, 100))
  speed_show_msg = font.render(speed_info, True, (255, 255, 255))
  screen.blit(speed_show_msg, (BLOCK_COL_NUM * SIZE + 10, 150))
  # 下一個圖形（文字提示）
  next_style_msg = font.render('下一個: ', True, (255, 255, 255))
  screen.blit(next_style_msg, (BLOCK_COL_NUM * SIZE + 10, 200))
  # 下一個圖形（圖形）
  for row, line in enumerate(next_block):
   for col, block in enumerate(line):
    if block != '.':
     pygame.draw.rect(screen, (20, 128, 200), (320 + SIZE * col, (BLOCK_COL_NUM + row) * SIZE, SIZE, SIZE), 0)
     # 顯示這個方格的4個邊的顏色
     # 左
     pygame.draw.line(screen, (0, 0, 0), (320 + SIZE * col, (BLOCK_COL_NUM + row) * SIZE), (320 + SIZE * col, (BLOCK_COL_NUM + row + 1) * SIZE), 1)
     # 上
     pygame.draw.line(screen, (0, 0, 0), (320 + SIZE * col, (BLOCK_COL_NUM + row) * SIZE), (320 + SIZE * (col + 1), (BLOCK_COL_NUM + row) * SIZE), 1)
     # 下
     pygame.draw.line(screen, (0, 0, 0), (320 + SIZE * col, (BLOCK_COL_NUM + row + 1) * SIZE), (320 + SIZE * (col + 1), (BLOCK_COL_NUM + row + 1) * SIZE), 1)
     # 右
     pygame.draw.line(screen, (0, 0, 0), (320 + SIZE * (col + 1), (BLOCK_COL_NUM + row) * SIZE), (320 + SIZE * (col + 1), (BLOCK_COL_NUM + row + 1) * SIZE), 1)

  # 顯示游戲結束畫面
  if game_over:
   game_over_tips = game_over_font.render('GAME OVER', True, RED)
   screen.blit(game_over_tips, ((SCREEN_WIDTH - game_over_font_width) // 2, (SCREEN_HEIGHT - game_over_font_height) // 2))
   # 顯示"鼠標點擊任意位置，再來一局"
   game_again = font.render('鼠標點擊任意位置，再來一局', True, RED)
   screen.blit(game_again, ((SCREEN_WIDTH - game_again_font_width) // 2, (SCREEN_HEIGHT - game_again_font_height) // 2 + 80))

  # 刷新顯示（此時窗口才會真正的顯示）
  pygame.display.update()
  # FPS（每秒鐘顯示畫面的次數）
  clock.tick(60) # 通過一定的延時，實現1秒鐘能夠循環60次


if __name__ == '__main__':
 main()

# S形方塊
block_s = [['.OO',
   'OO.',
   '...'],
   ['O..',
   'OO.',
   '.O.']]
# Z形方塊
block_z = [['OO.',
   '.OO',
   '...'],
   ['.O.',
   'OO.',
   'O..']]
# I型方塊
block_i = [['.O..',
   '.O..',
   '.O..',
   '.O..'],
   ['....',
   '....',
   'OOOO',
   '....']]
# O型方塊
block_o = [['OO',
   'OO']]
# J型方塊
block_j = [['O..',
   'OOO',
   '...'],
   ['.OO',
   '.O.',
   '.O.'],
   ['...',
   'OOO',
   '..O'],
   ['.O.',
   '.O.',
   'OO.']]
# L型方塊
block_l = [['..O',
   'OOO',
   '...'],
   ['.O.',
   '.O.',
   '.OO'],
   ['...',
   'OOO',
   'O..'],
   ['OO.',
   '.O.',
   '.O.']]
# T型方塊
block_t = [['.O.',
   'OOO',
   '...'],
   ['.O.',
   '.OO',
   '.O.'],
   ['...',
   'OOO',
   '.O.'],
   ['.O.',
   'OO.',
   '.O.']]