import matplotlib.pyplot as plt
from math import ceil

# Aglaea's Eidolons and LC level.
Aglaea = "E6S1"

# E2 Robin(0.16), E1 Huohuo(0.12), 2pc Amphoreus(0.08) etc
External_spdbuffs = 0

Aglaea_E = Aglaea[:2]
Aglaea_S = Aglaea[2:]

# From Aglaea's ULT and 4pc hero(0.06)
Agleae_spdbuff = [0.06]
for i in range(7):
    if Aglaea_E in 'E0''E1''E2''E3''E4':
    #E0~E4 Aglaea
        num = (i+1)*0.15 + 0.06
    elif Aglaea_E in 'E5''E6':
    # For E5 Aglaea
        num = (i+1)*0.16 + 0.06
    value = ceil(num * 1000) / 1000 
    Agleae_spdbuff.append(value)

# Total SPD buffs on aglaea at max stacks
if Aglaea_E in 'E0''E1''E2''E3':
    index = 6
elif Aglaea_E in 'E4''E5''E6':
    index = 7
else:
    index = 6 #Default
spd_buffs = Agleae_spdbuff[index] + External_spdbuffs

# Signature LC base SPD stats
sig_LC_BaseSPD = {'S0':102,'S1':114,'S2':116,'S3':118,'S4':120,'S5':122}
Base_spd = sig_LC_BaseSPD[Aglaea_S]

# Function to calculate Aglaea_spd
def calculate_aglaea_spd(Sunday_spd_breakpoint):
    return (2 * Sunday_spd_breakpoint) - (Base_spd * spd_buffs)

# Range of Sunday_spd values
Sunday_spd_breakpoint = [134,144,158,160,172]

# Lists to store results
aglaea_spd_values = []

# Calculate values for each Sunday_spd
for sunday_spd in Sunday_spd_breakpoint:
    aglaea_spd = calculate_aglaea_spd(sunday_spd)
    aglaea_spd_values.append(aglaea_spd)

# Plotting the results
plt.figure(figsize=(10, 6))

# Plot Aglaea_spd
plt.plot(Sunday_spd_breakpoint, aglaea_spd_values, label='Aglaea_spd', marker='o')

# Add labels and title
font1 = {'family':'serif','color':'Black','size':20}
font2 = {'family':'serif','color':'darkred','size':15}

plt.xlabel('Sunday_spd', fontdict=font2)
plt.ylabel("Aglaea's Speed", fontdict=font2)
plt.title(Aglaea+' Aglaea_spd vs Sunday_spd',fontdict=font1)
plt.legend()
plt.grid(True)

# Show plot
plt.show()

# '''