diff --git a/main/astrodatagui/FlareSummaryPlotGUI.py b/main/astrodatagui/FlareSummaryPlotGUI.py index 11eb740..2b4ebf3 100644 --- a/main/astrodatagui/FlareSummaryPlotGUI.py +++ b/main/astrodatagui/FlareSummaryPlotGUI.py @@ -1,5 +1,6 @@ from PyQt5 import QtCore -from PyQt5.QtWidgets import QWidget, QVBoxLayout +from PyQt5.QtWidgets import (QWidget, QVBoxLayout, QGridLayout, + QPushButton, QCheckBox, QLabel) from matplotlib.figure import Figure from matplotlib.backends.backend_qtagg import ( FigureCanvas, NavigationToolbar2QT as NavigationToolbar) @@ -19,7 +20,43 @@ class FlareSummaryPlotGUI(QWidget): self.qualityAxis = self.figureAxis.twinx() fc = FigureCanvas(self.figure) toolbar = NavigationToolbar(fc, self) + + self.buttonGridLayout = QGridLayout() + self.btShow = QPushButton("Show ") + + self.cbKepler = QCheckBox("Kepler") + self.cbKepler.setChecked(True) + self.cbK2 = QCheckBox("K2") + self.cbK2.setChecked(True) + self.cbTESS = QCheckBox("TESS") + self.cbTESS.setChecked(True) + + self.cbSpTypeL = QCheckBox("L") + self.cbSpTypeL.setChecked(True) + self.cbSpTypeM = QCheckBox("M") + self.cbSpTypeM.setChecked(True) + self.cbSpTypeK = QCheckBox("K") + self.cbSpTypeK.setChecked(True) + self.cbSpTypeG = QCheckBox("G") + self.cbSpTypeG.setChecked(True) + self.cbSpTypeF = QCheckBox("F") + self.cbSpTypeF.setChecked(True) + + self.buttonGridLayout.addWidget(QLabel("Plot Types: "), 0, 0) + self.buttonGridLayout.addWidget(self.btShow, 0, 1) + + self.buttonGridLayout.addWidget(QLabel("Sources: "), 1, 0) + self.buttonGridLayout.addWidget(self.cbKepler, 1, 1) + self.buttonGridLayout.addWidget(self.cbK2, 1, 2) + self.buttonGridLayout.addWidget(self.cbTESS, 1, 3) + + self.buttonGridLayout.addWidget(QLabel("Sp Types: "), 2, 0) + self.buttonGridLayout.addWidget(self.cbSpTypeL, 2, 1) + self.buttonGridLayout.addWidget(self.cbSpTypeM, 2, 2) + self.buttonGridLayout.addWidget(self.cbSpTypeK, 2, 3) + self.buttonGridLayout.addWidget(self.cbSpTypeG, 2, 4) + self.buttonGridLayout.addWidget(self.cbSpTypeF, 2, 5) + self.mainLayout.addLayout(self.buttonGridLayout) self.mainLayout.addWidget(toolbar) self.mainLayout.addWidget(fc) - - self.resize(1000, 800) + self.resize(1000, 800) \ No newline at end of file