astrodatagui: FlareSummaryPlotGUI: add file save/open support

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)