flaredetector: add quality flag to return lists

2024-05-13 14:40:20 +02:00
parent f68c635545
commit dbdda26399
1 changed files with 4 additions and 2 deletions
@@ -50,10 +50,12 @@ def calculateFlareFitsForLightcurve(lightcurve, num=100, distance=1, height=0.05
            Peak, flareFitDataPoints, fit = calculateFitForFlare(lightcurve, index)
            peaks.append({"FlarePeak": Peak,
                          "FlarePeakTime": lightcurve.time[index],
-                          "StandardIndex": index})
+                          "StandardIndex": index,
+                          "Quality": lightcurve.quality[index]})
            fits.append({"FlareFit": fit,
                         "FlareFitTime": lightcurve.time[flareFitDataPoints],
-                         "StandardIndex": flareFitDataPoints})
+                         "StandardIndex": flareFitDataPoints,
+                          "Quality": lightcurve.quality[flareFitDataPoints]})
        except Exception as error:
            pass
            #print("Flare fit exception: ", error)