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

[add] some hacks to display air pressure in plot

Browse Source
This commit is contained in:
Marcel Paffrath 2018-01-08 13:47:40 +01:00
parent 8c9fded873
commit 08703bf691
1 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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()