SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flaredetector: make sure we dont run into ranges without data

Browse Source
This commit is contained in:
SGCMarkus
2024-04-15 14:41:20 +02:00
parent 5bfa2b1898
commit 0f0001f7a3
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/flaredetector.py
+6
View File
@@ -36,6 +36,9 @@ def getFlareRange(flux, peakIndex):
if(minInd < 0): if(minInd < 0):
minInd = 0 minInd = 0
while(np.isnan(flux[minInd].unmasked)):
minInd += 1
deltaCount = 0 deltaCount = 0
delta = 100 delta = 100
prevFlux = flux[peakIndex] prevFlux = flux[peakIndex]
@@ -54,4 +57,7 @@ def getFlareRange(flux, peakIndex):
if(maxInd > len(flux)-1): if(maxInd > len(flux)-1):
maxInd = len(flux)-1 maxInd = len(flux)-1
while(np.isnan(flux[maxInd].unmasked)):
maxInd -= 1
return minInd, maxInd return minInd, maxInd