如何使用Qt代表进行自定义绘画?

时间:2011-01-21 01:16:34

标签: python qt pyqt

我试图覆盖将项目拖过它们时绘制QTreeWidgetItems的方式。我已经重写了拖动事件,将Qt.UserRole数据设置为1,用于我想要绘制的QTreeWidgetItems。在Item Delegate中,我阅读UserRole并相应地绘制。

我的自定义绘画完全符合预期(即粗线);但是,我无法弄清楚如何抑制标准画家拖动(即小矩形)所做的绘制而不会抑制所有其他绘画(即文本等)。

任何想法都会受到赞赏。

alt text

实施例

def dragMoveEvent(self, event):
    pos = event.pos()
    item = self.myTreeWidget.itemAt(pos)

    # If hovered over an item during drag, set UserRole = 1
    if item:
        index = self.myTreeWidget.indexFromItem(item)
        self.myTreeWidget.model().setData(index, 1, Qt.UserRole)

    # reset UserRole to 0 for all other indices
    for i in range(self.myTreeWidget.model().rowCount()):
        _index = self.myTreeWidget.model().index(i, 0)
        if not item or index != _index:
            self.myTreeWidget.model().setData(_index, 0, Qt.UserRole)


class MyDelegate(QStyledItemDelegate):

    def paint( self, painter, option, index ):
        QStyledItemDelegate.paint(self, painter, option, index)
        painter.save()
        data = index.model().data( index, Qt.UserRole ).toInt()
            # if UserRole = 1 draw custom line
        if data[1] and data[0] == 1:
            line = QLine( option.rect.topLeft(), option.rect.topRight() )
            painter.drawLine( line )
        painter.restore()

1 个答案:

答案 0 :(得分:1)

使用c ++可以很容易地解决这个问题:你可以定义一个新的样式,覆盖drawPrimitive方法并在那里通过元素接收到QStyle :: PE_IndicatorItemViewItemDrop常量时自定义绘制(或者什么都不做) 参数。以下是一个例子:

class TestStyle : public QProxyStyle
{
public:
    TestStyle(QStyle *baseStyle = 0) : QProxyStyle(baseStyle) {}

    void drawPrimitive(PrimitiveElement element, const QStyleOption *option, QPainter *painter, const QWidget *widget) const
    {
        if (element == QStyle::PE_IndicatorItemViewItemDrop)
        {
            //?? do nothing or do custom painting here
        }
        else
        {
            QProxyStyle::drawPrimitive(element, option, painter, widget);
        }
    }
};

.. 

ui->treeView->setStyle(new TestStyle(ui->treeView->style()));

现在坏消息是pyqt似乎对QProxyStyle一无所知;看起来它没有包裹在那里,所以为了使它能够工作,你需要自己包装样式类。

另一种解决方案是创建自定义QTreeView后代并覆盖其paintEvent方法。默认实现是调用drawTree和paintDropIndicator;其中paintDropIndicator负责拖放指示器,而drawTree则呈现树项。 drawTree受到保护,您可以从paintEvent调用:

class TestTreeView : public QTreeView
{
public:
    explicit TestTreeView(QWidget *parent = 0) : QTreeView(parent) {}

    void paintEvent(QPaintEvent * event)
    {
        QPainter painter(viewport());
        drawTree(&painter, event->region());
    }
};

这应该会取消默认的拖放指示器。如果您遇到麻烦将其转换为python,请告诉我。

希望这有帮助,尊重