play.py

import pygame
import sys
import random
from field import Field
from tetrimino import Tetrimino
from config import BLOCK_SIZE, BLOCK_IMG_SIZE


def make_colors():
    """7色のブロックをそれぞれ抽出する関数

    tile.png に 40x280の画像があり、40x40で切り出し、最終的には20x20でリサイズ
    :return: 7色のブロック画像配列 (pygame.Surface オブジェクト)
    """
    colors = []
    image = pygame.image.load("./tile.png")
    for i in range(7):
        tmp_color_surface = pygame.Surface((BLOCK_IMG_SIZE, BLOCK_IMG_SIZE))
        color_surface = pygame.Surface((BLOCK_SIZE, BLOCK_SIZE))
        tmp_color_surface.blit(image, (0, 0), (BLOCK_IMG_SIZE * i, 0, BLOCK_IMG_SIZE, BLOCK_IMG_SIZE))
        pygame.transform.scale(tmp_color_surface, (BLOCK_SIZE, BLOCK_SIZE), color_surface)
        colors.append(color_surface)
    return colors


def generate_tetrimino():
    """テトリミノをランダムに1つ生成する関数

    :return: ランダムに生成されたTetrimino インスタンス
    """
    random.seed()
    return Tetrimino(5, 2, 0, random.randrange(0, 7))


def play():
    """ Play関数

    キーボード操作でテトリスを遊べる関数
    :return:
    """
    pygame.init()  # pygameを初期化数する
    screen = pygame.display.set_mode((640, 480))  # 画面サイズを640, 480に
    pygame.display.set_caption("tetris_ai")
    clock = pygame.time.Clock()  # フレームレート(描画頻度)を更新するためのタイマー
    font = pygame.font.SysFont("Arial", 20)

    # 7色ブロックの準備
    colors = make_colors()

    # 盤面の定義
    field = Field()

    # 1つめ(初期)のTetrimino(mino)
    mino = generate_tetrimino()

    # minoの落下が確定して次のminoを生成する必要がある場合にTrueとする
    generate_flag = False

    # ゲームオーバの場合にTrue
    game_over_flag = False

    # トータル経過フレーム数
    num_frame = 0

    # 得点(消したライン数)
    score = 0
    while True:   # ゲームのメインループ
        num_frame += 1
        screen.fill((0, 0, 0))  # マイループで画面を真っ黒に一度する

        # 盤面の描画
        field.draw(screen, colors)
        mino.draw(screen, colors)

        moved_flag = False # キー操作等によりminoが移動したことを表すflag
        for event in pygame.event.get():  # イベントは一括で取得
            if event.type == pygame.QUIT:  # 終了ボタンを押したときにpygameを閉じるのはお作法
                pygame.quit()
                sys.exit()
            if event.type == pygame.KEYDOWN:  # これよりキーボードのイベントを捕捉
                if event.key == pygame.K_RIGHT:  # 右が押された場合...
                    next_mino = mino.clone(1, 0, 0)  # 右に移動したminoを仮に作成
                    if not next_mino.collision(field):  # 当たり判定
                        mino = next_mino  # 当たり判定無しであれば移動確定
                        moved_flag = True
                if event.key == pygame.K_LEFT:
                    next_mino = mino.clone(-1, 0, 0)  # 左に移動したminoを仮に作成(以下同様)
                    if not next_mino.collision(field):
                        mino = next_mino
                        moved_flag = True
                if event.key == pygame.K_UP:
                    next_mino = mino.clone(0, 0, 1)  # 1回転したminoを仮に作成(時計回り)(以下同様)
                    if not next_mino.collision(field):
                        mino = next_mino
                        moved_flag = True
                if event.key == pygame.K_DOWN:  # 下に移動(落下)は盤面的に可能なところまで落ちる
                    next_mino = mino.clone(0, 1, 0)
                    while not next_mino.collision(field):  # 1ブロックづつ落下させ当たり判定を繰り返す
                        mino = next_mino
                        next_mino = mino.clone(0, 1, 0)
                if event.key == pygame.K_SPACE:
                    pass

        if game_over_flag:  # game overであれば何もしない
            continue

        if moved_flag:  # キー操作(左右、回転）があった場合は 一旦その移動を完了し落下は次フレームでやる (落下は除く)
            continue

        next_mino = mino.clone(0, 1, 0)  # minoを下に1つずらしてみる
        if next_mino.collision(field):  # 盤面との接触判定
            field.set_blocks(mino.get_blocks())  # 盤面にブロックを置く(落下停止)
            s = field.line_erase()
            if s != 0:
                score += s
                print("Score: {}".format(score))
            generate_flag = True
        else:
            mino = next_mino  # 盤面との接触が無いので下に1つ移動を確定する

        if generate_flag:
            mino = generate_tetrimino()
            if mino.collision(field):
                print("GAME OVER")
                game_over_flag = True
            generate_flag = False

        pygame.display.update()  # 描画を更新する
        clock.tick(3)  # フレームレートを3 (秒間3回更新するぐらいの頻度に設定)


if __name__ == "__main__":
    play()