astrodatagui: FlaredetectorWidget: add option to plot multiple fits files at once

This commit is contained in:
2024-03-19 12:36:48 +01:00
parent 9fb54837fd
commit cd33dfbe29
@@ -32,6 +32,27 @@ class FlaredetectorWidget(QtWidgets.QWidget):
def setFitsFile(self, fitsFilePath, mainName: str): def setFitsFile(self, fitsFilePath, mainName: str):
print(f"Plotting: {fitsFilePath}") print(f"Plotting: {fitsFilePath}")
self.currentLC = lk.read(fitsFilePath) self.currentLC = lk.read(fitsFilePath)
self.currentLCCollection = None
self.currentMainName = mainName
self.updatePlot()
def setFitsFiles(self, mainName: str, fitsFilePaths: list[str]):
print(f"Combining {len(fitsFilePaths)} fits files...")
if(len(fitsFilePaths) == 0):
print("Error, need to provide atleast 1 fits file")
return
if(len(fitsFilePaths) == 1):
self.setFitsFile(fitsFilePaths[0], mainName)
return
lcList = []
for path in fitsFilePaths:
print(path)
lcList.append(lk.read(path))
self.currentLCCollection = lk.LightCurveCollection(lcList)
self.currentLC = self.currentLCCollection.stitch()
self.currentMainName = mainName self.currentMainName = mainName
self.updatePlot() self.updatePlot()