diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py index 96f4509..51ea6b4 100644 --- a/main/flaredetector/flaredetector.py +++ b/main/flaredetector/flaredetector.py @@ -57,6 +57,7 @@ def getFlareRange(flux, peakIndex): maxInd = np.inf maxDelta = 0.005 + maxDeltaToMean = 2*maxDelta meanVal = 1 maxBacklog = 20 maxForwardlog = 100 @@ -66,7 +67,7 @@ def getFlareRange(flux, peakIndex): prevFlux = flux[peakIndex] for i in range(peakIndex-1, peakIndex - maxBacklog, -1): delta = prevFlux - flux[i] - if(delta < maxDelta and abs(flux[i] - meanVal) < maxDelta): + if(delta < maxDelta and abs(flux[i] - meanVal) < maxDeltaToMean): deltaCount += 1 if(deltaCount > 3): minInd = i @@ -89,7 +90,7 @@ def getFlareRange(flux, peakIndex): prevFlux = flux[peakIndex] for i in range(peakIndex+1, peakIndex + maxForwardlog, 1): delta = prevFlux - flux[i] - if(delta < maxDelta and abs(flux[i] - meanVal) < maxDelta): + if(delta < maxDelta and abs(flux[i] - meanVal) < maxDeltaToMean): deltaCount += 1 if(deltaCount > 3): maxInd = i