[add] phase defaults for phasesTab

This commit is contained in:
Marcel Paffrath 2017-07-19 15:12:01 +02:00
parent 7748a39a72
commit c644095b5a
2 changed files with 371 additions and 225 deletions

View File

@ -2583,7 +2583,8 @@ class MainWindow(QMainWindow):
def PyLoTprefs(self):
if not self._props:
self._props = PropertiesDlg(self, infile=self.infile)
self._props = PropertiesDlg(self, infile=self.infile,
inputs=self._inputs)
if self._props.exec_():
self.init_wfWidget()

View File

@ -38,6 +38,7 @@ from PySide.QtWebKit import QWebView
from obspy import Stream, UTCDateTime
from obspy.core.util import AttribDict
from obspy.taup import TauPyModel
from obspy.taup.utils import get_phase_names
from pylot.core.io.data import Data
from pylot.core.io.inputs import FilterOptions, PylotParameter
from pylot.core.pick.utils import getSNR, earllatepicker, getnoisewin, \
@ -48,6 +49,7 @@ from pylot.core.util.utils import prepTimeAxis, full_range, scaleWFData, \
demeanTrace, isSorted, findComboBoxIndex, clims
from autoPyLoT import autoPyLoT
from pylot.core.util.thread import Thread
if sys.version_info.major == 3:
import icons_rc_3 as icons_rc
elif sys.version_info.major == 2:
@ -59,7 +61,8 @@ if pg:
pg.setConfigOption('background', 'w')
pg.setConfigOption('foreground', 'k')
pg.setConfigOptions(antialias=True)
#pg.setConfigOption('leftButtonPan', False)
# pg.setConfigOption('leftButtonPan', False)
def getDataType(parent):
type = QInputDialog().getItem(parent, "Select phases type", "Type:",
@ -72,13 +75,14 @@ def getDataType(parent):
return type
def plot_pdf(_axes, x, y, annotation, bbox_props, xlabel=None, ylabel=None,
title=None):
# try method or data
try:
_axes.plot(x, y()) # y provided as method
_axes.plot(x, y()) # y provided as method
except:
_axes.plot(x, y) # y provided as data
_axes.plot(x, y) # y provided as data
if title:
_axes.set_title(title)
@ -91,6 +95,7 @@ def plot_pdf(_axes, x, y, annotation, bbox_props, xlabel=None, ylabel=None,
return _axes
def createAction(parent, text, slot=None, shortcut=None, icon=None,
tip=None, checkable=False):
"""
@ -109,6 +114,7 @@ def createAction(parent, text, slot=None, shortcut=None, icon=None,
action.setCheckable(True)
return action
class ComparisonDialog(QDialog):
def __init__(self, c, parent=None):
self._data = c
@ -258,7 +264,7 @@ class ComparisonDialog(QDialog):
_ax1 = self.canvas.figure.add_subplot(_gs[2, 0])
_ax2 = self.canvas.figure.add_subplot(_gs[2, 1])
#_axes.cla()
# _axes.cla()
station = self.plotprops['station']
phase = self.plotprops['phase']
pdf = self.data.comparison[station][phase]
@ -268,11 +274,11 @@ class ComparisonDialog(QDialog):
annotation = "%s difference on %s\n" \
"expectation: %7.4f s\n" \
"std: %7.4f s" % (phase, station,
exp, std)
exp, std)
bbox_props = dict(boxstyle='round', facecolor='lightgrey', alpha=.7)
plot_pdf(_axes, x, y, annotation, bbox_props, 'time difference [s]',
'propability density [-]', phase)
'propability density [-]', phase)
pdf_a = copy.deepcopy(self.data.get('auto')[station][phase])
pdf_m = copy.deepcopy(self.data.get('manu')[station][phase])
@ -297,11 +303,11 @@ class ComparisonDialog(QDialog):
# set annotation text
mannotation = "probability density of manual pick\n" \
"expectation: %7.4f s\n" \
"std: %7.4f s" % (expmanu-x0.timestamp, stdmanu)
"std: %7.4f s" % (expmanu - x0.timestamp, stdmanu)
aannotation = "probability density of automatic pick\n" \
"expectation: %7.4f s\n" \
"std: %7.4f s" % (expauto-x0.timestamp, stdauto)
"std: %7.4f s" % (expauto - x0.timestamp, stdauto)
_ax1 = plot_pdf(_ax1, xmanu, ymanu, mannotation,
bbox_props=bbox_props, xlabel='seconds since '
@ -414,7 +420,7 @@ class PlotWidget(FigureCanvas):
class WaveformWidgetPG(QtGui.QWidget):
def __init__(self, parent=None, xlabel='x', ylabel='y', title='Title'):
QtGui.QWidget.__init__(self, parent)#, 1)
QtGui.QWidget.__init__(self, parent) # , 1)
self.setParent(parent)
self._parent = parent
# attribute plotdict is a dictionary connecting position and a name
@ -443,7 +449,7 @@ class WaveformWidgetPG(QtGui.QWidget):
if self.plotWidget.sceneBoundingRect().contains(pos):
mousePoint = self.plotWidget.getPlotItem().vb.mapSceneToView(pos)
x, y, = (mousePoint.x(), mousePoint.y())
#if x > 0:# and index < len(data1):
# if x > 0:# and index < len(data1):
wfID = self._parent.getWFID(y)
station = self._parent.getStationName(wfID)
if self._parent.get_current_event():
@ -482,7 +488,7 @@ class WaveformWidgetPG(QtGui.QWidget):
if not component == '*':
alter_comp = compclass.getCompPosition(component)
#alter_comp = str(alter_comp[0])
# alter_comp = str(alter_comp[0])
st_select = wfdata.select(component=component)
st_select += wfdata.select(component=alter_comp)
@ -499,9 +505,9 @@ class WaveformWidgetPG(QtGui.QWidget):
try:
self.plotWidget.getPlotItem().vb.setLimits(xMin=float(0),
xMax=float(wfend-wfstart),
xMax=float(wfend - wfstart),
yMin=-0.5,
yMax=len(nsc)+0.5)
yMax=len(nsc) + 0.5)
except:
print('Warning: Could not set zoom limits')
@ -511,7 +517,7 @@ class WaveformWidgetPG(QtGui.QWidget):
if mapping:
comp = channel[-1]
n = compclass.getPlotPosition(str(comp))
#n = n[0]
# n = n[0]
if n > nmax:
nmax = n
if verbosity:
@ -523,8 +529,8 @@ class WaveformWidgetPG(QtGui.QWidget):
if not scaleddata:
trace.detrend('constant')
trace.normalize(np.max(np.abs(trace.data)) * 2)
times = [time for index, time in enumerate(time_ax) if not index%nth_sample]
data = [datum + n for index, datum in enumerate(trace.data) if not index%nth_sample]
times = [time for index, time in enumerate(time_ax) if not index % nth_sample]
data = [datum + n for index, datum in enumerate(trace.data) if not index % nth_sample]
plots.append((times, data))
self.setPlotDict(n, (station, channel, network))
self.xlabel = 'seconds since {0}'.format(wfstart)
@ -572,7 +578,7 @@ class WaveformWidgetPG(QtGui.QWidget):
self.plotWidget.getPlotItem().setTitle(text)
self.draw()
def updateWidget(self):#, xlabel, ylabel, title):
def updateWidget(self): # , xlabel, ylabel, title):
self.updateXLabel(self.xlabel)
self.updateYLabel(self.ylabel)
self.updateTitle(self.title)
@ -636,7 +642,7 @@ class WaveformWidget(FigureCanvas):
if not component == '*':
alter_comp = compclass.getCompPosition(component)
#alter_comp = str(alter_comp[0])
# alter_comp = str(alter_comp[0])
st_select = wfdata.select(component=component)
st_select += wfdata.select(component=alter_comp)
@ -656,7 +662,7 @@ class WaveformWidget(FigureCanvas):
if mapping:
comp = channel[-1]
n = compclass.getPlotPosition(str(comp))
#n = n[0]
# n = n[0]
if n > nmax:
nmax = n
if verbosity:
@ -668,13 +674,13 @@ class WaveformWidget(FigureCanvas):
if not scaleddata:
trace.detrend('constant')
trace.normalize(np.max(np.abs(trace.data)) * 2)
times = [time for index, time in enumerate(time_ax) if not index%nth_sample]
data = [datum + n for index, datum in enumerate(trace.data) if not index%nth_sample]
times = [time for index, time in enumerate(time_ax) if not index % nth_sample]
data = [datum + n for index, datum in enumerate(trace.data) if not index % nth_sample]
self.getAxes().plot(times, data, 'k', linewidth=0.7)
if noiselevel is not None:
for level in noiselevel:
self.getAxes().plot([time_ax[0], time_ax[-1]],
[level, level], '--k')
for level in noiselevel:
self.getAxes().plot([time_ax[0], time_ax[-1]],
[level, level], '--k')
if iniPick:
ax = self.getAxes()
ax.vlines(iniPick, ax.get_ylim()[0], ax.get_ylim()[1],
@ -736,8 +742,69 @@ class WaveformWidget(FigureCanvas):
axann.set_bbox(dict(facecolor='lightgrey', alpha=.6))
class PhaseDefaults(QtGui.QDialog):
def __init__(self, parent=None, nrow=10,
phase_defaults=['P', 'S'],
current_phases=[]):
super(PhaseDefaults, self).__init__(parent)
self.nrow = nrow
self.main_layout = QtGui.QVBoxLayout()
self.sub_layout = QtGui.QGridLayout()
self.phase_names = phase_defaults
self.current_phases = current_phases
self.setButtons()
self.setupUi()
self.connectSignals()
self.setWindowTitle('Default Phases')
self.selected_phases = []
def setButtons(self):
self._buttonbox = QDialogButtonBox(QDialogButtonBox.Ok |
QDialogButtonBox.Cancel)
def setupUi(self):
self.setLayout(self.main_layout)
self.main_layout.addLayout(self.sub_layout)
self.init_phase_layout()
self.main_layout.addWidget(self._buttonbox)
def connectSignals(self):
self._buttonbox.accepted.connect(self.accept)
self._buttonbox.rejected.connect(self.reject)
def init_phase_layout(self):
self._checkboxes = {}
row = 0
column = 0
for index, phase in enumerate(self.phase_names):
if row > self.nrow:
column += 1
row = 0
checkbox = self.create_phase_box(phase)
self.sub_layout.addWidget(checkbox,
row, column)
self._checkboxes[phase] = checkbox
checkbox.setChecked(bool(phase in self.current_phases))
row += 1
def create_phase_box(self, phase_name):
checkbox = QtGui.QCheckBox(phase_name)
return checkbox
def update_selected_phases(self):
self.selected_phases = []
for phase in self.phase_names:
if self._checkboxes[phase].isChecked():
self.selected_phases.append(phase)
def accept(self):
self.update_selected_phases()
QtGui.QDialog.accept(self)
class PickDlg(QDialog):
update_picks = QtCore.Signal(dict)
def __init__(self, parent=None, data=None, station=None, network=None, picks=None,
autopicks=None, rotate=False, parameter=None, embedded=False, model='iasp91'):
super(PickDlg, self).__init__(parent)
@ -1048,8 +1115,8 @@ class PickDlg(QDialog):
phaseSelect = {'P': self.p_phase_select,
'S': self.s_phase_select}
nHotkey = 4 # max hotkeys per phase
hotkey = 1 # start hotkey
nHotkey = 4 # max hotkeys per phase
hotkey = 1 # start hotkey
# loop over P and S (use explicit list instead of iter over dict.keys to keep order)
for phaseIndex, phaseID in enumerate(['P', 'S']):
@ -1070,7 +1137,7 @@ class PickDlg(QDialog):
slot=slot,
shortcut=shortcut)
picksMenu.addAction(picksAction)
self.picksActions[str(phase)] = picksAction # save action in dictionary
self.picksActions[str(phase)] = picksAction # save action in dictionary
if phaseIndex == 0:
picksMenu.addSeparator()
@ -1167,7 +1234,7 @@ class PickDlg(QDialog):
self.activatePicking()
self.currentPhase = str(self.s_button.text())
def set_button_color(self, button, color = None):
def set_button_color(self, button, color=None):
if type(color) == QtGui.QColor:
palette = button.palette()
role = button.backgroundRole()
@ -1235,7 +1302,7 @@ class PickDlg(QDialog):
def getStation(self):
if self.network and self.station:
return self.network+'.'+self.station
return self.network + '.' + self.station
return self.station
def getPlotWidget(self):
@ -1365,7 +1432,7 @@ class PickDlg(QDialog):
itrace -= 1
stime = self.getStartTime()
stime_diff = wfdata[itrace].stats.starttime-stime
stime_diff = wfdata[itrace].stats.starttime - stime
# copy data for plotting
data = self.getWFData().copy()
@ -1382,7 +1449,7 @@ class PickDlg(QDialog):
'Denied', 'setIniPickP: Could not filter waveform: {}'.format(e))
self.qmb.show()
result = getSNR(wfdata, (noise_win, gap_win, signal_win), ini_pick-stime_diff, itrace)
result = getSNR(wfdata, (noise_win, gap_win, signal_win), ini_pick - stime_diff, itrace)
snr = result[0]
noiselevel = result[2]
@ -1423,7 +1490,7 @@ class PickDlg(QDialog):
signal_win = twins[2]
stime = self.getStartTime()
stime_diff = wfdata[0].stats.starttime-stime
stime_diff = wfdata[0].stats.starttime - stime
# copy data for plotting
data = self.getWFData().copy()
@ -1441,7 +1508,7 @@ class PickDlg(QDialog):
self.qmb.show()
# determine SNR and noiselevel
result = getSNR(wfdata, (noise_win, gap_win, signal_win), ini_pick-stime_diff)
result = getSNR(wfdata, (noise_win, gap_win, signal_win), ini_pick - stime_diff)
snr = result[0]
noiselevel = result[2]
@ -1460,7 +1527,7 @@ class PickDlg(QDialog):
horiz_comp = find_horizontals(data)
data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
x_res = getResolutionWindow(snr, parameter.get('extent'))
x_res = getResolutionWindow(snr, parameter.get('inp'))
self.setXLims(tuple([ini_pick - x_res, ini_pick + x_res]))
traces = self.getTraceID(horiz_comp)
@ -1506,8 +1573,6 @@ class PickDlg(QDialog):
'Denied', 'setPick: Could not filter waveform: {}'.format(e))
self.qmb.show()
# get earliest and latest possible pick and symmetric pick error
if wfdata[0].stats.channel[2] == 'Z' or wfdata[0].stats.channel[2] == '3':
nfac = parameter.get('nfacP')
@ -1518,10 +1583,10 @@ class PickDlg(QDialog):
# return absolute time values for phases
stime = self.getStartTime()
stime_diff = wfdata[0].stats.starttime-stime
stime_diff = wfdata[0].stats.starttime - stime
[epp, lpp, spe] = earllatepicker(wfdata, nfac, (TSNR[0], TSNR[1], TSNR[2]),
pick-stime_diff, verbosity=1)
pick - stime_diff, verbosity=1)
mpp = stime + pick
if epp:
@ -1645,7 +1710,7 @@ class PickDlg(QDialog):
if not self.picks and not self.autopicks:
return
# init empty list and get station starttime
X=[]
X = []
starttime = self.getStartTime()
# init dictionarys to iterate through and iterate over them
allpicks = [self.picks, self.autopicks]
@ -1655,7 +1720,7 @@ class PickDlg(QDialog):
# add relative pick time, phaseID and picktype index
X.append((pick_rel, phase, index_ptype))
# find index and value closest to x
index, value = min(enumerate([val[0] for val in X]), key=lambda y: abs(y[1]-x))
index, value = min(enumerate([val[0] for val in X]), key=lambda y: abs(y[1] - x))
# unpack the found value
pick_rel, phase, index_ptype = X[index]
# delete the value from corresponding dictionary
@ -1867,6 +1932,7 @@ class TuneAutopicker(QWidget):
:param: parent
:type: QtPyLoT Mainwindow
'''
def __init__(self, parent):
QtGui.QWidget.__init__(self, parent, 1)
self.parent = parent
@ -1890,8 +1956,8 @@ class TuneAutopicker(QWidget):
self.metadata = self.parent.metadata
else:
self.metadata = None
#self.setWindowModality(QtCore.Qt.WindowModality.ApplicationModal)
#self.setWindowFlags(self.windowFlags() | QtCore.Qt.WindowStaysOnTopHint)
# self.setWindowModality(QtCore.Qt.WindowModality.ApplicationModal)
# self.setWindowFlags(self.windowFlags() | QtCore.Qt.WindowStaysOnTopHint)
def init_main_layouts(self):
self.main_layout = QtGui.QVBoxLayout()
@ -1935,7 +2001,7 @@ class TuneAutopicker(QWidget):
stations.sort()
model = self.stationBox.model()
for network, station in stations:
item = QtGui.QStandardItem(network+'.'+station)
item = QtGui.QStandardItem(network + '.' + station)
if station in self.get_current_event().pylot_picks:
item.setBackground(self.parent._colors['ref'])
model.appendRow(item)
@ -2047,7 +2113,7 @@ class TuneAutopicker(QWidget):
st = self.data.getWFData()
tr = st.select(station=self.get_current_station())[0]
starttime = tr.stats.starttime
p_axes=[
p_axes = [
('mainFig', 0),
('aicFig', 0),
('slength', 0),
@ -2055,7 +2121,7 @@ class TuneAutopicker(QWidget):
('el_Ppick', 0),
('fm_picker', 0),
('fm_picker', 1)]
s_axes=[
s_axes = [
('mainFig', 1),
('mainFig', 2),
('aicARHfig', 0),
@ -2076,24 +2142,24 @@ class TuneAutopicker(QWidget):
self.plot_manual_Spick_to_ax(ax, (picks['S']['mpp'] - starttime))
def plot_manual_Ppick_to_ax(self, ax, pick):
y_top = 0.9*ax.get_ylim()[1]
y_bot = 0.9*ax.get_ylim()[0]
y_top = 0.9 * ax.get_ylim()[1]
y_bot = 0.9 * ax.get_ylim()[0]
ax.vlines(pick, y_bot, y_top,
color='teal', linewidth=2, label='manual P Onset')
ax.plot([pick-0.5, pick+0.5],
ax.plot([pick - 0.5, pick + 0.5],
[y_bot, y_bot], linewidth=2, color='teal')
ax.plot([pick-0.5, pick+0.5],
ax.plot([pick - 0.5, pick + 0.5],
[y_top, y_top], linewidth=2, color='teal')
ax.legend()
def plot_manual_Spick_to_ax(self, ax, pick):
y_top = 0.9*ax.get_ylim()[1]
y_bot = 0.9*ax.get_ylim()[0]
y_top = 0.9 * ax.get_ylim()[1]
y_bot = 0.9 * ax.get_ylim()[0]
ax.vlines(pick, y_bot, y_top,
color='magenta', linewidth=2, label='manual S Onset')
ax.plot([pick-0.5, pick+0.5],
ax.plot([pick - 0.5, pick + 0.5],
[y_bot, y_bot], linewidth=2, color='magenta')
ax.plot([pick-0.5, pick+0.5],
ax.plot([pick - 0.5, pick + 0.5],
[y_top, y_top], linewidth=2, color='magenta')
ax.legend()
@ -2162,7 +2228,7 @@ class TuneAutopicker(QWidget):
if project.getEventFromPath(index):
if not project.getEventFromPath(index).isTestEvent():
break
#in case all events are marked as test events and last event is reached
# in case all events are marked as test events and last event is reached
if index == nevents - 1:
index = -1
self.eventBox.setCurrentIndex(index)
@ -2200,7 +2266,7 @@ class TuneAutopicker(QWidget):
self.ap_thread.finished.connect(self.finish_picker)
self.figure_tabs.setCurrentIndex(4)
self.ap_thread.start()
#picks = autoPyLoT(self.parameter, fnames='None', iplot=2, fig_dict=self.fig_dict)
# picks = autoPyLoT(self.parameter, fnames='None', iplot=2, fig_dict=self.fig_dict)
def finish_picker(self):
self.enable(True)
@ -2212,8 +2278,8 @@ class TuneAutopicker(QWidget):
if not self.pylot_picks:
self._warn('No picks found. See terminal output.')
return
#renew tabs
#self.fill_figure_tabs()
# renew tabs
# self.fill_figure_tabs()
self.set_stretch()
self.update.emit('Update')
self.figure_tabs.setCurrentIndex(1)
@ -2241,7 +2307,7 @@ class TuneAutopicker(QWidget):
if hasattr(self, 'pickDlg'):
if self.pickDlg:
self.pickDlg.setParent(None)
del(self.pickDlg)
del (self.pickDlg)
if hasattr(self, 'overview'):
self.overview.setParent(None)
if hasattr(self, 'p_tabs'):
@ -2268,6 +2334,7 @@ class TuneAutopicker(QWidget):
class PylotParaBox(QtGui.QWidget):
accepted = QtCore.Signal(str)
rejected = QtCore.Signal(str)
def __init__(self, parameter, parent=None):
'''
Generate Widget containing parameters for PyLoT.
@ -2321,7 +2388,7 @@ class PylotParaBox(QtGui.QWidget):
def _init_dialog_buttons(self):
self._dialog_buttons = QtGui.QHBoxLayout()
self._okay = QtGui.QPushButton('Ok')
self._okay = QtGui.QPushButton('Ok')
self._close = QtGui.QPushButton('Close')
self._apply = QtGui.QPushButton('Apply')
self._dialog_buttons.addWidget(self._okay)
@ -2593,7 +2660,7 @@ class PylotParaBox(QtGui.QWidget):
continue
box = self.boxes[param]
value = self.parameter[param]
#self.parameter.checkValue(param, value)
# self.parameter.checkValue(param, value)
self.setValue(box, value)
def setValue(self, box, value):
@ -2681,18 +2748,19 @@ class PylotParaBox(QtGui.QWidget):
class PropertiesDlg(QDialog):
def __init__(self, parent=None, infile=None):
def __init__(self, parent=None, infile=None, inputs=None):
super(PropertiesDlg, self).__init__(parent)
self.infile = infile
self.inputs = inputs
self.setWindowTitle("PyLoT Properties")
self.tabWidget = QTabWidget()
self.tabWidget.addTab(InputsTab(self), "Inputs")
#self.tabWidget.addTab(OutputsTab(self), "Outputs")
self.tabWidget.addTab(PhasesTab(self), "Phases")
# self.tabWidget.addTab(OutputsTab(self), "Outputs")
self.tabWidget.addTab(PhasesTab(self, inputs), "Phases")
self.tabWidget.addTab(GraphicsTab(self), "Graphics")
#self.tabWidget.addTab(LocalisationTab(self), "Loc. Tools")
# self.tabWidget.addTab(LocalisationTab(self), "Loc. Tools")
self.tabWidget.addTab(LocalisationTab(self), "NonLinLoc")
self.tabWidget.addTab(ChannelOrderTab(self), "Channel Order")
self.buttonBox = QDialogButtonBox(QDialogButtonBox.Ok |
@ -2712,7 +2780,7 @@ class PropertiesDlg(QDialog):
self.buttonBox.button(QDialogButtonBox.RestoreDefaults).clicked.connect(self.restore)
def getinfile(self):
return self.infile
return self.infile
def accept(self, *args, **kwargs):
self.apply()
@ -2752,7 +2820,6 @@ class PropertiesDlg(QDialog):
for values in self._current_values():
self.setValues(values)
@staticmethod
def setValues(tabValues):
settings = QSettings()
@ -2844,9 +2911,9 @@ class InputsTab(PropTab):
para = PylotParameter(infile)
datstruct = para.get('datastructure')
if datstruct == 'SeisComp':
index = 0
index = 0
else:
index = 2
index = 2
datapath = para.get('datapath')
rootpath = para.get('rootpath')
database = para.get('database')
@ -2858,6 +2925,7 @@ class InputsTab(PropTab):
"data/Structure": self.structureSelect.setCurrentIndex(index)}
return values
class OutputsTab(PropTab):
def __init__(self, parent=None, infile=None):
super(OutputsTab, self).__init__(parent)
@ -2889,12 +2957,17 @@ class OutputsTab(PropTab):
class PhasesTab(PropTab):
def __init__(self, parent=None):
def __init__(self, parent=None, inputs=None):
super(PhasesTab, self).__init__(parent)
self.inputs = inputs
self.Pphases = 'P, Pg, Pn, PmP, P1, P2, P3'
self.Sphases = 'S, Sg, Sn, SmS, S1, S2, S3'
self.PphasesEdit = QLineEdit()
self.SphasesEdit = QLineEdit()
self.pickDefaultsButton = QtGui.QPushButton('Choose default phases...')
PphasesLabel = QLabel("P Phases to pick")
SphasesLabel = QLabel("S Phases to pick")
@ -2905,24 +2978,96 @@ class PhasesTab(PropTab):
self.PphasesEdit.setText("%s" % Pphases)
self.SphasesEdit.setText("%s" % Sphases)
self.main_layout = QtGui.QHBoxLayout()
layout = QGridLayout()
layout.addWidget(PphasesLabel, 0, 0)
layout.addWidget(SphasesLabel, 1, 0)
layout.addWidget(self.PphasesEdit, 0, 1)
layout.addWidget(self.SphasesEdit, 1, 1)
self.setLayout(layout)
self.main_layout.addLayout(layout)
self.main_layout.addWidget(self.pickDefaultsButton)
self.setLayout(self.main_layout)
self.connectSignals()
def connectSignals(self):
self.pickDefaultsButton.clicked.connect(self.get_defaults)
def get_defaults(self):
phases = [p.strip() for p in self.Pphases.split(',')] + [s.strip() for s in self.Sphases.split(',')]
p_current = [p.strip() for p in self.PphasesEdit.text().split(',')]
s_current = [s.strip() for s in self.SphasesEdit.text().split(',')]
current_phases = p_current + s_current
if self.inputs:
parameter = self.inputs
if parameter.get('extent') == 'global':
# get all default phase names known to obspy.taup
# in a list
phases = get_phase_names('ttall')
phaseDefaults = PhaseDefaults(self, phase_defaults=phases,
current_phases=current_phases)
if phaseDefaults.exec_():
phase_dict = self.sortPhases(phaseDefaults.selected_phases)
p_phases=''
s_phases=''
for index, p in enumerate(phase_dict['P']):
p_phases += p
if not index == len(phase_dict['P']) - 1:
p_phases += ', '
for index, s in enumerate(phase_dict['S']):
s_phases += s
if not index == len(phase_dict['S']) - 1:
s_phases += ', '
self.PphasesEdit.setText(p_phases)
self.SphasesEdit.setText(s_phases)
def sortPhases(self, phases):
sorted_phases = {'P': [],
'S': []}
# suffix for phase name if not phase identified by last letter (P, p, etc.)
alt_PS = ['diff', 'n', 'g', '1', '2', '3'] # alternative suffix
for phase in phases:
self.loopIdentifyPhase(phase, alt_PS, sorted_phases)
return sorted_phases
def loopIdentifyPhase(self, phase, alt_PS, sorted_phases):
phase_copy = phase
while not self.identifyPhase(phase_copy):
identified = False
for alt_suf in alt_PS:
if phase_copy.endswith(alt_suf):
phase_copy = phase_copy.split(alt_suf)[0]
identified = True
if not identified:
phase_copy = phase_copy[:-1]
if len(phase_copy) < 1:
print('Warning: Could not identify phase {}!'.format(phase))
return
sorted_phases[self.identifyPhase(phase_copy)].append(phase)
def identifyPhase(self, phase):
# common phase suffix for P and S
common_P = ['P', 'p']
common_S = ['S', 's']
if phase[-1] in common_P:
return 'P'
if phase[-1] in common_S:
return 'S'
else:
return False
def getValues(self):
values = {'p_phases': self.PphasesEdit.text(),
's_phases': self.SphasesEdit.text()}
p_phases = self.PphasesEdit.text()
s_phases = self.SphasesEdit.text()
values = {'p_phases': p_phases,
's_phases': s_phases}
return values
def resetValues(self, infile=None):
Pphases = 'P, Pg, Pn, PmP, P1, P2, P3'
Sphases = 'S, Sg, Sn, SmS, S1, S2, S3'
values = {'p_phases': self.PphasesEdit.setText(Pphases),
's_phases': self.SphasesEdit.setText(Sphases)}
values = {'p_phases': self.PphasesEdit.setText(self.Pphases),
's_phases': self.SphasesEdit.setText(self.Sphases)}
return values
@ -3057,9 +3202,9 @@ class ChannelOrderTab(PropTab):
"Channel E [east/west, default=2]": self.ChannelOrderEEdit.setText("%d" % Edefault)}
return values
# MP MP: No idea why this function exists!?
# def getComponents(self):
# self.CompName = dict(Z='10', N='11', E='12')
# MP MP: No idea why this function exists!?
# def getComponents(self):
# self.CompName = dict(Z='10', N='11', E='12')
class LocalisationTab(PropTab):
@ -3069,14 +3214,14 @@ class LocalisationTab(PropTab):
settings = QSettings()
curtool = settings.value("loc/tool", None)
#loctoollabel = QLabel("location tool")
# loctoollabel = QLabel("location tool")
self.locToolComboBox = QComboBox()
#loctools = LOCTOOLS.keys()
#self.locToolComboBox.addItems(loctools)
# loctools = LOCTOOLS.keys()
# self.locToolComboBox.addItems(loctools)
#toolind = findComboBoxIndex(self.locToolComboBox, curtool)
# toolind = findComboBoxIndex(self.locToolComboBox, curtool)
#self.locToolComboBox.setCurrentIndex(toolind)
# self.locToolComboBox.setCurrentIndex(toolind)
curroot = settings.value("{0}/rootPath".format(curtool), None)
curbin = settings.value("{0}/binPath".format(curtool), None)
@ -3098,13 +3243,13 @@ class LocalisationTab(PropTab):
binBrowse = QPushButton('...', self)
binBrowse.clicked.connect(lambda: self.selectDirectory(self.binedit))
#self.locToolComboBox.currentIndexChanged.connect(self.updateUi)
# self.locToolComboBox.currentIndexChanged.connect(self.updateUi)
self.updateUi()
layout = QGridLayout()
#layout.addWidget(loctoollabel, 0, 0)
#layout.addWidget(self.locToolComboBox, 0, 1)
# layout.addWidget(loctoollabel, 0, 0)
# layout.addWidget(self.locToolComboBox, 0, 1)
layout.addWidget(self.rootlabel, 1, 0)
layout.addWidget(self.rootedit, 1, 1)
layout.addWidget(rootBrowse, 1, 2)
@ -3116,7 +3261,7 @@ class LocalisationTab(PropTab):
def updateUi(self):
curtool = self.locToolComboBox.currentText()
#if curtool is not None:
# if curtool is not None:
self.rootlabel.setText("{0} root directory".format(curtool))
self.binlabel.setText("{0} bin directory".format(curtool))
@ -3130,7 +3275,7 @@ class LocalisationTab(PropTab):
loctool = self.locToolComboBox.currentText()
values = {"{0}/rootPath".format(loctool): self.rootedit.text(),
"{0}/binPath".format(loctool): self.binedit.text()}
#"loc/tool": loctool}
# "loc/tool": loctool}
return values
def resetValues(self, infile):