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

flaredetector: split peaks/fits for calculateFlareFitsForLightcurve

Browse Source
This commit is contained in:
SGCMarkus
2024-04-15 15:58:10 +02:00
parent c8ad11187e
commit af824867fb
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/flaredetector.py
+5 -4
View File
@@ -27,15 +27,16 @@ def calculateFitForFlare(lightcurve, peakIndex):
def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05): def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05):
maxIndices = findMaxIndices(lightcurve, num, distance=distance, height=height) maxIndices = findMaxIndices(lightcurve, num, distance=distance, height=height)
peaks = []
fits = [] fits = []
for index in maxIndices: for index in maxIndices:
Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index) Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index)
fits.append({"FlarePeak": Peak, peaks.append({"FlarePeak": Peak,
"FlarePeakTime": lightcurve.time[index], "FlarePeakTime": lightcurve.time[index]})
"FlareFit": fit, fits.append({"FlareFit": fit,
"FlareFitTime": lightcurve.time[flareFitDataPoints]}) "FlareFitTime": lightcurve.time[flareFitDataPoints]})
return fits return peaks, fits
def getFlareRange(flux, peakIndex): def getFlareRange(flux, peakIndex):
minInd = -np.inf minInd = -np.inf