[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

pylot/core/util/utils.py

@@ -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,

pylot/core/util/widgets.py

@@ -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,14 +898,9 @@ 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')
-            data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
+        # scale waveform for plotting
+        horiz_comp = find_horizontals(data)
+        data = scaleWFData(data, noiselevel * 2.5, horiz_comp)
 
         x_res = getResolutionWindow(snr)