def __init__(self, color): self._color = color self._player_pieces = [ Pieces.General(color), Pieces.Guard(color), Pieces.Guard(color), Pieces.Horse(color), Pieces.Horse(color), Pieces.Elephant(color), Pieces.Elephant(color), Pieces.Chariot(color), Pieces.Chariot(color), Pieces.Cannon(color), Pieces.Cannon(color), Pieces.Soldier(color), Pieces.Soldier(color), Pieces.Soldier(color), Pieces.Soldier(color), Pieces.Soldier(color) ] if self._color == 'red': # (row, col) self._initial_placements = [(1, 4), (0, 3), (0, 5), (0, 2), (0, 7), (0, 1), (0, 6), (0, 0), (0, 8), (2, 1), (2, 7), (3, 0), (3, 2), (3, 4), (3, 6), (3, 8)] else: self._initial_placements = [(8, 4), (9, 3), (9, 5), (9, 2), (9, 7), (9, 1), (9, 6), (9, 0), (9, 8), (7, 1), (7, 7), (6, 0), (6, 2), (6, 4), (6, 6), (6, 8)]