ガイスターアプリ開発(4) 初期盤面

Python でつくるガイスター、連載第 4 回です。

はマウス操作で駒の初期配置を決められるようになりました。

今回はゲーム本番での初期盤面を描画するところまでやっていきます。

モジュールに切り分ける

駒の初期配置を決定するモジュールと、ゲーム本番での動作を規定するモジュールを分けて書きたいので、

共通部分だけmain.pyに置いて、そこから各モジュールを呼び出す形にします。

setup.py
1
import sys
2

3
import pygame
4
from pygame.locals import *
5

6
from config import IVORY
7
import draw, mouse
8

9

10
def _check_color(colors):
11
    '''
12
    駒がすべて配置済みで赤と青が半分ずつあるか
13
    -> bool
14

15
    colors : list <- [str]
16
        色のリスト
17
    '''
18
    return (len(colors) == 8
19
        and len([s for s in colors if s == 'R'])
20
            == len([s for s in colors if s == 'B']))
21

22

23
def main(screen, font, select_snd, decide_snd, forbid_snd):
24
    # 配置を決定する側
25
    # 0 - 先攻, 1 - 後攻
26
    _state = 0
27
    # 決定された初期配置
28
    # [{(int, int): Piece}]
29
    _order = [{}, {}]
30

31
    while True:
32
        satisfied = _check_color(list(_order[_state].values()))
33

34
        screen.fill(IVORY)
35
        draw.setup(screen, font, _state, _order[_state], not satisfied)
36
        pygame.display.update()
37

38
        # イベントハンドリング
39
        for event in pygame.event.get():
40
            # 閉じるボタン
41
            if event.type == QUIT:
42
                pygame.quit()
43
                sys.exit()
44
            # マウスクリック
45
            if event.type == MOUSEBUTTONDOWN:
46
                # 左
47
                if event.button == 1:
48
                    _mouse_pos = event.pos
49
                    _square_pos = tuple(mouse.chcoord(_mouse_pos))
50

51
                    for i in range(1, 5):
52
                        for j in range(2, 4):
53
                            if _square_pos == (i, j):
54
                                select_snd.play()
55
                                _order[_state][(i, j)] = 'R'
56
                    
57
                    if mouse.on_area(*_mouse_pos, 500, 530, 80, 50):
58
                        if satisfied:
59
                            decide_snd.play()
60
                            _state += 1
61
                        else:
62
                            forbid_snd.play()
63
                # 右
64
                elif event.button == 3:
65
                    _mouse_pos = event.pos
66
                    _square_pos = tuple(mouse.chcoord(_mouse_pos))
67

68
                    for i in range(1, 5):
69
                        for j in range(2, 4):
70
                            if _square_pos == (i, j):
71
                                select_snd.play()
72
                                _order[_state][(i, j)] = 'B'
73
            # キー
74
            if event.type == KEYDOWN:
75
                # Esc キー
76
                if event.key == K_ESCAPE:
77
                    pygame.quit()
78
                    sys.exit()

game.py
1
import sys
2

3
import pygame
4
from pygame.locals import *
5

6
from config import IVORY
7

8

9
def main(screen):
10
    while True:
11
        screen.fill(IVORY)
12
        pygame.display.update()
13

14
        # イベントハンドリング
15
        for event in pygame.event.get():
16
            # 閉じるボタン
17
            if event.type == QUIT:
18
                pygame.quit()
19
                sys.exit()
20
            # キー
21
            if event.type == KEYDOWN:
22
                # Esc キー
23
                if event.key == K_ESCAPE:
24
                    pygame.quit()
25
                    sys.exit()

main.py
1
import sys
2

3
import pygame
4
from pygame.locals import *
5

6
from config import DISP_SIZE
7
import setup, game
8

9

10
if __name__ == '__main__':
11
    pygame.init()
12
    # 音声の設定
13
    snd = pygame.mixer.Sound
14
    select_snd = snd('./sounds/select.wav')
15
    decide_snd = snd('./sounds/decide.wav')
16
    forbid_snd = snd('./sounds/forbid.wav')
17

18
    pygame.display.set_caption('Geister')
19
    font = pygame.font.SysFont('hg丸ｺﾞｼｯｸmpro', 16)
20

21

22
    setup.main(screen, font, select_snd, decide_snd, forbid_snd)
23

24
    game.main(screen)

setup.pyでのwhileループが終了すればgame.pyでのループが開始します。

駒クラスを作成する

盤面のデータは、位置：駒の辞書型にすることにします。

駒はオブジェクトとして、どちら側の駒か、色は何かなどの属性を持たせます。

piece.py
1
import numpy as np
2

3
class Piece:
4
    def __init__(self, color, side):
5
        self.color = color  # 赤('R') or 青('B')
6
        self.side = side    # 先攻(0) or 後攻(1)
7
    
8
    def __repr__(self):
9
        return self.color + str(self.side)
10
    
11
    def covering_squares(self, pos):
12
        '''
13
        可能な移動先のマスのリスト
14
        -> list <- [(int, int)]
15

16
        pos : tuple <- (int, int)
17
            現在の位置
18
        '''
19
        pos_ = np.asarray(pos) + [(0, 1), (0, -1), (-1, 0), (1, 0)]
20
        dest = [(x, y) for x, y in pos_ if 0 <= x <= 5 and 0 <= y <= 5]
21
        if self.color == 'B':
22
            if self.side == 0:
23
                if pos == (0, 0):
24
                    dest += [(0, -1)]
25
                if pos == (5, 0):
26
                    dest += [(5, -1)]
27
            elif self.side == 1:
28
                if pos == (0, 5):
29
                    dest += [(0, 6)]
30
                if pos == (5, 5):
31
                    dest += [(5, 6)]
32

33
        return dest

__repr__メソッドはデバッグのときに役立ちます。

printしたときに出力される文字列を返しますが、これがないとオブジェクトidだけを返すので属性の情報がわかりづらくなります。

convering_squaresメソッドの条件分岐は、良いおばけが相手の陣地の角から脱出することができるようにするものです。

初期盤面を生成する

プレイヤーの配置から初期盤面を生成する関数を作ります。

setup.py
1
...
2
from piece import Piece
3
...
4

5
def _init_board(order1, order2):
6
    '''
7
    駒の配置から初期盤面を出力
8
    -> dict <- {(int, int): Piece}
9

10
    order1, order2 : dict <- {(int, int): str}
11
        駒の初期配置. order1 が先攻
12
    '''
13
    return {**{(5-x, 3-y): Piece(s, 1) for (x, y), s in order2.items()},
14
        **{(x, y+2): Piece(s, 0) for (x, y), s in order1.items()}}
15

盤面は左上が(0, 0)から右下が(5, 5)までで座標を取っていて、

駒の位置は先手が(1, 4)から(4, 5)、後手が(1, 0)から(4, 1)までとなります。

マウスで登録したときには左上(1, 2)、右下(4, 3)になっているので、座標を変換しています。

先手は平行移動だけですが、後手は180度回転もしています。

これで_init_board(*_order)とすれば、初期盤面を返すことができます。

これをsetup.main()の返り値としてmain.pyに渡し、

さらにgame.pyに引数として渡すことで盤面を描画していきます。

setup.py
1
...
2
        # イベントハンドリング
3
        for event in pygame.event.get():
4
            ...
5
            # マウスクリック
6
            if event.type == MOUSEBUTTONDOWN:
7
                # 左
8
                if event.button == 1:
9
                    ...
10
                    
11
                    if mouse.on_area(*_mouse_pos, 500, 530, 80, 50):
12
                        if satisfied:
13
                            decide_snd.play()
14
                            if _state == 1: return _init_board(*_order)
15
                            _state += 1
16
                        ...

main.py
1
...
2
if __name__ == '__main__':
3
    ...
4
    orders = setup.main(screen, font, select_snd, decide_snd, forbid_snd)
5
    game.main(screen, orders)

game.py
1
...
2
def main(screen, orders):
3
    ...

盤面を描画する

さて、盤面を描画する関数を定義していきます。

ガイスターでは良いおばけが盤の角から外に出ると勝ちというルールがあるので、

角に矢印を描いています。

draw.py
1
...
2
def _arrow(screen, coord, direction):
3
    '''
4
    矢印を描く
5

6
    screen : pygame.display.set_mode
7
    coord : tuple <- (int, int)
8
        矢先の座標
9
    direction : str <- 'U', 'D'
10
        'U' - 上, 'D' - 下
11
    '''
12
    assert direction == 'U' or direction == 'D',\
13
        'draw._arrow の引数 direction は "U", "D" の値を取ります'
14
    _coord = np.asarray(coord)
15
    if direction == 'D':
16
        pygame.draw.line(screen, BLACK,
17
            _coord, _coord + (PIECE_SIZE/2, -PIECE_SIZE/2), 2)
18
        pygame.draw.line(screen, BLACK,
19
            _coord, _coord - (PIECE_SIZE/2, PIECE_SIZE/2), 2)
20
        pygame.draw.line(screen, BLACK,
21
            _coord, _coord - (0, PIECE_SIZE), 2)
22
    else:
23
        pygame.draw.line(screen, BLACK,
24
            _coord, _coord + (PIECE_SIZE/2, PIECE_SIZE/2), 2)
25
        pygame.draw.line(screen, BLACK,
26
            _coord, _coord + (-PIECE_SIZE/2, PIECE_SIZE/2), 2)
27
        pygame.draw.line(screen, BLACK,
28
            _coord, _coord + (0, PIECE_SIZE), 2)
29
...
30
def board(screen, board, turn):
31
    '''
32
    ゲームボードと盤面上の駒を描く
33

34
    screen : pygame.display.set_mode
35
    board : dict <- {(int, int): Piece}
36
        駒の位置とオブジェクト
37
    turn : int <- 0, 1, 2
38
        0 - 先攻の駒を開く, 1 - 後攻の駒を開く, 2 - 両方開く,
39
    '''
40
    assert turn == 0 or turn == 1 or turn == 2, 'draw.board の引数 turn は 0, 1, 2 の値を取ります'
41
    # グリッド
42
    _grid(screen, MARGIN, 6, 6)
43
    # 角の矢印
44
    _padding = (SQUARE_SIZE - PIECE_SIZE)/2
45
    _arrow(screen, MARGIN+(SQUARE_SIZE/2, _padding), 'U')
46
    _arrow(screen, MARGIN+(11*SQUARE_SIZE/2, _padding), 'U')
47
    _arrow(screen, DISP_SIZE-MARGIN-(SQUARE_SIZE/2, _padding), 'D')
48
    _arrow(screen, DISP_SIZE-MARGIN-(11*SQUARE_SIZE/2, _padding), 'D')
49
    # 駒
50
    for pos, piece in board.items():
51
        if turn == 2:
52
            if piece.side == 0:
53
                _piece(screen, RED if piece.color == 'R' else BLUE, pos)
54
            else:
55
                _piece(screen, RED if piece.color == 'R' else BLUE, pos, True)
56
        elif turn == 0:
57
            if piece.side == 0:
58
                _piece(screen, RED if piece.color == 'R' else BLUE, pos)
59
            else:
60
                _piece(screen, GREY, pos, True)
61
        elif turn == 1:
62
            if piece.side == 0:
63
                _piece(screen, GREY, pos)
64
            else:
65
                _piece(screen, RED if piece.color == 'R' else BLUE, pos, True)

BLACKやGREYは色の定数で、config.pyに定義しています。

これをgame.pyで呼び出します。

game.py
1
...
2
import draw
3
...
4
def main(screen, orders):
5
    # ゲームボード {(int, int): Piece}
6
    _board = orders
7

8
    while True:
9
        screen.fill(IVORY)
10
        draw.board(screen, _board, 0)
11
        pygame.display.update()
12
        ...

これで駒の配置を決定したあとに、初期盤面が表示されます。

---

は駒の移動とターン交代時の画面を用意します。

読んでくれてありがとうございました。

では