diff --git a/main/flaredetector/util.py b/main/flaredetector/util.py index 4e2a5b4..f3f56b4 100644 --- a/main/flaredetector/util.py +++ b/main/flaredetector/util.py @@ -138,8 +138,6 @@ def getFoldedBestFit(foldedLc): def getFoldedFitPeakValley(sineFit): minima = argrelextrema(sineFit, np.less)[0] maxima = argrelextrema(sineFit, np.greater)[0] - #print(minima) - #print(maxima) if(len(minima) > 0 and len(maxima) > 0): if(minima[0] < maxima[0] and minima[-1] > maxima[-1]): if(sineFit[minima[0]] > sineFit[minima[-1]]): @@ -152,6 +150,17 @@ def getFoldedFitPeakValley(sineFit): else: maxima = np.delete(maxima, 0) + if(len(minima) == 0 and len(maxima) == 1): + if(sineFit[0] < sineFit[-1]): + minima = np.array((0)) + else: + minima = np.array((len(sineFit)-1)) + elif(len(minima) == 1 and len(maxima) == 0): + if(sineFit[0] > sineFit[-1]): + minima = np.array((0)) + else: + minima = np.array((len(sineFit)-1)) + return minima, maxima def findNearestIndexOfValue(array, value):