PyChess/move_sets.py at main · teeedor/PyChess · 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
# MOVESETS
# Basic movements,  no capturing yet
def pawn_ms(color, capture, x1, y1, x2, y2):
    # print("moving a pawn")
    valid = False

    # White Movement
    if (color == 'w') and (y1 == 2) and (y1 == y2-2):  # special first movement
        valid = True
    if (color == 'w') and (y1 == (y2-1)) and (x1 == x2):  # regular movement
        valid = True
    # Capturing Case
    if (color == 'w') and capture and (y1 == (y2-1)) and (x2 == x1-1 or x2 == x1+1):
        valid = True

    # Black Movement
    if (color == 'b') and (y1 == 7) and (y1 == y2+2):  # special first movement
        valid = True
    if (color == 'b') and (y1 == (y2+1)) and (x1 == x2):  # regular movement
        valid = True
    # Capturing Case
    if (color == 'b') and capture and (y1 == (y2+1)) and (x2 == x1-1 or x2 == x1+1):
        valid = True
    return valid


# def rook_ms - Only need start and end positions
def rook_ms(x1, y1, x2, y2):
    # print("moving a rook")
    valid = False
    if x2 - x1 == 0:  # in the same row or column
        valid = True
    if y2 - y1 == 0:  # in the same row or column
        valid = True
    return valid


def horse_ms(x1, y1, x2, y2):
    # print("moving a horse")
    valid = False
    if abs(x1 - x2) == 1:
        if abs(y1 - y2) == 2:
            valid = True
    if abs(x1 - x2) == 2:
        if abs(y1 - y2) == 1:
            valid = True
    return valid


# def bishop_ms -
def bishop_ms(x1, y1, x2, y2):
    # print("moving a bishop")
    valid = False
    xdel = x2 - x1
    ydel = y2 - y1
    # check to see if the deltas are the same magnitude. if so it's a valid move
    if abs(xdel) == abs(ydel):
        valid = True
    return valid


def king_ms(x1, y1, x2, y2):
    # print("moving a king")
    valid = False
    if (x2 == x1-1 or x2 == x1+1 or x2 == x1) and (y2 == y1-1 or y2 == y1+1 or y2 == y1):
        valid = True
    return valid


# combination of rook and Bishop movesets
def queen_ms(x1, y1, x2, y2):
    # print("moving a queen")
    valid = False
    valid_r = rook_ms(x1, y1, x2, y2)
    valid_b = bishop_ms(x1, y1, x2, y2)
    if valid_b or valid_r:
        valid = True
    return valid