"""

Example -- magnetics Tools for Coils
====================================

This script shows hwo to use the magnetics module
to display, fit, and feed forward 3D coil currents.

These shots where taken from J. King's 2013-23-01
experiment validating 3D Magnetics responses.

"""

# standard modules
import numpy as np

# local modules
import magnetics

############################################## Variables

il = magnetics.coil_arrays['IL']
shot = 153482 # vacuum I-Coils at 10kHz
xlim = (3e3,5e3) # coils are rotating, 2 amplitudes

############################################## Find Amplitude and Phase of Applied Coils

il.set_data(shot)
il.fit.update(ns=[1],ms=[0],xlim=xlim)
fdata = il.fit.plot()

############################################## Feed Forward a desired mode

n,m = 1,0
time = np.linspace(xlim[0],xlim[1],2e3)
amp = 1
phase = 10*2*np.pi*(time-time.min())/1000 # 10Hz

# set the mode, and plot sample coils
il.set_mode(amp,phase,n,m,time=time)
f1 = il.plotdata(search='30')	# will find IL30 and IL330
f1.axes[0].set_ylabel('kA')	# remind ourself these are not sensors

# add a jump to the amplitude and plot a sample coil
amp = np.ones_like(time)
amp[time>np.mean(xlim)]*=2
il.set_mode(amp,phase,n,m,time=time)
f2 = il.plotdata(search='30')
f2.axes[0].set_ylabel('kA')

magnetics.plt.show()