[fixes #181] now picking on horizontal components 1, 2 and N,E is possible

This commit is contained in:
Sebastian Wehling-Benatelli 2016-03-02 13:37:40 +01:00
parent 827a0da210
commit 848d11270b

View File

@ -422,6 +422,11 @@ class PickDlg(QDialog):
trace = selectTrace(trace, 'NE') trace = selectTrace(trace, 'NE')
if trace: if trace:
wfdata.append(trace) wfdata.append(trace)
elif component == '1' or component == '2':
for trace in self.getWFData():
trace = selectTrace(trace, '12')
if trace:
wfdata.append(trace)
elif component == 'Z': elif component == 'Z':
wfdata = self.getWFData().select(component=component) wfdata = self.getWFData().select(component=component)
return wfdata return wfdata
@ -527,10 +532,16 @@ class PickDlg(QDialog):
inoise = getnoisewin(t, ini_pick, noise_win, gap_win) inoise = getnoisewin(t, ini_pick, noise_win, gap_win)
trace = demeanTrace(trace, inoise) trace = demeanTrace(trace, inoise)
horiz_comp = ('n', 'e') try:
horiz_comp = ('n', 'e')
data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
except IndexError as e:
print('warning: {0}'.format(e))
horiz_comp = ('1', '2')
data = scaleWFData(data, noiselevel * 2.5, horiz_comp) data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
x_res = getResolutionWindow(snr) x_res = getResolutionWindow(snr)
self.setXLims(tuple([ini_pick - x_res, ini_pick + x_res])) self.setXLims(tuple([ini_pick - x_res, ini_pick + x_res]))