import matplotlib.pyplot as plt
import numpy as np
import matplotlib as mpl
# matplotlib.use('Qt5Agg')
from src import coil_class as BC

# %%

HH_Coil = BC.BCoil(HH=1, distance=54, radius=48, layers=8, windings=8, wire_height=0.5,
                   wire_width=0.5, insulation_thickness=(0.546 - 0.5) / 2, is_round=True,
                   winding_scheme=2)

HH_Coil.set_R_inner(45.6)
HH_Coil.set_d_min(47.4)
HH_Coil.print_info()

AHH_Coil = BC.BCoil(HH=-1, distance=54, radius=48, layers=HH_Coil.get_layers, windings=2 * HH_Coil.get_windings,
                    wire_height=0.5, wire_width=0.5, insulation_thickness=(0.546 - 0.5) / 2,
                    is_round=True, winding_scheme=2)

AHH_Coil.set_R_inner(45.6)
AHH_Coil.set_d_max(77.6)

AHH_Coil.print_info()

# %%

mpl.rcParams['xtick.direction'] = 'in'
mpl.rcParams['ytick.direction'] = 'in'

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

# mpl.rcParams['xtick.major.size'] = 10
# mpl.rcParams['xtick.major.width'] = 3
# mpl.rcParams['xtick.minor.size'] = 10
# mpl.rcParams['xtick.minor.width'] = 3
# mpl.rcParams['ytick.major.size'] = 10
# mpl.rcParams['ytick.major.width'] = 3
# mpl.rcParams['ytick.minor.size'] = 10
# mpl.rcParams['ytick.minor.width'] = 3
mpl.rcParams['font.size'] = 11

# mpl.rcParams.update({'font.size': 22, 'axes.linewidth': 3, 'lines.linewidth': 3})

# %%

#HH_Coil.plot_raster()

raster_value = 400
full_structure = HH_Coil.full_raster(raster_value, plot_radius_offset=0.00007) * 1e3

fig, ax = plt.subplots(figsize=(3.8, 3.8),dpi=600)

ax.scatter(full_structure[:, :, 1], full_structure[:, :, 0], linewidths=0.001)
ax.set_xlabel("radius ϕ [mm]")
ax.set_ylabel("z-axis [mm]")
ax.axvline(x=HH_Coil.get_R_inner() * 1e3 - 0.1, lw=5, color="red")

ext = 0.2
ax.set_xlim(1e3 * HH_Coil.get_R_inner() - ext- 0.2, 1e3 * (HH_Coil.get_R_inner()+HH_Coil.get_coil_height() )+ ext)
ax.set_ylim(1e3 * HH_Coil.get_zmin() - ext, 1e3 * HH_Coil.get_zmax() + ext)

plt.savefig('C:/Users/Joschka/Desktop/Magnetic_field_coils_project/Thesis_Plots/Coil_Design/Out/winding_scheme_2.png')
plt.show()