import numpy as np
import scipy
import sys
sys.path.insert(0, './../src/')
import proc_profile_bottollier as ppb
if len(sys.argv) < 1:
shot_number = int(open('shot_number.txt', 'r').read())
else:
if (len(sys.argv) == 1) & ("py" in sys.argv[0]):
shot_number = int(open('shot_number.txt', 'r').read())
else:
shot_number = int(sys.argv[1])
shot = ppb.ProcProfile(shot_number)
sweeps_average = 8
shot.reference_gd(all_shot=1, sw_clustersize=sweeps_average)
shot.eval_freq_overlap()
shot.plasma_gd(5000, sweeps_average, 1)
shot.eval_gd_overlap()
shot.init_gd()
shot.eval_phase()
fig = plt.figure()
fig.subplots_adjust(bottom=0.25)
ax1 = fig.add_subplot(111)
lns1 = ax1.plot(shot.freqs,
shot.gd,
marker='.',
import proc_profile_bottollier as ppb
import ref_functions as rf
if len(sys.argv) < 1:
shot_number = int(open('shot_number.txt', 'r').read())
else:
if (len(sys.argv) == 1) & ("py" in sys.argv[0]):
shot_number = int(open('shot_number.txt', 'r').read())
else:
shot_number = int(sys.argv[1])
factors = {'1': '1', '2/3': '2/3', '1/2': '1/2'}
pos = {}
cluster = 20
shot = ppb.ProcProfile(shot_number, save_locally=1)
sweep_ini = shot.time2sweep(100)
shot.reference_gd(all_shot=1, sw_clustersize=cluster)
shot.eval_freq_overlap()
shot.prepare_gd(sweep_ini, cluster, all_shot=1)
shot.eval_phase()
fig, ax = plt.subplots()
plt.subplots_adjust(bottom=0.25)
lines = {}
for f in factors.keys():
pos[f] = ppb.find_pos(shot.freqs[1:], shot.phi, factor=eval(f))
lines[f], = plt.plot(pos[f] * 1e2, shot.ne[1:], label=factors[f])
ax.legend(loc='upper left', title='Factor:')
ax.set_xlabel('r (cm)')
def main(argv):
"""Test maximum density."""
print(len(argv))
if len(argv) < 1:
shot_number = int(open('shot_number.txt', 'r').read())
else:
if (len(argv) == 1) & ("py" in argv[0]):
shot_number = int(open('shot_number.txt', 'r').read())
else:
shot_number = int(argv[1])
fig, ax = plt.subplots()
plt.subplots_adjust(bottom=0.25)
shot = ppb.ProcProfile(shot_number)
shot.reference_gd(all_shot=1)
cluster = 20
shot.plasma_gd(1, cluster, 1)
ax.pcolormesh(shot.X['K'], shot.Y['K'], shot.matrix_k_mean)
ax.pcolormesh(shot.X['Ka'], shot.Y['Ka'], shot.matrix_ka_mean)
plt.plot(shot.X['K'],
shot.Y['K'][shot.matrix_k_mean.argmax(axis=0)] - 3.36,
color='b',
linewidth=2.0)
plt.plot(shot.X['Ka'],
shot.Y['Ka'][shot.matrix_ka_mean.argmax(axis=0)] - 3.36,
color='b',
linewidth=2.0)
plt.plot(shot.X['K'],
shot.Y['K'][shot.matrix_k_mean.argmax(axis=0)],
color='b',
linewidth=2.0)
plt.plot(shot.X['Ka'],
shot.Y['Ka'][shot.matrix_ka_mean.argmax(axis=0)],
color='b',
linewidth=2.0)
l, = plt.plot(shot.X['K'],
shot.Y['K'][shot.matrix_k_mean.argmax(axis=0)],
color='r',
linewidth=2.0)
m, = plt.plot(shot.X['Ka'],
shot.Y['Ka'][shot.matrix_ka_mean.argmax(axis=0)],
color='r',
linewidth=2.0)
plt.xlabel("freq (GHz)")
plt.ylabel("group delay (ns)")
plt.title("# %s - time: %s ms" %
(shot.shot, shot.sweep2time(shot.sweep_cur)))
plt.ylim(0, 12)
plt.xlim(shot.X['K'].min(), shot.X['Ka'].max())
axcolor = 'lightgoldenrodyellow'
axfreq = plt.axes([0.25, 0.1, 0.65, 0.03], axisbg=axcolor)
sweep = Slider(axfreq,
'Sweep',
1,
len(shot.points) - 1 - cluster,
valinit=1,
valfmt='%1.f')
def update(val):
shot.plasma_gd(int(sweep.val), cluster, 1)
ax.pcolormesh(shot.X['K'], shot.Y['K'], shot.matrix_k_mean)
ax.pcolormesh(shot.X['Ka'], shot.Y['Ka'], shot.matrix_ka_mean)
l.set_ydata(shot.Y['K'][shot.matrix_k_mean.argmax(axis=0)])
m.set_ydata(shot.Y['Ka'][shot.matrix_ka_mean.argmax(axis=0)])
ax.set_title("# %s - time: %.3f ms" %
(shot.shot, shot.sweep2time(shot.sweep_cur)))
fig.canvas.draw_idle()
sweep.on_changed(update)
resetax = plt.axes([0.8, 0.025, 0.1, 0.04])
button = Button(resetax, 'Reset', color=axcolor, hovercolor='0.975')
def reset(event):
sweep.reset()
button.on_clicked(reset)
plt.show()