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())
#print(arr)
defence_2 = 2
#defence_1 = 1
defence_0 = 0
residual_energy_0  = 100
#residual_energy_1  = 100
residual_energy_2  = 100
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 = []
for i in range(len(arr['intensity'])):
    intensity = arr['intensity'][str(i)]
    if defence_0==0:
        if(intensity == 0):
            residual_energy_0 = residual_energy_0-energy_a0-energy_d0
        #elif (intensity == 1):
         #   residual_energy_0 = residual_energy_0-energy_a1-energy_d0
        elif(intensity == 2):
            residual_energy_0 = residual_energy_0-energy_a2-energy_d0
    if defence_2==2:
        if(intensity == 0):
            residual_energy_2 = residual_energy_2-energy_a0-energy_d2
        #elif (intensity == 1):
         #   residual_energy_2 = residual_energy_2-energy_a1-energy_d2
        elif(intensity == 2):
            residual_energy_2 = residual_energy_2-energy_a2-energy_d2
    x_point.append(i)
   
    y_point_0.append(residual_energy_0)
    #y_point_1.append(residual_energy_1)
    y_point_2.append(residual_energy_2)
    #print(f'i = {i} and RE = {residual_energy}')

plt_x = np.array(x_point)
plt_y = np.array(y_point_0)
#print(f'a-xis = {x_point} \n y-axis {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')
leg = plt.legend(loc='upper center')
plt.xlabel('Time')
plt.ylabel('Residual Energy')
plt.title('Resource Usage')
plt.show()