-
Notifications
You must be signed in to change notification settings - Fork 0
/
helpers.py
157 lines (121 loc) · 3.64 KB
/
helpers.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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
# helper functions
import numpy as np
# 2D board array
# https://www.geeksforgeeks.org/python-using-2d-arrays-lists-the-right-way/
rows, cols = (8, 8)
board = [['--' for i in range(rows)] for j in range(cols)]
savedLocations = {}
map={
'a': 0,
'b': 1,
'c': 2,
'd': 3,
'e': 4,
'f': 5,
'g': 6,
'h': 7
}
# TODO
# use like pieces.KING
pieces = {
'KING': 'K',
'QUEEN': 'Q',
'ROOK': 'R',
'KNIGHT': 'N',
'BISHOP': 'B',
'PAWN': 'p'
}
# can be used several times
colors = {
'WHITE' : 'W',
'BLACK': 'B'
}
# returns True if two given cells are on same diagonal
def checkOnSameDiagonal(previousCell, nextCell):
prevCellCol = map[previousCell[0]]
prevCellRow = int(previousCell[1])
prevCellDiff = prevCellCol - prevCellRow
nextCellCol = map[nextCell[0]]
nextCellRow = int(nextCell[1])
nextCellDiff = nextCellCol - nextCellRow
if (prevCellDiff == nextCellDiff):
return True
return False
# returns True if two given cells are on same row
def checkOnSameRow(previousCell, nextCell):
if(previousCell[1] == nextCell[1]):
return True
return False
# returns True if two given cells are on same column
def checkOnSameCol(previousCell, nextCell):
if (previousCell[0] == nextCell[0]):
return True
return False
def switchColor(color):
if color == colors['WHITE']:
return colors['BLACK']
else:
return colors['WHITE']
def mapCellToIndex(location):
desiredCol = location[:1]
desiredRow = location[1:]
indexRow = int(desiredRow) - 1
indexCol = map[desiredCol]
return [indexRow, indexCol]
def mapIndexToCell(index):
map=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
row=index[0]+1
col=map[index[1]]
return col+str(row)
# return the cell containing a certain piece by searching an entire row
def findCellByRowLocation(board, piece, locationHint):
for i in range(8):
col = i
row = int(locationHint) - 1
cell = mapIndexToCell([row, col])
if(board[row][col] == piece):
return cell
return False
# return the cell containing a certain piece by searching an entire col
def findCellByColLocation(board, piece, locationHint):
for i in range(8):
row= i
col = map[locationHint]
cell = mapIndexToCell([row, col])
if(board[row][col] == piece):
return cell
return False
# return the piece that is on the cell
def getCellValue(board, cell):
index = mapCellToIndex(cell)
return board[index[0]][index[1]]
# return the color of cell
def getCellColor(cell):
index=mapCellToIndex(cell)
if (index[0]+index[1]) % 2 == 0:
return colors['WHITE']
else:
return colors['BLACK']
# return saved locations of the given piece
def findSavedCellLocation(piece):
return savedLocations[piece]
# make cell empty
def emptyCell(previousCell):
previousIndex = mapCellToIndex(previousCell)
piece=board[previousIndex[0]][previousIndex[1]]
# https://stackoverflow.com/questions/10996140/how-to-remove-specific-elements-in-a-numpy-array
a = np.array(savedLocations[piece])
b = np.array([previousCell])
savedLocations[piece] = np.setdiff1d(a, b)
board[previousIndex[0]][previousIndex[1]] = '--'
return
# fill cell with a given piece
def fillCell(nextCell, piece):
nextIndex = mapCellToIndex(nextCell)
board[nextIndex[0]][nextIndex[1]] = piece
# https://stackoverflow.com/questions/9775297/concatenate-a-numpy-array-to-another-numpy-array
a = np.array([nextCell])
b = savedLocations[piece]
savedLocations[piece]=np.concatenate((a, b))
# print(savedLocations)
return