flaredetector: util: check if its a Periodogram

2024-03-26 12:37:17 +01:00
parent f756873687
commit e5db328b9a
2 changed files with 9 additions and 6 deletions
@@ -1,8 +1,11 @@
 import numpy as np
 from scipy.signal import find_peaks

-def findMaxIndices(data, num=3, distance=100):
-    peak_indices, peak_dict = find_peaks(data, height=(None, None), distance=distance)
-    peak_heights = peak_dict["peak_heights"]
-    maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
-    return maxIndices
+from lightkurve.periodogram import Periodogram
+
+def findMaxIndices(lc, num=3, distance=100):
+    if(isinstance(lc, Periodogram)):
+        peak_indices, peak_dict = find_peaks(lc.power, height=(None, None), distance=distance)
+        peak_heights = peak_dict["peak_heights"]
+        maxIndices = peak_indices[np.argsort(peak_heights)[-num:][::-1]]
+        return maxIndices