import random
import matplotlib.pyplot as plt

random.seed()
basechance = 0.02
failstack = basechance / 10;

#GREEN
costGreen = 0
#BOSS
costBoss = 0

currentFailstack = 40
stacks = 0

#prices
failstackcosts = {40: 75, 45: 119, 50:201, 55: 296, 60: 408, 65:541 , 70:704, 75: 906, 80:1159}

differentCosts = []
differentCosts2 = {}

for y in range(45, 81, 5):
    for x in range(1, 400000):
        roll = round(random.uniform(0, 1), 2);
        if(roll > basechance + failstack * currentFailstack):
            currentFailstack += 5
            costGreen += 120
            costBoss += 199
            if currentFailstack == y:
                stacks += 1
                currentFailstack = 40
        else:
            costGreen += failstackcosts[currentFailstack]
            costBoss += failstackcosts[currentFailstack]
            costGreen -= 150
            costBoss -= 707
            currentFailstack = 40
    currentFailstack = 40
    if stacks != 0:
        differentCosts.append((costBoss - costGreen) / stacks);
        differentCosts2.update({y: (costBoss - costGreen) / stacks})

        costGreen = 0
        costBoss = 0

        stacks = 0
    else:
        differentCosts.append(0);

#print ("COST ENHANCING WITH GRUNIL/HEVE:",costGreen)
#print ("COST ENHANCING WITH BOSS GEAR :",costBoss)
#print ("MILLONES SAVED PER 60 STACK:",(costBoss - costGreen) / stacks)

print (differentCosts2)
print (list(differentCosts2.keys()))
plt.plot(differentCosts2.keys(), differentCosts2.values())
plt.show()