UltimaTTT-Agent/main.py at main · shInNei/UltimaTTT-Agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from state import State, State_2
import time
from importlib import import_module


def main(player_X, player_O, rule = 2):
    dict_player = {1: 'X', -1: 'O'}
    if rule == 1:
        cur_state = State()
    else:
        cur_state = State_2()
    turn = 1

    limit = 81
    remain_time_X = 120
    remain_time_O = 120

    player_1 = import_module(player_X)
    player_2 = import_module(player_O)


    while turn <= limit:
        print("turn:", turn, end='\n\n')
        if cur_state.game_over:
            print("winner:", dict_player[cur_state.player_to_move * -1])
            break

        start_time = time.time()
        if cur_state.player_to_move == 1:
            new_move = player_1.select_move(cur_state, remain_time_X)
            elapsed_time = time.time() - start_time
            remain_time_X -= elapsed_time
        else:
            new_move = player_2.select_move(cur_state, remain_time_O)
            elapsed_time = time.time() - start_time
            remain_time_O -= elapsed_time

        if new_move == None:
            break

        if remain_time_X < 0 or remain_time_O < 0:
            print("out of time")
            print("winner:", dict_player[cur_state.player_to_move * -1])
            break

        if elapsed_time > 10.0:
            print("elapsed time:", elapsed_time)
            print("winner: ", dict_player[cur_state.player_to_move * -1])
            break

        cur_state.act_move(new_move)
        print(cur_state)

        turn += 1

    print("X:", cur_state.count_X)
    print("O:", cur_state.count_O)


main('random_agent', '_MSSV')