From 3465c2237035f11b2d5a8ffb128cffc82435c603 Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Tue, 12 Mar 2024 13:37:08 +0100 Subject: [PATCH] astrodatagui: switch to lightkurve plotting --- main/astrodatagui/AstrodataGUI.py | 2 +- main/astrodatagui/ui/FlaredetectorWidget.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/main/astrodatagui/AstrodataGUI.py b/main/astrodatagui/AstrodataGUI.py index 6732255..a9baf2a 100644 --- a/main/astrodatagui/AstrodataGUI.py +++ b/main/astrodatagui/AstrodataGUI.py @@ -159,4 +159,4 @@ class AstrodataGUI(QtWidgets.QMainWindow): source = seqText[0] seq = seqText[1] filePath = self.starDB.getFilePath(mainName, source, seq) - self.flaredetectorPreview.plotFitsFile(filePath) \ No newline at end of file + self.flaredetectorPreview.plotFitsFile(filePath, mainName) \ No newline at end of file diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index d24da1a..6261261 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -4,6 +4,7 @@ from matplotlib.backends.backend_qtagg import ( FigureCanvas, NavigationToolbar2QT as NavigationToolbar) from astropy.io import fits +import lightkurve as lk class FlaredetectorWidget(QtWidgets.QWidget): def __init__(self, name): @@ -19,11 +20,10 @@ class FlaredetectorWidget(QtWidgets.QWidget): self.mainLayout.addWidget(toolbar) self.mainLayout.addWidget(fc) - def plotFitsFile(self, fitsFilePath): + def plotFitsFile(self, fitsFilePath, mainName: str): print(f"Plotting: {fitsFilePath}") self.figureAxis.clear() - hdu = fits.open(fitsFilePath, mode="readonly") - time = hdu[1].data["TIME"] - data = hdu[1].data["SAP_FLUX"] - self.figureAxis.plot(time, data) + lc = lk.read(fitsFilePath) + label = f"{mainName} - SAP Flux" + lc.plot(column="sap_flux", label=label, ax=self.figureAxis) self.figure.canvas.draw_idle() \ No newline at end of file