-
Notifications
You must be signed in to change notification settings - Fork 4
/
utils.py
82 lines (73 loc) · 3.26 KB
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# -*- coding: utf-8 -*-
# This file is part of the gunyanza.
# Copyright (C) 2014- Erik Bernhardsson
# Copyright (C) 2015- Tasuku SUENAGA <[email protected]>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from __future__ import unicode_literals, print_function
import numpy
import shogi
def board2arrays(b, flip=False):
# 9 * 9 + 持ち駒 7 types * 2 color
x = numpy.zeros(9 * 9 + 7 * 2, dtype=numpy.int8)
# 駒をマスに配置
for pos, piece in enumerate(b.pieces):
if piece != shogi.NONE:
color = int(bool(b.occupied[shogi.BLACK] & shogi.BB_SQUARES[pos]))
col = int(pos % 9)
row = int(pos / 9)
if flip:
row = 8 - row
color = 1 - color
piece = color * len(shogi.PIECE_TYPES) + piece
x[row * 9 + col] = piece
# 持ち駒それぞれの数を保存
for color in shogi.COLORS:
for piece_type in range(shogi.PAWN, shogi.KING):
# 駒のタイプは1オリジンなので1引く
x[81 + color * 7 + (piece_type - 1)] = b.pieces_in_hand[color].get(piece_type, 0)
return x
def boardarrays2bitarray(board_arrays):
# 駒は、1-14(先手)と16-29(後手)。
# それぞれの駒の数値かどうかを0, 1の配列にして、hstackでつなげる。
return numpy.hstack(numpy.array([
(board_arrays[...,:81]==1).astype(int),
(board_arrays[...,:81]==2).astype(int),
(board_arrays[...,:81]==3).astype(int),
(board_arrays[...,:81]==4).astype(int),
(board_arrays[...,:81]==5).astype(int),
(board_arrays[...,:81]==6).astype(int),
(board_arrays[...,:81]==7).astype(int),
(board_arrays[...,:81]==8).astype(int),
(board_arrays[...,:81]==9).astype(int),
(board_arrays[...,:81]==10).astype(int),
(board_arrays[...,:81]==11).astype(int),
(board_arrays[...,:81]==12).astype(int),
(board_arrays[...,:81]==13).astype(int),
(board_arrays[...,:81]==14).astype(int),
(board_arrays[...,:81]==16).astype(int),
(board_arrays[...,:81]==17).astype(int),
(board_arrays[...,:81]==18).astype(int),
(board_arrays[...,:81]==19).astype(int),
(board_arrays[...,:81]==20).astype(int),
(board_arrays[...,:81]==21).astype(int),
(board_arrays[...,:81]==22).astype(int),
(board_arrays[...,:81]==23).astype(int),
(board_arrays[...,:81]==24).astype(int),
(board_arrays[...,:81]==25).astype(int),
(board_arrays[...,:81]==26).astype(int),
(board_arrays[...,:81]==27).astype(int),
(board_arrays[...,:81]==28).astype(int),
(board_arrays[...,:81]==29).astype(int),
]))