From e64135c7a839c6d4f74036d5d32b6eb2aef75975 Mon Sep 17 00:00:00 2001
From: Ludger Kueperkoch <kueperkoch@igem-energie.de>
Date: Wed, 29 Jul 2020 12:14:42 +0200
Subject: [PATCH] [Bugfix]: Check for key S and be sure to get element weight
 for pick dictionary.

---
 pylot/core/io/phases.py  | 26 +++++++++++++++-----------
 pylot/core/pick/utils.py |  9 +++++++--
 2 files changed, 22 insertions(+), 13 deletions(-)

diff --git a/pylot/core/io/phases.py b/pylot/core/io/phases.py
index f657bea4..39416ee6 100644
--- a/pylot/core/io/phases.py
+++ b/pylot/core/io/phases.py
@@ -269,21 +269,24 @@ def picksdict_from_picks(evt):
                 msg = str(e) + ',\n falling back to symmetric uncertainties'
                 lpp = mpp + spe
                 epp = mpp - spe
-                # get onset weight from uncertainty
-                infile = os.path.join(os.path.expanduser('~'), '.pylot', 'pylot.in')
-                print('Using default input file {}'.format(infile))
-                parameter = PylotParameter(infile)
-                if pick.phase_hint == 'P':
-                    errors = parameter['timeerrorsP']
-                elif pick.phase_hint == 'S':
-                    errors = parameter['timeerrorsS']
-                weight = get_quality_class(spe, errors)
             warnings.warn(msg)
         phase['mpp'] = mpp
         phase['epp'] = epp
         phase['lpp'] = lpp
         phase['spe'] = spe
-        phase['weight'] = weight
+        try:
+            phase['weight'] = weight
+        except:
+            # get onset weight from uncertainty
+            infile = os.path.join(os.path.expanduser('~'), '.pylot', 'pylot.in')
+            print('Using default input file {}'.format(infile))
+            parameter = PylotParameter(infile)
+            if pick.phase_hint == 'P':
+                errors = parameter['timeerrorsP']
+            elif pick.phase_hint == 'S':
+                errors = parameter['timeerrorsS']
+            weight = get_quality_class(spe, errors)
+            phase['weight'] = weight
         phase['channel'] = channel
         phase['network'] = network
         phase['picker'] = picker
@@ -782,7 +785,8 @@ def writephases(arrivals, fformat, filename, parameter=None, eventinfo=None):
                     Ponset = arrivals[key]['P']['mpp']
                     Prt = Ponset - stime  # onset time relative to source time
                     fid.write('%s    %6.3f  1  P\n' % (key, Prt))
-                    # S onsets
+            if arrivals[key].has_key('S'):
+                # S onsets
                 if arrivals[key]['S']['weight'] < 4:
                     Sonset = arrivals[key]['S']['mpp']
                     Srt = Sonset - stime  # onset time relative to source time
diff --git a/pylot/core/pick/utils.py b/pylot/core/pick/utils.py
index 29b35b59..de290978 100644
--- a/pylot/core/pick/utils.py
+++ b/pylot/core/pick/utils.py
@@ -316,8 +316,13 @@ def fmpicker(Xraw, Xfilt, pickwin, Pick, iplot=0, fig=None, linecolor='k'):
             islope2 = np.where((t >= Pick) & (t <= Pick + t[imax2]))
             # calculate slope as polynomal fit of order 1
             xslope2 = np.arange(0, len(xfilt[islope2]), 1)
-            P2 = np.polyfit(xslope2, xfilt[islope2], 1)
-            datafit2 = np.polyval(P2, xslope2)
+            try:
+                P2 = np.polyfit(xslope2, xfilt[islope2], 1)
+                datafit2 = np.polyval(P2, xslope2)
+            except ValueError as e:
+                emsg = 'fmpicker: polyfit failed: {}'.format(e)
+                print(emsg)
+                return FM
 
         # compare results
         if P1 is not None and P2 is not None: