import matplotlib.pyplot as plt, numpy as np
import pandas as pd

#Read in Data from the field
df = pd.read_excel('c:/myData/PhD/code/attackintesity_3.xlsx')
#print(df.loc[df['sn']>0,['intensity']])
arr = eval(df.to_json())
#print(arr)

#Label the defence levels
defence_2 = 2
defence_1 = 1
defence_0 = 0

#Set Initial Residual Energy  
residual_energy_0  = 100
residual_energy_1  = 100
residual_energy_2  = 100

#Set Assumption for Attacker
energy_a0 = 0.0/100
energy_a1 = 0.001/100
energy_a2 = 0.002/100

#Set Assumption for Defender
energy_d0 = 0.0/100
energy_d1 = 0.015/100
energy_d2 = 0.025/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_1==1:
        if(intensity == 0):
            residual_energy_1 = residual_energy_1-energy_a0-energy_d1
        elif (intensity == 1):
            residual_energy_1 = residual_energy_1-energy_a1-energy_d1
        elif(intensity == 2):
            residual_energy_1 = residual_energy_1-energy_a2-energy_d1
    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)
    if (i%100==0):
        print(f'i = {i} and RE0 = {residual_energy_0} and RE1 = {residual_energy_1} and RE2 = {residual_energy_2}')
#print(y_point_0,'' ,y_point_1,' ' ,y_point_2)
print()
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 (Cycle)')
plt.ylabel('Residual Energy (%)')
plt.title('Resource Usage = Memory Usage + Power Consumption')
plt.show()