def runHELIOSprep(SHOT,pathNAM,fnamH5,fnamMAT,CHDsetupNum=6,downSAMPfac=100,bacPNTSM=5,bacinterppnt=350,bacfrac=30,quiet=True): rawDF,header = prep.loadtoPDS(SHOT,quiet=False) downsamDF,smDF,bacoffDF = prep.makWavs(rawDF,header,baseNum=downSAMPfac,bacPNTSM=bacPNTSM,bacinterppnt=bacinterppnt,bacfrac=bacfrac,quiet=quiet) truncDF = prep.truncWavs(SHOT,bacoffDF,fnamH5,quiet=quiet) SHEdata = gST.load2DICT(fnamH5) finDF = mR.run(SHOT,truncDF,CHDsetNum=CHDsetupNum,CHDpath=pathNAM,quiet=quiet) return rawDF,downsamDF,smDF,bacoffDF,truncDF,finDF,SHEdata
def getTimes(shot,fnamH5,truncVal=0.9,fnamMAT=None,load2HDF=False,fixSHE=True,quiet=True,debug=False):
if load2HDF:
if fnamMAT == None:
fnamMAT='/Users/unterbee/Desktop/shot_118800.mat'
gST.getShotmat(118800,fnamMAT)
dataDICT = gST.load2DICT(fnamH5)
if fixSHE:
oldX=dataDICT['x_she1']
oldY=dataDICT['y_she1']
dataDICT['x_she_corr']=oldX[:]+0.04
dataDICT['y_she_corr']=oldY[:]-np.mean(oldY[0:10])
ind = np.where((dataDICT['y_she_corr']/dataDICT['y_she_corr'].max()) > truncVal)
else:
ind = np.where((dataDICT['y_she1']/dataDICT['y_she1'].max()) > truncVal)
tValues = dataDICT['x_she_corr'][ind]
if debug and not quiet:
print "num data, num trunc {} {}".format(dataDICT['x_she_corr'].shape,tValues.shape)
return tValues,ind
