flaredetector: add markFlare function

2024-04-17 08:54:46 +02:00
parent ffacde3386
commit ef455dcfbb
1 changed files with 4 additions and 0 deletions
@@ -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