source: sasview/src/sas/qtgui/Perspectives/Fitting/ViewDelegate.py @ 2241130

from PyQt4 import QtGui
from PyQt4 import QtCore

import sas.qtgui.Utilities.GuiUtils as GuiUtils

# Main parameter table view columns
PARAM_PROPERTY=0
PARAM_VALUE=1
PARAM_MIN=2
PARAM_MAX=3
PARAM_UNIT=4

# polydispersity functions
POLYDISPERSE_FUNCTIONS=['rectangle', 'array', 'lognormal', 'gaussian', 'schulz']
# polydispersity columns
POLY_PARAMETER=0
POLY_PD=1
POLY_MIN=2
POLY_MAX=3
POLY_NPTS=4
POLY_NSIGS=5
POLY_FUNCTION=6

class CustomLineEdit(QtGui.QLineEdit):
    editingFinished = QtCore.pyqtSignal()
    def __init__(self, parent=None, old_value=None):
        super(CustomLineEdit, self).__init__(parent)
        self.setAutoFillBackground(True)
        self.old_value = old_value
    def focusOutEvent(self, event):
        self.editingFinished.emit()

class ModelViewDelegate(QtGui.QStyledItemDelegate):
    """
    Custom delegate for appearance and behavior control of the model view
    """
    def paint(self, painter, option, index):
        """
        Overwrite generic painter for certain columns
        """
        if index.column() in (PARAM_UNIT, PARAM_MIN, PARAM_MAX):
            # Units - present in nice HTML
            options = QtGui.QStyleOptionViewItemV4(option)
            self.initStyleOption(options,index)
            style = QtGui.QApplication.style() if options.widget is None else options.widget.style()

            # Prepare document for inserting into cell
            doc = QtGui.QTextDocument()

            # Convert the unit description into HTML
            text_html = GuiUtils.convertUnitToHTML(str(options.text))
            doc.setHtml(text_html)

            # delete the original content
            options.text = ""
            style.drawControl(QtGui.QStyle.CE_ItemViewItem, options, painter);

            context = QtGui.QAbstractTextDocumentLayout.PaintContext()
            textRect = style.subElementRect(QtGui.QStyle.SE_ItemViewItemText, options)

            painter.save()
            painter.translate(textRect.topLeft())
            painter.setClipRect(textRect.translated(-textRect.topLeft()))
            # Draw the QTextDocument in the cell
            doc.documentLayout().draw(painter, context)

            painter.restore()
        else:
            # Just the default paint
            QtGui.QStyledItemDelegate.paint(self, painter, option, index)

    def createEditor(self, widget, option, index):
        """
        Overwrite generic editor for certain columns
        """
        if not index.isValid():
            return 0
        if index.column() == PARAM_VALUE: #only in the value column
            editor = QtGui.QLineEdit(widget)
            validator = QtGui.QDoubleValidator()
            editor.setValidator(validator)
            return editor
        elif index.column() in (PARAM_MIN, PARAM_MAX):
            # Save current value in case we need to revert
            #self._old_value = index.data().toFloat()[0]
            self._old_value = index.data().toString()
            editor = CustomLineEdit(widget, old_value=self._old_value)
            editor.editingFinished.connect(self.commitAndCloseEditor)
            return editor

        return super(ModelViewDelegate, self).createEditor(widget, option, index)

    #def setEditorData(self, editor, index):
    #    if index.column() == MIN:
    #        #value = index.data().toString()[0]
    #        value = index.model().data(index, QtCore.Qt.DisplayRole).toString()
    #        print "VALUE = ", value
    #        editor.setText('['+value+']')
    #        return editor

    #    return super(ModelViewDelegate, self).setEditorData(editor, index)

    def commitAndCloseEditor(self):
            editor = self.sender()
            content = editor.text()
            try:
                value_float = float(content)
            except ValueError:
                # TODO: Notify the user
                # <scary popup>
                # Do nothing
                return
            self.commitData.emit(editor)
            self.closeEditor.emit(editor, QtGui.QAbstractItemDelegate.NoHint)


class PolyViewDelegate(QtGui.QStyledItemDelegate):
    """
    Custom delegate for appearance and behavior control of the polydisperisty view
    """
    def createEditor(self, parent, option, index):
        # TODO: set it to correct index on creation
        if index.column() == POLY_FUNCTION:
            editor = QtGui.QComboBox(parent)
            for function in POLYDISPERSE_FUNCTIONS:
                editor.addItem(function)
            return editor
        else:
            QtGui.QStyledItemDelegate.createEditor(self, parent, option, index)