From ad8c22978ebf5f5a4292db8b99a6d0f4160a2a4b Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Wed, 17 Apr 2024 08:35:58 +0200
Subject: [PATCH] flaredetector: support plotting peaks onto given
 non-normalized fluxes

---
 main/flaredetector/flaredetector.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py
index 58aebe4..856d7c6 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -32,15 +32,21 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05
     for index in maxIndices:
         Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index)
         peaks.append({"FlarePeak": Peak,
-                      "FlarePeakTime": lightcurve.time[index]})
+                      "FlarePeakTime": lightcurve.time[index],
+                      "StandardIndex": index})
         fits.append({"FlareFit": fit,
-                     "FlareFitTime": lightcurve.time[flareFitDataPoints]})
+                     "FlareFitTime": lightcurve.time[flareFitDataPoints],
+                     "StandardIndex": flareFitDataPoints})
 
     return peaks, fits
 
-def plotFlarePeaks(ax, peaks):
-    for peak in peaks:
-        ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red")
+def plotFlarePeaks(ax, peaks, flux = None):
+    if(flux is None):
+        for peak in peaks:
+            ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red")
+    else:
+        for peak in peaks:
+            ax.plot(peak["FlarePeakTime"], flux[peak["StandardIndex"]], "x", color="red")
 
 def plotFlareFits(ax, fits):
     for fit in fits: