# %%
import matplotlib.pyplot as plt
import numpy as np
import matplotlib as mpl
from matplotlib.ticker import AutoMinorLocator

# %%
# Import data without water cooling

I = np.array([0, 0.5, 1, 1.25, 1.5, 1.75, 2,2.25])

HH = np.array([25, 26, 30.9, 36.6, 39.8, 46.8, 53.9, 62.3])
AHH = ([25,26.5,35.2, 42.7, 52.2])

# %%
# Data with water cooling

I_w = np.arange(0,5.25,0.5)
I_w = np.append(I_w, 5.25)

HH_w = np.array([17.8, 18.2, 18.9, 20, 21.8, 24.3, 27.3, 31.4, 36.8, 42.5, 49.3, 56.2])
AHH_w = np.array([17.2, 17.7,19.1, 21.3, 24.7, 29.3, 35.7, 43.9, 55.6])

# %%
my_colors = {'light_green': '#97e144',
             'orange': '#FF914D',
             'light_grey': '#545454',
             'pastel_blue': '#1b64d1',
             'light_blue': '#71C8F4',
             'purple': '#7c588c'}


mpl.rcParams.update({'font.size': 11, 'axes.linewidth': 1, 'lines.linewidth': 2, 'text.usetex': False, 'font.family': 'arial'})
mpl.rcParams['xtick.direction'] = 'in'
mpl.rcParams['ytick.direction'] = 'in'

mpl.rcParams['xtick.top'] = True
mpl.rcParams['ytick.right'] = True

# %%
fig, ax = plt.subplots(figsize=(5, 3.6),dpi=400)
ax.scatter(I, HH, label='HH, no water')
ax.scatter(I[0:len(AHH)], AHH, label='AHH, no water',marker='v')
ax.scatter(I_w, HH_w, label='HH, with water',marker = 's')
ax.scatter(I_w[0:len(AHH_w)], AHH_w, label='AHH, with water', marker='d')
ax.legend(fontsize=10.5)

ax.grid(alpha = 0.2)
ax.set_xlim(-0.3, 5.7)
ax.set_ylim(11, 65)
ax.set_ylabel('temperature coil T (°C)')
ax.set_xlabel('current I (A)')

ax.xaxis.set_minor_locator(AutoMinorLocator(2))
ax.yaxis.set_minor_locator(AutoMinorLocator(2))
fig.tight_layout()
plt.savefig('C:/Users/Joschka/Desktop/Master_Thesis/Figures/Coil_measurements/Final/heating.png')
plt.savefig('C:/Users/Joschka/Desktop/Master_Thesis/Figures/Coil_measurements/Final_low/heating.png',dpi=96)
plt.show()