astrodatagui: switch to lightkurve plotting

This commit is contained in:
2024-03-12 13:37:08 +01:00
parent d4db3c163f
commit 3465c22370
2 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -159,4 +159,4 @@ class AstrodataGUI(QtWidgets.QMainWindow):
source = seqText[0] source = seqText[0]
seq = seqText[1] seq = seqText[1]
filePath = self.starDB.getFilePath(mainName, source, seq) filePath = self.starDB.getFilePath(mainName, source, seq)
self.flaredetectorPreview.plotFitsFile(filePath) self.flaredetectorPreview.plotFitsFile(filePath, mainName)
+5 -5
View File
@@ -4,6 +4,7 @@ from matplotlib.backends.backend_qtagg import (
FigureCanvas, NavigationToolbar2QT as NavigationToolbar) FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from astropy.io import fits from astropy.io import fits
import lightkurve as lk
class FlaredetectorWidget(QtWidgets.QWidget): class FlaredetectorWidget(QtWidgets.QWidget):
def __init__(self, name): def __init__(self, name):
@@ -19,11 +20,10 @@ class FlaredetectorWidget(QtWidgets.QWidget):
self.mainLayout.addWidget(toolbar) self.mainLayout.addWidget(toolbar)
self.mainLayout.addWidget(fc) self.mainLayout.addWidget(fc)
def plotFitsFile(self, fitsFilePath): def plotFitsFile(self, fitsFilePath, mainName: str):
print(f"Plotting: {fitsFilePath}") print(f"Plotting: {fitsFilePath}")
self.figureAxis.clear() self.figureAxis.clear()
hdu = fits.open(fitsFilePath, mode="readonly") lc = lk.read(fitsFilePath)
time = hdu[1].data["TIME"] label = f"{mainName} - SAP Flux"
data = hdu[1].data["SAP_FLUX"] lc.plot(column="sap_flux", label=label, ax=self.figureAxis)
self.figureAxis.plot(time, data)
self.figure.canvas.draw_idle() self.figure.canvas.draw_idle()