def dataFunc(args):
shot = args
pulses = sat.findColdPulses(shot)
print shot, "data started"
elecTree = MDSplus.Tree('electrons', shot)
teNode = elecTree.getNode('\gpc_t0')
te = teNode.data()
time = teNode.dim_of().data()
peaks = sat.findSawteeth(time, te, 0.57, 1.43)
print shot, "data done"
return shot, te, time , peaks, pulses
midRot = simps(td.pro[1,:,minRad:maxRad]*td.rho[minRad:maxRad], x=np.sqrt(td.rho[minRad:maxRad]), axis=1)
"""
specTree = MDSplus.Tree('spectroscopy', shot)
rotNode = specTree.getNode('\SPECTROSCOPY::TOP.HIREX_SR.ANALYSIS.Z:VEL')
if ncols > 1:
ax = axarr[i,j]
else:
ax = axarr[i]
ax.plot(rotNode.dim_of().data(), rotNode.data()[0])
pulses = sat.findColdPulses(shot)
for p in pulses:
ax.axvline(x=p, c='b', ls='--')
ax.set_title(str(shot), y=0.8)
k += 1
f.subplots_adjust(hspace=0, wspace=0)
plt.show()
