diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index 9194e1c..bf8abc3 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -33,6 +33,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): print(f"Plotting: {fitsFilePath}") self.currentLC = lk.read(fitsFilePath) self.currentMainName = mainName + self.updatePlot() def setNormalizeState(self, enabled: bool, scale: str): self.NormalizeState["Enabled"] = enabled @@ -40,6 +41,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): print(f"Scale not supported: {scale}, setting default unscaled") scale = "unscaled" self.NormalizeState["Scale"] = scale + self.updatePlot() def setRemoveOutliersState(self, enabled: bool, sigma: float): self.RemoveOutliersState["Enabled"] = enabled @@ -47,9 +49,11 @@ class FlaredetectorWidget(QtWidgets.QWidget): print(f"Negative sigma detected, setting default 5.0") sigma = 5.0 self.RemoveOutliersState["Sigma"] = sigma + self.updatePlot() def setRemoveNansState(self, enabled: bool): self.RemoveNansState["Enabled"] = enabled + self.updatePlot() def setBinState(self, enabled: bool, binSize: float): self.BinState["Enabled"] = enabled @@ -57,6 +61,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): print(f"Negative binSize detected, setting default 1") binSize = 1 self.BinState["Size"] = binSize + self.updatePlot() def setFlattenState(self, enabled: bool, windowLength: int, polyorder: int): self.FlattenState["Enabled"] = enabled @@ -71,6 +76,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): if(self.FlattenState["Enabled"]): self.FoldState["Enabled"] = False self.PeriodogramState["Enabled"] = False + self.updatePlot() def setFoldState(self, enabled: bool, period: float, epoch: float): self.FoldState["Enabled"] = enabled @@ -85,6 +91,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): if(self.FoldState["Enabled"]): self.FlattenState["Enabled"] = False self.PeriodogramState["Enabled"] = False + self.updatePlot() def setPeriodogramState(self, enabled: bool, method: str): self.PeriodogramState["Enabled"] = enabled @@ -95,6 +102,7 @@ class FlaredetectorWidget(QtWidgets.QWidget): if(self.PeriodogramState["Enabled"]): self.FoldState["Enabled"] = False self.FlattenState["Enabled"] = False + self.updatePlot() def updatePlot(self): lc = self.currentLC