Merge branch 'develop' of ariadne.geophysik.ruhr-uni-bochum.de:/data/git/pylot into develop

pylot/core/analysis/magnitude.py

@@ -192,6 +192,14 @@ class LocalMagnitude(Magnitude):
 
     def peak_to_peak(self, st, t0):
 
+        try:
+            iplot = int(self.plot_flag)
+        except:
+            if self.plot_flag == True or self.plot_flag == 'True':
+                iplot = 2
+            else:
+                iplot = 0
+
         # simulate Wood-Anderson response
         st.simulate(paz_remove=None, paz_simulate=self._paz)
 
@@ -224,7 +232,7 @@ class LocalMagnitude(Magnitude):
 
         # check for plot flag (for debugging only)
         fig = None
-        if self.plot_flag > 1:
+        if iplot > 1:
             st.plot()
             fig = plt.figure()
             ax = fig.add_subplot(111)
@@ -455,6 +463,14 @@ def calcsourcespec(wfstream, onset, vp, delta, azimuth, incidence,
     if verbosity:
         print("Calculating source spectrum for station %s ...." % wfstream[0].stats.station)
 
+    try:
+        iplot = int(iplot)
+    except:
+        if iplot == True or iplot == 'True':
+            iplot = 2
+        else:
+            iplot = 0
+
     # get Q value
     Q, A = qp
 
@@ -651,6 +667,14 @@ def fitSourceModel(f, S, fc0, iplot, verbosity=False):
     :type:   float
     '''
 
+    try:
+        iplot = int(iplot)
+    except:
+        if iplot == True or iplot == 'True':
+            iplot = 2
+        else:
+            iplot = 0
+
     w0 = []
     stdw0 = []
     fc = []

pylot/core/io/phases.py

@@ -205,6 +205,8 @@ def picksdict_from_picks(evt):
         station = pick.waveform_id.station_code
         channel = pick.waveform_id.channel_code
         network = pick.waveform_id.network_code
+        mpp = pick.time
+        spe = pick.time_errors.uncertainty
         try:
             picker = str(pick.method_id)
             if picker.startswith('smi:local/'):
@@ -216,8 +218,6 @@ def picksdict_from_picks(evt):
         except KeyError as e:
             # print(e)
             onsets = {}
-        mpp = pick.time
-        spe = pick.time_errors.uncertainty
         try:
             lpp = mpp + pick.time_errors.upper_uncertainty
             epp = mpp - pick.time_errors.lower_uncertainty

pylot/core/pick/utils.py

@@ -1073,6 +1073,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