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

tmp

Browse Source
This commit is contained in:
SGCMarkus
2024-04-29 08:59:29 +02:00
parent e835964857
commit 24979f97cd
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/flaredetector.py
+18 -5
View File
@@ -16,10 +16,10 @@ def moveFluxForFit(curPeakIndex, flux):
flarePoints = np.array(np.linspace(minInd, maxInd, num=maxInd-minInd+1), dtype=int) flarePoints = np.array(np.linspace(minInd, maxInd, num=maxInd-minInd+1), dtype=int)
flareFitDataPoints = flarePoints - curPeakIndex flareFitDataPoints = flarePoints - curPeakIndex
Peak = flux[curPeakIndex] - 1 Peak = flux[curPeakIndex] - 1
return flarePoints, flareFitDataPoints, Peak, maxInd - curPeakIndex return flarePoints, flareFitDataPoints, Peak, curPeakIndex - minInd
def isValidFlarePoint(flareData, newPeakIndex): def isValidFlarePoint(flareData, newPeakIndex):
threshold = 0.005 threshold = 0.003
if((flareData[newPeakIndex-1] > threshold and flareData[newPeakIndex+1] > threshold) or if((flareData[newPeakIndex-1] > threshold and flareData[newPeakIndex+1] > threshold) or
(flareData[newPeakIndex+1] > threshold and flareData[newPeakIndex+2] > threshold)): (flareData[newPeakIndex+1] > threshold and flareData[newPeakIndex+2] > threshold)):
return True return True
@@ -27,9 +27,21 @@ def isValidFlarePoint(flareData, newPeakIndex):
def calculateFitForFlare(lightcurve, peakIndex): def calculateFitForFlare(lightcurve, peakIndex):
flarePoints, flareFitDataPoints, Peak, newPeakIndex = moveFluxForFit(peakIndex, lightcurve.flux) flarePoints, flareFitDataPoints, Peak, newPeakIndex = moveFluxForFit(peakIndex, lightcurve.flux)
if(not isValidFlarePoint(lightcurve.flux[flarePoints]-1, newPeakIndex)): flareData = lightcurve.flux[flarePoints]-1
PeakMinus1 = flareData[newPeakIndex-1]
PeakMinus2 = flareData[newPeakIndex-2]
if(not isValidFlarePoint(flareData, newPeakIndex)):
raise Exception("Flare not valid: not enough datapoints above mean") 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) if(PeakMinus1 > 0.01 and PeakMinus1 < PeakMinus2):
if(Peak > 0.025):
print(lightcurve.time[peakIndex], lightcurve.flux[peakIndex])
print(lightcurve.flux[peakIndex], lightcurve.flux[peakIndex-1], lightcurve.flux[peakIndex-2])
raise Exception(f"Flare not valid: peak-2 > peak-1, {PeakMinus2} > {PeakMinus1}")
# TODO: allow peak-1 to be around/lower than mean too if theres enough points after
popt, pcov = curve_fit(lambda x, l, w: GaussExpo(x, Peak, l, w), flareFitDataPoints, flareData)
pcovDiag = np.diag(pcov)
if(pcovDiag[0] > 1 and pcovDiag[1] > 1):
raise Exception("Flare not valid: could not properly fit")
fit = GaussExpo(flareFitDataPoints, Peak, *popt) fit = GaussExpo(flareFitDataPoints, Peak, *popt)
flareFitDataPoints = flareFitDataPoints + peakIndex flareFitDataPoints = flareFitDataPoints + peakIndex
return Peak+1, flareFitDataPoints, fit+1 return Peak+1, flareFitDataPoints, fit+1
@@ -48,7 +60,8 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05
"FlareFitTime": lightcurve.time[flareFitDataPoints], "FlareFitTime": lightcurve.time[flareFitDataPoints],
"StandardIndex": flareFitDataPoints}) "StandardIndex": flareFitDataPoints})
except Exception as error: except Exception as error:
print("Flare fit exception: ", error) pass
#print("Flare fit exception: ", error)
return peaks, fits return peaks, fits