import matplotlib.pyplot as plt
import numpy as np

# base attack of a unit, should be around 10k for SJW and 7.5k for hunters
base_attack = 10000.00 
# total attack of a unit
attack = 28000.00 
# how many seconds for one expert stack
# e.g. 2.5 seconds => 40 stacks in 100 seconds, 100 stacks in 250 seconds
exp_rate = 2.5

ex = ((0.016*base_attack)/attack*100.00)/exp_rate
ex_max = exp_rate * 100.00

x = np.linspace(0, 300, 600)

# 8 piece curse
y1 = np.where(x <= 100, 30 + 0.2*x, 30 + 0.2*100)

# 4C + 4E
y2 = np.where(x <= 100, 20 + 0.1*x + (ex*x)*(1.2 + 0.001*x), 
              np.where(x <= ex_max, 30 + ex*x*1.3, 30 + ex*ex_max*1.3))

diff = np.abs(y1 - y2)
intersection_index = np.argmin(diff)
intersection_x = x[intersection_index]
intersection_y = y1[intersection_index]

plt.figure(figsize=(8, 6))
plt.plot(x, y1, label='8 piece Curse', color='blue')
plt.plot(x, y2, label='4 Curse + 4 Expert', color='red')

plt.scatter(intersection_x, intersection_y, color='green', zorder=5)
plt.text(intersection_x + 5, intersection_y - 8, f'time={intersection_x:.2f} s\ndmg={intersection_y:.2f} %', 
         fontsize=10, color='green')

plt.xlabel('Time (seconds)')
plt.ylabel('Damage Increase (%)')
plt.ylim(0, None)

plt.title('Damage Increase Over Time (Curse vs Expert)')

plt.text(180, 15, f"Parameters\nAttack: {int(attack/1000)}k\nExpert stacks rate: {exp_rate}s", fontsize=10, bbox=dict(facecolor='white', alpha=0.5))

plt.legend()

plt.grid(True)
plt.show()