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

FlaredetectorWidget: signal epoch_time and support multiple minima/maxima

Browse Source
This commit is contained in:
SGCMarkus
2024-07-15 11:51:54 +02:00
parent 0de8071803
commit 51e0d81c28
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/astrodatagui/ui/FlaredetectorWidget.py
+5 -2
View File
@@ -13,6 +13,7 @@ from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve
class FlaredetectorWidget(QtWidgets.QWidget): class FlaredetectorWidget(QtWidgets.QWidget):
periodsCalculated = pyqtSignal(list) periodsCalculated = pyqtSignal(list)
epochCalculated = pyqtSignal(float)
def __init__(self, name): def __init__(self, name):
super(FlaredetectorWidget, self).__init__() super(FlaredetectorWidget, self).__init__()
@@ -197,7 +198,9 @@ class FlaredetectorWidget(QtWidgets.QWidget):
self.foldedLC = lc.to_periodogram(method=self.PeriodogramState["Method"]) self.foldedLC = lc.to_periodogram(method=self.PeriodogramState["Method"])
self.peaks, self.fits = calculateFlareFitsForLightcurve(self.currentFlattenLC, num=100) self.peaks, self.fits = calculateFlareFitsForLightcurve(self.currentFlattenLC, num=100)
self.periods = [self.foldedLC.period[i] for i in findMaxIndices(self.foldedLC, num=4, distance=100, sortByHighest=True)] self.periods = [self.foldedLC.period[i] for i in findMaxIndices(self.foldedLC, num=4, distance=100, sortByHighest=True)]
self.epoch_time = getEpochTime(lc)
self.periodsCalculated.emit(self.periods) self.periodsCalculated.emit(self.periods)
self.epochCalculated.emit(self.epoch_time)
def updatePlot(self): def updatePlot(self):
lc = self.currentLC lc = self.currentLC
@@ -253,8 +256,8 @@ class FlaredetectorWidget(QtWidgets.QWidget):
lc.flux[lc.cycle == cycle][foldedIndex], "x", color="red") lc.flux[lc.cycle == cycle][foldedIndex], "x", color="red")
phase, sineFit = getFoldedBestFit(lc) phase, sineFit = getFoldedBestFit(lc)
self.figureAxis.plot(phase, sineFit, color="red") self.figureAxis.plot(phase, sineFit, color="red")
minPhaseBoundsIndices, maxPhaseBoundsIndices = getPhaseRangesNearPeak(getFoldedFitPeakValley(sineFit), phase) minPhasesBoundsIndices, maxPhasesBoundsIndices = getPhaseRangesNearPeak(getFoldedFitPeakValley(sineFit), phase)
plotPhaseRangesNearPeak((minPhaseBoundsIndices, maxPhaseBoundsIndices), phase, ax=self.figureAxis) plotPhaseRangesNearPeak((minPhasesBoundsIndices, maxPhasesBoundsIndices), phase, ax=self.figureAxis)
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"):