flaredetector: util: findMaxIndices: set data for light curve objects to their flux

2024-03-26 13:32:57 +01:00
parent cbfbd7e976
commit cf9958a1af
1 changed files with 3 additions and 0 deletions
@@ -2,10 +2,13 @@ import numpy as np
 from scipy.signal import find_peaks

 from lightkurve.periodogram import Periodogram
+from lightkurve.lightcurve import LightCurve

 def findMaxIndices(lc, num=3, distance=100):
    if(isinstance(lc, Periodogram)):
        data = lc.power
+    elif(isinstance(lc, LightCurve)):
+        data = lc.flux
    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]]