chess/pieces_logic.py at master · Davidooff/chess · GitHub

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
def get_enemy_color(color):
    if color == 'white':
        enemy_color = 'black'
    else:
        enemy_color = 'white'
    return enemy_color


def check_vertical_and_horizontal(x, y, color, board):
    enemy_color = get_enemy_color(color)

    addHighlitesDict = []
    x_temp = x
    while x_temp < 8: # check on x on board (from x to 8)
        x_temp += 1
        if "square-" + str(x_temp) + str(y) in board[color]:
            break
        elif "square-" + str(x_temp) + str(y) in board[enemy_color]:
            addHighlitesDict.append({'x': x_temp, 'y':y})
            break
    x_temp = x
    while x_temp > 0: # check on x on board (from x to 0)
        x_temp -= 1
        if "square-" + str(x_temp) + str(y) in board[color]:
            break
        elif "square-" + str(x_temp) + str(y) in board[enemy_color]:
            addHighlitesDict.append({'x': x_temp, 'y':y})
            break

    y_temp = y
    while y_temp < 8: # check on y on board (from y to 8)
        y_temp += 1
        if "square-" + str(x) + str(y_temp) in board[color]:
            break
        elif "square-" + str(x) + str(y_temp) in board[enemy_color]:
            addHighlitesDict.append({'x': x, 'y':y_temp})
            break
    y_temp = y
    while y_temp > 0: # check on y on board (from y to 0)
        y_temp -= 1
        if "square-" + str(x) + str(y_temp) in board[color]:
            break
        elif "square-" + str(x) + str(y_temp) in board[enemy_color]:
            addHighlitesDict.append({'x': x, 'y':y_temp})
            break
    return addHighlitesDict

def check_diagonal(x,y, color, board):
    enemy_color = get_enemy_color(color)
    addHighlitesDict = []
    temp_x = x
    temp_y = y
    while temp_x < 8 and temp_y < 8: # check on x,y on board (from x,y to 8)
        temp_x += 1
        temp_y += 1
        if "square-" + str(temp_x) + str(temp_y) in board[color]:
            break
        elif "square-" + str(temp_x) + str(temp_y) in board[enemy_color]:
            addHighlitesDict.append({'x': temp_x, 'y':temp_y})
            break
    temp_x = x
    temp_y = y
    while temp_x < 8 and temp_y > 0: # check on x,y on board (from x to 8, y to 0)
        temp_x += 1
        temp_y -= 1
        if "square-" + str(temp_x) + str(temp_y) in board[color]:
            break
        elif "square-" + str(temp_x) + str(temp_y) in board[enemy_color]:
            addHighlitesDict.append({'x': temp_x, 'y':temp_y})
            break

    temp_x = x
    temp_y = y
    while temp_x > 0 and temp_y > 0: # check on x,y on board (from x, y to 0)
        temp_x -= 1
        temp_y -= 1
        if "square-" + str(temp_x) + str(temp_y) in board[color]:
            break
        elif "square-" + str(temp_x) + str(temp_y) in board[enemy_color]:
            addHighlitesDict.append({'x': temp_x, 'y':temp_y})
            break

    temp_x = x
    temp_y = y
    while temp_x > 0 and temp_y < 8: # check on x,y on board (from x to 0, y to 8)
        temp_x -= 1
        temp_y += 1
        if "square-" + str(temp_x) + str(temp_y) in board[color]:
            break
        elif "square-" + str(temp_x) + str(temp_y) in board[enemy_color]:
            addHighlitesDict.append({'x': temp_x, 'y':temp_y})
            break
    return addHighlitesDict

def rook(x,y, color, board):
    return check_vertical_and_horizontal(x, y, color, board)

def queen(x,y, color, board):
    addHighlightsDict = []
    addHighlightsDict.extend(check_vertical_and_horizontal(x, y, color, board))
    addHighlightsDict.extend(check_diagonal(x, y, color, board))
    return addHighlightsDict

def bishop(x, y, color, board):
    return check_diagonal(x, y, color, board)

def knight(x, y, color, board):
    enemy_color = get_enemy_color(color)
    addHighlitesDict = []
    step_long = 2
    step_short = 1

    x_temp = x + step_long
    y_temp = y + step_short
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x + step_long
    y_temp = y - step_short
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x - step_long
    y_temp = y + step_short
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x - step_long
    y_temp = y - step_short
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})


    x_temp = x + step_short
    y_temp = y + step_long
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x + step_short
    y_temp = y - step_long
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x - step_short
    y_temp = y + step_long
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    x_temp = x - step_short
    y_temp = y - step_long
    if "square-" + str(x_temp) + str(y_temp) in board[enemy_color]:
        addHighlitesDict.append({'x': x_temp, 'y':y_temp})

    return addHighlitesDict