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)
