From 08703bf691b374e07210a56d271d1a5f7e4f95d3 Mon Sep 17 00:00:00 2001
From: Marcel <marcel.paffrath@rub.de>
Date: Mon, 8 Jan 2018 13:47:40 +0100
Subject: [PATCH] [add] some hacks to display air pressure in plot

---
 pylot/core/util/widgets.py | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/pylot/core/util/widgets.py b/pylot/core/util/widgets.py
index f755965e..e2b2b6ab 100644
--- a/pylot/core/util/widgets.py
+++ b/pylot/core/util/widgets.py
@@ -901,7 +901,8 @@ class PylotCanvas(FigureCanvas):
 
     def plotWFData(self, wfdata, title=None, zoomx=None, zoomy=None,
                    noiselevel=None, scaleddata=False, mapping=True,
-                   component='*', nth_sample=1, iniPick=None, verbosity=0):
+                   component='*', nth_sample=1, iniPick=None, verbosity=0,
+                   plot_additional=False):
         ax = self.axes[0]
         ax.cla()
 
@@ -965,6 +966,17 @@ class PylotCanvas(FigureCanvas):
                                 color = linecolor,
                                 linestyle = 'dashed')
                 self.setPlotDict(n, (station, channel, network))
+        if plot_additional:
+            pressure = wfdata.select(channel='MDO')
+            if pressure:
+                p_data = pressure[0].data
+                #normalize
+                p_max = max(abs(p_data))
+                p_data /= p_max
+                for index in range(3):
+                    ax.plot(times, p_data, color='red', alpha=0.5, linewidth=0.7)
+                    p_data += 1
+
         if iniPick:
             ax.vlines(iniPick, ax.get_ylim()[0], ax.get_ylim()[1],
                       colors='m', linestyles='dashed',
@@ -1166,6 +1178,8 @@ class PickDlg(QDialog):
             self.filteroptions = FILTERDEFAULTS
         self.pick_block = False
         self.nextStation = QtGui.QCheckBox('Continue with next station.')
+        self.additionalChannel = QtGui.QCheckBox('Additional Channel')
+        self.additionalChannel.stateChanged.connect(self.resetPlot)
 
         # initialize panning attributes
         self.press = None
@@ -1326,6 +1340,7 @@ class PickDlg(QDialog):
                                    'padding-left:5px}')
         _dialtoolbar.addWidget(est_label)
         _dialtoolbar.addWidget(self.plot_arrivals_button)
+        _dialtoolbar.addWidget(self.additionalChannel)
 
         # layout the innermost widget
         _innerlayout = QVBoxLayout()
@@ -1614,7 +1629,8 @@ class PickDlg(QDialog):
         self.reset_p_button()
         self.reset_s_button()
         self.multicompfig.plotWFData(wfdata=self.getWFData(),
-                                        title=self.getStation())
+                                     title=self.getStation(),
+                                     plot_additional=self.additionalChannel.isChecked())
         self.drawAllPicks()
         self.drawArrivals()
         self.setPlotLabels()
@@ -1822,7 +1838,8 @@ class PickDlg(QDialog):
                                         zoomy=self.getYLims(),
                                         noiselevel=(trace_number + noiselevel,
                                                     trace_number - noiselevel),
-                                        iniPick=ini_pick)
+                                        iniPick=ini_pick,
+                                        plot_additional=self.additionalChannel.isChecked())
 
     def setIniPickS(self, gui_event, wfdata):
 
@@ -1890,7 +1907,8 @@ class PickDlg(QDialog):
                                         zoomy=self.getYLims(),
                                         noiselevel=noiselevels,
                                         scaleddata=True,
-                                        iniPick=ini_pick)
+                                        iniPick=ini_pick,
+                                        plot_additional=self.additionalChannel.isChecked())
 
     def setPick(self, gui_event):
 
@@ -2154,7 +2172,8 @@ class PickDlg(QDialog):
             title = old_title
         self.multicompfig.plotWFData(wfdata=data, title=title,
                                         zoomx=self.getXLims(),
-                                        zoomy=self.getYLims())
+                                        zoomy=self.getYLims(),
+                                        plot_additional=self.additionalChannel.isChecked())
         self.setPlotLabels()
         self.drawAllPicks()
         self.draw()
@@ -2165,7 +2184,8 @@ class PickDlg(QDialog):
         title = self.multicompfig.axes[0].get_title()
         self.multicompfig.plotWFData(wfdata=data, title=title,
                                         zoomx=self.getXLims(),
-                                        zoomy=self.getYLims())
+                                        zoomy=self.getYLims(),
+                                        plot_additional=self.additionalChannel.isChecked())
         self.setPlotLabels()
         self.drawAllPicks()
         self.draw()