[bugfix] saving data did not work properly due to changed variable signature; picking on just one horizontal component now possible, crash was caused by mismatching component labels -> new function added to find these labels from the available data
This commit is contained in:
parent
d953609419
commit
49b07163ad
33
QtPyLoT.py
33
QtPyLoT.py
@ -493,18 +493,18 @@ class MainWindow(QMainWindow):
|
||||
fname = str(action.data().toString())
|
||||
return fname
|
||||
|
||||
def get_fnames(self):
|
||||
return self.fname
|
||||
def get_fnames(self, type='manual'):
|
||||
return self.fname[type]
|
||||
|
||||
def set_fname(self, fname, type):
|
||||
if self.get_fnames()[type] is not None:
|
||||
self.add_recentfile(self.get_fnames())
|
||||
if self.get_fnames(type) is not None:
|
||||
self.add_recentfile(self.get_fnames(type))
|
||||
self.fname[type] = fname
|
||||
|
||||
def getEventFileName(self):
|
||||
if self.get_fnames() is None:
|
||||
self.set_fname(self.get_data().getEventFileName())
|
||||
return self.get_fnames()
|
||||
def getEventFileName(self, type='manual'):
|
||||
if self.get_fnames(type) is None:
|
||||
self.set_fname(self.get_data().getEventFileName(), type)
|
||||
return self.get_fnames(type)
|
||||
|
||||
def saveData(self):
|
||||
|
||||
@ -551,9 +551,26 @@ class MainWindow(QMainWindow):
|
||||
fbasename, exform = getSavePath(e)
|
||||
except AttributeError as e:
|
||||
fbasename, exform = getSavePath(e)
|
||||
|
||||
# catch all possible cases before going on
|
||||
if not fbasename:
|
||||
return False
|
||||
# warn overwriting
|
||||
elif os.path.exists(fbasename + exform):
|
||||
ans = QMessageBox.question(self, self.tr("Overwrite file..."),
|
||||
self.tr("File already exists: {0}\n".format(fbasename + exform) + \
|
||||
"Overwrite file anyway?"),
|
||||
QMessageBox.Cancel, QMessageBox.Yes | QMessageBox.No,
|
||||
QMessageBox.Cancel)
|
||||
# only negative answers have to be caught
|
||||
if ans == QMessageBox.No:
|
||||
self.saveData()
|
||||
elif ans == QMessageBox.Cancel:
|
||||
return False
|
||||
|
||||
# export to given path
|
||||
self.get_data().exportEvent(fbasename, exform)
|
||||
# all files save (ui clean)
|
||||
self.setDirty(False)
|
||||
self.update_status('Event saved as %s' % (fbasename + exform))
|
||||
return True
|
||||
|
@ -8,7 +8,7 @@ import os
|
||||
import pwd
|
||||
import re
|
||||
import subprocess
|
||||
from obspy.core import UTCDateTime
|
||||
from obspy import UTCDateTime, read
|
||||
|
||||
|
||||
def _pickle_method(m):
|
||||
@ -354,6 +354,29 @@ def prepTimeAxis(stime, trace):
|
||||
return time_ax
|
||||
|
||||
|
||||
def find_horizontals(data):
|
||||
"""
|
||||
takes `obspy.core.stream.Stream` object and returns a list containing the component labels of the horizontal components available
|
||||
:param data: waveform data
|
||||
:type data: `obspy.core.stream.Stream`
|
||||
:return: components list
|
||||
:rtype: list
|
||||
|
||||
..example::
|
||||
|
||||
>>> st = read()
|
||||
>>> find_horizontals(st)
|
||||
[u'N', u'E']
|
||||
"""
|
||||
rval = []
|
||||
for tr in data:
|
||||
if tr.stats.channel[-1].upper() in ['Z', '3']:
|
||||
continue
|
||||
else:
|
||||
rval.append(tr.stats.channel[-1].upper())
|
||||
return rval
|
||||
|
||||
|
||||
def scaleWFData(data, factor=None, components='all'):
|
||||
"""
|
||||
produce scaled waveforms from given waveform data and a scaling factor,
|
||||
|
@ -33,7 +33,7 @@ from pylot.core.pick.compare import Comparison
|
||||
from pylot.core.util.defaults import OUTPUTFORMATS, FILTERDEFAULTS, LOCTOOLS, \
|
||||
COMPPOSITION_MAP
|
||||
from pylot.core.util.utils import prepTimeAxis, getGlobalTimes, scaleWFData, \
|
||||
demeanTrace, isSorted, findComboBoxIndex, clims
|
||||
demeanTrace, isSorted, findComboBoxIndex, clims, find_horizontals
|
||||
|
||||
|
||||
def getDataType(parent):
|
||||
@ -898,13 +898,8 @@ class PickDlg(QDialog):
|
||||
inoise = getnoisewin(t, ini_pick, noise_win, gap_win)
|
||||
trace = demeanTrace(trace, inoise)
|
||||
|
||||
# account for non-oriented horizontal waveforms
|
||||
try:
|
||||
horiz_comp = ('n', 'e')
|
||||
data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
|
||||
except IndexError as e:
|
||||
print('warning: {0}'.format(e))
|
||||
horiz_comp = ('1', '2')
|
||||
# scale waveform for plotting
|
||||
horiz_comp = find_horizontals(data)
|
||||
data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
|
||||
|
||||
x_res = getResolutionWindow(snr)
|
||||
|
Loading…
Reference in New Issue
Block a user