我在使用指定字段中的值对QListView
中的项目进行排序时遇到问题。
基本上我想做的是这样的:
检测照片集中的人脸并将其显示在QListView
中
聚类人脸(图像)
通过将列表中属于同一群集的项(人脸图像)放在一起来更新视图。具体来说,如果项目1、3、5在一个群集中,而项目2、4、6在另一个群集中,则应该在显示项目2、4、6中的任何一个之前显示项目1、3、5(以任何排列)或相反亦然。
我执行此操作的方式是将列表中每个UserRole
的QStandardItem
字段之一设置为群集标签,然后尝试获取QStandardModel
根据此UserRole
进行排序。然后,这将显示彼此相邻的同一群集中的项目(即UserRole
中具有相同的群集标签)。
我可以为项目成功设置UserRole
,但是即使将sort角色设置为默认的QStandardModel
(即根据每个面孔的文字标签)可以正常工作。
谁能告诉我我的代码出了什么问题或提供替代方法?我已经搜索了排序列表,并且在QSortFilterProxyModel上找到了以下链接,但是由于我对Qt还是很陌生,所以我无法使其适应我的情况。
在此先感谢您的任何答复。
以下是相关代码:
import os
from PySide.QtGui import QListView, QStandardItemModel, QStandardItem, QIcon
from PySide.QtCore import Qt
class FacesView(QListView):
"""
View to display detected faces for user to see and label.
"""
UNCLUSTERED_LABEL = -1
CLUSTER_ROLE = Qt.UserRole + 1
def __init__(self, *args):
super(FacesView, self).__init__(*args)
self._dataModel = QStandardItemModel()
self.setModel(self._dataModel)
# Layout items in batches instead of waiting for all items to be
# loaded before user is allowed to interact with them.
self.setLayoutMode(QListView.Batched)
def updateFaceClusters(self, labels):
"""Update the cluster label for each face.
@param labels: [1 x N] array where each element is an integer
for the cluster the face belongs to."""
assert(len(labels) == self._dataModel.rowCount())
# Put the cluster label each item/face belong to in the
# CLUSTER_ROLE field.
for i in xrange(self._dataModel.rowCount()):
index = self._dataModel.index(i, 0)
self._dataModel.setData(index, labels[i], self.CLUSTER_ROLE)
# Use cluster label as sort role
self._dataModel.setSortRole(self.CLUSTER_ROLE)
# This does NOT seem to sort the items even though it works fine
# when sort role is the default Qt.DisplayRole.
self._dataModel.sort(0)
print("Finished updating face clusters")
def itemsInList(self):
"""Returns the label for a face and the path to its image.
@return: (label, path)"""
items = []
for i in xrange(self._dataModel.rowCount()):
label = self._dataModel.index(i, 0).data(Qt.DisplayRole)
imagePath = self._dataModel.index(i, 0).data(Qt.UserRole)
clusterLabel = self._dataModel.index(i, 0).data(self.CLUSTER_ROLE)
items.append((imagePath, label, clusterLabel))
return items
def addItem(self, label, imagePath):
"""Add an item to list view
@param label: The label associated with the item.
@param imagePath: Path to image for the icon."""
if os.path.exists(imagePath):
icon = QIcon(imagePath)
else:
icon = QIcon(':/res/Unknown-person.gif')
item = QStandardItem(icon, label)
item.setEditable(True)
# Add image path to the UserRole field.
item.setData(imagePath, Qt.UserRole)
# Add cluster label to image. CLUSTER_ROLE is where I intend
# to put the item's cluster label.
item.setData(self.UNCLUSTERED_LABEL, self.CLUSTER_ROLE)
# Prevent an item from dropping into another item.
item.setDropEnabled(False)
# Add item to list indirectly by adding it to the model.
self._dataModel.appendRow(item)
def clear(self):
self._dataModel.clear()
最佳答案
您发布的代码没有错。因此,您的使用方式一定存在问题。您如何生成群集标签?
这是一个使用您的FacesView
类的测试脚本,该脚本按预期排序:
from random import randint
from PySide.QtGui import QWidget, QPushButton, QVBoxLayout, QApplication
from facesview import FacesView
class Window(QWidget):
def __init__(self):
QWidget.__init__(self)
self.list = FacesView(self)
self.button = QPushButton('Test', self)
self.button.clicked.connect(self.handleButton)
layout = QVBoxLayout(self)
layout.addWidget(self.list)
layout.addWidget(self.button)
def handleButton(self):
labels = []
self.list.model().setRowCount(0)
for row in range(10):
labels.append(randint(0, 3))
text = 'Item(%d) - Cluster(%d)' % (row, labels[-1])
self.list.addItem(text, 'icon.png')
self.list.updateFaceClusters(labels)
if __name__ == '__main__':
import sys
app = QApplication(sys.argv)
window = Window()
window.show()
sys.exit(app.exec_())
关于python - 如何使用Qt.UserRole对Qt QListview中的项目进行排序,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/12882980/