import matplotlib.pyplot as plt, numpy as np
import pandas as pd
df = pd.read_excel('c:/myData/PhD/code/attackintesity.xlsx')
#print(df.loc[df['sn']>0,['intensity']])
arr = eval(df.to_json())
defence_2 = 2
#defence_1 = 1
defence_0 = 0

energy_a0 = 0.0/100
#energy_a1 = 0.01/100
energy_a2 = 0.02/100

energy_d0 = 0.0/100
#energy_d1 = 0.15/100
energy_d2 = 0.25/100

x_point = []
y_point_0 = []
y_point_1 = []
y_point_2 = []
success_0_count=0
#success_1_count=0
success_2_count=0
for i in range(len(arr['intensity'])):
    intensity = arr['intensity'][str(i)]
    if defence_0==0:
        if(intensity <= defence_0):
            success_0 = 1
            success_0_count+=1
        
   # if defence_1==1:
    #    if(intensity <= defence_1):
     #       success_1 = 1
      #      success_1_count+=1
    
    if defence_2==2:
        if(intensity <= defence_2):
            success_2 = 1
            success_2_count+=1
    
    x_point.append(i)
#y_point_0.append(success_0_count/i)
#y_point_1.append(success_1_count/i)
#y_point_2.append(success_2_count/i)
for j in range(i+1):
    y_point_0.append(success_0_count/i*100)
   # y_point_1.append(success_1_count/i*50)
    y_point_2.append(success_2_count/i*100)
    #print(f'i = {i} and RE = {residual_energy}')
print(len(y_point_2), i, len(x_point))


plt_x = np.array(x_point)
plt_y = np.array(y_point_0)
key0, = plt.plot(plt_x, plt_y, label ='Defence: Level-0')

#plt_x = np.array(x_point)
#plt_y = np.array(y_point_1)
#key1, = plt.plot(plt_x, plt_y, label ='Defence: Level-1')

plt_x = np.array(x_point)
plt_y = np.array(y_point_2)
key2, = plt.plot(plt_x, plt_y, label ='Defence: Level-2')
# Set Axes
plt.xlim(left=0)

leg = plt.legend(loc='upper center')
plt.xlabel('Time')
plt.ylabel('Sucess Rate')
plt.title('Success Rate')
plt.show()