import numpy as np
import matplotlib.pyplot as plt
import src.coil_class as BC

I_current = 1.25

# scale = 1000 --> μm
scale = 1000
lim = 5
nr_points = (2 * lim) * scale + 1
x = np.linspace(-lim, lim, nr_points)
z = np.linspace(-lim, lim, nr_points)
print(x)


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


print(x[mu_it(-60)])

# standard
C1 = BC.BCoil(HH=1, distance=54, radius=48, layers=8, windings=8, wire_height=0.5, wire_width=0.5,
              insulation_thickness=0.06, is_round=True, winding_scheme=True)
C1.set_R_outer(49.8)
C1.set_d_min(49.8)
C1.print_info()

Bz, Bx = C1.B_tot_along_axis(I_current, x, z, raster=2)

# plt.figure(5)
# plt.plot(z,Bz)
# plt.plot(x,Bx, label = "B_tot along x-axis")
# plt.show()
# plt.close(5)


# coil is ~ 500 μm thicker
shift_radius = 1  # 0.125
C_shift = BC.BCoil(HH=1, distance=54, radius=48, layers=8, windings=8, wire_height=0.5, wire_width=0.5,
                   insulation_thickness=0.06, is_round=True, winding_scheme=True)
C_shift.set_R_outer(49.8 + shift_radius)
C_shift.set_d_min(49.8)

# shift_radius = 0.125

Bz2, By = C_shift.B_tot_along_axis(I_current + 0.00082, x, z, raster=2)

shift_int = int(shift_radius * scale - 1)
print(shift_int)
By_shift = By[shift_int:]
y_shift = x[:-shift_int]
print(By_shift)
print(y_shift)
B_sum = (By_shift + Bx[:-shift_int]) / 2
# shift By shift with shift radius

plt.figure(6)
# plt.plot(z,Bz)
plt.plot(x, Bx, label="B_tot along x-axis")
plt.plot(y_shift, By_shift, label="Shiftet tot B")
plt.plot(y_shift, B_sum, label="B_sum")
plt.legend()
plt.show()
# plt.close(5)

it = mu_it(int(-shift_radius * 1e3 // 2))
# it = μm_it(-60)
print(it)
zero = mu_it(0)
print(y_shift[it])
print(y_shift[zero])

tot_diff = B_sum[it] - B_sum[zero]

print(f"diff {y_shift[it]} μm --> 0: {tot_diff} G, relative = {tot_diff / B_sum[it]}")