QTreeView中选择缓慢，为什么？

Question

我最近在使用PyQt开发的一个项目中遇到了问题。我有一个连接到QAbstractItemModel的QTreeView，其中通常有成千上万个节点。到目前为止，它可以正常工作，但是我今天意识到选择很多节点非常缓慢。经过一些调查，发现QAbstractItemModel.parent()被频繁调用。我创建了最小化的代码来复现这个问题：

#!/usr/bin/env python
import sys
import cProfile
import pstats

from PyQt4.QtCore import Qt, QAbstractItemModel, QVariant, QModelIndex
from PyQt4.QtGui import QApplication, QTreeView

# 200 root nodes with 10 subnodes each

class TreeNode(object):
    def __init__(self, parent, row, text):
        self.parent = parent
        self.row = row
        self.text = text
        if parent is None: # root node, create subnodes
            self.children = [TreeNode(self, i, unicode(i)) for i in range(10)]
        else:
            self.children = []

class TreeModel(QAbstractItemModel):
    def __init__(self):
        QAbstractItemModel.__init__(self)
        self.nodes = [TreeNode(None, i, unicode(i)) for i in range(200)]

    def index(self, row, column, parent):
        if not self.nodes:
            return QModelIndex()
        if not parent.isValid():
            return self.createIndex(row, column, self.nodes[row])
        node = parent.internalPointer()
        return self.createIndex(row, column, node.children[row])

    def parent(self, index):
        if not index.isValid():
            return QModelIndex()
        node = index.internalPointer()
        if node.parent is None:
            return QModelIndex()
        else:
            return self.createIndex(node.parent.row, 0, node.parent)

    def columnCount(self, parent):
        return 1

    def rowCount(self, parent):
        if not parent.isValid():
            return len(self.nodes)
        node = parent.internalPointer()
        return len(node.children)

    def data(self, index, role):
        if not index.isValid():
            return QVariant()
        node = index.internalPointer()
        if role == Qt.DisplayRole:
            return QVariant(node.text)
        return QVariant()


app = QApplication(sys.argv)
treemodel = TreeModel()
treeview = QTreeView()
treeview.setSelectionMode(QTreeView.ExtendedSelection)
treeview.setSelectionBehavior(QTreeView.SelectRows)
treeview.setModel(treemodel)
treeview.expandAll()
treeview.show()
cProfile.run('app.exec_()', 'profdata')
p = pstats.Stats('profdata')
p.sort_stats('time').print_stats()

为了重现问题，只需运行代码（进行性能分析）并选择树部件中的所有节点（通过Shift选择或Cmd-A）。当您退出应用程序时，性能分析统计数据将显示类似以下内容：

Fri May  8 20:04:26 2009    profdata

         628377 function calls in 6.210 CPU seconds

   Ordered by: internal time

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    4.788    4.788    6.210    6.210 {built-in method exec_}
   136585    0.861    0.000    1.182    0.000 /Users/hsoft/Desktop/slow_selection.py:34(parent)
   142123    0.217    0.000    0.217    0.000 {built-in method createIndex}
    17519    0.148    0.000    0.164    0.000 /Users/hsoft/Desktop/slow_selection.py:52(data)
   162198    0.094    0.000    0.094    0.000 {built-in method isValid}
     8000    0.055    0.000    0.076    0.000 /Users/hsoft/Desktop/slow_selection.py:26(index)
   161357    0.047    0.000    0.047    0.000 {built-in method internalPointer}
       94    0.000    0.000    0.000    0.000 /Users/hsoft/Desktop/slow_selection.py:46(rowCount)
      404    0.000    0.000    0.000    0.000 /Users/hsoft/Desktop/slow_selection.py:43(columnCount)
       94    0.000    0.000    0.000    0.000 {len}
        1    0.000    0.000    6.210    6.210 <string>:1(<module>)
        1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}

这个数据中奇怪的部分是parent()函数被调用的次数：2000个节点中竟然调用了136k次！有人知道为什么吗？

- Virgil Dupras

Answer 1

尝试为您的树视图调用setUniformRowHeights(true)： https://doc.qt.io/qt-4.8/qtreeview.html#uniformRowHeights-prop 此外，有一个名为modeltest的C++工具来自qt labs。 我不确定是否有适用于Python的类似工具： https://wiki.qt.io/Model_Test

- Mark Beckwith

---

谢谢提示，但遗憾的是它没有帮助。虽然它减少了父调用的数量，但只有134k次。至于Modeltest，它看起来很有趣，但我不知道如何在PyQt中导入第三方C++组件（我需要搜索一下）。但无论如何，这个模型对我来说似乎是正确的，不是吗？ - Virgil Dupras

Answer 2

我把您的很好的示例代码转换为PyQt5并在Qt5.2下运行，可以确认数字仍然相似，即调用次数非常大。例如，这里是启动报告的顶部部分，cmd-A选择全部，滚动一页，退出：

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1   14.880   14.880   15.669   15.669 {built-in method exec_}
   196712    0.542    0.000    0.703    0.000 /Users/dcortes1/Desktop/scratch/treeview.py:36(parent)
   185296    0.104    0.000    0.104    0.000 {built-in method createIndex}
    20910    0.050    0.000    0.056    0.000 /Users/dcortes1/Desktop/scratch/treeview.py:54(data)
   225252    0.036    0.000    0.036    0.000 {built-in method isValid}
   224110    0.034    0.000    0.034    0.000 {built-in method internalPointer}
     7110    0.020    0.000    0.027    0.000 /Users/dcortes1/Desktop/scratch/treeview.py:28(index)

虽然计数确实过多（我无法解释），但请注意cumtime值并不太大。这些函数也可以重新编码以运行得更快；例如，在index()中，“if not self.nodes”是否为真？同样，注意parent()和createIndex()的计数几乎相同，因此index.isValid()通常为true（合理，因为末节点比父节点数量多得多）。重新编码以先处理该情况将进一步缩短parent()的cumtime。编辑：再想一想，这样的优化是“在泰坦尼克号上重新排列甲板椅子”。

- user405