flaredetector: util: check if its a Periodogram

main/astrodatagui/ui/FlaredetectorWidget.py

@@ -196,7 +196,7 @@ class FlaredetectorWidget(QtWidgets.QWidget):
             lc = lc.to_periodogram(method=self.PeriodogramState["Method"])
             label = "Periodogram"
 
-            maxPowers = findMaxIndices(lc.power, 4)
+            maxPowers = findMaxIndices(lc, 4)
             print(maxPowers)
             self.lbMaxPowerPeriod1.setText(str(lc.period[maxPowers[0]]))
             self.lbMaxPowerPeriod2.setText(str(lc.period[maxPowers[1]]))

main/flaredetector/util.py

@@ -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