diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index 9e1bedb..e34d4b3 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -52,6 +52,10 @@ def plotFlareFits(ax, fits): for fit in fits: ax.plot(fit["FlareFitTime"], fit["FlareFit"], "g--") +def markFlare(ax, fits, flux): + for fit in fits: + ax.plot(fit["FlareFitTime"], flux[fit["StandardIndex"]], color="red") + def getFlareRange(flux, peakIndex): minInd = -np.inf maxInd = np.inf