From 0ca7aaecc01ea03c7b4e04410208b46e03db34ee Mon Sep 17 00:00:00 2001 From: jbrussell Date: Fri, 23 Apr 2021 11:58:10 -0400 Subject: [PATCH] Allow flexibility using N,E in place of 1,2. Will check first for 1,2 and if don't exist will use N,E --- orientpy/utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/orientpy/utils.py b/orientpy/utils.py index 0a6620e..9c781cf 100644 --- a/orientpy/utils.py +++ b/orientpy/utils.py @@ -481,8 +481,12 @@ def DLcalc(stream, Rf, LPF, HPF, epi, baz, A, winlen=10., ptype=0): st = stream.slice(starttime=dt+r1, endtime=dt+r2) # Extract waveform data for each component - tr1 = st.select(component='1')[0].data - tr2 = st.select(component='2')[0].data + try: + tr1 = st.select(component='1')[0].data + tr2 = st.select(component='2')[0].data + except: + tr1 = st.select(component='N')[0].data + tr2 = st.select(component='E')[0].data trZ = st.select(component='Z')[0].data # Calculate Hilbert transform of vertical trace data