Merge branch 'master' of https://github.com/nfsi-canada/OrientPy

nfsi-canada · May 13, 2021 · 2ce408f · 2ce408f
2 parents 159763b + ba15599
commit 2ce408f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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