astrodatagui: FlaredetectorWidget: add plot fits file function
This commit is contained in:
@@ -3,6 +3,7 @@ from matplotlib.figure import Figure
|
|||||||
from matplotlib.backends.backend_qtagg import (
|
from matplotlib.backends.backend_qtagg import (
|
||||||
FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
|
FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
|
||||||
|
|
||||||
|
from astropy.io import fits
|
||||||
|
|
||||||
class FlaredetectorWidget(QtWidgets.QWidget):
|
class FlaredetectorWidget(QtWidgets.QWidget):
|
||||||
def __init__(self, name):
|
def __init__(self, name):
|
||||||
@@ -17,3 +18,12 @@ class FlaredetectorWidget(QtWidgets.QWidget):
|
|||||||
toolbar = NavigationToolbar(fc, self)
|
toolbar = NavigationToolbar(fc, self)
|
||||||
self.mainLayout.addWidget(toolbar)
|
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()
|
||||||
Reference in New Issue
Block a user