flaredetector: make sure we dont run into ranges without data

2024-04-15 14:41:20 +02:00
parent 5bfa2b1898
commit 0f0001f7a3
1 changed files with 6 additions and 0 deletions
@@ -36,6 +36,9 @@ def getFlareRange(flux, peakIndex):
    if(minInd < 0):
        minInd = 0

+    while(np.isnan(flux[minInd].unmasked)):
+        minInd += 1
+
    deltaCount = 0
    delta = 100
    prevFlux = flux[peakIndex]
@@ -54,4 +57,7 @@ def getFlareRange(flux, peakIndex):
    if(maxInd > len(flux)-1):
        maxInd = len(flux)-1

+    while(np.isnan(flux[maxInd].unmasked)):
+        maxInd -= 1
+
    return minInd, maxInd