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)