Python的俄罗斯方块源码

#: pip install pygameimport randomimport sysimport pygame
#: 颜色定义COLOR_WHITE = (255, 255, 255)COLOR_BLACK = (0, 0, 0)
class Block:    """小块"""    width = 24    height = 24
    @staticmethod    def draw(s, left, top, color, bg_color):        pygame.draw.rect(s, bg_color, pygame.Rect(left, top, Block.width, Block.height))        pygame.draw.rect(s, color, pygame.Rect(left, top, Block.width - 1, Block.height - 1))

class Building:    """积木"""
    def __init__(self):        """        方块的7种基本形状        每次初始化随机选择一个形状        @:return True / False        """        self.form = random.choice(            [                [                    [0, 0, 0, 0, 0],                    [0, 0, 1, 0, 0],                    [0, 1, 1, 1, 0],                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0],                    [1, 1, 1, 1, 0],                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 1, 1, 0, 0],                    [0, 0, 1, 1, 0],                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 0, 1, 1, 0],                    [0, 1, 1, 0, 0],                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 1, 1, 0, 0],                    [0, 0, 1, 0, 0],                    [0, 0, 1, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 0, 1, 1, 0],                    [0, 0, 1, 0, 0],                    [0, 0, 1, 0, 0],                    [0, 0, 0, 0, 0]                ],                [                    [0, 0, 0, 0, 0],                    [0, 1, 1, 0, 0],                    [0, 1, 1, 0, 0],                    [0, 0, 0, 0, 0],                    [0, 0, 0, 0, 0]                ]            ])
    def __getitem__(self, pos):        return self.form[pos]
    def __setitem__(self, key, value):        self.form[key] = value

class Layout:    """棋盘"""
    def __init__(self):        self.block_x_count = 16;        self.block_y_count = 22;        self.layout = [[0 if 1 < i < self.block_x_count - 2 and j < self.block_y_count - 2 else 1                        for i in range(self.block_x_count)] for j in range(self.block_y_count)]
    @property    def size(self):        """返回棋盘屏幕大小(width,height)"""        return (self.block_x_count * Block.width, self.block_y_count * Block.height)
    def create_new_building(self):        """        创建新的积木,初始化位置为第5,0格, 速度为4        :return: 返回是否无空间创建了        """        self.building = Building()        self.building_left, self.building_top = 5, 0  #        self.drop_speed = 3        print(self.test_building_touch_wall())        return self.test_building_touch_wall()        @property    def speed(self):        return self.drop_speed
    def test_building_touch_wall(self, x_offset=0, y_offset=0):        """        积木是否已经触底/墙壁        具体操作：        判断积木最后一排的1，是否在当前棋牌对应的位置是也是1        @:param x_offset: x的偏移量  移动时可以传入1/-1来判断        @:param y_offset: y的偏移量  正常下落时可以传入1来判断        """        for i in range(4, -1, -1):            for j in range(5):                if self.building[i][j]:                    if self.layout[i + self.building_top + y_offset][j + self.building_left + x_offset]:                        return True        return False
    def move_left_right(self, x):        """        左右移动        @:param x:  移动量 x_offset        """        #: 移动时不能撞墙        if not self.test_building_touch_wall(x_offset=x):            self.building_left += x
    def down_build(self):        """ 盒子的自动下移 """        self.building_top += 1
    def direct_down(self):        """ 手动快速降落 """        self.drop_speed = 50
    def convert_building(self):        """        * 扭转盒子的总方位 (右转)        具体操作：        把第一竖排的倒序给第一横排的        把第二竖排的倒序给第二横排的        后面同理.        """        new_box = [[0 for i in range(5)] for j in range(5)]        for i in range(5):            for j in range(4, -1, -1):                new_box[i][j] = self.building[4 - j][i]        self.building = new_box
    def clear_full_lines(self):        """消除满行的所有行"""        new_layout = [[0 if 1 < i < self.block_x_count - 2 and j < self.block_y_count - 2 else 1                       for i in range(self.block_x_count)] for j in range(self.block_y_count)]
        row_len = self.block_x_count - 4        new_row = self.block_y_count - 2 - 1        for cur_row in range(self.block_y_count - 2 - 1, 0, -1):            if sum(self.layout[cur_row][2:self.block_x_count - 2]) < row_len:                new_layout[new_row] = self.layout[cur_row]                new_row -= 1        self.layout = new_layout
    def put_building_to_layout(self):        """将积木放到棋盘里"""        for i in range(4, -1, -1):            for j in range(5):                if self.building[i][j]:                    self.layout[i + self.building_top][j + self.building_left] = 1        #: 这里会调用消除函数        self.clear_full_lines()
    def draw_building(self, s):        """        显示积木        @:param s : pygame = screen         """        cur_left, cur_top = self.building_left * Block.width, self.building_top * Block.height        for i in range(5):            for j in range(5):                # 只画积木实体，不管盒子本身                if self.building[j][i]:                    Block.draw(s, cur_left + i * Block.width, cur_top + j * Block.height, COLOR_BLACK, COLOR_WHITE)
    def draw(self, s):        """        显示棋盘        @:param s : pygame = screen         """        for i in range(self.block_x_count):            for j in range(self.block_y_count):                if self.layout[j][i] == 0:                    Block.draw(s, i * Block.width, j * Block.height, COLOR_WHITE, COLOR_BLACK)                else:                    Block.draw(s, i * Block.width, j * Block.height, COLOR_BLACK, COLOR_WHITE)

# -------------------------------------------------------------------# Main# -------------------------------------------------------------------def main():    #: 初始化    while True:        layout = Layout()        layout.create_new_building()        pygame.init()        pygame.display.set_caption('俄罗斯方块')        screen = pygame.display.set_mode((layout.size), 0, 32)        is_over = False        #: 单局游戏循环开始 [结束后直接重新开始]        while not is_over:            #: 处理游戏消息            for e in pygame.event.get():                if e.type == pygame.QUIT:                    sys.exit()                #: 处理按键                if e.type == pygame.KEYDOWN:                    if e.key == pygame.K_UP:                        layout.convert_building()                    if e.key == pygame.K_DOWN:                        layout.direct_down()                    if e.key == pygame.K_LEFT:                        layout.move_left_right(-1)                    if e.key == pygame.K_RIGHT:                        layout.move_left_right(1)            #: 是否碰触底部地面了，是 -> 融合背景   否 -> 继续下落            if layout.test_building_touch_wall(y_offset=1):                layout.put_building_to_layout()                is_over = layout.create_new_building()            else:                layout.down_build()            #: 绘制            layout.draw(screen)            layout.draw_building(screen)            pygame.display.update()            #: 速度            pygame.time.Clock().tick(layout.speed)

if __name__ == '__main__':    main()