diff --git a/QtPyLoT.py b/QtPyLoT.py index a8ccdfb5..2cf0e6bb 100755 --- a/QtPyLoT.py +++ b/QtPyLoT.py @@ -64,7 +64,7 @@ from pylot.core.io.data import Data from pylot.core.io.inputs import FilterOptions, PylotParameter from autoPyLoT import autoPyLoT from pylot.core.pick.compare import Comparison -from pylot.core.pick.utils import symmetrize_error +from pylot.core.pick.utils import symmetrize_error, getQualityfromUncertainty from pylot.core.io.phases import picksdict_from_picks import pylot.core.loc.nll as nll from pylot.core.util.defaults import FILTERDEFAULTS, SetChannelComponents @@ -1951,13 +1951,19 @@ class MainWindow(QMainWindow): } stat_picks = self.getPicks(type=picktype)[station] - stime = self.getStime() for phase in stat_picks: picks = stat_picks[phase] if type(stat_picks[phase]) is not dict and type(stat_picks[phase]) is not AttribDict: return + + # get quality classes + if phase[0] == 'P': + quality = getQualityfromUncertainty(picks['spe'], self._inputs['timeerrorsP']) + elif phase[0] == 'S': + quality = getQualityfromUncertainty(picks['spe'], self._inputs['timeerrorsS']) + colors = phase_col[phase[0].upper()] mpp = picks['mpp'] - stime @@ -1994,7 +2000,8 @@ class MainWindow(QMainWindow): else: pw.plot([mpp, mpp], ylims, pen=colors[0], name='{}-Pick (NO PICKERROR)'.format(phase)) elif picktype == 'auto': - pw.plot([mpp, mpp], ylims, pen=colors[3]) + if quality < 4: + pw.plot([mpp, mpp], ylims, pen=colors[3]) else: raise TypeError('Unknown picktype {0}'.format(picktype)) else: @@ -2009,9 +2016,10 @@ class MainWindow(QMainWindow): else: ax.plot([mpp, mpp], ylims, colors[6], label='{}-Pick (NO PICKERROR)'.format(phase)) elif picktype == 'auto': - ax.plot(mpp, ylims[1], colors[3], - mpp, ylims[0], colors[4]) - ax.vlines(mpp, ylims[0], ylims[1], colors[5], linestyles='dotted') + if quality < 4: + ax.plot(mpp, ylims[1], colors[3], + mpp, ylims[0], colors[4]) + ax.vlines(mpp, ylims[0], ylims[1], colors[5], linestyles='dotted') else: raise TypeError('Unknown picktype {0}'.format(picktype)) diff --git a/pylot/core/pick/utils.py b/pylot/core/pick/utils.py index 41a5509a..5f1a8d43 100644 --- a/pylot/core/pick/utils.py +++ b/pylot/core/pick/utils.py @@ -1069,6 +1069,31 @@ def checkZ4S(X, pick, zfac, checkwin, iplot, fig=None): ax.set_xlabel('Time [s] since %s' % zdat[0].stats.starttime) return returnflag +def getQualityfromUncertainty(uncertainty, Errors): + '''Script to transform uncertainty into quality classes 0-4 + regarding adjusted time errors Errors. + ''' + + if uncertainty == None or uncertainty == 'None': + quality = 4 + else: + if uncertainty <= Errors[0]: + quality = 0 + elif (uncertainty > Errors[0]) and \ + (uncertainty < Errors[1]): + quality = 1 + elif (uncertainty > Errors[1]) and \ + (uncertainty < Errors[2]): + quality = 2 + elif (uncertainty > Errors[2]) and \ + (uncertainty < Errors[3]): + quality = 3 + elif uncertainty > Errors[3]: + quality = 4 + else: + pass + + return quality if __name__ == '__main__': import doctest