From b1a7c36f2251be2c3047932b76f3fc9214f02fed Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Wed, 19 Jun 2024 10:56:47 +0200 Subject: [PATCH] astrodatagui: FlaredetectorWidget: show folded fit --- main/astrodatagui/ui/FlaredetectorWidget.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index 6093a19..9d7e45b 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -4,13 +4,13 @@ from matplotlib.figure import Figure from matplotlib.backends.backend_qtagg import ( FigureCanvas, NavigationToolbar2QT as NavigationToolbar) -from astropy.io import fits from astropy.visualization import time_support import lightkurve as lk from ...flaredetector.util import (findMaxIndices, plotFlarePeaks, plotFlareFits, markFlare, - convertStarndardIndexToFoldedIndex) + convertStarndardIndexToFoldedIndex, + getFoldedBestFit) from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve class FlaredetectorWidget(QtWidgets.QWidget): @@ -254,6 +254,8 @@ class FlaredetectorWidget(QtWidgets.QWidget): cycle, foldedIndex = convertStarndardIndexToFoldedIndex(lc, peak["StandardIndex"]) self.figureAxis.plot(lc.phase[lc.cycle == cycle][foldedIndex].value, lc.flux[lc.cycle == cycle][foldedIndex], "x", color="red") + phase, sineFit = getFoldedBestFit(lc) + self.figureAxis.plot(phase, sineFit, color="red") elif(self.PeriodogramState["Enabled"]): lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"]) if(self.PeriodogramState["View"] == "period"):