def main():
selected_game_mode = -1
while (selected_game_mode < 0 or selected_game_mode > 3):
print('Select game mode:')
print('1. Normal chess')
print('2. Reverse chess (suicide mode)')
selected_game_mode = int(input('Selection: '))
print('')
selected_scenario = -1
while (selected_scenario < 0 or selected_scenario > 7):
print('Select game scenario:')
print('1. Train AI')
print('2. Watch trained AI vs trained AI')
print('3. Watch trained AI vs normal AI')
print('4. Watch normal AI vs normal AI')
print('5. Play against trained AI')
print('6. Play against normal AI')
selected_scenario = int(input('Selection: '))
# Select playing color if human vs ai
selected_color = -1
if (selected_scenario > 4 and selected_scenario <= 6):
while (selected_color < 0 or selected_color > 3):
print('Select Color:')
print('1. White')
print('2. Black')
selected_color = int(input('Selection: '))
game_variant = None
if (selected_game_mode == 1):
game_variant = 'chess'
elif (selected_game_mode == 2):
game_variant = 'suicide'
else:
raise Exception('Invalid game mode!')
# Load train data if a scenario including 'trained AI' is selected
# train_data = np.load('train_file.npy')
# [1. Train AI]
if (selected_scenario == 1):
# todoooooooooooooooooooooooooo------------------------------------------
pass
else:
white_player = None
black_player = None
# [2. Watch trained AI vs trained AI]
if (selected_scenario == 2):
white_player = AIPlayer(is_white=True,
depth=3,
piece_square_table=None)
black_player = AIPlayer(is_white=False,
depth=3,
piece_square_table=None)
# [3. Watch trained AI vs normal AI]
elif (selected_scenario == 3):
white_player = AIPlayer(is_white=True,
depth=3,
piece_square_table=None)
black_player = AIPlayer(is_white=False,
depth=3,
piece_square_table=None)
# [4. Watch normal AI vs normal AI]
elif (selected_scenario == 4):
white_player = AIPlayer(is_white=True,
depth=3,
piece_square_table=None)
black_player = AIPlayer(is_white=False,
depth=3,
piece_square_table=None)
# [5. Play against trained AI]
elif (selected_scenario == 5):
# [1. White]
if (selected_color == 1):
white_player = HumanPlayer(is_white=True)
black_player = AIPlayer(is_white=False,
depth=3,
piece_square_table=None)
# [2. Black]
elif (selected_color == 2):
white_player = AIPlayer(is_white=True,
depth=3,
piece_square_table=None)
black_player = HumanPlayer(is_white=False)
else:
raise Exception('Invalid color!')
# [6. Play against normal AI]
elif (selected_scenario == 6):
# [1. White]
if (selected_color == 1):
white_player = HumanPlayer(is_white=True)
black_player = AIPlayer(is_white=False,
depth=3,
piece_square_table=None)
# [2. Black]
elif (selected_color == 2):
white_player = AIPlayer(is_white=True,
depth=3,
piece_square_table=None)
black_player = HumanPlayer(is_white=False)
else:
raise Exception('Invalid color!')
else:
raise Exception('Invalid scenario!')
chess_game = ChessGame(white_player=white_player,
black_player=black_player,
variant=game_variant,
max_turn=None)
gameover = False
result = None
while (not gameover):
gameover, result = chess_game.play_half_turn()
print(gameover)
print(result)
