def setup_game():
# create board
board = Board()
# create teams
white = Team('white')
black = Team('black')
teams = (white, black)
# create pieces
white_pawns = [Piece(white, 'pawn', number_to_letter(i)+str(2), 'p') for i in range(1, 9)]
black_pawns = [Piece(black, 'pawn', number_to_letter(i)+str(7), 'P') for i in range(1, 9)]
white_rooks = [Piece(white, 'rook', number_to_letter(i)+str(1), 'r') for i in range(1, 9, 7)]
black_rooks = [Piece(black, 'rook', number_to_letter(i)+str(8), 'R') for i in range(1, 9, 7)]
white_knights = [Piece(white, 'knight', number_to_letter(i)+str(1), 'n') for i in range(2, 8, 5)]
black_knights = [Piece(black, 'knight', number_to_letter(i)+str(8), 'N') for i in range(2, 8, 5)]
white_bishops = [Piece(white, 'bishop', number_to_letter(i)+str(1), 'b') for i in range(3, 7, 3)]
black_bishops = [Piece(black, 'bishop', number_to_letter(i)+str(8), 'B') for i in range(3, 7, 3)]
white_queen = [Piece(white, 'queen', 'd1', 'q')]
black_queen = [Piece(black, 'queen', 'd8', 'Q')]
white_king = [Piece(white, 'king', 'e1', 'k')]
black_king = [Piece(black, 'king', 'e8', 'K')]
white_pieces = [white_pawns, white_rooks, white_knights, white_bishops, white_queen, white_king]
black_pieces = [black_pawns, black_rooks, black_knights, black_bishops, black_queen, black_king]
# initialize teams
white.initialize(white_pieces)
black.initialize(black_pieces)
# initialize board
board.initialize((white_pieces, black_pieces))
return board, teams
def create_preview_board(self):
# Create a new board
vars.screen.calc_board_spacing(self.presets[0].BOARD_SIZE)
self.preview_board = Board(self.presets[0].BOARD_SIZE, self.presets[0].SHIPS)
# Place ships on board
for index, ship in enumerate(self.preview_board.ships):
ship.set_initial_pos(index)
# Set 50% of ships to be sunken
for ship in random.sample(self.preview_board.ships,
min(len(self.preview_board.ships), 2)):
ship.sunk = True
for pos in ship.get_coords(None):
self.preview_board.set_hit(pos)
# Populate board with shots
for i in range(self.presets[0].BOARD_SIZE):
pos = (random.randint(0, self.presets[0].BOARD_SIZE - 1),
random.randint(0, self.presets[0].BOARD_SIZE - 1))
for ship in self.preview_board.ships:
if pos in ship.get_coords():
if ship.hits.count(True) == ship.length - 1:
break
ship.hits[ship.get_coords().index(pos)] = True
self.preview_board.set_hit(pos)
break
else:
self.preview_board.set_miss(pos)
def __init__(self, player, width, height, speed):
self.speed = speed
self.player = player
self.width = width
self.height = height
self.board = Board(width, height)
self.snake = Snake(random.randint(0, width - 1),
random.randint(0, height - 1))
self.fruit = Fruit(random.randint(0, width - 1),
random.randint(0, height - 1))
self.score = 0
self.game_is_dead = False
# Cecil was a caterpilla, Cecil was my friend - Mum
self.window = GraphWin("Cecil", width=300, height=300, autoflush=False)
self.window.setBackground('white')
from objects.board import Board from algorithms.genetic import genetic b = Board() b.readExternalFile() b.show() genetic(b)
