astrodatagui: switch to lightkurve plotting
This commit is contained in:
@@ -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)
|
||||
self.flaredetectorPreview.plotFitsFile(filePath, mainName)
|
||||
@@ -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()
|
||||
Reference in New Issue
Block a user