From ef455dcfbbdfee1999b1f30b10309016f4d92eeb Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Wed, 17 Apr 2024 08:54:46 +0200
Subject: [PATCH] flaredetector: add markFlare function

---
 main/flaredetector/flaredetector.py | 4 ++++
 1 file changed, 4 insertions(+)

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