From d6a871901456e9fb2cb09fa9b6dc346acb2af2d0 Mon Sep 17 00:00:00 2001 From: SGCMarkus Date: Fri, 7 Jun 2024 15:07:05 +0200 Subject: [PATCH] astrodatagui: FlareSummaryPlotGUI: add file save/open support --- main/astrodatagui/FlareSummaryPlotGUI.py | 22 +++++++++++++++++++++- 1 file changed, 21 insertions(+), 1 deletion(-) diff --git a/main/astrodatagui/FlareSummaryPlotGUI.py b/main/astrodatagui/FlareSummaryPlotGUI.py index d6f7bd0..8e99de2 100644 --- a/main/astrodatagui/FlareSummaryPlotGUI.py +++ b/main/astrodatagui/FlareSummaryPlotGUI.py @@ -1,4 +1,4 @@ -from PyQt5 import QtCore +from PyQt5 import QtCore, QtWidgets from PyQt5.QtWidgets import (QWidget, QVBoxLayout, QGridLayout, QPushButton, QCheckBox, QLabel) from matplotlib.figure import Figure @@ -16,6 +16,18 @@ class FlareSummaryPlotGUI(QWidget): self.starFLareDictList = starFLareDictList + self.menubar = QtWidgets.QMenuBar(self) + #self.setMenuBar(self.menubar) + self.fileMenu = QtWidgets.QMenu("File", self) + self.saveAction = QtWidgets.QAction("Save As", self) + self.openAction = QtWidgets.QAction("Open", self) + self.fileMenu.addAction(self.saveAction) + self.fileMenu.addAction(self.openAction) + self.menubar.addMenu(self.fileMenu) + + self.saveAction.triggered.connect(self.saveActionTriggered) + self.openAction.triggered.connect(self.openActionTriggered) + self.mainLayout = QVBoxLayout() self.setLayout(self.mainLayout) @@ -71,6 +83,14 @@ class FlareSummaryPlotGUI(QWidget): self.mainLayout.addWidget(fc) self.resize(1000, 800) + def saveActionTriggered(self): + fileName = QtWidgets.QFileDialog.getSaveFileName(self, "Save as", filter="Counted Flares File (*.cff)")[0] + self.starFLareDictList.to_pickle(fileName) + + def openActionTriggered(self): + fileName = QtWidgets.QFileDialog.getOpenFileName(self, "Open", filter="Counted Flares File (*.cff)")[0] + self.starFLareDictList = pd.read_pickle(fileName) + def btShowFlaresPerFileClicked(self): data = self.starFLareDictList x = np.arange(start=1, stop=len(data)+1, step=1, dtype=int)