チェスアプリ開発(20) プロモーションの拡張

Python プログラムで動かすフェアリーチェスアプリ開発、連載第 20 回です。

でフェアリーチェスへのキャスリングの拡張が完了しました。

今回はポーンのプロモーションの拡張をしていきます。

拡張前のプロモーションの実装は第11回に載せています。

  関連記事チェスアプリ開発(11) プロモーションの実装

ゲーム毎にプロモーション先候補を指定

まずはゲームによって異なるプロモーション先候補を、各ゲームクラスの

として定義することが必要です。

これはですでに完了しています。

games.py
1
class Normal:
2
    '''通常のチェス'''
3
    # 盤面のサイズ
4
    size = 8
5
    # キャスリングの有無
6
    castling = True
7
    # プロモーション先
8
    promote2 = [Knight, Bishop, Rook, Queen]
9
    # 駒の配置
10
    placers = {1: [Rook, Knight, Bishop, Queen, King, Bishop, Knight, Rook],
11
        2: [Pawn] * size}
12
    # 画像IDの割り当て
13
    ID = {}
14
    for rk in placers:
15
        for fl in range(size):
16
            if placers[rk][fl] is not None:
17
                ID['W' + placers[rk][fl].abbr] = size * rk + fl
18
                ID['B' + placers[rk][fl].abbr] = -(size * rk + fl)

promote2というリストの中からポーンのプロモーション先が選択されます。

プロモーション時に表示する選択肢を一般化

次にプロモーションの選択肢として表示される駒を、このリストに合わせて変更しなければなりません。

現在プロモーション選択肢の表示部分のコードは次のようになっています。

main.py
1
...
2
class Game:
3
    ...
4
    def draw(self):
5
        '''描画コールバック'''
6
        ...
7
            # プロモーション
8
            if self.prom:
9
                draw_balloon(*self.endpos)
10
                piece_color = self.gameboard[self.endpos].color
11
                glEnable(GL_TEXTURE_2D)
12
                draw_img(2.0, 3.5, piece_ID[piece_color + 'N'])
13
                draw_img(3.0, 3.5, piece_ID[piece_color + 'B'])
14
                draw_img(4.0, 3.5, piece_ID[piece_color + 'R'])
15
                draw_img(5.0, 3.5, piece_ID[piece_color + 'Q'])
16
                glDisable(GL_TEXTURE_2D)
17
            ...

12-15行目がプロモーション先を固定してしまっているので、この部分を変更します。

どの位置座標にどの駒の画像を配置するかもゲームによって変化しますので、

表示幅は 4 列までとして行を増やしていく形で描画します。

main.py
1
# プロモーション
2
if self.prom:
3
    draw_balloon(*self.endpos)
4
    piece_color = self.gameboard[self.endpos].color
5
    glEnable(GL_TEXTURE_2D)
6
    draw_img(2.0, 3.5, piece_ID[piece_color + 'N'])
7
    draw_img(3.0, 3.5, piece_ID[piece_color + 'B'])
8
    draw_img(4.0, 3.5, piece_ID[piece_color + 'R'])
9
    draw_img(5.0, 3.5, piece_ID[piece_color + 'Q'])
10
    for i in range(len(self.kind.promote2)):
11
        draw_img(2.0 + i % 4,
12
            3.5 + ((len(self.kind.promote2) - 1)//4)/2 - i//4,
13
            self.kind.ID[piece_color +
14
                self.kind.promote2[i].abbr])
15
    glDisable(GL_TEXTURE_2D)

また、それに伴って吹き出しの縦幅も増やす必要があります。

吹き出しはdraw_balloon関数が描画していますので、そちらの中身を書き換えます。

utils.py
1
def draw_balloon(x, y):
2
def draw_balloon(x, y, num=4):
3
    '''
4
    プロモーションのときの吹き出しを描画する
5

6
    Parameters
7
    ----------
8
    x, y : int
9
        駒の座標．
10
    num : int, default 4
11
        プロモーション選択肢数．
12
    '''
13
    glColor(0.5, 0.5, 0.5)  # 色の指定
14
    glBegin(GL_QUADS)       # 四角形を描画
15
    glVertex(1.0, 2.5)
16
    glVertex(1.0, 4.5)
17
    glVertex(6.0, 4.5)
18
    glVertex(6.0, 2.5)
19
    glVertex(1.0, 2.5 - ((num - 1) // 4) / 2)
20
    glVertex(1.0, 4.5 + ((num - 1) // 4) / 2)
21
    glVertex(2.0 + num if num <= 4 else 6.0, 4.5 + ((num - 1) // 4) / 2)
22
    glVertex(2.0 + num if num <= 4 else 6.0, 2.5 - ((num - 1) // 4) / 2)
23
    glEnd()
24
    glBegin(GL_TRIANGLES)   # 三角形を描画
25
    glVertex(3.0, 3.5)
26
    glVertex(4.0, 3.5)
27
    glVertex(x, y)
28
    glEnd()

glBeginからglEndまでの間の４つのglVertexが四角形の各頂点の位置を指定しています。

新たに追加いた引数num=プロモーション選択肢の数によってこの座標を調整しています。

(num - 1) // 4はnumが4までのとき0、8までのとき1というように増えていくので、

縦方向の座標2.5と4.5を基準に縦幅が増えていくことになります。

横幅は2.0 + num if num <= 4 else 6.0というふうに、選択肢が４つ以下かどうかで変えています。

吹き出し描画部分のコードも変更します。

main.py
1
# プロモーション
2
if self.prom:
3
    draw_balloon(*self.endpos)
4
    draw_balloon(*self.endpos, num=len(self.kind.promote2))
5
    piece_color = self.gameboard[self.endpos].color
6
    glEnable(GL_TEXTURE_2D)
7
    for i in range(len(self.kind.promote2)):
8
        draw_img(2.0 + i % 4,
9
            3.5 + ((len(self.kind.promote2) - 1)//4)/2 - i//4,
10
            self.kind.ID[piece_color +
11
                self.kind.promote2[i].abbr])
12
    glDisable(GL_TEXTURE_2D)

これで通常のゲームのプロモーションでは下のようになり、

ユニコーンが加わったゲーム（promote2 = [Knight, Bishop, Rook, Queen, Unicorn]）では下のようになります。

縦幅が大きくなりました。

マウスクリックを調整

マウスクリックでこの選択肢の中から駒を選択するわけですが、今のコードは次のようになっています。

main.py
1
...
2
class Game:
3
    ...
4
    def mouse(self, button, state, x, y):
5
        '''
6
        マウス入力コールバック
7
        '''
8
        ...
9
            # プロモーション
10
            if self.prom:
11
                piece_color = self.gameboard[self.endpos].color
12
                if on_square(*self.mousepos, 1.5, 2.5, 3.0, 4.0):
13
                    self.gameboard[self.endpos] = Knight(piece_color)
14
                    self.prom = False
15
                if on_square(*self.mousepos, 2.5, 3.5, 3.0, 4.0):
16
                    self.gameboard[self.endpos] = Bishop(piece_color)
17
                    self.prom = False
18
                if on_square(*self.mousepos, 3.5, 4.5, 3.0, 4.0):
19
                    self.gameboard[self.endpos] = Rook(piece_color)
20
                    self.prom = False
21
                if on_square(*self.mousepos, 4.5, 5.5, 3.0, 4.0):
22
                    self.gameboard[self.endpos] = Queen(piece_color)
23
                    self.prom = False
24
            ...

座標も駒も固定になっていますので、これを変更します。

main.py
1
# プロモーション
2
if self.prom:
3
    piece_color = self.gameboard[self.endpos].color
4
    for i in range(len(self.kind.promote2)):
5
        if on_square(*self.mousepos,
6
                        1.5 + i % 4,
7
                        2.5 + i % 4,
8
                        3.0 + ((len(self.kind.promote2) - 1)//4)/2 - i//4,
9
                        4.0 + ((len(self.kind.promote2) - 1)//4)/2 - i//4):
10
            self.gameboard[self.endpos] = self.kind.promote2[i](piece_color)
11
            self.prom = False

on_square関数の最後の４つの引数left, right, bottom, topが有効範囲を指定しています。

この範囲内にself.mousepos=クリック時のマウスカーソル位置があれば、ポーンが指定の駒に置き換わります。

これでプロモーションの拡張が完了しました！

今回の変更も GitHub に上げておきます。

---

これまで20回かけてチェスのコードの編集やフェアリーチェスへの拡張をやってきました。

これにさらにチェックやメイトを知らせるダイアログを表示したり、手数を戻せるようにしたり、

あるいはコンピュータ対戦を導入しても面白いでしょう。

が、いったんこのシリーズは一区切りついたことにしたいと思います。

お読みいただきありがとうございました～

では