astrodatagui: FlaredetectorWidget: set up setFluxType function

2024-03-18 15:32:41 +01:00
parent 029fd1c776
commit 003a821789
1 changed files with 8 additions and 0 deletions
@@ -35,6 +35,13 @@ class FlaredetectorWidget(QtWidgets.QWidget):
        self.currentMainName = mainName
        self.updatePlot()

+    def setFluxType(self, fluxType: str):
+        if(fluxType not in ["sap_flux", "pdcsap_flux"]):
+            print(f"FluxType not supported: {fluxType}, setting default sap_flux")
+            fluxType = "sap_flux"
+        self.fluxType = fluxType
+        self.updatePlot()
+
    def setNormalizeState(self, enabled: bool, scale: str):
        self.NormalizeState["Enabled"] = enabled
        if(scale not in ["unscaled", "percent", "ppt", "ppm"]):
@@ -107,6 +114,7 @@ class FlaredetectorWidget(QtWidgets.QWidget):
    def updatePlot(self):
        lc = self.currentLC
        label = f"{self.currentMainName}"
+        lc.flux = lc[self.fluxType]
        if(self.NormalizeState["Enabled"]):
            lc = lc.normalize(unit=self.NormalizeState["Scale"])
            label += " - normalized"