From 7816e6342f3abe3b402381bd1629a0f2822b6ec6 Mon Sep 17 00:00:00 2001
From: Sebastian Wehling-Benatelli <sebastian.wehling@rub.de>
Date: Mon, 13 Apr 2015 09:42:17 +0200
Subject: [PATCH] zooming for 3-component window changed now zooming is done by
 using the mouse wheel bugfix: calculation of the snr corrected

---
 pylot/core/pick/utils.py   | 10 ++++++----
 pylot/core/util/widgets.py | 34 +++++++++++++++++++++++++++++-----
 2 files changed, 35 insertions(+), 9 deletions(-)

diff --git a/pylot/core/pick/utils.py b/pylot/core/pick/utils.py
index 15de2312..e97dab3f 100644
--- a/pylot/core/pick/utils.py
+++ b/pylot/core/pick/utils.py
@@ -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
 
diff --git a/pylot/core/util/widgets.py b/pylot/core/util/widgets.py
index 1d62c65e..7124f565 100644
--- a/pylot/core/util/widgets.py
+++ b/pylot/core/util/widgets.py
@@ -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):