From 57e454cb3b86d25a892876d7bcdf46d280f5228d Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Wed, 17 Apr 2024 08:55:42 +0200
Subject: [PATCH] flaredetector: move plotting functions to util

---
 main/flaredetector/flaredetector.py | 16 ----------------
 main/flaredetector/util.py          | 18 +++++++++++++++++-
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py
index e34d4b3..6064299 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -40,22 +40,6 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05
 
     return peaks, fits
 
-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:
-        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
diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py
index 2735594..d4c8f8f 100644
--- a/main/flaredetector/util.py
+++ b/main/flaredetector/util.py
@@ -17,4 +17,20 @@ def findMaxIndices(lc, num=3, distance=100, height=(None, None), sortByHighest=F
     if(sortByHighest):
         return peak_indices[np.argsort(peak_heights)[-num:][::-1]]
     else:
-        return peak_indices[0:num]
\ No newline at end of file
+        return peak_indices[0:num]
+
+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:
+        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")
\ No newline at end of file