marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[update] EVERY trace now scaled relative to its own noiselevel

Browse Source
This commit is contained in:
Marcel Paffrath 2018-01-17 11:47:52 +01:00
parent d2437b5014
commit 473529961e
2 changed files with 42 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pylot/RELEASE-VERSION
View File

@ -1 +1 @@
98f7-dirty
d243-dirty

39
pylot/core/util/widgets.py
View File

@ -1921,6 +1921,12 @@ class PickDlg(QDialog):
st += data.select(channel=action.text())
return st
def calcNoiseScaleFactor(self, noiselevel, zoomfactor=5., norm=1):
# calculate factor to upscale a trace normed to 'norm' in a way that all values
# zoomfactor*noiselevel are found within -0.5*norm and 0.5*norm
scaleFactor = (norm/2.) / (zoomfactor * noiselevel)
return scaleFactor
def setIniPick(self, gui_event):
self.multicompfig.set_frame_color('green')
trace_number = round(gui_event.ydata)
@ -1955,18 +1961,13 @@ class PickDlg(QDialog):
noise_win = twins[0]
gap_win = twins[1]
signal_win = twins[2]
itrace = int(trace_number)
while itrace > len(wfdata) - 1:
itrace -= 1
stime = self.getStartTime()
stime_diff = wfdata[itrace].stats.starttime - stime
# copy data for plotting
data = self.getWFData().copy()
data = self.getPickPhases(data, 'P')
data.normalize(global_max=True)
data.normalize()
if not data:
QtGui.QMessageBox.warning(self, 'No channel to plot',
'No channel to plot for phase: {}.'.format('P'))
@ -1985,12 +1986,12 @@ class PickDlg(QDialog):
'Denied', 'setIniPickP: Could not filter waveform: {}'.format(e))
self.qmb.show()
snr = []
noiselevels = {}
# determine SNR and noiselevel
for trace in data.traces:
st = data.select(channel=trace.stats.channel)
stime_diff = trace.stats.starttime - stime
result = getSNR(st, (noise_win, gap_win, signal_win), ini_pick - stime_diff)
snr.append(result[0])
noiselevel = result[2]
@ -2005,6 +2006,12 @@ class PickDlg(QDialog):
t = prepTimeAxis(trace.stats.starttime - stime, trace)
inoise = getnoisewin(t, ini_pick, noise_win, gap_win)
trace = demeanTrace(trace, inoise)
# upscale trace data in a way that each trace is vertically zoomed to noiselevel*factor
channel = trace.stats.channel
noiselevel = noiselevels[channel]
noiseScaleFactor = self.calcNoiseScaleFactor(noiselevel, zoomfactor=5.)
trace.data *= noiseScaleFactor
noiselevels[channel] *= noiseScaleFactor
x_res = getResolutionWindow(np.mean(snr), parameter.get('extent'))
@ -2013,7 +2020,6 @@ class PickDlg(QDialog):
plot_additional = bool(self.compareChannel.currentText())
additional_channel = self.compareChannel.currentText()
scale_channel = self.scaleChannel.currentText()
self.multicompfig.plotWFData(wfdata=data,
title=self.getStation() +
' picking mode',
@ -2023,8 +2029,7 @@ class PickDlg(QDialog):
scaleddata=True,
iniPick=ini_pick,
plot_additional=plot_additional,
additional_channel=additional_channel,
scaleToChannel=scale_channel)
additional_channel=additional_channel)
def setIniPickS(self, gui_event, wfdata):
@ -2038,12 +2043,11 @@ class PickDlg(QDialog):
signal_win = twins[2]
stime = self.getStartTime()
stime_diff = wfdata[0].stats.starttime - stime
# copy data for plotting
data = self.getWFData().copy()
data = self.getPickPhases(data, 'S')
data.normalize(global_max=True)
data.normalize()
if not data:
QtGui.QMessageBox.warning(self, 'No channel to plot',
'No channel to plot for phase: {}.'.format('S'))
@ -2067,6 +2071,7 @@ class PickDlg(QDialog):
# determine SNR and noiselevel
for trace in data.traces:
st = data.select(channel=trace.stats.channel)
stime_diff = trace.stats.starttime - stime
result = getSNR(st, (noise_win, gap_win, signal_win), ini_pick - stime_diff)
snr.append(result[0])
noiselevel = result[2]
@ -2081,10 +2086,12 @@ class PickDlg(QDialog):
t = prepTimeAxis(trace.stats.starttime - stime, trace)
inoise = getnoisewin(t, ini_pick, noise_win, gap_win)
trace = demeanTrace(trace, inoise)
# scale waveform for plotting
#horiz_comp = find_horizontals(data)
#data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
# upscale trace data in a way that each trace is vertically zoomed to noiselevel*factor
channel = trace.stats.channel
noiselevel = noiselevels[channel]
noiseScaleFactor = self.calcNoiseScaleFactor(noiselevel, zoomfactor=5.)
trace.data *= noiseScaleFactor
noiselevels[channel] *= noiseScaleFactor
x_res = getResolutionWindow(np.mean(snr), parameter.get('extent'))