From 537b7cb4a393336f058e2342ebd6e6c4d485daa6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ludger=20K=C3=BCperkoch?= Date: Fri, 11 Aug 2017 16:57:10 +0200 Subject: [PATCH] [Bugfix] Take into account different formats of input arguments iplot and fig. --- pylot/core/pick/utils.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/pylot/core/pick/utils.py b/pylot/core/pick/utils.py index 5f1a8d43..212928d0 100644 --- a/pylot/core/pick/utils.py +++ b/pylot/core/pick/utils.py @@ -975,6 +975,15 @@ def checkZ4S(X, pick, zfac, checkwin, iplot, fig=None): are shown : type: int ''' + + try: + iplot = int(iplot) + except: + if iplot == True or iplot == 'True': + iplot = 2 + else: + iplot = 0 + assert isinstance(X, Stream), "%s is not a stream object" % str(X) @@ -1052,10 +1061,14 @@ def checkZ4S(X, pick, zfac, checkwin, iplot, fig=None): t = np.arange(diff_dict[key], trace.stats.npts / trace.stats.sampling_rate + diff_dict[key], trace.stats.delta) if i == 0: + if fig == None or fig == 'None': + fig = plt.figure() # self.iplot) ### WHY? MP MP ax1 = fig.add_subplot(3, 1, i + 1) ax = ax1 ax.set_title('CheckZ4S, Station %s' % zdat[0].stats.station) else: + if fig == None or fig == 'None': + fig = plt.figure() # self.iplot) ### WHY? MP MP ax = fig.add_subplot(3, 1, i + 1, sharex=ax1) ax.plot(t, abs(trace.data), color='b', label='abs') ax.plot(t, trace.data, color='k')