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