From 9ba9f29331fc44bf424e9dc91afb9e2d84c1963f Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Thu, 13 Jun 2024 14:49:47 +0200 Subject: [PATCH] astrodatagui: FlaredetectorWidget: show flare peaks in folded view --- main/astrodatagui/ui/FlaredetectorWidget.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index cb442d6..789d892 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -8,7 +8,8 @@ from astropy.visualization import time_support import lightkurve as lk from ...flaredetector.util import (findMaxIndices, plotFlarePeaks, - plotFlareFits, markFlare) + plotFlareFits, markFlare, + convertStarndardIndexToFoldedIndex) from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve class FlaredetectorWidget(QtWidgets.QWidget): @@ -242,6 +243,10 @@ class FlaredetectorWidget(QtWidgets.QWidget): time_support() if(self.FoldState["Enabled"]): lc.scatter(label=label, ax=self.figureAxis) + for peak in self.peaks: + 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") elif(self.PeriodogramState["Enabled"]): lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"]) if(self.PeriodogramState["View"] == "period"):