astrodatagui: FlaredetectorWidget: add plot fits file function

This commit is contained in:
2024-03-08 11:29:43 +01:00
parent 044aa47d35
commit 8f7fb78c8f
@@ -3,6 +3,7 @@ from matplotlib.figure import Figure
from matplotlib.backends.backend_qtagg import (
FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
from astropy.io import fits
class FlaredetectorWidget(QtWidgets.QWidget):
def __init__(self, name):
@@ -17,3 +18,12 @@ class FlaredetectorWidget(QtWidgets.QWidget):
toolbar = NavigationToolbar(fc, self)
self.mainLayout.addWidget(toolbar)
self.mainLayout.addWidget(fc)
def plotFitsFile(self, fitsFilePath):
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)
self.figure.canvas.draw_idle()