flaredetector: add calculateFlareFitsForLightcurve

this function fits all flares for a lightcurve

main/flaredetector/flaredetector.py

@@ -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