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

scale = 10
lim = 20
nr_points = (2 * lim) * scale + 1
x = np.linspace(-lim,lim,nr_points)
z = np.linspace(-lim,lim,nr_points)

def mu_it(x_pos):
    it = nr_points//2 + x_pos
    return it



Wire_1 = [0.5, 0.546]
scale = 10
#I_current = 0.94
HH_Coil = BC.BCoil(HH = 1, distance = 80, radius = 80, layers = 8, windings = 8, wire_height = Wire_1[0],
                   wire_width = Wire_1[0], insulation_thickness=(Wire_1[1] - Wire_1[0]) / 2, is_round = True,
                   winding_scheme= 2)
HH_Coil.set_R_inner(83.3-5)
HH_Coil.set_d_min(70.6 + 4)
print(HH_Coil.get_wire_length())

HH_Coil.print_basic_info()

I = 0.75
HH_Coil.B_quick_plot(I)
HH_Coil.B_curv_quick_plot(I)

HH_Coil.cooling(I,23)
print(" ")
HH_Coil.max_field(I)
print("\n")
AHH_Coil = BC.BCoil(HH = -1, distance = 80, radius = 80, layers = 10, windings = 10, wire_height = 1.18,
                   wire_width = 1.18, insulation_thickness=0.06, is_round = True,
                   winding_scheme= 2)


AHH_Coil.set_R_inner(HH_Coil.get_R_inner()*1e3+1)
AHH_Coil.set_d_min(HH_Coil.get_zmax()*2 * 1e3 + 4)

# Position 2
# AHH_Coil.set_R_inner(62)
# AHH_Coil.set_d_min(116)
#
# print(f"wire length = {AHH_Coil.get_wire_length()} m")

# AHH_Coil.print_info()

AHH_Coil.print_basic_info()
I = 1
#print(f"current I = {I} A")
AHH_Coil.cooling(I,23)
print("")
AHH_Coil.max_gradient(I)
# AHH_Coil.B_quick_plot(I)
# AHH_Coil.B_grad_quick_plot(I, nr_points = 200)
#AHH_Coil.B_curv_quick_plot(I, nr_points = scale)


HH_Coil.print_info()
AHH_Coil.print_info()


print(f"inductivity AHH: {AHH_Coil.induct_perry()*2*1e3} mH" )
print(f"inductivity HH: {HH_Coil.induct_perry() * 2*1e3} mH" )

print(f"resistance AHH: {2*AHH_Coil.resistance(22)} Ω")
print(f"resistance HH: {2*HH_Coil.resistance(22)} Ω")