From 062729672794f71724d603b9a9ef89d404058a2c Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Mon, 15 Apr 2024 16:00:21 +0200
Subject: [PATCH] flaredetector: add function to plot fit peaks to an axis

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

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