pyqtgraph/widgets/CheckTable.py

93 lines
3.3 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
from pyqtgraph.Qt import QtGui, QtCore
2012-05-11 22:05:41 +00:00
from . import VerticalLabel
__all__ = ['CheckTable']
class CheckTable(QtGui.QWidget):
sigStateChanged = QtCore.Signal(object, object, object) # (row, col, state)
def __init__(self, columns):
QtGui.QWidget.__init__(self)
self.layout = QtGui.QGridLayout()
self.layout.setSpacing(0)
self.setLayout(self.layout)
self.headers = []
self.columns = columns
col = 1
for c in columns:
label = VerticalLabel.VerticalLabel(c, orientation='vertical')
self.headers.append(label)
self.layout.addWidget(label, 0, col)
col += 1
self.rowNames = []
self.rowWidgets = []
self.oldRows = {} ## remember settings from removed rows; reapply if they reappear.
def updateRows(self, rows):
for r in self.rowNames[:]:
if r not in rows:
self.removeRow(r)
for r in rows:
if r not in self.rowNames:
self.addRow(r)
def addRow(self, name):
label = QtGui.QLabel(name)
row = len(self.rowNames)+1
self.layout.addWidget(label, row, 0)
checks = []
col = 1
for c in self.columns:
check = QtGui.QCheckBox('')
check.col = c
check.row = name
self.layout.addWidget(check, row, col)
checks.append(check)
col += 1
if name in self.oldRows:
check.setChecked(self.oldRows[name])
#QtCore.QObject.connect(check, QtCore.SIGNAL('stateChanged(int)'), self.checkChanged)
check.stateChanged.connect(self.checkChanged)
self.rowNames.append(name)
self.rowWidgets.append([label] + checks)
def removeRow(self, name):
row = self.rowNames.index(name)
self.oldRows[name] = self.saveState()['rows'][row] ## save for later
self.rowNames.pop(row)
for w in self.rowWidgets[row]:
w.setParent(None)
#QtCore.QObject.disconnect(w, QtCore.SIGNAL('stateChanged(int)'), self.checkChanged)
if isinstance(w, QtGui.QCheckBox):
w.stateChanged.disconnect(self.checkChanged)
self.rowWidgets.pop(row)
for i in range(row, len(self.rowNames)):
widgets = self.rowWidgets[i]
for j in range(len(widgets)):
widgets[j].setParent(None)
self.layout.addWidget(widgets[j], i+1, j)
def checkChanged(self, state):
check = QtCore.QObject.sender(self)
#self.emit(QtCore.SIGNAL('stateChanged'), check.row, check.col, state)
self.sigStateChanged.emit(check.row, check.col, state)
def saveState(self):
rows = []
for i in range(len(self.rowNames)):
row = [self.rowNames[i]] + [c.isChecked() for c in self.rowWidgets[i][1:]]
rows.append(row)
return {'cols': self.columns, 'rows': rows}
def restoreState(self, state):
rows = [r[0] for r in state['rows']]
self.updateRows(rows)
for r in state['rows']:
rowNum = self.rowNames.index(r[0])
for i in range(1, len(r)):
self.rowWidgets[rowNum][i].setChecked(r[i])