From d614f5ed7440468e7901fee9d31a4ea8e9191b69 Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Tue, 2 Jul 2024 12:03:15 +0200 Subject: [PATCH] astrodatagui: FlaredetectorWidget: show vlines are border for folded lc fit --- main/astrodatagui/ui/FlaredetectorWidget.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index 1ded0ed..a1666fd 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -7,10 +7,7 @@ from matplotlib.backends.backend_qtagg import ( from astropy.visualization import time_support import lightkurve as lk -from ...flaredetector.util import (findMaxIndices, plotFlarePeaks, - plotFlareFits, markFlare, - convertStarndardIndexToFoldedIndex, - getFoldedBestFit) +from ...flaredetector.util import * from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve class FlaredetectorWidget(QtWidgets.QWidget): @@ -256,6 +253,8 @@ class FlaredetectorWidget(QtWidgets.QWidget): lc.flux[lc.cycle == cycle][foldedIndex], "x", color="red") phase, sineFit = getFoldedBestFit(lc) self.figureAxis.plot(phase, sineFit, color="red") + minPhaseBoundsIndices, maxPhaseBoundsIndices = getPhaseRangesNearPeak(getFoldedFitPeakValley(sineFit), phase) + plotPhaseRangesNearPeak((minPhaseBoundsIndices, maxPhaseBoundsIndices), phase, ax=self.figureAxis) elif(self.PeriodogramState["Enabled"]): lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"]) if(self.PeriodogramState["View"] == "period"):