SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

astrodatagui: FlaredetectorWidget: show peaks in flatten

Browse Source
This commit is contained in:
SGCMarkus
2024-03-28 12:02:21 +01:00
parent 54629bf5cd
commit 0e5c4e7a58
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/astrodatagui/ui/FlaredetectorWidget.py
+7
View File
@@ -4,6 +4,7 @@ from matplotlib.backends.backend_qtagg import (
FigureCanvas, NavigationToolbar2QT as NavigationToolbar) FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from astropy.io import fits from astropy.io import fits
from astropy.visualization import time_support
import lightkurve as lk import lightkurve as lk
from ...flaredetector.util import findMaxIndices from ...flaredetector.util import findMaxIndices
@@ -188,6 +189,8 @@ class FlaredetectorWidget(QtWidgets.QWidget):
lc = lc.flatten(window_length=self.FlattenState["WindowLength"], lc = lc.flatten(window_length=self.FlattenState["WindowLength"],
polyorder=self.FlattenState["PolynomialOrder"]) polyorder=self.FlattenState["PolynomialOrder"])
label += " - flattened" label += " - flattened"
maxPowers = findMaxIndices(lc, 100, height=1.005, distance=1)
if(self.FoldState["Enabled"]): if(self.FoldState["Enabled"]):
lc = lc.fold(period=self.FoldState["Period"], lc = lc.fold(period=self.FoldState["Period"],
epoch_time=self.FoldState["EpochTime"]) epoch_time=self.FoldState["EpochTime"])
@@ -214,6 +217,10 @@ class FlaredetectorWidget(QtWidgets.QWidget):
self.figureAxis.plot(lc.period[maxPowers], lc.power[maxPowers], "x") self.figureAxis.plot(lc.period[maxPowers], lc.power[maxPowers], "x")
if(self.PeriodogramState["View"] == "frequency"): if(self.PeriodogramState["View"] == "frequency"):
self.figureAxis.plot(lc.frequency[maxPowers], lc.power[maxPowers], "x") self.figureAxis.plot(lc.frequency[maxPowers], lc.power[maxPowers], "x")
elif(self.FlattenState["Enabled"]):
time_support()
lc.plot(label=label, ax=self.figureAxis)
self.figureAxis.plot(lc.time[maxPowers], lc.flux[maxPowers], "x")
else: else:
lc.plot(label=label, ax=self.figureAxis) lc.plot(label=label, ax=self.figureAxis)
self.figure.canvas.draw_idle() self.figure.canvas.draw_idle()