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

flaredetector: use correct peak index for checking if flare is valid

Browse Source
This commit is contained in:
SGCMarkus
2024-04-25 15:49:44 +02:00
parent a15e771845
commit 31dfaebc3e
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/flaredetector.py
+6 -7
View File
@@ -16,19 +16,18 @@ def moveFluxForFit(curPeakIndex, flux):
flarePoints = np.array(np.linspace(minInd, maxInd, num=maxInd-minInd+1), dtype=int)
flareFitDataPoints = flarePoints - curPeakIndex
Peak = flux[curPeakIndex] - 1
return flarePoints, flareFitDataPoints, Peak
return flarePoints, flareFitDataPoints, Peak, maxInd - curPeakIndex
def isValidFlarePoint(flareData):
def isValidFlarePoint(flareData, newPeakIndex):
threshold = 0.005
peakIndex = np.argmax(flareData)
if((flareData[peakIndex-1] > threshold and flareData[peakIndex+1] > threshold) or
(flareData[peakIndex+1] > threshold and flareData[peakIndex+2] > threshold)):
if((flareData[newPeakIndex-1] > threshold and flareData[newPeakIndex+1] > threshold) or
(flareData[newPeakIndex+1] > threshold and flareData[newPeakIndex+2] > threshold)):
return True
return False
def calculateFitForFlare(lightcurve, peakIndex):
flarePoints, flareFitDataPoints, Peak = moveFluxForFit(peakIndex, lightcurve.flux)
if(not isValidFlarePoint(lightcurve.flux[flarePoints]-1)):
flarePoints, flareFitDataPoints, Peak, newPeakIndex = moveFluxForFit(peakIndex, lightcurve.flux)
if(not isValidFlarePoint(lightcurve.flux[flarePoints]-1, newPeakIndex)):
raise Exception("Flare not valid: not enough datapoints above mean")
popt, pcov = curve_fit(lambda x, l, w: GaussExpo(x, Peak, l, w), flareFitDataPoints, lightcurve.flux[flarePoints]-1)
fit = GaussExpo(flareFitDataPoints, Peak, *popt)