astrodatagui: FlaredetectorWidget: add plot fits file function

This commit is contained in:
2024-03-08 11:29:43 +01:00
parent 044aa47d35
commit 8f7fb78c8f
+11 -1
View File
@@ -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):
@@ -16,4 +17,13 @@ class FlaredetectorWidget(QtWidgets.QWidget):
fc = FigureCanvas(self.figure)
toolbar = NavigationToolbar(fc, self)
self.mainLayout.addWidget(toolbar)
self.mainLayout.addWidget(fc)
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()