class CustomTreeItem(QtGui.QTreeWidgetItem):
def __init__(self, window, parent, name, view=None, color=True):
#print "creating ", name, " under ", parent
if view:
self.view = view
if not view:
try:
self.view = parent.view
except AttributeError:
self.view = parent
self.window = window
self._name = name
## Init super class ( QtGui.QTreeWidgetItem )
super(CustomTreeItem, self).__init__(parent)
## Column 0 - Text:
self.setText(0, name)
## Column 1 - Color:
if color:
#print type(self.view.view)
self.colorChooser = KColorButton(self.view)
self.colorChooser.setColor(QtGui.QColor(randint(0, 255), randint(0, 255), randint(0, 255)))
#self.colorChooser.setGeometry(QtCore.QRect(0, 0, 10, 10))
self.colorChooser.setFixedSize(30,20)
self.view.setItemWidget(self, 1, self.colorChooser)
self.view.connect(self.colorChooser, QtCore.SIGNAL("changed (const QColor&)"), self.colorChanged)
## Column 2 - CheckBox:
self.button = QtGui.QCheckBox(self.view)
self.button.setChecked(True)
self.view.setItemWidget(self, 2, self.button)
## Column 3 - Current value:
if color:
self.setText(3, 'N/A')
else:
self.setText(3, '')
## Signals
self.view.connect(self.button, QtCore.SIGNAL("clicked()"), self.buttonPressed)
def setDataValue(self, value):
self.setText(3, str(value))
def buttonPressed(self):
if self.childCount():
for i in range(self.childCount()):
self.child(i).button.setChecked(self.button.isChecked())
self.child(i).buttonPressed()
elif self.button.isChecked():
self.window.enablePlot(self._name)
self.window.setPlotColor(self._name, self.colorChooser.color())
else:
self.window.disablePlot(self._name)
def colorChanged(self):
self.window.setPlotColor(self._name, self.colorChooser.color())
class CustomTreeItem(QtGui.QTreeWidgetItem):
def __init__(self, window, parent, name, view=None, color=True):
#print "creating ", name, " under ", parent
if view:
self.view = view
if not view:
try:
self.view = parent.view
except AttributeError:
self.view = parent
self.window = window
self._name = name
## Init super class ( QtGui.QTreeWidgetItem )
super(CustomTreeItem, self).__init__(parent)
## Column 0 - Text:
self.setText(0, name)
## Column 1 - Color:
if color:
#print type(self.view.view)
self.colorChooser = KColorButton(self.view)
self.colorChooser.setColor(
QtGui.QColor(randint(0, 255), randint(0, 255), randint(0,
255)))
#self.colorChooser.setGeometry(QtCore.QRect(0, 0, 10, 10))
self.colorChooser.setFixedSize(30, 20)
self.view.setItemWidget(self, 1, self.colorChooser)
self.view.connect(self.colorChooser,
QtCore.SIGNAL("changed (const QColor&)"),
self.colorChanged)
## Column 2 - CheckBox:
self.button = QtGui.QCheckBox(self.view)
self.button.setChecked(True)
self.view.setItemWidget(self, 2, self.button)
## Column 3 - Current value:
if color:
self.setText(3, 'N/A')
else:
self.setText(3, '')
## Signals
self.view.connect(self.button, QtCore.SIGNAL("clicked()"),
self.buttonPressed)
def setDataValue(self, value):
self.setText(3, str(value))
def buttonPressed(self):
if self.childCount():
for i in range(self.childCount()):
self.child(i).button.setChecked(self.button.isChecked())
self.child(i).buttonPressed()
elif self.button.isChecked():
self.window.enablePlot(self._name)
self.window.setPlotColor(self._name, self.colorChooser.color())
else:
self.window.disablePlot(self._name)
def colorChanged(self):
self.window.setPlotColor(self._name, self.colorChooser.color())
