def run_mcts_player(self, num_games=10): full_mobility = True scores = [] self.game_state.update_state(True) game_number = 0 while 1: for event in pygame.event.get(): if event.type == pygame.QUIT: return scores if not full_mobility: if self.game_state.frame_count % 10 == 0: action = mcts.get_MCTS_action(self.game_state) else: action = False else: action = mcts.get_MCTS_action(self.game_state) #print "Action: " + str(action) prev_game_state = deepcopy(self.game_state) self.game_state.update_state(action) if self.game_state.game_over: scores.append(self.game_state.frame_count) print "Game number: " + str(game_number) print "Frame Count: " + str(self.game_state.frame_count) game_number += 1 if game_number == num_games: return scores self.restart() if self.display_game: self.draw_state()
import mcts from game_state import GameState from player import Player from swing_copter_constants import SC from collections import deque player = Player(SC.player_start_pos, 0, 0) walls = deque() hammers = deque() game_state = GameState(player, walls, hammers, 0, False) print mcts.get_MCTS_action(game_state)