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


Wires = [[0.45, 0.514],[0.475, 0.543],[0.5, 0.568]]

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 = 9, 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)
HH_Coil.set_d_min(70.6 + 4)
print(HH_Coil.get_wire_length())

HH_Coil.print_info()

I = 0.67
# HH_Coil.B_quick_plot(I,nr_points=scale)
# HH_Coil.B_curv_quick_plot(I,nr_points = scale)

HH_Coil.cooling(I,23)

AHH_Coil = BC.BCoil(HH = -1, distance = 80, radius = 80, layers = 10, windings = 16, 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)
print(AHH_Coil.get_wire_length())
print(HH_Coil.get_R_inner()*1e3)
AHH_Coil.set_R_inner(HH_Coil.get_R_inner()*1e3)
AHH_Coil.set_d_min(HH_Coil.get_zmax()*2 * 1e3 + 4)

AHH_Coil.print_info()

I = 2
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)

AHH_Coil.cooling(I,23)