每日一个小题

import pygame

import random

# 初始化 Pygame

pygame.init()

# 屏幕大小

screen_width = 300

screen_height = 600

block_size = 30

# 颜色定义

colors = [

(0, 0, 0),

(255, 0, 0),

(0, 150, 0),

(0, 0, 255),

(255, 120, 0),

(255, 255, 0),

(180, 0, 255),

(0, 220, 220)

]

# 形状定义

shapes = [

[[1, 1, 1],

[0, 1, 0]],

[[0, 2, 2],

[2, 2, 0]],

[[3, 3, 0],

[0, 3, 3]],

[[4, 4],

[4, 4]],

[[0, 5, 0],

[5, 5, 5]],

[[6, 6, 6, 6]],

[[7, 7],

[7, 7]]

]

class Tetris:

def __init__(self):

self.width = screen_width // block_size

self.height = screen_height // block_size

self.board = [[0 for x in range(self.width)] for y in range(self.height)]

self.score = 0

self.gameover = False

self.current_piece = self.new_piece()

self.next_piece = self.new_piece()

self.clock = pygame.time.Clock()

self.fall_time = 0

self.level = 1

self.lines = 0

def new_piece(self):

shape = random.choice(shapes)

color = colors[shapes.index(shape) + 1]

return {'shape': shape, 'color': color, 'x': self.width // 2 - len(shape[0]) // 2, 'y': 0}

def rotate_piece(self):

piece = self.current_piece['shape']

new_piece = list(zip(*piece[::-1]))

self.current_piece['shape'] = new_piece

def valid_space(self, piece, offset):

off_x, off_y = offset

for y, row in enumerate(piece):

for x, cell in enumerate(row):

if cell and (x + off_x < 0 or x + off_x >= self.width or y + off_y >= self.height or self.board[y + off_y][x + off_x]):

return False

return True

def freeze(self):

piece = self.current_piece['shape']

for y, row in enumerate(piece):

for x, cell in enumerate(row):

if cell:

self.board[y + self.current_piece['y']][x + self.current_piece['x']] = self.current_piece['color']

self.clear_lines()

self.current_piece = self.next_piece

self.next_piece = self.new_piece()

if not self.valid_space(self.current_piece['shape'], (self.current_piece['x'], self.current_piece['y'])):

self.gameover = True

def clear_lines(self):

lines = 0

for i, row in enumerate(self.board):

if all(row):

del self.board[i]

self.board.insert(0, [0 for _ in range(self.width)])

lines += 1

self.score += lines ** 2 * 100

self.lines += lines

if self.lines >= 10:

self.level += 1

self.lines -= 10

pygame.time.set_timer(pygame.USEREVENT+1, max(100, 1000 - (self.level-1)*100))

def drop(self):

self.current_piece['y'] += 1

if not self.valid_space(self.current_piece['shape'], (self.current_piece['x'], self.current_piece['y'])):

self.current_piece['y'] -= 1

self.freeze()

def move(self, dx):

self.current_piece['x'] += dx

if not self.valid_space(self.current_piece['shape'], (self.current_piece['x'], self.current_piece['y'])):

self.current_piece['x'] -= dx

def draw_grid(self, surface):

for y in range(self.height):

for x in range(self.width):

pygame.draw.rect(surface, colors[self.board[y][x]], (x*block_size, y*block_size, block_size, block_size), 0)

pygame.draw.rect(surface, (255, 255, 255), (x*block_size, y*block_size, block_size, block_size), 1)

def draw_piece(self, surface):

shape = self.current_piece['shape']

for y, row in enumerate(shape):

for x, cell in enumerate(row):

if cell:

pygame.draw.rect(surface, self.current_piece['color'], ((self.current_piece['x'] + x) * block_size, (self.current_piece['y'] + y) * block_size, block_size, block_size), 0)

pygame.draw.rect(surface, (255, 255, 255), ((self.current_piece['x'] + x) * block_size, (self.current_piece['y'] + y) * block_size, block_size, block_size), 1)

def run(self):

screen = pygame.display.set_mode((screen_width, screen_height))

pygame.display.set_caption('Tetris')

pygame.time.set_timer(pygame.USEREVENT+1, 1000)

while not self.gameover:

for event in pygame.event.get():

if event.type == pygame.QUIT:

pygame.quit()

return

elif event.type == pygame.KEYDOWN:

if event.key == pygame.K_LEFT:

self.move(-1)