class ColoredFloatEditor(CustomEditor):
"""Widget for editing a float field, with a calculator"""
def __init__(self,
parent,
precision=2,
reverse=False,
neutral=False,
**kwargs):
CustomEditor.__init__(self, parent)
action = QtGui.QAction(self)
action.setShortcut(Qt.Key_F3)
self.setFocusPolicy(Qt.StrongFocus)
self.spinBox = CustomDoubleSpinBox(parent)
self.spinBox.setDecimals(precision)
self.spinBox.setAlignment(Qt.AlignRight | Qt.AlignVCenter)
self.spinBox.addAction(action)
self.spinBox.setButtonSymbols(QtGui.QAbstractSpinBox.NoButtons)
self.arrow = QtGui.QLabel()
self.arrow.setPixmap(
Icon('tango/16x16/actions/go-up.png').getQPixmap())
self.arrow.setFixedHeight(self.get_height())
self.arrow.setAutoFillBackground(False)
self.arrow.setMaximumWidth(19)
self.calculatorButton = QtGui.QToolButton()
icon = Icon('tango/16x16/apps/accessories-calculator.png').getQIcon()
self.calculatorButton.setIcon(icon)
self.calculatorButton.setAutoRaise(True)
self.calculatorButton.setFixedHeight(self.get_height())
self.calculatorButton.clicked.connect(
lambda: self.popupCalculator(self.spinBox.value()))
action.triggered.connect(
lambda: self.popupCalculator(self.spinBox.value()))
self.spinBox.editingFinished.connect(self.spinbox_editing_finished)
self.releaseKeyboard()
layout = QtGui.QHBoxLayout()
layout.setMargin(0)
layout.setSpacing(0)
layout.addSpacing(3.5)
layout.addWidget(self.arrow)
layout.addWidget(self.spinBox)
layout.addWidget(self.calculatorButton)
self.reverse = reverse
self.neutral = neutral
self.setFocusProxy(self.spinBox)
self.setLayout(layout)
if not self.reverse:
if not self.neutral:
self.icons = {
-1:
Icon('tango/16x16/actions/go-down-red.png').getQPixmap(),
1: Icon('tango/16x16/actions/go-up.png').getQPixmap(),
0: Icon('tango/16x16/actions/zero.png').getQPixmap()
}
else:
self.icons = {
-1:
Icon('tango/16x16/actions/go-down-blue.png').getQPixmap(),
1: Icon('tango/16x16/actions/go-up-blue.png').getQPixmap(),
0: Icon('tango/16x16/actions/zero.png').getQPixmap()
}
else:
self.icons = {
1: Icon('tango/16x16/actions/go-down-red.png').getQPixmap(),
-1: Icon('tango/16x16/actions/go-up.png').getQPixmap(),
0: Icon('tango/16x16/actions/zero.png').getQPixmap()
}
def set_field_attributes(self,
editable=True,
background_color=None,
prefix='',
suffix='',
minimum=constants.camelot_minfloat,
maximum=constants.camelot_maxfloat,
single_step=1.0,
**kwargs):
self.set_enabled(editable)
self.set_background_color(background_color)
self.spinBox.setPrefix(u'%s ' % (unicode(prefix).lstrip()))
self.spinBox.setSuffix(u' %s' % (unicode(suffix).rstrip()))
self.spinBox.setRange(minimum, maximum)
self.spinBox.setSingleStep(single_step)
def set_enabled(self, editable=True):
self.spinBox.setReadOnly(not editable)
self.spinBox.setEnabled(editable)
self.calculatorButton.setShown(editable)
if editable:
self.spinBox.setButtonSymbols(QtGui.QAbstractSpinBox.UpDownArrows)
else:
self.spinBox.setButtonSymbols(QtGui.QAbstractSpinBox.NoButtons)
def set_value(self, value):
value = CustomEditor.set_value(self, value) or 0.0
self.spinBox.setValue(value)
self.arrow.setPixmap(self.icons[cmp(value, 0)])
def get_value(self):
self.spinBox.interpretText()
value = self.spinBox.value()
return CustomEditor.get_value(self) or value
def popupCalculator(self, value):
from camelot.view.controls.calculator import Calculator
calculator = Calculator()
calculator.setValue(value)
calculator.calculation_finished_signal.connect(
self.calculation_finished)
calculator.exec_()
def calculation_finished(self, value):
self.spinBox.setValue(float(unicode(value)))
self.editingFinished.emit()
@QtCore.pyqtSlot()
def spinbox_editing_finished(self):
self.editingFinished.emit()
def set_background_color(self, background_color):
if background_color not in (None, ValueLoading):
selfpalette = self.spinBox.palette()
sbpalette = self.spinBox.palette()
lepalette = self.spinBox.lineEdit().palette()
for x in [
QtGui.QPalette.Active, QtGui.QPalette.Inactive,
QtGui.QPalette.Disabled
]:
for y in [
self.backgroundRole(), QtGui.QPalette.Window,
QtGui.QPalette.Base
]:
selfpalette.setColor(x, y, background_color)
for y in [
self.spinBox.backgroundRole(), QtGui.QPalette.Window,
QtGui.QPalette.Base
]:
sbpalette.setColor(x, y, background_color)
for y in [
self.spinBox.lineEdit().backgroundRole(),
QtGui.QPalette.Window, QtGui.QPalette.Base
]:
lepalette.setColor(x, y, background_color)
self.setPalette(selfpalette)
self.spinBox.setPalette(sbpalette)
self.spinBox.lineEdit().setPalette(lepalette)
return True
else:
return False
