From b56a2e3701a17900a4c8ec62ce072b2a274908d5 Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Mon, 15 Apr 2024 16:01:52 +0200
Subject: [PATCH] flaredetector: add function to plot fits 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 8086d03..58aebe4 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -42,6 +42,10 @@ def plotFlarePeaks(ax, peaks):
     for peak in peaks:
         ax.plot(peak["FlarePeakTime"], peak["FlarePeak"], "x", color="red")
 
+def plotFlareFits(ax, fits):
+    for fit in fits:
+        ax.plot(fit["FlareFitTime"], fit["FlareFit"], "g--")
+
 def getFlareRange(flux, peakIndex):
     minInd = -np.inf
     maxInd = np.inf