def getCol(colName):
theCol = [k for k, v in DATA_HEADINGS.iteritems() if v == colName]
return theCol[0]
def initUI(self):
# set window size and position on screen
self.setGeometry(300, 200, 1000, 600)
# get variables from spreadsheet
global DATA_HEADINGS
self.vars = []
for index in range(3, len(DATA_HEADINGS)):
self.vars += [DATA_HEADINGS.get(index)]
# initialize default graph
self.graph = graph.Graph(self, xvarname="Time",
yvarname=self.vars[0])
self.toolbar = graph.NavigationToolbar(self.graph, self)
self.updating = True
self.setWindowTitle(self.vars[0])
self.plotOptionMenu = QtGui.QComboBox()
self.plotOptionMenu.addItem('Switch graph')
self.plotOptionMenu.addItem('Add to left axis')
# make drop-down menu for selecting graphs
self.selectVar = QtGui.QComboBox()
for var in self.vars:
self.selectVar.addItem(var)
self.selectVar.activated[str].connect(self.selectGraph)
# set up layout and sub-layouts
self.layout = QtGui.QVBoxLayout(self)
self.optionslayout = QtGui.QGridLayout()
self.gridlayout = QtGui.QGridLayout()
self.axeslayout = QtGui.QGridLayout()
self.timelayout = QtGui.QGridLayout()
# this exists so auto axis buttons can move if necessary
self.autowidget = QtGui.QWidget(self)
self.autolayout = QtGui.QGridLayout()
# first column holds graph, second column holds graph options
# set the column stretches - 0 is the default
# set minimum column widths
self.gridlayout.setColumnStretch(0, 4)
self.gridlayout.setColumnStretch(1, 0)
self.gridlayout.setColumnMinimumWidth(0,300)
self.gridlayout.setRowMinimumHeight(0,375)
# add drop-down menus and MPL toolbar to top of window
self.layout.addLayout(self.optionslayout)
self.optionslayout.addWidget(self.plotOptionMenu, 0, 0)
self.optionslayout.addWidget(self.selectVar, 0, 1, 1, 3)
self.optionslayout.addWidget(self.toolbar, 1, 0, 1, 4)
# initialize checkbox that acts as pause button
self.hold_cb = QtGui.QCheckBox('Hold', self)
self.hold_cb.stateChanged.connect(self.hold)
# initialize input boxes for axis limits
self.minutes = QtGui.QLineEdit(self)
self.minutes.setFixedWidth(40)
self.hours = QtGui.QLineEdit(self)
self.hours.setFixedWidth(40)
self.days = QtGui.QLineEdit(self)
self.days.setFixedWidth(40)
self.Ymin = QtGui.QLineEdit(self)
self.Ymax = QtGui.QLineEdit(self)
self.YminR = QtGui.QLineEdit(self)
self.YmaxR = QtGui.QLineEdit(self)
# create labels for the input boxes
self.label_time = QtGui.QLabel('Show data from the last:')
self.label_minutes = QtGui.QLabel('minutes')
self.label_hours = QtGui.QLabel('hours')
self.label_days = QtGui.QLabel('days')
self.label_Ymin = QtGui.QLabel('Y Min (left):')
self.label_Ymax = QtGui.QLabel('Y Max (left):')
self.label_YminR = QtGui.QLabel('Y Min (right):')
self.label_YmaxR = QtGui.QLabel('Y Max (right):')
# initialize buttons and their connections
self.set_axes = QtGui.QPushButton('Enter')
self.auto_xaxes = QtGui.QPushButton('Auto X')
self.auto_yaxes = QtGui.QPushButton('Auto Y (left)')
self.auto_yraxes = QtGui.QPushButton('Auto Y (right)')
self.set_axes.clicked.connect(self.setAxes)
self.auto_xaxes.clicked.connect(self.autoXAxes)
self.auto_yaxes.clicked.connect(self.autoYAxes)
self.auto_yraxes.clicked.connect(self.autoYRAxes)
# set the possible streches of input boxes
self.Ymin.setSizePolicy(QtGui.QSizePolicy.Maximum, QtGui.QSizePolicy.Preferred)
self.Ymax.setSizePolicy(QtGui.QSizePolicy.Maximum, QtGui.QSizePolicy.Preferred)
self.YminR.setSizePolicy(QtGui.QSizePolicy.Maximum, QtGui.QSizePolicy.Preferred)
self.YmaxR.setSizePolicy(QtGui.QSizePolicy.Maximum, QtGui.QSizePolicy.Preferred)
# initialize menu to choose variable for right-hand axis
self.label_raxis = QtGui.QLabel('Choose a variable to plot on the right-hand axis:')
self.choose_var = QtGui.QComboBox()
for var in self.vars:
self.choose_var.addItem(var)
self.set_raxis = QtGui.QPushButton('Plot')
self.set_raxis.clicked.connect(self.addRAxis)
# place the main grid layout inside layout for window
self.layout.addLayout(self.gridlayout)
# add graph and options to main grid layout
self.gridlayout.addWidget(self.graph, 0, 0)
self.gridlayout.addLayout(self.axeslayout, 0, 1)
# set alignment for the widgets
self.axeslayout.setAlignment(QtCore.Qt.AlignTop)
# create spacers to separate fields in graph options layout
self.spacer1 = QtGui.QSpacerItem(1, 20)
self.spacer2 = QtGui.QSpacerItem(1, 20)
self.spacer3 = QtGui.QSpacerItem(1, 20)
self.spacer4 = QtGui.QSpacerItem(1, 20)
# add items to the graph options layout
self.axeslayout.addItem(self.spacer1, 0, 0)
self.axeslayout.addWidget(self.hold_cb, 1, 0)
self.axeslayout.addItem(self.spacer2, 2, 0)
self.axeslayout.addItem(self.spacer3, 3, 0)
self.axeslayout.addWidget(self.label_raxis, 4, 0)
self.axeslayout.addWidget(self.choose_var, 5, 0)
self.axeslayout.addWidget(self.set_raxis, 6, 0)
self.axeslayout.addItem(self.spacer4, 7, 0)
self.axeslayout.addWidget(self.label_time, 8, 0)
self.axeslayout.addLayout(self.timelayout, 9, 0)
# add options for time axis to a sub-grid
self.timelayout.addWidget(self.minutes, 0, 0)
self.timelayout.addWidget(self.label_minutes, 0, 1)
self.timelayout.addWidget(self.hours, 1, 0)
self.timelayout.addWidget(self.label_hours, 1, 1)
self.timelayout.addWidget(self.days, 2, 0)
self.timelayout.addWidget(self.label_days, 2, 1)
# add more items to graph options layout
self.axeslayout.addWidget(self.label_Ymin, 13, 0)
self.axeslayout.addWidget(self.Ymin, 14, 0)
self.axeslayout.addWidget(self.label_Ymax, 15, 0)
self.axeslayout.addWidget(self.Ymax, 16, 0)
self.axeslayout.addWidget(self.label_YminR, 17, 0)
self.axeslayout.addWidget(self.YminR, 18, 0)
self.axeslayout.addWidget(self.label_YmaxR, 19, 0)
self.axeslayout.addWidget(self.YmaxR, 20, 0)
self.axeslayout.addWidget(self.set_axes, 21, 0)
# hide options for second axis initially
self.label_YminR.hide()
self.YminR.hide()
self.label_YmaxR.hide()
self.YmaxR.hide()
# add widget that holds auto axis buttons
self.axeslayout.addWidget(self.autowidget, 22, 0, 1, 2)
self.autowidget.setLayout(self.autolayout)
# add auto axis buttons
self.autolayout.addWidget(self.auto_xaxes, 0 , 0)
self.autolayout.addWidget(self.auto_yaxes, 0 , 1)
self.autolayout.addWidget(self.auto_yraxes, 0 , 2)
# hide option for auto right axis initially
self.auto_yraxes.hide()
self.show()
def getCol(colName):
theCol = [k for k, v in DATA_HEADINGS.iteritems() if v == colName]
return theCol[0]
