astrodatagui: FlaredetectorWidget: show flare peaks in folded view

This commit is contained in:
2024-06-13 14:49:47 +02:00
parent b5479b5dc1
commit 9ba9f29331
+6 -1
View File
@@ -8,7 +8,8 @@ from astropy.visualization import time_support
import lightkurve as lk import lightkurve as lk
from ...flaredetector.util import (findMaxIndices, plotFlarePeaks, from ...flaredetector.util import (findMaxIndices, plotFlarePeaks,
plotFlareFits, markFlare) plotFlareFits, markFlare,
convertStarndardIndexToFoldedIndex)
from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve
class FlaredetectorWidget(QtWidgets.QWidget): class FlaredetectorWidget(QtWidgets.QWidget):
@@ -242,6 +243,10 @@ class FlaredetectorWidget(QtWidgets.QWidget):
time_support() time_support()
if(self.FoldState["Enabled"]): if(self.FoldState["Enabled"]):
lc.scatter(label=label, ax=self.figureAxis) 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"]): elif(self.PeriodogramState["Enabled"]):
lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"]) lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"])
if(self.PeriodogramState["View"] == "period"): if(self.PeriodogramState["View"] == "period"):