flaredetector: add calculateFlareFitsForLightcurve

this function fits all flares for a lightcurve
2024-04-15 15:19:54 +02:00
parent 77e23f480a
commit b8c38762b1
1 changed files with 12 additions and 0 deletions
@@ -22,6 +22,18 @@ def calculateFitForFlare(lightcurve, peakIndex):
    flareFitDataPoints = flareFitDataPoints + peakIndex
    return Peak+1, flareFitDataPoints, fit+1

+def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05):
+    maxIndices = findMaxIndices(lightcurve, num, distance=distance, height=height)
+    fits = []
+    for index in maxIndices:
+        Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index)
+        fits.append({"FlarePeak": Peak,
+                     "FlarePeakTime": lightcurve.time[index],
+                     "FlareFit": fit,
+                     "FlareFitTime": lightcurve.time[flareFitDataPoints]})
+
+    return fits
+
 def getFlareRange(flux, peakIndex):
    minInd = -np.inf
    maxInd = np.inf