diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index bf87446..8086d03 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -38,6 +38,10 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05 return peaks, fits +def plotFlarePeaks(ax, peaks): + for peak in peaks: + ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red") + def getFlareRange(flux, peakIndex): minInd = -np.inf maxInd = np.inf