Chess/placed_objects.py at master · bradwyatt/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
import pygame
import load_images_sounds as lis
import board

PLACED_SPRITES = pygame.sprite.Group()

def remove_placed_object(mousepos):
    for placed_item_list in (PlacedPawn.white_pawn_list, PlacedBishop.white_bishop_list,
                             PlacedKnight.white_knight_list, PlacedRook.white_rook_list,
                             PlacedQueen.white_queen_list, PlacedKing.white_king_list,
                             PlacedPawn.black_pawn_list, PlacedBishop.black_bishop_list,
                             PlacedKnight.black_knight_list, PlacedRook.black_rook_list,
                             PlacedQueen.black_queen_list, PlacedKing.black_king_list):
        for placed_item in placed_item_list:
            if placed_item.rect.collidepoint(mousepos):
                PLACED_SPRITES.remove(placed_item)
                placed_item_list.remove(placed_item)
    return

def remove_all_placed():
    for spr_list in [PlacedPawn.white_pawn_list, PlacedBishop.white_bishop_list,
                     PlacedKnight.white_knight_list, PlacedRook.white_rook_list,
                     PlacedQueen.white_queen_list, PlacedKing.white_king_list,
                     PlacedPawn.black_pawn_list, PlacedBishop.black_bishop_list,
                     PlacedKnight.black_knight_list, PlacedRook.black_rook_list,
                     PlacedQueen.black_queen_list, PlacedKing.black_king_list]:
        for obj in spr_list:
            obj.kill()
    PlacedPawn.white_pawn_list = []
    PlacedBishop.white_bishop_list = []
    PlacedKnight.white_knight_list = []
    PlacedRook.white_rook_list = []
    PlacedQueen.white_queen_list = []
    PlacedKing.white_king_list = []
    PlacedPawn.black_pawn_list = []
    PlacedBishop.black_bishop_list = []
    PlacedKnight.black_knight_list = []
    PlacedRook.black_rook_list = []
    PlacedQueen.black_queen_list = []
    PlacedKing.black_king_list = []

class PlacedPiece(pygame.sprite.Sprite):
    def __init__(self, piece_type, coord, color, own_list):
        pygame.sprite.Sprite.__init__(self)
        self.color = color
        self.image = lis.IMAGES[f"SPR_{color.upper()}_{piece_type.upper()}"]
        PLACED_SPRITES.add(self)
        own_list.append(self)
        self._own_list = own_list
        self.coordinate = coord
        self.rect = self.image.get_rect()
        self.rect.topleft = board.Grid.grid_dict[self.coordinate].rect.topleft
    def update(self):
        self.rect.topleft = board.Grid.grid_dict[self.coordinate].rect.topleft
    def destroy(self):
        self._own_list.remove(self)
        self.kill()

class PlacedPawn(PlacedPiece):
    white_pawn_list = []
    black_pawn_list = []
    def __init__(self, coord, color):
        lst = PlacedPawn.white_pawn_list if color == "white" else PlacedPawn.black_pawn_list
        super().__init__("pawn", coord, color, lst)

class PlacedBishop(PlacedPiece):
    white_bishop_list = []
    black_bishop_list = []
    def __init__(self, coord, color):
        lst = PlacedBishop.white_bishop_list if color == "white" else PlacedBishop.black_bishop_list
        super().__init__("bishop", coord, color, lst)

class PlacedKnight(PlacedPiece):
    white_knight_list = []
    black_knight_list = []
    def __init__(self, coord, color):
        lst = PlacedKnight.white_knight_list if color == "white" else PlacedKnight.black_knight_list
        super().__init__("knight", coord, color, lst)

class PlacedRook(PlacedPiece):
    white_rook_list = []
    black_rook_list = []
    def __init__(self, coord, color):
        lst = PlacedRook.white_rook_list if color == "white" else PlacedRook.black_rook_list
        super().__init__("rook", coord, color, lst)

class PlacedQueen(PlacedPiece):
    white_queen_list = []
    black_queen_list = []
    def __init__(self, coord, color):
        lst = PlacedQueen.white_queen_list if color == "white" else PlacedQueen.black_queen_list
        super().__init__("queen", coord, color, lst)

class PlacedKing(PlacedPiece):
    white_king_list = []
    black_king_list = []
    def __init__(self, coord, color):
        lst = PlacedKing.white_king_list if color == "white" else PlacedKing.black_king_list
        super().__init__("king", coord, color, lst)


PLACED_PIECE_CLASS = {
    "pawn":   PlacedPawn,
    "bishop": PlacedBishop,
    "knight": PlacedKnight,
    "rook":   PlacedRook,
    "queen":  PlacedQueen,
    "king":   PlacedKing,
}