import random
import math

def play(totals, player, win, can_attack, hand):
    if win != -1:
        return totals, win, can_attack, hand[player]
    else:
        for i in range(3):
            tar = math.floor(random.random() * float(len(totals[0]) + len(totals[1])))
            if ((tar < len(totals[0])) and (len(totals[0]) < 9)):
                totals[0][tar] -= 1
                if ((totals[0][tar] == 0) and (tar != 0)):
                    totals[0].pop(tar)
                    can_attack[0].pop(tar)
                elif ((totals[0][tar] == 0) and (tar <= 0)):
                    totals[player].append(2)
                    return totals, 1, can_attack, (hand[player]-1)
            elif ((tar >= len(totals[0])) and (len(totals[0]) < 9)):
                totals[1][tar-len(totals[0])] -= 1
                if ((totals[1][tar-len(totals[0])] == 0) and ((tar-len(totals[0])) != 0)):
                    totals[1].pop(tar-len(totals[0]))
                    can_attack[1].pop(tar-len(totals[0]))
                elif ((totals[1][tar-len(totals[0])] == 0) and ((tar-len(totals[0])) <= 0)):
                    totals[player].append(2)
                    return totals, 0, can_attack, (hand[player]-1)
            else:
                return totals, -1, can_attack, hand
        totals[player].append(2)
        can_attack[player].append(False)
        return totals, -1, can_attack, (hand[player]-1)

def attack(can_attack, player):
    dmg = 0
    for i in can_attack[player]:
        if i:
            dmg += 3
    return dmg

def wakeup(can_attack, player):
    for i in range(1,len(can_attack[player])):
        if not can_attack[player][i]:
            can_attack[player][i] = True
    return can_attack

def hero_power(totals, player, toon):
    if toon == "huntard":
        totals[int(math.cos((math.pi/2.0) * player))][0] -= 2
    elif toon == "wario":
        totals[player][0] += 2
    else:
        return totals
    return totals

win_counter = 0
turn_counter = 0
iterations = 10000
toon = ["wario", "huntard"]

for i in range(iterations):
    boards = [[30], [30]]
    attack_list = [[False], [False]]
    mana = 2
    win = -1
    n=2
    hand_size = [5,5]

    boards, win, attack_list, hand_size[1] = play(boards, 1, win, attack_list, hand_size)
    while win == -1:
        attack_list = wakeup(attack_list, 0)
        hand_size[0] += 1

        hp = True
        for i in range(math.floor(mana/2)):
            if (((math.floor(mana/2) - 1) > hand_size[0]) and hp):
                boards = hero_power(boards, 0, toon[0])
                hp = False
            elif hand_size[0] > 0:
                boards, win, attack_list, hand_size[0] = play(boards, 0, win, attack_list, hand_size)
        boards[1][0] -= attack(attack_list, 0)
        if boards[1][0] <= 0:
            win = 0
        n += 1

        attack_list = wakeup(attack_list, 1)
        hand_size[1] += 1
        hp = True
        for i in range(math.floor(mana/2)):
            if (((math.floor(mana/2) - 1) > hand_size[1]) and hp):
                boards = hero_power(boards, 1, toon[1])
                hp = False
            elif hand_size[1] > 0:
                boards, win, attack_list, hand_size[1] = play(boards, 1, win, attack_list, hand_size)
        boards[0][0] -= attack(attack_list, 1)
        if boards[0][0] <= 0:
            win = 1
        n += 1

        mana += 1

    win_counter += win
    turn_counter += n

print("coin winrate: " + str(100.0 * (float(win_counter)/float(iterations))) + "\navg turn count: " + str(float(turn_counter)/float(iterations)) + "\ntested over " + str(iterations) + " iterations")