[bugfix] closes #233 closes #234

can cope with stations without horizontal components now, removed dangerous try/except construct

PyLoT.py

@@ -2440,11 +2440,13 @@ class MainWindow(QMainWindow):
             canvas.setZoomBorders2content()
         if self.tap.pylot_picks:
             station = self.tap.get_current_station()
-            p_pick = self.tap.pylot_picks[station]['P']
-            s_pick = self.tap.pylot_picks[station]['S']
+            p_pick = self.tap.pylot_picks[station].get('P')
+            if p_pick:
                 self.tap.pickDlg.autopicks['P_tuning'] = p_pick
-            self.tap.pickDlg.autopicks['S_tuning'] = s_pick
                 self.tap.pickDlg.drawPicks(phase='P_tuning', picktype='auto', picks=p_pick)
+            s_pick = self.tap.pylot_picks[station].get('S')
+            if s_pick:
+                self.tap.pickDlg.autopicks['S_tuning'] = s_pick
                 self.tap.pickDlg.drawPicks(phase='S_tuning', picktype='auto', picks=s_pick)
 
     def autoPick(self):

pylot/core/pick/autopick.py

@@ -8,6 +8,7 @@ function conglomerate utils.
 
 :author: MAGS2 EP3 working group / Ludger Kueperkoch
 """
+import traceback
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -975,15 +976,14 @@ def autopickstation(wfstream, pickparam, verbose=False,
         ax1.set_ylim([-1.5, 1.5])
         ax1.set_ylabel('Normalized Counts')
         # fig.suptitle(tr_filt.stats.starttime)
-        try:
-            len(edat[0])
-        except:
+        # only continue if one horizontal stream exists
+        if (ndat or edat) and Sflag == 1:
+            # mirror components in case one does not exist
+            if not edat:
                 edat = ndat
-        try:
-            len(ndat[0])
-        except:
+            if not ndat:
                 ndat = edat
-        if len(edat[0]) > 1 and len(ndat[0]) > 1 and Sflag == 1:
+            if len(edat[0]) > 1 and len(ndat[0]) > 1:
                 # plot horizontal traces
                 ax2 = fig.add_subplot(3, 1, 2, sharex=ax1)
                 th1data = np.arange(0,
@@ -1101,12 +1101,22 @@ def autopickstation(wfstream, pickparam, verbose=False,
         epickP = zdat[0].stats.starttime - timeerrorsP[3]
         mpickP = zdat[0].stats.starttime
 
+    # create dictionary
+    # for P phase
+    ccode = zdat[0].stats.channel
+    ncode = zdat[0].stats.network
+    ppick = dict(channel=ccode, network=ncode, lpp=lpickP, epp=epickP, mpp=mpickP, spe=Perror, snr=SNRP,
+                 snrdb=SNRPdB, weight=Pweight, fm=FM, w0=None, fc=None, Mo=None,
+                 Mw=None, picker=picker, marked=Pmarker)
+
     if edat:
         hdat = edat[0]
     elif ndat:
         hdat = ndat[0]
     else:
-        return
+        # no horizontal components given
+        picks = dict(P=ppick)
+        return picks
 
     if lpickS is not None and lpickS == mpickS:
         lpickS += hdat.stats.delta
@@ -1123,13 +1133,6 @@ def autopickstation(wfstream, pickparam, verbose=False,
         epickS = hdat.stats.starttime - timeerrorsS[3]
         mpickS = hdat.stats.starttime
 
-    # create dictionary
-    # for P phase
-    ccode = zdat[0].stats.channel
-    ncode = zdat[0].stats.network
-    ppick = dict(channel=ccode, network=ncode, lpp=lpickP, epp=epickP, mpp=mpickP, spe=Perror, snr=SNRP,
-                 snrdb=SNRPdB, weight=Pweight, fm=FM, w0=None, fc=None, Mo=None,
-                 Mw=None, picker=picker, marked=Pmarker)
     # add S phase
     ccode = hdat.stats.channel
     ncode = hdat.stats.network