flaredetector: split peaks/fits for calculateFlareFitsForLightcurve

2024-04-15 15:58:10 +02:00
parent c8ad11187e
commit af824867fb
1 changed files with 5 additions and 4 deletions
@@ -27,15 +27,16 @@ def calculateFitForFlare(lightcurve, peakIndex):

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

-    return fits
+    return peaks, fits

 def getFlareRange(flux, peakIndex):
    minInd = -np.inf