astrodatagui: FlaredetectorWidget: show flare peaks in folded view
This commit is contained in:
@@ -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"):
|
||||||
|
|||||||
Reference in New Issue
Block a user