def test_next_move(): cur_ai = AI() current_board = BoardState.initial_state() answer1_board = BoardState.initial_state() answer2_board = BoardState.initial_state() current_board.load('board_3_in_a_row.pickle') answer1_board.load('board_3_in_a_row_ans1.pickle') answer2_board.load('board_3_in_a_row_ans2.pickle') assert (cur_ai.next_move(current_board, None)[0].board.all() == answer1_board.board.all() or cur_ai.next_move( current_board, None)[0].board == answer2_board.board.all())
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 player = 1 while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return start_time = time.time() while True: new_board = ai.next_move(board if player == 1 else board.inverted()) if new_board is None: break else: board = new_board if player == 1 else new_board.inverted() if new_board.in_the_process_of_taking == False: player = player % 2 + 1 break if time.time() - start_time > 30.0: print('TL') break draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 for i, j in product(range(8), range(8)): board.board[i][j] = 0 board.board[6][5] = 1 board.board[6][3] = 1 board.board[5][2] = 1 board.board[4][3] = 1 board.board[4][5] = 1 board.board[3][0] = -2 board.board[1][2] = -1 board.board[1][4] = -1 board.board[2][3] = -1 player = 1 while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.KEYDOWN: if event.key == pygame.K_SPACE: while True: new_board = ai.next_move(board if player == 1 else board.inverted()) if new_board is None: break else: board = new_board if player == 1 else new_board.inverted() if new_board.in_the_process_of_taking == False: player = player % 2 + 1 break draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()
class DraughtsTest(unittest.TestCase): MAX_MOVE_TIME = 3 def setUp(self): self.ai = AI(PositionEvaluation(), search_depth=2) self.board = BoardState() def test_ai_vs_ai(self): board = BoardState.initial_state() while not board.ended(): start_time = time() board = self.ai.next_move(board) finish_time = time() self.assertIsNotNone(board) elp_time = finish_time - start_time self.assertLess(elp_time, DraughtsTest.MAX_MOVE_TIME) def test_simple_move_forward(self): self.board.board[7, 0] = 1 result = self.ai.next_move(self.board) self.assertEqual(result.board[7, 0], 0) self.assertEqual(result.board[6, 1], 1) def test_simple_capt(self): self.board.board[7, 0] = 1 self.board.board[6, 1] = -1 result = self.ai.next_move(self.board) self.assertEqual(result.board[7, 0], 0) self.assertEqual(result.board[6, 1], 0) self.assertEqual(result.board[5, 2], 1) def test_king_creation(self): self.board.board[1, 0] = 1 result = self.ai.next_move(self.board) self.assertEqual(result.board[1, 0], 0) self.assertEqual(result.board[0, 1], 2) def test_king_capt(self): self.board.board[7, 0] = 2 self.board.board[1, 6] = -1 result = self.ai.next_move(self.board) self.assertEqual(result.board[7, 0], 0) self.assertEqual(result.board[1, 6], 0) self.assertEqual(result.board[0, 7], 2) def test_multi_move(self): self.board.board[7, 0] = 1 self.board.board[6, 1] = -1 self.board.board[4, 3] = -1 self.board.board[4, 5] = -1 result = self.board.copy() for i in range(3): result = self.ai.next_move(result) self.assertEqual(result.board[7, 0], 0) self.assertEqual(result.board[6, 1], 0) self.assertEqual(result.board[4, 3], 0) self.assertEqual(result.board[4, 5], 0) self.assertEqual(result.board[5, 6], 1)
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.KEYDOWN: if event.key == pygame.K_s: with open('save', 'wb') as f: f.write(pickle.dumps(board)) if event.key == pygame.K_l: with open('save', 'rb') as f: board = pickle.loads(f.read()) if board.ended(): continue if board.current_player == -1: pygame.display.set_caption(CAPTION + ' [Computing...]') start_time = time.time() new_board = ai.next_move(board) finish_time = time.time() elp_time = finish_time - start_time sleep_time = MOVE_TIME - elp_time if sleep_time > 0: time.sleep(sleep_time) pygame.display.set_caption(CAPTION) if new_board is not None: board = new_board if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1: mouse_click_position = event.pos if event.type == pygame.MOUSEBUTTONUP and event.button == 1: new_x, new_y = [p // grid_size for p in event.pos] old_x, old_y = [p // grid_size for p in mouse_click_position] new_board = board.do_move(old_x, old_y, new_x, new_y) if new_board is not None: board = new_board draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // game_set.board_size while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.MOUSEBUTTONUP: x, y = [p // grid_size for p in event.pos] if event.button == 1: # do move board.notification = None board = board.do_move(y, x) elif event.button == 3: # change figure board.board[y, x] = ((board.board[y, x] + 2) % 3) - 1 if event.type == pygame.KEYDOWN: if event.key == pygame.K_r: board = BoardState.initial_state() if event.key == pygame.K_s: board.save() if event.key == pygame.K_l: board = BoardState.load() if event.key == pygame.K_z: pass if (game_set.is_ai_enabled and event.key == pygame.K_SPACE and game_set.first_player_is_ai == board.is_first_player_turn): new_board = ai.next_move(board, None)[0] if new_board is not None: board = new_board if board.is_game_finished: board.notification = "Second" if board.is_first_player_turn \ else "First" board.notification += " player won! Press R to restart" draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1: mouse_click_position = event.pos if event.type == pygame.MOUSEBUTTONUP and event.button == 1: new_x, new_y = [p // grid_size for p in event.pos] old_x, old_y = [p // grid_size for p in mouse_click_position] new_board = board.do_move(old_x, old_y, new_x, new_y) if new_board is not None: board = new_board if event.type == pygame.MOUSEBUTTONUP and event.button == 3: x, y = [p // grid_size for p in event.pos] board.board[y, x] = (board.board[y, x] + 1 + 2) % 5 - 2 # change figure if event.type == pygame.KEYDOWN: if event.key == pygame.K_r: board = board.inverted() if event.key == pygame.K_s: board.save_board("checkers.txt") if event.key == pygame.K_l: board.open_saved_board("checkers.txt") if event.key == pygame.K_SPACE: new_board = board.inverted() while True: new_board = ai.next_move(new_board) if new_board is None or new_board.in_the_process_of_taking == False: break if new_board is not None: board = new_board.inverted() draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 previous_board = board visible_board = board step = 0 flag = True while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1: mouse_click_position = event.pos if event.type == pygame.MOUSEBUTTONUP and event.button == 1: new_x, new_y = [p // grid_size for p in event.pos] old_x, old_y = [p // grid_size for p in mouse_click_position] new_board = board.do_move(old_x, old_y, new_x, new_y) if new_board is not None: previous_board = board board = new_board.inverted() if step % 2 == 0: visible_board = new_board else: visible_board = board step += 1 if board.get_winner != 0: flag = False #if event.type == pygame.MOUSEBUTTONUP and event.button == 3: #x, y = [p // grid_size for p in event.pos] #board.board[y, x] = (board.board[y, x] + 1 + 2) % 5 - 2 # change figure if event.type == pygame.KEYDOWN: if event.key == pygame.K_r: visible_board = visible_board.inverted() elif event.key == pygame.K_n: board = board.initial_state() visible_board = board step = 0 elif event.key == pygame.K_s: save_board(board, './src/save_board.bin') save_board(visible_board, './src/save_visible.bin') elif event.key == pygame.K_l: board = load_board('./src/save_board.bin') visible_board = load_board('./src/save_visible.bin') flag = True if board.current_player == visible_board.current_player: step = 0 else: step = 1 elif event.key == pygame.K_1: board = load_board( './src/test_boards/test_board_1_visible') visible_board = load_board( './src/test_boards/test_board_1_board') flag = True if board.current_player == visible_board.current_player: step = 0 else: step = 1 elif event.key == pygame.K_2: board = load_board( './src/test_boards/test_board_2_visible') visible_board = load_board( './src/test_boards/test_board_2_board') flag = True if board.current_player == visible_board.current_player: step = 0 else: step = 1 elif event.key == pygame.K_z: board = previous_board elif event.key == pygame.K_SPACE: new_board = ai.next_move(board) if new_board is not None: previous_board = board board = new_board.inverted() if step % 2 == 0: visible_board = new_board else: visible_board = board step += 1 if board.get_winner != 0: flag = False draw_board(screen, 0, 0, grid_size, visible_board) if not flag: draw_final_message(screen, board.get_winner) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): max_y, max_x = screen.get_size()[1], screen.get_size()[0] elem_size = max_y // 16 # размер одной клетки save_box = InputBox('SAVE', max_y + (max_x - max_y) * 625 // 10000, max_y * 4 // 5, 250, 30) load_box = InputBox('LOAD', max_y + (max_x - max_y) * 625 // 10000, max_y * 4 // 5 + 50, 250, 30) while True: for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.KEYDOWN and event.key == pygame.K_RETURN: return save_name = save_box.handle_event(event) load_name = load_box.handle_event(event) if save_name: db = shelve.open(save_name) db[save_name] = board db.close() elif load_name: if os.path.exists(os.path.join(os.getcwd(), load_name + '.db')): db = shelve.open(load_name) board = db[load_name] db.close() else: print('No such file:(') if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1: mouse_click_position = event.pos to_x, to_y = [(p - elem_size // 2) // elem_size for p in mouse_click_position] try: new_board = board.do_move(to_x, to_y) except EndGame as EG: print(EG.winner, 'bruh') board = EG.board board.current_player *= 2 new_board = None # draw_board(screen, elem_size, elem_size, elem_size, board) # screen.fill((23, 123, 234)) # pygame.display.flip() # print('--------------') # print(new_board) # pygame.time.wait(1000) # return # screen.fill(color) # screen.blit(InputBox.FONT.render(('black' if EG.winner == 1 else 'white') + 'player won!!!', True, (0, 0, 0)), (0.7 * screen.get_size()[0], 0.2 * screen.get_size()[1])) # time.sleep(3) # return if new_board is not None: board = new_board if event.type == pygame.KEYDOWN and event.key == pygame.K_SPACE: print('kek') # print(ai.next_move(board)) again = True while again: try: position_evalution, *move = ai.next_move(board, (0, 0)) again = False except TypeError: pass # print(position_evalution) # print(move) # print((ai.next_move(board))) print( '+=+=+=++=++++++++=+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++=' ) # break move = move[0] print(f'my move = {move}') try: new_board = board.do_move(move[1], move[0]) except EndGame as EG: print(EG.winner, 'bruh') board = EG.board board.current_player *= 2 new_board = None if new_board is not None: board = new_board save_box.update() load_box.update() draw_board(screen, elem_size, elem_size, elem_size, board) save_box.draw(screen) load_box.draw(screen) pygame.display.flip()
def game_loop(screen: Surface, board: BoardState, ai: AI): grid_size = screen.get_size()[0] // 8 # размер одной клетки count = -1 while True: # print(pygame.event.get()) for event in pygame.event.get(): if event.type == pygame.QUIT: return if event.type == pygame.MOUSEBUTTONDOWN and event.button == 1: mouse_click_position = event.pos # print(f'event.pos = {event.pos}') if event.type == pygame.MOUSEBUTTONUP and event.button == 1: new_x, new_y = [p // grid_size for p in event.pos] old_x, old_y = [p // grid_size for p in mouse_click_position] try: new_board = board.do_move(old_x, old_y, new_x, new_y) except EndGame: return if new_board is not None: board = new_board if event.type == pygame.MOUSEBUTTONUP and event.button == 3: # ???????????????????????? x, y = [p // grid_size for p in event.pos] board.board[ y, x] = (board.board[y, x] + 1 + 2) % 5 - 2 # change figure if event.type == pygame.KEYDOWN: if event.key == pygame.K_r: board = board.inverted() elif event.key == pygame.K_SPACE: new_board = ai.next_move(board) if new_board is not None: board = new_board elif event.key == pygame.K_s: count += 1 count %= 5 name = f'game{count}' # name = input_name() db = shelve.open(name) db[name] = board db.close() # todo elif event.key == pygame.K_l: load = True elif event.key in (pygame.K_0, pygame.K_1, pygame.K_2, pygame.K_3, pygame.K_4) and load: name = 'game' + (chr(event.key)) if os.path.exists(os.path.join(os.getcwd(), name + '.db')): db = shelve.open(name) board = db[name] db.close() load = False else: print('No such file:(') draw_board(screen, 0, 0, grid_size, board) pygame.display.flip()