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

Merge branch 'develop' of ariadne.geophysik.rub.de:/data/git/pylot into develop

Browse Source
This commit is contained in:
Ludger Küperkoch 2015-04-13 09:42:32 +02:00
commit fee0d05253
2 changed files with 35 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pylot/core/pick/utils.py
View File

@ -293,7 +293,8 @@ def getSNR(X, TSNR, t1):
'''
Function to calculate SNR of certain part of seismogram relative to
given time (onset) out of given noise and signal windows. A safety gap
between noise and signal part can be set. Returns SNR and SNR [dB].
between noise and signal part can be set. Returns SNR and SNR [dB] and
noiselevel.
:param: X, time series (seismogram)
:type: `~obspy.core.stream.Stream`
@ -324,9 +325,10 @@ def getSNR(X, TSNR, t1):
return
#calculate ratios
noiselevel = np.mean(abs(x[inoise]))
SNR = max(abs(x[isignal])) / noiselevel
SNRdB = 20 * np.log10(SNR)
noiselevel = np.sqrt(np.mean(np.square(x[inoise])))
signallevel = np.sqrt(np.mean(np.square(x[isignal])))
SNR = signallevel / noiselevel
SNRdB = 10 * np.log10(SNR)
return SNR, SNRdB, noiselevel

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

@ -25,7 +25,7 @@ from PySide.QtWebKit import QWebView
from obspy import Stream, UTCDateTime
from obspy.core.event import Pick, WaveformStreamID
from pylot.core.read import FilterOptions
from pylot.core.pick.utils import getSNR
from pylot.core.pick.utils import getSNR, earllatepicker
from pylot.core.util.defaults import OUTPUTFORMATS
from pylot.core.util import prepTimeAxis, getGlobalTimes
@ -85,7 +85,8 @@ class MPLWidget(FigureCanvas):
def setParent(self, parent):
self._parent = parent
def plotWFData(self, wfdata, title=None, zoomx=None, zoomy=None):
def plotWFData(self, wfdata, title=None, zoomx=None, zoomy=None,
noiselevel=None):
self.axes.cla()
self.clearPlotDict()
wfstart = getGlobalTimes(wfdata)[0]
@ -100,6 +101,11 @@ class MPLWidget(FigureCanvas):
trace.detrend('demean')
trace.normalize(trace.data.max() * 2)
self.axes.plot(time_ax, trace.data + n, 'k')
if noiselevel is not None:
self.axes.plot([time_ax[0], time_ax[-1]],
[noiselevel, noiselevel], '--k')
self.axes.plot([time_ax[0], time_ax[-1]],
[-noiselevel, -noiselevel], '--k')
xlabel = 'seconds since {0}'.format(wfstart)
ylabel = ''
self.updateWidget(xlabel, ylabel, title)
@ -381,6 +387,11 @@ class PickDlg(QDialog):
self.disconnectScrollEvent()
self.disconnectMotionEvent()
self.reconnectPressEvent(self.setIniPick)
else:
self.cidpress = self.connectPressEvent(self.panPress)
self.cidmotion = self.connectMotionEvent()
self.cidrelease = self.connectReleaseEvent()
self.cidscroll = self.connectScrollEvent()
def getComponents(self):
return self.components
@ -446,7 +457,7 @@ class PickDlg(QDialog):
'VLRW' : 15.
}
result = getSNR(wfdata, (10., 2., 1.5), ini_pick)
result = getSNR(wfdata, (5., .5, 1.), ini_pick)
snr = result[0]
noiselevel = result[2] * 1.5
@ -467,19 +478,32 @@ class PickDlg(QDialog):
title=self.getStation() +
' picking mode',
zoomx=zoomx,
zoomy=zoomy)
zoomy=zoomy,
noiselevel=noiselevel)
self.updateAPD(wfdata)
# reset labels
self.setPlotLabels()
def setPick(self, gui_event):
pick = gui_event.xdata
# setting pick
pick = gui_event.xdata # get pick time relative to the traces timeaxis not to the global
channel = self.getChannelID(round(gui_event.ydata))
wfdata = self.getAPD().copy().select(channel=channel)
# get earliest and latest possible pick
[epp, lpp, pickerror] = earllatepicker(wfdata, 1.5, (5., .5, 1.), pick, 1)
# plotting picks
ax = self.getPlotWidget().axes
ylims = ax.get_ylim()
ax.plot([pick, pick], ylims, 'r--')
ax.plot([epp, epp], ylims, 'c--')
ax.plot([lpp, lpp], ylims, 'm--')
self.getPlotWidget().draw()
def panPress(self, gui_event):