From 8f7fb78c8f709aada493c6c5761509ad9fcd6dcc Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Fri, 8 Mar 2024 11:29:43 +0100 Subject: [PATCH] astrodatagui: FlaredetectorWidget: add plot fits file function --- main/astrodatagui/ui/FlaredetectorWidget.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/main/astrodatagui/ui/FlaredetectorWidget.py b/main/astrodatagui/ui/FlaredetectorWidget.py index ec4b004..d24da1a 100644 --- a/main/astrodatagui/ui/FlaredetectorWidget.py +++ b/main/astrodatagui/ui/FlaredetectorWidget.py @@ -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) \ No newline at end of file + 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() \ No newline at end of file