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()